An explainable deep-learning model reveals clinical clues in diabetic retinopathy through SHAP

作者：María Herrero-Tudela, Roberto Romero-Oraá, Roberto Hornero, Gonzalo C. Gutiérrez‐Tobal, María Isabel López Gálvez, María García · 发表于：Biomedical Signal Processing and Control · 年份：2024 · DOI：10.1016/j.bspc.2024.107328 · 被引用次数：35 · 研究领域：Retinal Imaging and Analysis、Artificial Intelligence in Healthcare、Machine Learning in Healthcare

Diabetic retinopathy (DR) is one of the leading causes of blindness globally. Several studies indicate that 90% of cases are preventable through early detection and appropriate treatment. Due to the increasing number of diabetic patients, the number of images that ophthalmologists have to manually analyze is becoming unaffordable. In this study, we propose a robust method for the automatic grading of the DR, while emphasizing the importance of providing visual explanations. The proposed method leans on a modified layer architecture of the ResNet-50 network. It also includes additional techniques such as data augmentation, regularization, early stopping criteria, transfer learning, and fine-tuning. In addition, in order to assist in the interpretation of the results of the deep-learning model, we introduce a visual Explainable Artificial Intelligence approach using SHapley Additive exPlanations (SHAP). We evaluated the effectiveness of our method using five publicly available databases of retinal images: APTOS-2019, EyePACS, DDR, IDRiD, and SUSTech-SYSU, achieving accuracy rates of 94.64%, 86.36%, 84.23%, 82.79%, and 85.65%, respectively. Notably, SHAP analysis revealed insights into our results, suggesting that retinal vasculature changes are potential DR risk indicators. We also found that peripheral retinal observations proved crucial in predicting DR progression, with initial lesions often found there. Moreover, this work overcomes the challenges of a highly imbalanced dat...