Application research on YOLOv5 model based on Lightweight Atrous Attention Module in brain tumor MRI image segmentation

基于轻量级空洞注意力模块的YOLOv5模型在脑肿瘤MRI图像分割中的应用研究

阅读:1

Abstract

OBJECTIVE: To enhance the segmentation accuracy and computational efficiency of brain tumor magnetic resonance imaging (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.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。