Automated Detection of Diabetic Retinopathy Lesions in Ultra-Widefield fundus Images Using an Attention-Augmented YOLOv8 Framework

Lei-Si Hu¹Jie Wang²Ming-Heng Zhang³Hai-Yu Huang^3*

• ¹First Clinical Medical College, Chongqing Medical University, Chongqing, Chongqing, China
• ²Chengdu University of Traditional Chinese Medicine, Chengdu,Sichuan, China
• ³School of Computer and Artificial Intelligence, Southwest Jiaotong University, Chengdu,Sichuan, China

Objective: To enhance the automatic detection precision of diabetic retinopathy (DR) lesions, this study introduces an improved YOLOv8 model specifically designed for the precise identification of DR lesions. Method: This study integrated two attention mechanisms, convolutional Exponential Moving Average (convEMA) and convolutional Simple Attention Module (convSimAM), into the backbone of the YOLOv8 model. A dataset consisting of 3,388 ultra-widefield (UWF) fundus images, obtained from patients with DR, each with a resolution of 2600 × 2048 pixels, was utilized for both training and testing purposes. The performances of three models-YOLOv8, YOLOv8+ convEMA, and YOLOv8+convSimAM-were systematically compared. Results: A comparative analysis of the three models revealed that the original YOLOv8 model suffers from missing detection issues, achieving a precision of 0.815 for hemorrhage spot detection. YOLOv8+convEMA improved hemorrhage detection precision to 0.906, while YOLOv8+convSimAM achieved the highest value of 0.910, demonstrating the enhanced sensitivity of spatial attention. The proposed model also maintained comparable precision in detecting hard exudates while improving recall to 0.804. It demonstrated the best performance in detecting cotton wool spots and epiretinal membrane. Overall, the proposed method provides a fine-tuned model specialized in subtle lesion detection, providing an improved solution for DR lesion assessment. Conclusion: In this study, we proposed two attention-augmented YOLOv8 models-YOLOv8+convEMA and YOLOv8+convSimAM-for the automated detection of DR lesions in UWF fundus images. Both models outperformed the baseline YOLOv8 in terms of detection precision, average precision, and recall. Among them, YOLOv8+convSimAM achieved the most balanced and accurate results across multiple lesion types, demonstrating an enhanced capability to detect small, lowcontrast, and structurally complex features. These findings support the effectiveness of lightweight attention mechanisms in optimizing deep learning models for highprecision DR lesion detection.