GBT-SAM: Adapting a Foundational Deep Learning Model for Generalizable Brain Tumor Segmentation via Efficient Integration of Multi-Parametric MRI Data
Journal:
arXiv
Published Date:
Mar 6, 2025
Abstract
Gliomas are aggressive brain tumors that require accurate imaging-based
diagnosis, with segmentation playing a critical role in evaluating morphology
and treatment decisions. Manual delineation of gliomas is time-consuming and
prone to variability, motivating the use of deep learning to improve
consistency and alleviate clinical workload. However, existing methods often
fail to fully exploit the information available in multi-parametric MRI
(mp-MRI), particularly inter-slice contextual features, and typically require
considerable computational resources while lacking robustness across tumor type
variations. We present GBT-SAM, a parameter-efficient deep learning framework
that adapts the Segment Anything Model (SAM), a large-scale vision model, to
volumetric mp-MRI data. GBT-SAM reduces input complexity by selecting fewer
than 2.6\% of slices per scan while incorporating all four MRI modalities,
preserving essential tumor-related information with minimal cost. Furthermore,
our model is trained by a two-step fine-tuning strategy that incorporates a
depth-aware module to capture inter-slice correlations and lightweight
adaptation layers, resulting in just 6.5M trainable parameters, which is the
lowest among SAM-based approaches. GBT-SAM achieves a Dice Score of 93.54 on
the BraTS Adult Glioma dataset and demonstrates robust performance on
Meningioma, Pediatric Glioma, and Sub-Saharan Glioma datasets. These results
highlight GBT-SAM's potential as a computationally efficient and domain-robust
framework for brain tumor segmentation using mp-MRI. Our code and models are
available at https://github.com/vpulab/med-sam-brain .