AI-based object detection for brain tumour using YOLO model
- Muhammad Ramzan(corresponding author),
- University of Bedfordshire,
- ,
Abstract
Brain tumour detection in MRI is still hard because tumours look very different, good labels are limited, and clinics need fast results. The purpose of this study is to evaluate whether a lightweight Deep learning-based YOLOv8 object detection pipeline can deliver reliable brain tumour classification with coarse localisation when only image-level labels are available. Using the public Kaggle Brain Tumour MRI Dataset (glioma, meningioma, pituitary tumour, and no tumour), we convert classification labels into weak bounding boxes and fine-tune a YOLOv8n detector. We compare a baseline YOLOv8n model against an enhanced version that adds CBAM attention and an Inner GIoU-based IoU loss. Performance is reported using precision, recall, F1 score, and mAP at IoU 0.5 and 0.5 to 0.95, alongside GPU latency. The baseline achieves of 0.991 and F1 of 0.953. The CBAM plus IoU loss variant improves throughput (10.88 ms per image vs. 12.90 ms), but reduces accuracy metrics (of 0.974). Overall, the results show that Brain Tumour Object Detection with YOLOv8 under weak supervision can work as a fast slice-level classifier, while localisation-focused add-ons such as CBAM and IoU loss need tighter spatial labels to consistently improve accuracy.
Publication Information
Output type
Original language
EnglishPages from-to (Number of pages)
Pages 290-304 (15 pages)Publication milestones
- Published - 02/07/2026
Publication status
Publisher
Springer, Japan, India, Australia, Germany, United States, United Arab Emirates, Austria, Switzerland, Italy, China, United Kingdom, Netherlands, Brazil, France, SingaporePublication series
- Publication series name: Lecture Notes in Networks and Systems
ISSN (Print): 2367-3370
ISSN (Electronic): 2367-3389
Volume: 1951 LNNS
ISBN (Print)
9783032248091Publication IDs
- Scopus: 105047318673
Host publication title
Intelligent Computing - Proceedings of the 2026 Computing ConferenceHost publication editors
- Kohei Arai
- Pascal Lorenz
