Abstract:Background and objective Early detection and grading of Diabetic Retinopathy (DR) is essential to determine an adequate treatment and prevent severe vision loss. However, the manual analysis of fundus images is time consuming and DR screening programs are challenged by the availability of human graders. Current automatic approaches for DR grading attempt the joint detection of all signs at the same time. However, the classification can be optimized if red lesions and bright lesions are independently processed since the task gets divided and simplified. Furthermore, clinicians would greatly benefit from explainable artificial intelligence (XAI) to support the automatic model predictions, especially when the type of lesion is specified. As a novelty, we propose an end-to-end deep learning framework for automatic DR grading (5 severity degrees) based on separating the attention of the dark structures from the bright structures of the retina. As the main contribution, this approach allowed us to generate independent interpretable attention maps for red lesions, such as microaneurysms and hemorrhages, and bright lesions, such as hard exudates, while using image-level labels only. Methods Our approach is based on a novel attention mechanism which focuses separately on the dark and the bright structures of the retina by performing a previous image decomposition. This mechanism can be seen as a XAI approach which generates independent attention maps for red lesions and bright lesions. The framework includes an image quality assessment stage and deep learning-related techniques, such as data augmentation, transfer learning and fine-tuning. We used the architecture Xception as a feature extractor and the focal loss function to deal with data imbalance. Results The Kaggle DR detection dataset was used for method development and validation. The proposed approach achieved 83.7% accuracy and a Quadratic Weighted Kappa of 0.78 to classify DR among 5 severity degrees, which outperforms several state-of-the-art approaches. Nevertheless, the main result of this work is the generated attention maps, which reveal the pathological regions on the image distinguishing the red lesions and the bright lesions. These maps provide explainability to the model predictions. Conclusions Our results suggest that our framework is effective to automatically grade DR. The separate attention approach has proven useful for optimizing the classification. On top of that, the obtained attention maps facilitate visual interpretation for clinicians. Therefore, the proposed method could be a diagnostic aid for the early detection and grading of DR.

BAM: Block attention mechanism for OCT image classification

MHANet: A hybrid attention mechanism for retinal diseases classification

Interpretable Diabetic Retinopathy Diagnosis based on Biomarker Activation Map

Introducing Feature Attention Module on Convolutional Neural Network for Diabetic Retinopathy Detection

Diabetic Retinal Grading Using Attention-Based Bilinear Convolutional Neural Network and Complement Cross Entropy

CABNet: Category Attention Block for Imbalanced Diabetic Retinopathy Grading

DR-IIXRN : Detection Algorithm of Diabetic Retinopathy Based on Deep Ensemble Learning and Attention Mechanism

Potential of Attention Mechanism for Classification of Optical Coherence Tomography Images.

BiRA-Net: Bilinear Attention Net for Diabetic Retinopathy Grading

Multi-Scale Class Attention Network for Diabetes Retinopathy Grading

Attention-based Deep Learning Framework for Automatic Fundus Image Processing to Aid in Diabetic Retinopathy Grading

Attention-Driven Cascaded Network for Diabetic Retinopathy Grading from Fundus Images

Attention‐based deep learning system for automated diagnoses of age‐related macular degeneration in optical coherence tomography images

Composite deep neural network with gated-attention mechanism for diabetic retinopathy severity classification

A deep integrative approach for diabetic retinopathy classification with synergistic channel-spatial and self-attention mechanism

Retinal image assessment using bi-level adaptive morphological component analysis

Joint disease classification and lesion segmentation via one-stage attention-based convolutional neural network in OCT images

A novel approach for automatic classification of macular degeneration OCT images

A Framework for Identifying Diabetic Retinopathy Based on Anti-noise Detection and Attention-Based Fusion.

A Novel Dual Attention Approach for DNN Based Automated Diabetic Retinopathy Grading

Recognition of Diabetic Retinopathy Grades Based on Data Augmentation and Attention Mechanisms