The impact of backbone selection in YOLOv8 models on brain tumor localization

Abstract

Abstract This study investigates the utilization of the You Only Look Once (YOLOv8) deep learning framework for accurately identifying the location of brain tumors in medical imaging. We investigate the effects of model size and pretraining on the accuracy and computational efficiency of tumor detection by utilizing different setups of the YOLOv8 model. These setups include various configurations, ranging from very small to large, and can be pretrained on the COCO dataset or not. The experimental results, carried out on Google Colaboratory using NVIDIA Tesla T4 GPUs, show that pretrained models often achieve better performance by utilizing the extensive feature representations learned from the COCO dataset, resulting in increased precision in tumor location. For instance, the YOLOv8-XS model pretrained on COCO achieves an IoU of 0.278 and a Dice coefficient of 0.435, whereas its non-pretrained counterpart attains only 0.241 IoU and 0.388 Dice, indicating a 15% improvement in tumor localization accuracy. Similarly, pretrained YOLOv8-L achieves 0.269 IoU, outperforming standard object detection models such as Mask R-CNN (IoU: 0.212) and Faster R-CNN (IoU: 0.228). These results highlight the impact of pretraining on model performance, particularly for lightweight architectures, while also revealing diminishing returns for larger models. The research uncovers a subtle connection between the complexity of the model, pretraining, and the time required for training. It emphasizes the possible advantages and constraints of pretraining for various sizes of models.

Topics & Concepts