Objective To evaluate the performance of two lightweight neural network models in the diagnosis of common fundus diseases and make comparison to another two classical models. Methods A total of 16,000 color fundus photography were collected, including 2000 each of glaucoma, diabetic retinopathy (DR), high myopia, central retinal vein occlusion (CRVO), age-related macular degeneration (AMD), optic neuropathy, and central serous chorioretinopathy (CSC), in addition to 2000 normal fundus. Fundus photography was obtained from patients or physical examiners who visited the Ophthalmology Department of Beijing Tongren Hospital, Capital Medical University. Each fundus photography has been diagnosed and labeled by two professional ophthalmologists. Two classical classification models (ResNet152 and DenseNet121), and two lightweight classification models (MobileNetV3 and ShufflenetV2), were trained. Area under the curve (AUC), sensitivity, specificity, accuracy, positive predictive value, and negative predictive value were used to evaluate the performance of the four models. Results Compared with the classical classification model, the total size and number of parameters of the two lightweight classification models were significantly reduced, and the classification speed was sharply improved. Compared with the DenseNet121 model, the ShufflenetV2 model took 50.7% less time to make a diagnosis on a fundus photography. The classical models performed better than lightweight classification models, and Densenet121 showed highest AUC in five out of the seven common fundus diseases. However, the performance of lightweight classification models is satisfying. The AUCs using MobileNetV3 model to diagnose AMD, diabetic retinopathy, glaucoma, CRVO, high myopia, optic atrophy, and CSC were 0.805, 0.892, 0.866, 0.812, 0.887, 0.868, and 0.803, respectively. For ShufflenetV2model, the AUCs for the above seven diseases were 0.856, 0.893, 0.855, 0.884, 0.891, 0.867, and 0.844, respectively. Conclusion The training of light-weight neural network models based on color fundus photography for the diagnosis of common fundus diseases is not only fast but also has a significant reduction in storage size and parameter number compared with the classical classification model, and can achieve satisfactory accuracy.
基金:
priming scientific research foundation for the junior researcher in Beijing Tongren Hospital, Capital Medical University
第一作者机构:[1]Capital Med Univ, Beijing Tongren Hosp, Beijing Tongren Eye Ctr, Beijing, Peoples R China
通讯作者:
推荐引用方式(GB/T 7714):
Wang Jin-Da,Liu Mei-Rui,Liu Mei-Ling,et al.An auxiliary diagnostic tool for common fundus diseases based on fundus color photography and light-weight classification models[J].GRAEFES ARCHIVE FOR CLINICAL AND EXPERIMENTAL OPHTHALMOLOGY.2024,262(1):223-229.doi:10.1007/s00417-023-06182-2.
APA:
Wang, Jin-Da,Liu, Mei-Rui,Liu, Mei-Ling,Zhang, Ran,Chen, Chang-Xi&Cao, Kai.(2024).An auxiliary diagnostic tool for common fundus diseases based on fundus color photography and light-weight classification models.GRAEFES ARCHIVE FOR CLINICAL AND EXPERIMENTAL OPHTHALMOLOGY,262,(1)
MLA:
Wang, Jin-Da,et al."An auxiliary diagnostic tool for common fundus diseases based on fundus color photography and light-weight classification models".GRAEFES ARCHIVE FOR CLINICAL AND EXPERIMENTAL OPHTHALMOLOGY 262..1(2024):223-229