Convolutional Neural Networks for Diabetic Retinopathy Fundus Image Classification Using Transfer Learning
摘要
Diabetic retinopathy is an incurable ocular disease stemming from elevated blood glucose levels, typical of diabetes mellitus. This ailment affects the retina’s blood vessels, potentially leading to blindness if not treated promptly. Computer-aided diagnosis (CAD) systems based on artificial intelligence are designed to assist specialists in the early diagnosis of such diseases, thereby facilitating prognosis. In this study, we utilized the DDR and IDRiD datasets for the binary classification of Diabetic Retinopathy using the K-Fold method for cross-validation and transfer learning. The first dataset was used for model training, while the second was reserved exclusively for external validation. The VGG16, ResNet50, and MobileNetV2 networks were evaluated in accuracy, precision, F1 score, AUC ROC score, recall, and specificity. MobileNetV2 exhibited the best performance, achieving values of 78.68%, 76.68%, 86.15%, 93.38%, 98.28%, and 38.10% for accuracy, precision, F1 score, AUC ROC score, recall, and specificity, respectively, when tested with images from IDRiD, which had never been seen by the model before.