Detection and diagnosis of diabetic eye diseases using two phase transfer learning approach

PeerJ Comput Sci. 2024 Sep 19:10:e2135. doi: 10.7717/peerj-cs.2135. eCollection 2024.

Abstract

Background: Early diagnosis and treatment of diabetic eye disease (DED) improve prognosis and lessen the possibility of permanent vision loss. Screening of retinal fundus images is a significant process widely employed for diagnosing patients with DED or other eye problems. However, considerable time and effort are required to detect these images manually.

Methods: Deep learning approaches in machine learning have attained superior performance for the binary classification of healthy and pathological retinal fundus images. In contrast, multi-class retinal eye disease classification is still a difficult task. Therefore, a two-phase transfer learning approach is developed in this research for automated classification and segmentation of multi-class DED pathologies.

Results: In the first step, a Modified ResNet-50 model pre-trained on the ImageNet dataset was transferred and learned to classify normal diabetic macular edema (DME), diabetic retinopathy, glaucoma, and cataracts. In the second step, the defective region of multiple eye diseases is segmented using the transfer learning-based DenseUNet model. From the publicly accessible dataset, the suggested model is assessed using several retinal fundus images. Our proposed model for multi-class classification achieves a maximum specificity of 99.73%, a sensitivity of 99.54%, and an accuracy of 99.67%.

Keywords: Deep learning; Defected region; Diabetic eye disease; Early diagnosis; Retinal fundus images; Transfer learning.

Associated data

  • figshare/10.6084/m9.figshare.22760705.v2

Grants and funding

The authors received no funding for this work.