Efficient Brain Tumor Segmentation for MRI Images Using YOLO-BT

Abstract

Aiming at the problems of inaccurate segmentation and low detection efficiency caused by irregular tumor shape and large size differences in brain MRI images, this study proposes a brain tumor segmentation algorithm, YOLO-BT, based on YOLOv11. YOLO-BT uses UNetV2 as the backbone network to enhance the feature extraction ability of key regions through the attention mechanism. The BiFPN structure is introduced into the neck network to replace the traditional feature splicing method, realize the two-way fusion of cross-scale features, improve detection accuracy, and reduce the amount of calculations required. The D-LKA mechanism is introduced into the C3k2 structure, and the large convolution kernel is used to process complex image information to enhance the model’s ability to characterize different scales and irregular tumors. In this study, multiple sets of experiments were performed on the Figshare Brain Tumor dataset to test the performance of YOLO-BT. The data results show that YOLO-BT improves Precision by 2.7%, Recall, mAP50 by 0.9%, and mAP50-95 by 0.3% in the candidate box-based evaluation compared to YOLOv11. In mask-based evaluations, Precision improved by 2.5%, Recall by 2.8%, mAP50 by 1.1%, and mAP50-95 by 0.5%. At the same time, the mIOU increased by 6.1%, and the Dice coefficient increased by 3.6%. It can be seen that the YOLO-BT algorithm is suitable for brain tumor detection and segmentation.