SegMoTE: Token-Level Mixture of Experts for Medical Image Segmentation
About
Medical image segmentation is vital for clinical diagnosis and quantitative analysis, yet remains challenging due to the heterogeneity of imaging modalities and the high cost of pixel-level annotations. Although general interactive segmentation models like SAM have achieved remarkable progress, their transfer to medical imaging still faces two key bottlenecks: (i) the lack of adaptive mechanisms for modality- and anatomy-specific tasks, which limits generalization in out-of-distribution medical scenarios; and (ii) current medical adaptation methods fine-tune on large, heterogeneous datasets without selection, leading to noisy supervision, higher cost, and negative transfer. To address these issues, we propose SegMoTE, an efficient and adaptive framework for medical image segmentation. SegMoTE preserves SAM's original prompt interface, efficient inference, and zero-shot generalization while introducing only a small number of learnable parameters to dynamically adapt across modalities and tasks. In addition, we design a progressive prompt tokenization mechanism that enables fully automatic segmentation, significantly reducing annotation dependence. Trained on MedSeg-HQ, a curated dataset less than 1% of existing large-scale datasets, SegMoTE achieves SOTA performance across diverse imaging modalities and anatomical tasks. It represents the first efficient, robust, and scalable adaptation of general segmentation models to the medical domain under extremely low annotation cost, advancing the practical deployment of foundation vision models in clinical applications.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Medical Image Segmentation | ISIC 2018 | Dice Score93.02 | 92 | |
| Medical Image Segmentation | SegThy | Dice83.91 | 7 | |
| Medical Image Segmentation | TotalSegmentator MRI | Dice: Adrenal gland67.05 | 7 | |
| Medical Image Segmentation | ISLES | Lesion Score0.773 | 7 | |
| Medical Image Segmentation | SegTHOR | Esophagus Score81.84 | 7 | |
| Medical Image Segmentation | AMOS CT | Dice Coefficient85.16 | 5 | |
| Medical Image Segmentation | AMOS MRI | Dice Coefficient0.8027 | 5 | |
| Medical Image Segmentation | BTCV | Dice Coefficient84.51 | 5 | |
| Medical Image Segmentation | CHAOS T1 | Dice Coefficient89 | 5 | |
| Medical Image Segmentation | SZ-CXR | Dice Coefficient95.04 | 5 |