Deep hybrid architectures for diabetic retinopathy classification

ABSTRACT

Diabetic retinopathy (DR) is the most severe ocular complication of diabetes. It leads to serious eye complications such as vision impairment and blindness. A computer-aided diagnosis may help in the early detection of this disease, which increases the chances of treating it efficiently. This paper carried out an empirical evaluation of the performances of 28 deep hybrid architectures for an automatic binary classification of the referable diabetic retinopathy, and compared them to seven end-to-end deep learning (DL) architectures. For the hybrid architectures, we combined seven DL techniques for feature extraction (DenseNet201, VGG16, VGG19, MobileNet_V2, Inception_V3, Inception_ResNet_V2 and ResNet50) and four classifiers (SVM, MLP, DT and KNN). For the end-to-end DL architectures, we used the same techniques used for the feature extraction in the hybrid architectures. The architectures were compared in terms of accuracy, sensitivity, precision and F1-score using the Scott Knott test and the Borda count voting method. All the empirical evaluations were over three datasets: APTOS, Kaggle DR and Messidor-2, using a k-fold cross validation method. The results showed the potential of combining deep learning techniques for feature extraction and classical machine learning techniques to classify referable diabetic retinopathy. The hybrid architecture using the SVM classifier and MobileNet_V2 for feature extraction was the top performing architecture and it was classified with the best performing end-to-end architectures in the best clusters of APTOS, Kaggle DR and Messidor-2 datasets with an accuracy equal to 88.80%, 84.01% and 84.05% respectively. Note that the two end-to-end architectures DenseNet201 and MobileNet_V2 outperformed all the hybrid architectures over the three datasets. However, we recommend the use of the hybrid architecture designed with SVM and MobileNet_V2 since it is promising and less time consuming, and requires less parameter tuning compared to the end-to-end techniques.

KEYWORDS:

• Medical images
• diabetic retinopathy
• hybrid architecture

Disclosure statement

No potential conflict of interest was reported by the author(s).

Data availability statement

All experiment datasets, APTOS, Kaggle DR and Messidor-2, are public and available on:

• APTOS website: https://www.kaggle.com/c/aptos2019-blindness-detection/data

• Kaggle DR website: https://www.kaggle.com/c/diabetic-retinopathy-detection

• Messidor-2 website: https://www.adcis.net/fr/logiciels-tiers/messidor2-fr/, https://www.kaggle.com/google-brain/messidor2-d