Review of the performance of diabetic retinopathy image classification using convolutional neural network models

被引:0
作者
Ibarra Belmonte, Isaul [1 ]
Jaramillo Avila, Uziel [2 ]
Cardona Reyes, Hector [2 ]
机构
[1] Ctr Invest Matemat, Ingeniero Software, Zacatecas, Zacatecas, Mexico
[2] Ctr Invest Matemat, Dept Ciencias Comp, Zacatecas, Zacatecas, Mexico
来源
2023 12TH INTERNATIONAL CONFERENCE ON SOFTWARE PROCESS IMPROVEMENT, CIMPS 2023 | 2023年
关键词
Diabetic Retinopathy; Image Classification; Image Capture; Convolutional Neural Networks; Nvidia Jetson Nano; Classification Algorithms; VISUAL IMPAIRMENT; PREVALENCE;
D O I
10.1109/CIMPS61323.2023.10528849
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
This study examines the critical importance of early detection in diabetic retinopathy (DR), a prevalent condition among people with diabetes that can lead to vision loss. It explores how convolutional neural networks (CNNs) can help identify key features in fundus images to detect early signs of DR. CNNs are powerful tools for processing unstructured data, especially images, skillfully decoding intricate data for DR detection. This article presents the architecture of CNNs, including input, convolutional, pooling, and fully connected layers, and examines two prevalent CNN architectures: Inceptionv3 [1][2] and ResNet34 [3][4][5]. These architectures were evaluated using two public diabetic retinopathy datasets, totaling 37,770 images at different stages of various resolutions and quality. Instances of ResNet34 and InceptionV3 models were created, each trained with both datasets. The results reveal that the accuracy of the ResNet34 model trained with dataset 1 ranges from 63.79% to 95.93%, while the ResNet34 model trained with dataset 2 performs between 73.04% and 75.37%. Regarding InceptionV3, its accuracy on dataset 1 varies from 53.77% to 74.44%, and on dataset 2, accuracy ranges from 71.76% to 77.61%. The results suggest that the ResNet34 model trained with dataset 2 is more robust and generalizable compared to other models, indicating its effectiveness in detecting diabetic retinopathy where image conditions vary.
引用
收藏
页码:161 / 170
页数:10
相关论文
共 21 条
[1]  
Balajiai, 2021, Diabetic Retinopathy Detection using PyTorch
[2]  
Chan W., 2022, Albert and Jakobiec's Principles and Practice of Ophthalmology, P2935, DOI [10.1007/978-3-030-42634-735, DOI 10.1007/978-3-030-42634-735]
[3]  
Congdon N, 2004, ARCH OPHTHALMOL-CHIC, V122, P477
[4]   Visual Impairment and Mental Health: Unmet Needs and Treatment Options [J].
Demmin, Docia L. ;
Silverstein, Steven M. .
CLINICAL OPHTHALMOLOGY, 2020, 14 :4229-4251
[5]  
Fanbyprinciple, 2021, pytorch diabetic retinopathy.
[6]  
Forouhi N.G., 2010, MEDICINE, V38, P602, DOI [DOI 10.1016/J.MPMED.2018.10.004, DOI 10.1016/J.MPMED.2014.09.007]
[7]   Automated Identification of Diabetic Retinopathy Using Deep Learning [J].
Gargeya, Rishab ;
Leng, Theodore .
OPHTHALMOLOGY, 2017, 124 (07) :962-969
[8]   Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs [J].
Gulshan, Varun ;
Peng, Lily ;
Coram, Marc ;
Stumpe, Martin C. ;
Wu, Derek ;
Narayanaswamy, Arunachalam ;
Venugopalan, Subhashini ;
Widner, Kasumi ;
Madams, Tom ;
Cuadros, Jorge ;
Kim, Ramasamy ;
Raman, Rajiv ;
Nelson, Philip C. ;
Mega, Jessica L. ;
Webster, R. .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2016, 316 (22) :2402-2410
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269