Deep convolutional neural networks for diabetic retinopathy detection by image classification

被引:231
|
作者
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
相关论文
共 50 条
  • [21] Diabetic Retinopathy Detection Based on Deep Convolutional Neural Networks for Localization of Discriminative Regions
    Pan, Junjun
    Yong, Zhifan
    Sui, Dong
    Qin, Hong
    [J]. 2018 8TH INTERNATIONAL CONFERENCE ON VIRTUAL REALITY AND VISUALIZATION (ICVRV), 2018, : 46 - 52
  • [22] Improving the Curvelet Saliency and Deep Convolutional Neural Networks for Diabetic Retinopathy Classification in Fundus Images
    Vo Thi Hong Tuyet
    Nguyen Thanh Binh
    Tin, Dang Thanh
    [J]. ENGINEERING TECHNOLOGY & APPLIED SCIENCE RESEARCH, 2022, 12 (01) : 8204 - 8209
  • [23] Enhancing Diabetic Retinopathy Detection Accuracy with Convolutional Neural Networks
    Kumari, H. M. L. S.
    Walgampaya, C. K.
    [J]. ENGINEER-JOURNAL OF THE INSTITUTION OF ENGINEERS SRI LANKA, 2024, 57 (03): : 45 - 59
  • [24] Histopathological Image Classification with Deep Convolutional Neural Networks
    Alom, Md Zahangir
    Aspiras, Theus
    Taha, Tarek M.
    Asari, Vijayan K.
    [J]. APPLICATIONS OF MACHINE LEARNING, 2019, 11139
  • [25] Convolutional Neural Networks for Diabetic Retinopathy
    Pratt, Harry
    Coenen, Frans
    Broadbent, Deborah M.
    Harding, Simon P.
    Zheng, Yalin
    [J]. 20TH CONFERENCE ON MEDICAL IMAGE UNDERSTANDING AND ANALYSIS (MIUA 2016), 2016, 90 : 200 - 205
  • [26] Evolving Deep Convolutional Neural Networks for Image Classification
    Sun, Yanan
    Xue, Bing
    Zhang, Mengjie
    Yen, Gary G.
    [J]. IEEE TRANSACTIONS ON EVOLUTIONARY COMPUTATION, 2020, 24 (02) : 394 - 407
  • [27] Deep Convolutional Neural Networks for Hyperspectral Image Classification
    Hu, Wei
    Huang, Yangyu
    Wei, Li
    Zhang, Fan
    Li, Hengchao
    [J]. JOURNAL OF SENSORS, 2015, 2015
  • [28] Hierarchical Pruning for Simplification of Convolutional Neural Networks in Diabetic Retinopathy Classification
    Hajabdollahi, Mohsen
    Esfandiarpoor, Reza
    Najarian, Kayvan
    Karimi, Nader
    Samavi, Shadrokh
    Soroushmehr, S. M. Reza
    [J]. 2019 41ST ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY (EMBC), 2019, : 970 - 973
  • [29] Deep Learning Solution for Diabetic Retinopathy Diagnosis based on Convolutional Neural Networks and Image Processing Algorithms
    Ion, Tomozei Cosmin
    Elena, Nechita
    Lazar, Dorian
    [J]. 2021 INTERNATIONAL CONFERENCE ON E-HEALTH AND BIOENGINEERING (EHB 2021), 9TH EDITION, 2021,
  • [30] Early Detection of Diabetic Retinopathy Using Deep Convolutional Neural Network
    Kannan, Rajeswari
    Vispute, S. R.
    Kharat, Reena
    Salunkhe, Dipti
    Vivekanandan, N.
    [J]. COMMUNICATIONS IN MATHEMATICS AND APPLICATIONS, 2023, 14 (03): : 1283 - 1292