arXiv:2502.03746cs.CVcs.LG2025-02被引 10

改进YOLOv8实现脑肿瘤精准定位,提升检测效率与精度。

Brain Tumor Identification using Improved YOLOv8

  • 用RT-DETR替代NMS,自动过滤重叠框,减少人工设定。
  • 引入鬼魂卷积与视觉变换器,降低计算开销并增强特征提取。
  • 在公开数据集上达0.91 [email protected],优于多个主流检测模型。

脑肿瘤范围识别是癌症治疗中的重大挑战,主要难点在于肿瘤大小的精确判定。磁共振成像(MRI)已成为关键诊断工具,但手动勾画肿瘤边界耗时且依赖经验。深度学习与计算机辅助检测技术为此带来显著进展。本文提出一种改进的YOLOv8模型,用于精准检测MRI图像中的脑肿瘤。首先,在检测头中以实时检测变压器(RT-DETR)替代非最大抑制(NMS)算法,消除人工预设组件;其次,用鬼魂卷积替换常规卷积模块,降低计算与内存消耗,支持实时部署;第三,在主干网络中引入视觉变换器块,增强上下文特征提取能力。实验使用公开脑肿瘤数据集,所提模型性能优于原始YOLOv8及其他主流目标检测器(如Faster R-CNN、Mask R-CNN、YOLOv3~v5、SSD、RetinaNet、EfficientDet、DETR),[email protected]达0.91。

原文摘要 · Abstract (English)

Identifying the extent of brain tumors is a significant challenge in brain cancer treatment. The main difficulty is in the approximate detection of tumor size. Magnetic resonance imaging (MRI) has become a critical diagnostic tool. However, manually detecting the boundaries of brain tumors from MRI scans is a labor-intensive task that requires extensive expertise. Deep learning and computer-aided detection techniques have led to notable advances in machine learning for this purpose. In this paper, we propose a modified You Only Look Once (YOLOv8) model to accurately detect the tumors within the MRI images. The proposed model replaced the Non-Maximum Suppression (NMS) algorithm with a Real-Time Detection Transformer (RT- DETR) in the detection head. NMS filters out redundant or overlapping bounding boxes in the detected tumors, but they are hand-designed and pre-set. RT-DETR removes hand-designed components. The second improvement was made by replacing the normal convolution block with ghost convolution. Ghost Convolution reduces computational and memory costs while maintaining high accuracy and enabling faster inference, making it ideal for resource-constrained environments and real-time applications. The third improvement was made by introducing a vision transformer block in the backbone of YOLOv8 to extract context-aware features. We used a publicly available dataset of brain tumors in the proposed model. The proposed model performed better than the original YOLOv8 model and also performed better than other object detectors (Faster R- CNN, Mask R-CNN, YOLO, YOLOv3, YOLOv4, YOLOv5, SSD, RetinaNet, EfficientDet, and DETR). The proposed model achieved 0.91 mAP (mean Average Precision)@0.5.

脑肿瘤YOLOv8医学图像目标检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。