XCapsNet: A deep neural network for automated detection of diabetic retinopathy

被引:4
作者
Gour, Mahesh [1 ,2 ]
Jain, Sweta [1 ]
Kaushal, Sushant [1 ]
机构
[1] Maulana Azad Natl Inst Technol, Bhopal, India
[2] Maulana Azad Natl Inst Technol, Bhopal, MP, India
关键词
Capsule network; CLAHE; deep learning; diabetic retinopathy; fundus image; Xception model; IMAGES;
D O I
10.1002/ima.22842
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Diabetic retinopathy (DR) is a retinal condition in which the blood vessels of the retina are damaged. If DR is not detected in its early stage, it can lead to visual impairment and even blindness. For DR diagnosis, ophthalmologists manually examine the condition of the retina using fundus images. This manual way of DR detection is cumbersome, error-prone, and time-consuming, and it requires skilled ophthalmologists. To address these issues, we propose a deep learning-based hybrid model named XCapsNet, which combines the discriminative ability of the Xception and Capsule networks for automated detection of DR from the fundus images. In the proposed approach, we applied the CLAHE image preprocessing technique to improve the discriminative information in fundus images by enhancing the contrast of the images. Furthermore, in the proposed XCapNet model, we utilized the pretrained Xception and CapsNet models to learn the discriminative and hierarchically deep features of the DR fundus images at multiple labels of abstraction for diagnosing the DR disease. The performance of the proposed method is evaluated on the two publicly available APTOS2019 and Messidor datasets. The proposed method achieved a classification accuracy of 83.06% and 98.91% on the APTOS2019 dataset for multiclass and binary-class classification of DR images, respectively. On the Messidor dataset, the proposed approach achieved an accuracy of 98.33% for the classification of fundus images into DR and Normal classes. Additionally, we have also investigated the performance of different pretrained CNN models for DR detection. The proposed method shows its superiority over the existing methods.
引用
收藏
页码:1014 / 1027
页数:14
相关论文
共 39 条
[1]   Improved Automated Detection of Diabetic Retinopathy on a Publicly Available Dataset Through Integration of Deep Learning [J].
Abramoff, Michael David ;
Lou, Yiyue ;
Erginay, Ali ;
Clarida, Warren ;
Amelon, Ryan ;
Folk, James C. ;
Niemeijer, Meindert .
INVESTIGATIVE OPHTHALMOLOGY & VISUAL SCIENCE, 2016, 57 (13) :5200-5206
[2]  
[Anonymous], 2019, Diabetes mellitus
[3]  
APTOS, APT 2019 BLINDN DET
[4]   Efficient multi-kernel multi-instance learning using weakly supervised and imbalanced data for diabetic retinopathy diagnosis [J].
Cao, Peng ;
Ren, Fulong ;
Wan, Chao ;
Yang, Jinzhu ;
Zaiane, Osmar .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2018, 69 :112-124
[5]   Multi-categorical deep learning neural network to classify retinal images: A pilot study employing small database [J].
Choi, Joon Yul ;
Yoo, Tae Keun ;
Seo, Jeong Gi ;
Kwak, Jiyong ;
Um, Terry Taewoong ;
Rim, Tyler Hyungtaek .
PLOS ONE, 2017, 12 (11)
[6]  
CHOLLET F, 2017, PROC CVPR IEEE, P1800, DOI [DOI 10.1109/CVPR.2017.195, 10.1109/CVPR.2017.195]
[7]   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
[8]   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
[9]  
Ghosh R, 2017, 2017 4TH INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING AND INTEGRATED NETWORKS (SPIN), P550, DOI 10.1109/SPIN.2017.8050011
[10]  
Gour M., 2019, P INT C COMP VIS IM, P243