Detection of Interpretable and Fine-Grained Brain Tumor Magnetic Resonance Imaging Based on Progressive Pruning: Machine Learning Model Development and Validation Study

基于渐进式剪枝的可解释且精细化脑肿瘤磁共振成像检测:机器学习模型开发与验证研究

阅读:1

Abstract

BACKGROUND: Brain tumor is one of the most malignant diseases of the central nervous system, and early accurate detection is of great significance for improving patient survival rate. However, the heterogeneity of brain tumors in terms of morphology, size, and location on magnetic resonance imaging (MRI) image, as well as their similarity to surrounding normal brain tissue, poses significant challenges for tumor detection. OBJECTIVE: This study aims to develop a high-performance brain tumor detection framework that integrates feature enhancement, channel attention, and progressive pruning, achieving an optimal balance between detection accuracy, model efficiency, and interpretability for slice-level MRI tumor localization tasks. METHODS: This paper proposes a convolution Prewitt-and-pooling-based preprocessing (CSPP) approach, based on the "you only look once" version 11 (YOLOv11) framework, which highlights important structural detail more effectively than traditional statistics. A dynamic convolution-based C3k2 (DCC) module was integrated to more efficiently capture both local and global features. A channel prior convolutional attention (CPCA) module was introduced before the detection head, enabling the network to specifically focus on information-rich channels and key spatial regions. Through a progressive hybrid pruning strategy (PHPS), the model was optimized for efficient inference. Furthermore, Eigen-class activation mapping (Eigen-CAM) was used to interpret the prediction result, making them more transparent. RESULTS: Extensive experiments on 3 brain tumor MRI datasets demonstrated the superior performance of CDCP-YOLO (CSPP-DCC-CPCA-PHPS-YOLO). On Br35H, the mean average precision (mAP) at an intersection-over-union (IoU) threshold of 0.5 (mAP0.5) increased by 2.6%, average mAP over several IoU thresholds (0.50-0.95; mAP0.5:0.95) increased by 5.9%, and number of floating-point operations (×10⁹; GFLOPs) decreased by 47.7%. On Roboflow, mAP0.5 increased by 19.5%, mAP0.5:0.95 increased by 7.7%, and GFLOPs decreased by 47.7%. On Capstone, mAP0.5 increased by 6.9%, mAP0.5:0.95 increased by 5.8%, and GFLOPs decreased by 47.7%. CONCLUSIONS: The proposed CDCP-YOLO framework achieves an optimal balance between accuracy, efficiency, and interpretability, providing a lightweight and reliable solution for slice-level brain tumor detection in MRI images.

特别声明

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

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

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

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