AI-Based Decision-Support System for Diagnosing Acanthamoeba Keratitis Using In Vivo Confocal Microscopy Images.

Abstract

In vivo confocal microscopy (IVCM) of the cornea is a valuable tool for clinical assessment of the cornea but does not provide stand-alone diagnostic support. The aim of this work was to develop an artificial intelligence (AI)-based decision-support system (DSS) for automated diagnosis of Acanthamoeba keratitis (AK) using IVCM images. The automated workflow for the AI-based DSS was defined and implemented using deep learning models, image processing techniques, rule-based decisions, and valuable input from domain experts. The models were evaluated with 5-fold-cross validation on a dataset of 85 patients (47,734 IVCM images from healthy, AK, and other disease cases) collected at a single eye clinic in Sweden. The developed DSS was validated on an additional 26 patients (21,236 images). Overall, the DSS uses as input raw unprocessed IVCM image data, successfully separates artefacts from true images (93% accuracy), then classifies the remaining images by their corneal layer (90% accuracy). The DSS subsequently predicts if the cornea is healthy or diseased (95% model accuracy). In disease cases, the DSS detects images with AK signs with 84% accuracy, and further localizes the regions of diagnostic value with 76.5% accuracy. The proposed AI-based DSS can automatically and accurately preprocess IVCM images (separating artefacts and sorting images into corneal layers) which decreases screening time. The accuracy of AK detection using raw IVCM images must be further explored and improved. The proposed automated DSS for experienced specialists assists in diagnosing AK using IVCM images.