BaT-SAM: Introducing Inductive Bias into SAM via a Mixture of Convolutional Experts for Brain Tumor Segmentation
Organizations: Mymensingh Medical College Mymensingh, Bangladesh · Department of LCSEE West Virginia University Morgantown, WV, USA · Department of CS Florida Polytechnic University Lakeland, Florida, USA · Department of CSE Manarat International University Dhaka, Bangladesh
Abstract
Accurate brain tumor segmentation from magnetic resonance imaging (MRI) is essential for diagnosis and treatment planning; however, manual delineation is labor-intensive and time-consuming. Existing approaches are either task-specific with limited generalization or rely on user-provided prompts during inference. We present BaT-SAM, a parameter-efficient adaptation framework that enables fully automatic brain tumor segmentation with the Segment Anything Model (SAM). BaT-SAM injects local inductive bias into SAM's frozen Vision Transformer (ViT) encoder via Conv-LoRA, where image features are processed by a sparsely gated mixture of multi-scale convolutional experts. This design compensates for the limited translation-equivariant inductive bias of vanilla ViTs while fine-tuning fewer than 5M parameters (less than 1% of the full model). To remove the need for user prompts, we further introduce a lightweight self-prompting box regressor that is jointly optimized with the adapter using scheduled sampling and predicts tumor locations directly from image embeddings. On the BRISC2025 benchmark, BaT-SAM achieves a weighted mean IoU of 0.832 while outperforming state-of-the-art benchmarks. It further demonstrates strong cross-dataset generalization, achieving Dice scores of 0.702 on Figshare and 0.784 on the LGG MRI Segmentation dataset.
Figures & tables
| Class | Train / Test | Area (%) | ( ) | ( ) |
|---|---|---|---|---|
| Glioma | 1147 / 254 | |||
| Meningioma | 1329 / 306 | |||
| Pituitary | 1457 / 300 | |||
| Total | 3933 / 860 | – | – | – |
| Generalization | Evaluation | Method | Base | Type | Dice | IoU |
| in-domain | Fine-tuning | Swin-HAFUNet [ 11 ] | UNet variant | BRISC2025 BRISC2025 | 0.8624 | 0.8240 |
| in-domain | Fine-tuning | BaT-SAM (ours) | SAM | BRISC2025 BRISC2025 | 0.8865 | 0.8325 |
| in-domain | Fine-tuning | Attention U-Net [ 29 ] | UNet variant | LGG LGG | 0.9016 | 0.8078 |
| in-domain | Fine-tuning | Res-UNet [ 14 ] | UNet variant | LGG LGG | 0.8530 | 0.7590 |
| in-domain | Fine-tuning | BaT-SAM (ours) | SAM | LGG LGG | 0.8561 | 0. 8090 |
| in-domain | Fine-tuning | Residual-UNet [ 3 ] | UNet variant | Figshare Figshare | 0.7369 | 0.6151 |
| Model | Glioma | Meningioma | Pituitary | mIoU |
|---|---|---|---|---|
| UNet [ 25 ] | 0.697 | 0.771 | 0.793 | 0.757 |
| UNet++ [ 32 ] | 0.717 | 0.742 | 0.797 | 0.753 |
| MANet [ 10 ] | 0.724 | 0.775 | 0.780 | 0.762 |
| LinkNet [ 6 ] | 0.717 | 0.748 | 0.790 | 0.753 |
| DeepLabV3+ [ 7 ] | 0.720 | 0.775 | 0.787 | 0.763 |
| PAN [ 18 ] | 0.720 | 0.745 | 0.807 | 0.759 |