Deep convolutional neural networks for diabetic retinopathy detection by image classification

被引:257
作者
Wan, Shaohua [1 ]
Liang, Yan [1 ]
Zhang, Yin [1 ,2 ]
机构
[1] Zhongnan Univ Econ & Law, Sch Informat & Safety Engn, Wuhan 430073, Hubei, Peoples R China
[2] Hubei Key Lab Med Informat Anal & Tumor Diag & Tr, Wuhan 430074, Hubei, Peoples R China
关键词
Diabetic retinopathy; Fundus images classification; Convolutional neural networks; Transfer learning;
D O I
10.1016/j.compeleceng.2018.07.042
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Diabetic retinopathy (DR) is a common complication of diabetes and one of the major causes of blindness in the active population. Many of the complications of DR can be prevented by blood glucose control and timely treatment. Since the varieties and the complexities of DR, it is really difficult for DR detection in the time-consuming manual diagnosis. This paper is to attempt towards finding an automatic way to classify a given set of fundus images. We bring convolutional neural networks (CNNs) power to DR detection, which includes 3 major difficult challenges: classification, segmentation and detection. Coupled with transfer learning and hyper-parameter tuning, we adopt AlexNet, VggNet, GoogleNet, ResNet, and analyze how well these models do with the DR image classification. We employ publicly available Kaggle platform for training these models. The best classification accuracy is 95.68% and the results have demonstrated the better accuracy of CNNs and transfer learning on DR image classification. (C) 2018 Elsevier Ltd. All rights reserved.
引用
收藏
页码:274 / 282
页数:9
相关论文
共 25 条
[1]  
[Anonymous], IEEE ACCESS
[2]  
[Anonymous], PROC CVPR IEEE
[3]  
[Anonymous], 2015, Very Deep Convolu- tional Networks for Large-Scale Image Recognition
[4]  
[Anonymous], 2018, IEEE Access
[5]  
Azizpour H., 2015, CVPRW DEEPVISION WOR
[6]   A non-local algorithm for image denoising [J].
Buades, A ;
Coll, B ;
Morel, JM .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 2, PROCEEDINGS, 2005, :60-65
[7]  
Hariharan B, 2015, PROC CVPR IEEE, P447, DOI 10.1109/CVPR.2015.7298642
[8]  
Krizhevsky A., 2017, COMMUN ACM, V60, P84, DOI [DOI 10.1145/3065386, 10.1145/3065386]
[9]   Learning and Transferring Mid-Level Image Representations using Convolutional Neural Networks [J].
Oquab, Maxime ;
Bottou, Leon ;
Laptev, Ivan ;
Sivic, Josef .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :1717-1724
[10]  
Parkhi O. M., 2015, P BRIT MACH VIS C BM