ISSN 1004-4140
    CN 11-3017/P

    用于脑肿瘤检测的轻量级目标检测网络改进算法

    Improved Lightweight Object Detection Network for Brain Tumor Detection

    • 摘要: 目的:通过对轻量化实时目标检测模型YOLOv7-tiny进行优化改进,提出针对脑肿瘤检测任务的优化框架,从而提升对脑肿瘤影像的检测效率。方法:采用包含1000例脑肿瘤MRI影像的基准数据集上进行实验,在主干网络中引入InceptionNeXt模块,扩大感受野,通过多尺度特征融合增强模型对不同尺寸和形态肿瘤的表征能力;其次,引入解耦头结构,采用独立分支优化策略降低复杂背景的干扰,以提升针对小尺寸以及边界模糊肿瘤的定位精度,加速模型收敛,提升检测性能。结果:改进模型较基线YOLOv7-tiny实现了较为显著的性能提升,其平均检测精度mAP@0.5由85.6%提升至88.2%,mAP@0.5:0.95由63.8%提升至65.4%,分别实现了2.6%、1.6%的精度增益。实验表明,改进后的模型能够更准确地识别脑肿瘤,具有比较精准的检测性能。结论:改进后的模型与原模型相比,精度具有比较大的提升,验证了该模型在脑肿瘤精准检测方面的可行性。

       

      Abstract: Objective:This study aimed to enhance the detection efficiency of brain tumor images by optimizing the lightweight real-time object detection model YOLOv7-tiny, proposing an improved framework tailored for brain tumor detection tasks. Methods:Experiments were conducted using a benchmark dataset comprising 1,000 brain tumor images acquired using magnetic resonance imaging (MRI). The InceptionNeXt module was introduced into the backbone of the YOLO network to expand the receptive field. Multiscale feature fusion was employed to enhance the model’s ability to characterize tumors of varying sizes and shapes. Furthermore, a decoupled head structure was incorporated, adopting an independent branch optimization strategy to mitigate interference from complex backgrounds. This improved localization accuracy for small and boundary-blurred tumors, accelerated model convergence, and enhanced detection performance. Results:The improved model achieved a notable performance gain compared to the baseline YOLOv7-tiny. Mean average precision (mAP)@0.5 increased from 85.6% to 88.2%, and mAP@0.5:0.95 increased from 63.8% to 65.4%, corresponding to improvements of 2.6% and 1.6%, respectively. The experimental results demonstrate that the enhanced model can identify brain tumors more accurately and exhibits a relatively precise detection performance. Conclusion:The improved model was substantially more accurate than the original model, which validates its feasibility for accurate brain tumor detection.

       

    /

    返回文章
    返回