RT info:eu-repo/semantics/article
T1 Attention-based deep learning framework for automatic fundus image processing to aid in diabetic retinopathy grading
A1 Romero Oraa, Roberto
A1 Herrero Tudela, María
A1 López Gálvez, María Isabel
A1 Hornero Sánchez, Roberto
A1 García Gadañón, María
K1 Diabetic retinopathy grading
K1 Fundus images
K1 Deep learning
K1 Attention mechanism
K1 Explainable artificial intelligence
AB 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.
PB Elsevier
SN 0169-2607
YR 2024
FD 2024
LK https://uvadoc.uva.es/handle/10324/72951
UL https://uvadoc.uva.es/handle/10324/72951
LA eng
NO Computer Methods and Programs in Biomedicine, 2024, vol. 249, 108160
NO Producción Científica
DS UVaDOC
RD 23-dic-2024