Open Access
28 August 2020 Explainable end-to-end deep learning for diabetic retinopathy detection across multiple datasets
Mohamed Chetoui, Moulay A. Akhloufi
Author Affiliations +
Abstract

Purpose: Diabetic retinopathy (DR) is characterized by retinal lesions affecting people having diabetes for several years. It is one of the leading causes of visual impairment worldwide. To diagnose this disease, ophthalmologists need to manually analyze retinal fundus images. Computer-aided diagnosis systems can help alleviate this burden by automatically detecting DR on retinal images, thus saving physicians’ precious time and reducing costs. The objective of this study is to develop a deep learning algorithm capable of detecting DR on retinal fundus images. Nine public datasets and more than 90,000 images are used to assess the efficiency of the proposed technique. In addition, an explainability algorithm is developed to visually show the DR signs detected by the deep model.

Approach: The proposed deep learning algorithm fine-tunes a pretrained deep convolutional neural network for DR detection. The model is trained on a subset of EyePACS dataset using a cosine annealing strategy for decaying the learning rate with warm up, thus improving the training accuracy. Tests are conducted on the nine datasets. An explainability algorithm based on gradient-weighted class activation mapping is developed to visually show the signs selected by the model to classify the retina images as DR.

Result: The proposed network leads to higher classification rates with an area under curve (AUC) of 0.986, sensitivity = 0.958, and specificity = 0.971 for EyePACS. For MESSIDOR, MESSIDOR-2, DIARETDB0, DIARETDB1, STARE, IDRID, E-ophtha, and UoA-DR, the AUC is 0.963, 0.979, 0.986, 0.988, 0.964, 0.957, 0.984, and 0.990, respectively.

Conclusions: The obtained results achieve state-of-the-art performance and outperform past published works relying on training using only publicly available datasets. The proposed approach can robustly classify fundus images and detect DR. An explainability model was developed and showed that our model was able to efficiently identify different signs of DR and detect this health issue.

© 2020 Society of Photo-Optical Instrumentation Engineers (SPIE) 2329-4302/2020/$28.00 © 2020 SPIE
Mohamed Chetoui and Moulay A. Akhloufi "Explainable end-to-end deep learning for diabetic retinopathy detection across multiple datasets," Journal of Medical Imaging 7(4), 044503 (28 August 2020). https://doi.org/10.1117/1.JMI.7.4.044503
Received: 6 January 2020; Accepted: 7 August 2020; Published: 28 August 2020
Lens.org Logo
CITATIONS
Cited by 32 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Eye models

Retina

Data modeling

Surface plasmons

Algorithm development

Visual process modeling

Blood vessels

RELATED CONTENT

What and where you have seen? Bag of Words based...
Proceedings of SPIE (April 17 2019)
Human visual performance model for crewstation design
Proceedings of SPIE (August 01 1991)

Back to Top