vesselFM: A Foundation Model for Universal 3D Blood Vessel Segmentation
About
Segmenting 3D blood vessels is a critical yet challenging task in medical image analysis. This is due to significant imaging modality-specific variations in artifacts, vascular patterns and scales, signal-to-noise ratios, and background tissues. These variations, along with domain gaps arising from varying imaging protocols, limit the generalization of existing supervised learning-based methods, requiring tedious voxel-level annotations for each dataset separately. While foundation models promise to alleviate this limitation, they typically fail to generalize to the task of blood vessel segmentation, posing a unique, complex problem. In this work, we present vesselFM, a foundation model designed specifically for the broad task of 3D blood vessel segmentation. Unlike previous models, vesselFM can effortlessly generalize to unseen domains. To achieve zero-shot generalization, we train vesselFM on three heterogeneous data sources: a large, curated annotated dataset, data generated by a domain randomization scheme, and data sampled from a flow matching-based generative model. Extensive evaluations show that vesselFM outperforms state-of-the-art medical image segmentation foundation models across four (pre-)clinically relevant imaging modalities in zero-, one-, and few-shot scenarios, therefore providing a universal solution for 3D blood vessel segmentation.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Blood Vessel Segmentation | MSD8 | Dice4.50e+3 | 12 | |
| Vessel segmentation | Fe-MRA (test) | Dice60.31 | 8 | |
| Vessel segmentation | AVT (test) | Dice0.2819 | 8 | |
| Artery Segmentation | DynaVessel1 | mDC41.4 | 4 | |
| Artery Segmentation | DynaVessel+ | mDC0.226 | 4 | |
| Artery Segmentation | TopBrain | mDC0.374 | 4 | |
| Binary vessel segmentation | ISLES (test) | mDC74.4 | 4 | |
| Binary vessel segmentation | VesselVerse (test) | mDC (Mean Dice Coefficient)32 | 4 | |
| Vein Segmentation | TopBrain | mDC0.083 | 4 | |
| Vein Segmentation | DynaVessel1 | mDC0.036 | 4 |