Comparative Study of Transfer Learning Models for Retinal Disease Diagnosis from Fundus Images

• Published in: Computers, materials & continua Vol. 70; no. 3; pp. 5821 - 5834
• Main Authors: Pin, Kuntha, Ho Chang, Jee, Nam, Yunyoung
• Format: Journal Article
• Language: English
• Published: Henderson Tech Science Press 2022
• Subjects: Age related diseases; Artificial neural networks; Comparative studies; Deep learning; Diabetic retinopathy; Digital imaging; Eye diseases; Glaucoma; Image classification; Image enhancement; Image filters; Image quality; Machine learning; Macular degeneration; Medical imaging; Ophthalmology; Quality assessment; Retinal images
• ISSN: 1546-2226; 1546-2218; 1546-2226
• DOI: 10.32604/cmc.2022.021943

While the usage of digital ocular fundus image has been widespread in ophthalmology practice, the interpretation of the image has been still on the hands of the ophthalmologists which are quite costly. We explored a robust deep learning system that detects three major ocular diseases: diabetic retinopathy (DR), glaucoma (GLC), and age-related macular degeneration (AMD). The proposed method is composed of two steps. First, an initial quality evaluation in the classification system is proposed to filter out poor-quality images to enhance its performance, a technique that has not been explored previously. Second, the transfer learning technique is used with various convolutional neural networks (CNN) models that automatically learn a thousand features in the digital retinal image, and are based on those features for diagnosing eye diseases. Comparison performance of many models is conducted to find the optimal model which fits with fundus classification. Among the different CNN models, DenseNet-201 outperforms others with an area under the receiver operating characteristic curve of 0.99. Furthermore, the corresponding specificities for healthy, DR, GLC, and AMD patients are found to be 89.52%, 96.69%, 89.58%, and 100%, respectively. These results demonstrate that the proposed method can reduce the time-consumption by automatically diagnosing multiple eye diseases using computer-aided assistance tools.