Geometry-aware Prototype Learning for Cross-domain Few-shot Medical Image Segmentation
About
Cross-domain few-shot medical image segmentation (CD-FSMIS) requires a model to generalise simultaneously to novel anatomical categories and unseen imaging domains from only a handful of annotated examples. Existing prototypical approaches inevitably entangle anatomical structure with domain-specific appearance variations, and thus lack a stable reference for reliable matching under domain shift. We observe that the geometric structure of human anatomy constitutes a reliable, domain-transferable prior that has been overlooked. Building on this insight, we propose GeoProto, a geometry-aware CD-FSMIS framework that enriches prototypical matching with explicit structural priors. The core component, Geometry-Aware Prototype Enrichment (GAPE), augments each local appearance prototype with a learned geometric offset encoding its ordinal position within the organ's interior topology. This offset is derived from an auxiliary Ordinal Shape Branch (OSB) trained under an ordinally consistent objective that enforces monotonic variation of geometric embeddings across interior strata, requiring no annotation beyond standard segmentation masks. Extensive experiments across seven datasets spanning three evaluation settings (cross-modality, cross-sequence, and cross-context) demonstrate that GeoProto achieves state-of-the-art performance.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Medical Image Segmentation | ISIC | DICE47.45 | 79 | |
| Cardiac Segmentation | Cardiac b-SSFP MRI | DSC (LV-BP)71.26 | 24 | |
| Medical Image Segmentation | Efficiency Analysis | Params (M)43.4 | 16 | |
| Medical Image Segmentation | CXR | DSC84.5 | 12 | |
| Medical Image Segmentation | MI-PMR | Bladder Segmentation Score76.43 | 12 | |
| Medical Image Segmentation | MICCAI Multi-sequence Cardiac MR b-SSFP → LGE 2019 | LV-BP Score74.91 | 12 | |
| Medical Image Segmentation | MICCAI Multi-sequence Cardiac MR LGE → b-SSFP 2019 | LV-BP Score85.45 | 12 | |
| Abdominal Organ Segmentation | Abdominal CT MICCAI 2015 | Liver DSC69.65 | 12 | |
| Abdominal Organ Segmentation | Abdominal MRI ISBI 2019 | Liver DSC62.31 | 12 | |
| Few-Shot Medical Image Segmentation | Abdominal CT | Liver Score71.49 | 12 |