SAMRI: Segment Any MRI
About
Summary: SAMRI is an MRI-specialized adaptation of the Segment Anything Model achieving superior whole-body MRI segmentation, particularly for small and clinically critical structures, through box and point prompts for rapid annotation. Purpose: Existing SAM adaptations treat MRI as a generic modality, overlooking variable tissue contrast, intensity inhomogeneity, and clinically important small structures. We propose an MRI-specialized foundation model with strong whole-body segmentation and zero-shot generalization for direct use on any MRI annotation task. Methods: SAMRI fine-tunes only the mask decoder of SAM (ViT-B/16), keeping encoders frozen to preserve pretrained representations and eliminate redundant passes-reducing training time by 94%, trainable parameters by 96%, and FLOPs by ~99% versus full-model retraining. Training used 1.1 million 2D slice-mask pairs from 30 datasets spanning 47 targets, T1/T2/FLAIR/DWI contrasts, and whole-body anatomy, with focal-Dice loss and bounding-box (with optional point) prompts. Sizes were stratified by mask area (small: <0.5%; medium: 0.5-3.5%; large: >3.5%), and significance assessed by the Wilcoxon signed-rank test. Results: SAMRI with box+point prompts achieved mean DSC 0.87 +/- 0.11 across 47 targets, outperforming MedSAM (0.74 +/- 0.24) by 17.6% (p < 0.05), with largest gains for small (+42.4%) and medium (+26.9%) structures. On six zero-shot datasets, SAMRI achieved mean DSC 0.85, outperforming baselines. Inference requires only ~4.5 GB VRAM through an interactive interface on standard hardware. Conclusion: Decoder-only fine-tuning on a large, MRI-specific corpus delivers superior whole-body segmentation with strong zero-shot generalization, particularly for small and clinically salient structures. Public code, pretrained models, and an interactive interface make SAMRI deployable for MRI segmentation research and clinical workflows.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Medical Image Segmentation | MRI (47 anatomical and pathological targets) SAMRI Performance Evaluation (test) | Dice Score98 | 20 | |
| Medical Image Segmentation | PICAI | Dice0.96 | 19 | |
| Medical Image Segmentation | MSD Prostate | Dice Coefficient0.86 | 18 | |
| Medical Image Segmentation | CrossM | Dice Score89 | 12 | |
| Medical Image Segmentation | QUBIQ Kidney | BDice98 | 9 | |
| Medical Image Segmentation | CHAOS T1 | Dice Coefficient94 | 9 | |
| 3D Medical Image Segmentation | Prostate MR-T2 | DSC95 | 7 | |
| Medical Image Segmentation | MRI anatomical structures Small (<0.5% area) | DSC84 | 5 | |
| Medical Image Segmentation | MRI anatomical structures Medium (0.5–3.5% area) | DSC85 | 5 | |
| Medical Image Segmentation | MRI anatomical structures Large (>3.5% area) | DSC95 | 5 |