Deep Learning-Based Automatic Diagnosis of Keratoconus with Corneal Endothelium Image

Abstract

Introduction: The primary objective of this study was to develop an end-to-end model that can accurately identify corneal endothelial cells and diagnose keratoconus based on corneal endothelial images acquired from a non-contact specular microscope. Methods: This was a retrospective case–control study performed at the Refractive Surgery Center of West China Hospital. A total of 403 keratoconus eyes (221 patients) and 370 myopic eyes (185 normal controls) were consecutively recruited from January 2021 to September 2022. Specular microscopy was used to image and measure the morphometric parameters of the corneal endothelial cells. A Fully Convolutional Network model with a ResNet50 (FCN_ResNet50) was established to perform the endothelial segmentation. The images were then classified using an ensemble machine learning system consisting of four pre-trained deep learning networks: DenseNet121, ResNet50, Inception_v3, and MobileNet_v2. The performance of the models was evaluated based on different metrics, such as accuracy, intersection over union (IoU), and mean IoU. Results: We established a fully end-to-end deep-learning model for the segmentation of endothelial and diagnosis of keratoconus. For endothelial segmentation, the accuracy of the FCN_ResNet50 model achieved near 90% with mean IoU converging to about 80%. The ensemble machine learning system can achieve over 92% accuracy, and > 98% area under curve (AUC) values to diagnose keratoconus with endothelial cell images. In addition, we constructed a diagnostic model based on deep-learning features and developed an associated nomogram which manifested an excellent performance for diagnosis and monitoring the progression of keratoconus. Conclusions: Our research developed an end-to-end model to automatically identify and assess corneal endothelial morphological changes in keratoconus eyes. Moreover, we also constructed a novel nomogram, which can provide valuable information for the diagnosis, monitoring, and management of the disease.