Application Research on YOLOv5 Model Based on Lightweight Atrous Attention Module in Brain Tumor MRI Image Segmentation

Tao Yang¹Chen Jinghui¹Lianxin Xie¹Lanlan Yang¹Chengbin Ye²Hongjia Zhao^2*

• ¹Fujian University of Traditional Chinese Medicine, Fuzhou, China
• ²People's Hospital of Fujian Province, Fuzhou, China

[Abstract] Objective: To enhance the segmentation accuracy and computational efficiency of brain tumor MRI images, this study proposes a novel Lightweight Atrous Attention Module (LAAM) that integrates the Convolutional Block Attention Module (CBAM) with an Atrous Spatial Pyramid Pooling (ASPP) structure. The LAAM was integrated into the YOLOv5s model to enhance its performance, aiming to boost accuracy and recall while keeping computational efficiency. Methods: This study utilized two publicly available meningioma and glioma MRI datasets from Kaggle. The LAAM incorporates depthwise separable convolutions, dual attention mechanisms, and residual connections to reduce computational complexity while enhancing feature extraction capabilities. The modified YOLOv5s model was trained and validated via five-fold cross-validation, with performance comparisons conducted against the original YOLOv5s architecture and other optimized models. Results: The enhanced YOLOv5s-LAAM model demonstrated superior performance, achieving a precision of 92.3%, a recall rate of 90.4%, and an mAP@50 score of 0.925. Concurrently, the model exhibited significantly reduced computational demands, with the GFLOPs reduced by 15% compared to the original YOLOv5s-ASPP baseline. Conclusion: The integration of the LAAM significantly enhances the YOLOv5s model's segmentation capabilities for brain tumor MRI images, making it a valuable tool for clinical diagnosis and treatment planning. The lightweight design ensures effective deployment in resource-constrained environments while maintaining high computational performance.