Effective methods of diabetic retinopathy detection based on deep convolutional neural networks

被引:4
作者
Gu, Yunchao [1 ,3 ,4 ]
Wang, Xinliang [1 ]
Pan, Junjun [1 ,2 ]
Yong, Zhifan [1 ]
Guo, Shihui [5 ]
Pan, Tianze [1 ]
Jiao, Yonghong [6 ]
Zhou, Zhong [1 ]
机构
[1] Beihang Univ, State Key Lab Virtual Real Technol & Syst, Beijing 100191, Peoples R China
[2] Peng Cheng Lab, Shenzhen 518000, Peoples R China
[3] Beihang Univ, Hangzhou Innovat Res Inst, Hangzhou 100191, Peoples R China
[4] Beihang Univ, Beijing Adv Innovat Ctr Big Data & Brain Comp BDB, Beijing 100191, Peoples R China
[5] Xiamen Univ, Sch Informat, Xiamen 361005, Peoples R China
[6] Capital Med Univ, Beijing Tongren Hosp, Beijing Tongren Eye Ctr, Beijing 100730, Peoples R China
基金
国家重点研发计划; 北京市自然科学基金; 中国国家自然科学基金;
关键词
Diabetic retinopathy; Fundus image analysis; Deep learning; Convolutional neural networks;
D O I
10.1007/s11548-021-02498-8
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Purpose Diabetic retinopathy (DR) has become the leading cause of blindness worldwide. In clinical practice, the detection of DR often takes a lot of time and effort for ophthalmologist. It is necessary to develop an automatic assistant diagnosis method based on medical image analysis techniques. Methods Firstly, we design a feature enhanced attention module to capture focus lesions and regions. Secondly, we propose a stage sampling strategy to solve the problem of data imbalance on datasets and avoid the CNN ignoring the focus features of samples that account for small parts. Finally, we treat DR detection as a regression task to keep the gradual change characteristics of lesions and output the final classification results through the optimization method on the validation set. Results Extensive experiments are conducted on open-source datasets. Our methods achieve 0.851 quadratic weighted kappa which outperforms first place in the Kaggle DR detection competition based on the EyePACS dataset and get the accuracy of 0.914 in the referable/non-referable task and 0.913 in the normal/abnormal task based on the Messidor dataset. Conclusion In this paper, we propose three novel automatic DR detection methods based on deep convolutional neural networks. The results illustrate that our methods can obtain comparable performance compared with previous methods and generate visualization pictures with potential lesions for doctors and patients.
引用
收藏
页码:2177 / 2187
页数:11
相关论文
共 44 条
[1]   Using filter banks in Convolutional Neural Networks for texture classification [J].
Andrearczyk, Vincent ;
Whelan, Paulf. .
PATTERN RECOGNITION LETTERS, 2016, 84 :63-69
[2]   Speeded-Up Robust Features (SURF) [J].
Bay, Herbert ;
Ess, Andreas ;
Tuytelaars, Tinne ;
Van Gool, Luc .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2008, 110 (03) :346-359
[3]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[4]   A systematic study of the class imbalance problem in convolutional neural networks [J].
Buda, Mateusz ;
Maki, Atsuto ;
Mazurowski, Maciej A. .
NEURAL NETWORKS, 2018, 106 :249-259
[5]   SMOTE: Synthetic minority over-sampling technique [J].
Chawla, Nitesh V. ;
Bowyer, Kevin W. ;
Hall, Lawrence O. ;
Kegelmeyer, W. Philip .
2002, American Association for Artificial Intelligence (16)
[6]   FEEDBACK ON A PUBLICLY DISTRIBUTED IMAGE DATABASE: THE MESSIDOR DATABASE [J].
Decenciere, Etienne ;
Zhang, Xiwei ;
Cazuguel, Guy ;
Lay, Bruno ;
Cochener, Beatrice ;
Trone, Caroline ;
Gain, Philippe ;
Ordonez-Varela, John-Richard ;
Massin, Pascale ;
Erginay, Ali ;
Charton, Beatrice ;
Klein, Jean-Claude .
IMAGE ANALYSIS & STEREOLOGY, 2014, 33 (03) :231-234
[7]   Classification of Diabetic Retinopathy Images by Using Deep Learning Models [J].
Dutta, Suvajit ;
Manideep, Bonthala C. S. ;
Basha, Syed Muzamil ;
Caytiles, Ronnie D. ;
Iyengar, N. Ch. S. N. .
INTERNATIONAL JOURNAL OF GRID AND DISTRIBUTED COMPUTING, 2018, 11 (01) :89-106
[8]  
Freeman WT, 1995, INT WORKSH AUT FAC G, V12, P296
[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]   Bag of Tricks for Image Classification with Convolutional Neural Networks [J].
He, Tong ;
Zhang, Zhi ;
Zhang, Hang ;
Zhang, Zhongyue ;
Xie, Junyuan ;
Li, Mu .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :558-567