Transfer Learning for Diabetic Retinopathy Detection: A Study of Dataset Combination and Model Performance

被引:16
作者
Mutawa, A. M. [1 ]
Alnajdi, Shahad [1 ]
Sruthi, Sai [1 ]
机构
[1] Kuwait Univ, Coll Engn & Petr, Dept Comp Engn, PO Box 5969, Safat 13060, Kuwait
来源
APPLIED SCIENCES-BASEL | 2023年 / 13卷 / 09期
关键词
convolutional neural network; deep learning; diabetic retinopathy; image classification; medical imaging; transfer learning; CONVOLUTIONAL NEURAL-NETWORKS; DIAGNOSIS;
D O I
10.3390/app13095685
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Diabetes' serious complication, diabetic retinopathy (DR), which can potentially be life-threatening, might result in vision loss in certain situations. Although it has no symptoms in the early stages, this illness is regarded as one of the "silent diseases" that go unnoticed. The fact that various datasets have varied retinal features is one of the significant difficulties in this field of study. This information impacts the models created for this purpose. This study's method can efficiently learn and classify DR from three diverse datasets. Four models based on transfer learning Convolution Neural Network (CNN)-Visual Geometry Group (VGG) 16, Inception version 3 (InceptionV3), Dense Network (DenseNet) 121, and Mobile Network version 2 (MobileNetV2)-are employed in this work, with evaluation parameters, including loss, accuracy, recall, precision, and specificity. The models are also tested by combining the images from the three datasets. The DenseNet121 model performs better with 98.97% accuracy on the combined image set. The study concludes that combining multiple datasets improves performance compared to individual datasets. The obtained model can be utilized globally to accommodate more tests that clinics perform for diabetic patients to prevent DR. It helps health workers refer patients to ophthalmologists before DR becomes serious.
引用
收藏
页数:17
相关论文
共 82 条
[1]  
Abadi M., 2016, PREPRINT, DOI DOI 10.48550/ARXIV.1603.04467
[2]   A survey on recent developments in diabetic retinopathy detection through integration of deep learning [J].
Agarwal, Shalini ;
Bhat, Aruna .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (11) :17321-17351
[3]  
Ahmad M, 2019, I S BIOMED IMAGING, P573, DOI [10.1109/ISBI.2019.8759417, 10.1109/isbi.2019.8759417]
[4]  
Al-Smadi M., 2021, Int. J. Electr. Comput. Eng. (IJECE), V11, P3492
[5]   Medical Image Analysis using Convolutional Neural Networks: A Review [J].
Anwar, Syed Muhammad ;
Majid, Muhammad ;
Qayyum, Adnan ;
Awais, Muhammad ;
Alnowami, Majdi ;
Khan, Muhammad Khurram .
JOURNAL OF MEDICAL SYSTEMS, 2018, 42 (11)
[6]  
APTOS, 2019, Blindness Detection
[7]   Deep learning based computer-aided diagnosis systems for diabetic retinopathy: A survey [J].
Asiri, Norah ;
Hussain, Muhammad ;
Al Adel, Fadwa ;
Alzaidi, Nazih .
ARTIFICIAL INTELLIGENCE IN MEDICINE, 2019, 99
[8]  
Bagadi Lavanya, 2022, Innovations in Electronics and Communication Engineering: Proceedings of the 9th ICIECE 2021. Lecture Notes in Networks and Systems (355), P253, DOI 10.1007/978-981-16-8512-5_28
[9]   AI-Based Automatic Detection and Classification of Diabetic Retinopathy Using U-Net and Deep Learning [J].
Bilal, Anas ;
Zhu, Liucun ;
Deng, Anan ;
Lu, Huihui ;
Wu, Ning .
SYMMETRY-BASEL, 2022, 14 (07)
[10]   A Transfer Learning and U-Net-based automatic detection of diabetic retinopathy from fundus images [J].
Bilal, Anas ;
Sun, Guangmin ;
Mazhar, Sarah ;
Imran, Azhar ;
Latif, Jahanzaib .
COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2022, 10 (06) :663-674