An interpretable dual attention network for diabetic retinopathy grading: IDANet

被引:9
作者
Bhati, Amit [1 ]
Gour, Neha [2 ]
Khanna, Pritee [1 ]
Ojha, Aparajita [1 ]
Werghi, Naoufel [2 ]
机构
[1] PDPM Indian Inst Informat Technol Design & Mfg, Jabalpur 482005, India
[2] Khalifa Univ, Dept Elect Engn & Comp Sci, Abu Dhabi, U Arab Emirates
关键词
Diabetic retinopathy; Dual attention network; Bi-directional spatial attention (BSA); Channel-wise parallel attention (CPA); Model interpretability; AUTOMATED DETECTION; VALIDATION; IMAGES; SYSTEM;
D O I
10.1016/j.artmed.2024.102782
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Diabetic retinopathy (DR) is the most prevalent cause of visual impairment in adults worldwide. Typically, patients with DR do not show symptoms until later stages, by which time it may be too late to receive effective treatment. DR Grading is challenging because of the small size and variation in lesion patterns. The key to fine-grained DR grading is to discover more discriminating elements such as cotton wool, hard exudates, hemorrhages, microaneurysms etc. Although deep learning models like convolutional neural networks (CNN) seem ideal for the automated detection of abnormalities in advanced clinical imaging, small-size lesions are very hard to distinguish by using traditional networks. This work proposes a bi-directional spatial and channelwise parallel attention based network to learn discriminative features for diabetic retinopathy grading. The proposed attention block plugged with a backbone network helps to extract features specific to fine-grained DR-grading. This scheme boosts classification performance along with the detection of small-sized lesion parts. Extensive experiments are performed on four widely used benchmark datasets for DR grading, and performance is evaluated on different quality metrics. Also, for model interpretability, activation maps are generated using the LIME method to visualize the predicted lesion parts. In comparison with state-of-the-art methods, the proposed IDANet exhibits better performance for DR grading and lesion detection.
引用
收藏
页数:11
相关论文
共 41 条
[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]   Computer-based detection of diabetes retinopathy stages using digital fundus images [J].
Acharya, U. R. ;
Lim, C. M. ;
Ng, E. Y. K. ;
Chee, C. ;
Tamura, T. .
PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART H-JOURNAL OF ENGINEERING IN MEDICINE, 2009, 223 (H5) :545-553
[3]   A General Survey on Attention Mechanisms in Deep Learning [J].
Brauwers, Gianni ;
Frasincar, Flavius .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (04) :3279-3298
[4]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[5]   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
[6]   On generating trustworthy counterfactual explanations [J].
Del Ser, Javier ;
Barredo-Arrieta, Alejandro ;
Diaz-Rodriguez, Natalia ;
Herrera, Francisco ;
Saranti, Anna ;
Holzinger, Andreas .
INFORMATION SCIENCES, 2024, 655
[7]  
Howard AG, 2017, Arxiv, DOI [arXiv:1704.04861, 10.48550/arXiv.1704.04861]
[8]   Automated Identification of Diabetic Retinopathy Using Deep Learning [J].
Gargeya, Rishab ;
Leng, Theodore .
OPHTHALMOLOGY, 2017, 124 (07) :962-969
[9]   Challenges for ocular disease identification in the era of artificial intelligence [J].
Gour, Neha ;
Tanveer, M. ;
Khanna, Pritee .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (31) :22887-22909
[10]   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