Use of machine learning to achieve keratoconus detection skills of a corneal expert

Abstract

Purpose: To construct an automatic machine-learning derived algorithm discriminating between normal corneas and suspect irregular or keratoconic corneas.

Methods: A total of 8526 corneal tomography images of 4904 eyes obtained between November 2010 and July 2017 using a combined Scheimpflug/Placido tomographer were retrospectively evaluated. Each image was evaluated for acquisition quality and was labeled as normal, suspect irregular or keratoconic by a cornea specialist. Two algorithms were built. The first was based on 94 instrument-derived output parameters, and the second integrated keratoconus prediction indices of the device with the 94 instrument-derived output parameters. Both models were compared with the tomographer's keratoconus detection algorithms. Out of the 8526 images evaluated, 7104 images of 3787 eyes had sufficient acquisition quality. Of those, 5904 examinations were randomly chosen for construction of the models using the random forest algorithm. The models were then validated using the remaining 1200 examinations.

Results: Both RF algorithms had a larger AUC compared with any of the tomographer's KC detection algorithms (p < 10^-9). The first constructed model had 90.2% accuracy, sensitivity of 94.2%, and specificity of 89.6% (Youden 0.838). Calculated AUC was 0.964. The second model had 91.5% accuracy, sensitivity of 94.7%, and specificity of 89.8% (Youden 0.846). Calculated AUC was 0.969.

Conclusion: Using the RF machine-learning algorithm, accuracy of discrimination between normal, suspect irregular and keratoconic corneas approximates that of an experienced corneal expert. Applying machine learning to corneal tomography can facilitate keratoconus screening in large populations as well as off-site screening of refractive surgery candidates.

Keywords: Artificial intelligence; Detection; Dual Scheimpflug/Placido; Galilei; Keratoconus; Machine learning; Random forest.