[论文解读] BGF-YOLO: Enhanced YOLOv8 with Multiscale Attentional Feature Fusion for Brain Tumor Detection

Ming Kang, Chee‐Ming Ting+2 more|arXiv (Cornell University)|Sep 22, 2023

Brain Tumor Detection and Classification被引用 7

一句话总结

tldr: BGF-YOLO 在 YOLOv8 的基础上加入双层路由注意力、广义特征金字塔以及第四检测头，以提升脑肿瘤检测，在 Br35H 上相比 YOLOv8x 取得绝对 mAP50 提高 4.7%。

ABSTRACT

You Only Look Once (YOLO)-based object detectors have shown remarkable accuracy for automated brain tumor detection. In this paper, we develop a novel BGF-YOLO architecture by incorporating Bi-level routing attention, Generalized feature pyramid networks, and Fourth detecting head into YOLOv8. BGF-YOLO contains an attention mechanism to focus more on important features, and feature pyramid networks to enrich feature representation by merging high-level semantic features with spatial details. Furthermore, we investigate the effect of different attention mechanisms and feature fusions, detection head architectures on brain tumor detection accuracy. Experimental results show that BGF-YOLO gives a 4.7% absolute increase of mAP$_{50}$ compared to YOLOv8x, and achieves state-of-the-art on the brain tumor detection dataset Br35H. The code is available at https://github.com/mkang315/BGF-YOLO.

研究动机与目标

• Motivate improved automatic brain tumor detection using advanced YOLO architectures.
• Investigate the impact of attention mechanisms and feature fusion on detection accuracy.
• Evaluate the benefits of a redesigned detection head within the YOLOv8 framework.

提出的方法

• Incorporates Bi-level routing attention to emphasize important features.
• Uses Generalized feature pyramid networks to merge high-level semantics with spatial details.
• Introduces a Fourth detecting head to augment detection capability.
• Evaluates different attention mechanisms and feature fusion strategies within YOLOv8-based architecture.
• Assesses detection performance on brain tumor datasets, notably Br35H.

实验结果

研究问题

• RQ1 Does Bi-level routing attention improve brain tumor detection accuracy in YOLOv8-based architectures?
• RQ2 How do generalized feature pyramid networks affect feature representation for brain tumor detection?
• RQ3 What is the impact of adding a fourth detection head on detection performance?
• RQ4 Which attention mechanisms and feature fusion schemes yield the best mAP on Br35H?

主要发现

• BG F-YOLO achieves a 4.7% absolute increase of mAP50 over YOLOv8x.
• The architecture attains state-of-the-art performance on the Br35H brain tumor dataset.
• The study analyzes the effects of different attention mechanisms and feature fusions on detection accuracy.