Automated Brain Tumor Segmentation and Classification in MRI Using YOLO-Based Deep Learning

Abstract

Recent advancements in image processing and computer vision have brought significant transformations in healthcare technology, leading to significant improvements in diagnosis accuracy, costeffectiveness, and time efficiency. Magnetic Resonance Imaging (MRI) is employed by the radiologist for its remarkable ability to detect even the most subtle brain abnormalities. This study considers a comprehensive analysis of the two prominent object identification frameworks, YOLOv5 and YOLOv7, leveraging state-ofthe-art deep learning architectures to classify and detect brain cancers within MRI. The brain tumor dataset encompasses three distinct classes, including meningiomas, gliomas and pituitary tumors. To ensure precise segmentation of the tumor regions, the preprocessing phase incorporates advanced mask alignment techniques. This preprocessed dataset has been used to evaluate the performance of the deep learning models for brain tumor detection and classification. From the numerical results of YOLOv5, it was noticed that a recall score of 0.905 for box detection and 0.906 for mask segmentation, with a precision score of 0.94 and 0.936, respectively. At an IoU threshold of 0.5, both box detection and mask segmentation achieve a mAP of 0.947, whereas, at an IoU threshold of 0.5 to 0.95, they achieve mAPs of 0.666 and 0.657, respectively. In comparison, YOLOv7 exhibits strong performance with box detection accuracy of 0.936 and a mask segmentation accuracy of 0.935. The recall score are 0.904 for box detection and mask segmentation is 0.903. Notably, the mAP result at the IoU threshold of 0.5 are 0.94 for box detection and mask segmentation is 0.941. Over the broader IoU spectrum of 0.5 to 0.95, the mAP was 0.677 for box detection and 0.659 for mask segmentation. To underscore the novelty of the approach, the performance of the proposed framework is systematically compared with established methods such as RCNN, Faster RCNN, and Mask RCNN.INDEX TERMS brain tumor, deep learning, image processing, MRI, YOLO