Enhancing Diabetic Retinopathy Classification: A Fusion of ResNet50 with Attention Mechanism

Mohamed Touati,Laurent Nana,Faouzi Benzarti
DOI: https://doi.org/10.1109/CoDIT62066.2024.10708076
2024-07-01
Abstract:Diabetic Retinopathy (DR) is a critical health concern that, if diagnosed early, can significantly reduce the risk of blindness among individuals with diabetes. Leveraging the power of artificial intelligence, this paper introduces an innovative approach for DR detection using Deep Learning techniques. Our methodology employs Transfer Learning, adapting a modified ResNet50 model integrated with a self-attention mechanism. This novel strategy not only enhances detection efficiency but also reduces associated costs. The modified model is specifically tailored to address challenges stemming from unbalanced datasets, such as the APTOS 2019 Blindness Detection dataset. By incorporating a self-attention mechanism, the model can better focus on pertinent features, thereby improving classification and detection accuracy. Performance metrics demonstrate the efficacy of our approach, with a training accuracy of 98.24%, a test accuracy of 0.89, and a F1 Score of 0.94.
Computer Science,Medicine
What problem does this paper attempt to address?