Automated identification and grading system of diabetic retinopathy using deep neural networks

• Published in: Knowledge-based systems 2019-07, Vol.175, p.12-25
• Main Authors: Zhang, Wei, Zhong, Jie, Yang, Shijun, Gao, Zhentao, Hu, Junjie, Chen, Yuanyuan, Yi, Zhang
• Format: Article
• Language: English
• Subjects: Artificial neural networks; Automation; Classifiers; Deep learning; Diabetes; Diabetic retinopathy; Diagnostic systems; Ensemble learning; Evaluation; Fundus images; Image classification; Image detection; Image quality; Medical imaging; Medical personnel; Neural networks; Reliability analysis; Sensitivity; Transfer learning
• ISSN: 0950-7051; 1872-7409
• DOI: 10.1016/j.knosys.2019.03.016

Diabetic retinopathy (DR) is a major cause of human vision loss worldwide. Slowing down the progress of the disease requires early screening. However, the clinical diagnosis of DR presents a considerable challenge in low-resource settings where few ophthalmologists are available to care for all patients with diabetes. In this study, an automated DR identification and grading system called DeepDR is proposed. DeepDR directly detects the presence and severity of DR from fundus images via transfer learning and ensemble learning. It comprises a set of state-of-the-art neural networks based on combinations of popular convolutional neural networks and customised standard deep neural networks. The DeepDR system is developed by constructing a high-quality dataset of DR medical images and then labelled by clinical ophthalmologists. We further explore the relationship between the number of ideal component classifiers and the number of class labels, as well as the effects of different combinations of component classifiers on the best integration performance to construct an optimal model. We evaluate the models on the basis of validity and reliability using nine metrics. Results show that the identification model performs best with a sensitivity of 97.5%, a specificity of 97.7% and an area under the curve of 97.7%. Meanwhile, the grading model achieves a sensitivity of 98.1% and a specificity of 98.9%. On the basis of the methods above, DeepDR can detect DR satisfactorily. Experiment results indicate the importance and effectiveness of the ideal number and combinations of component classifiers in relation to model performance. DeepDR provides reproducible and consistent detection results with high sensitivity and specificity instantaneously. Hence, this work provides ophthalmologists with insights into the diagnostic process. •We established a high-quality labelled dataset of DR medical images.•We developed a novel and well-performing DR recognition and classification system.•The optimal combination of ensemble model was explored experimentally.