Geometry-aware Prototype Learning for Cross-domain Few-shot Medical Image Segmentation

• URL: http://arxiv.org/abs/2605.10885v1
• Date: Mon, 11 May 2026 17:32:26 GMT
• Title: Geometry-aware Prototype Learning for Cross-domain Few-shot Medical Image Segmentation
• Abstract summary: Cross-domain few-shot medical image segmentation (CD-FSMIS) requires a model to generalise simultaneously to novel anatomical categories and unseen imaging domains.<n>Existing prototypical approaches inevitably entangle anatomical structure with domain-specific appearance variations.<n>We propose GeoProto, a geometry-aware CD-FSMIS framework that enriches matching with explicit structural priors.
• Score: 11.12183614587322
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: 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 papers

• PSP: Harnessing Position and Shape Priors for Cross-Domain Few-Shot Medical Image Segmentation [12.86282390925016]
  Few-Shot Medical Image (FSMIS) offers a powerful solution to data scarcity but struggles to generalize across different imaging modalities.<n>This performance collapse stems primarily from the drastic texture discrepancies between domains, which mislead models trained on source-specific intensity distributions.<n>We propose a novel framework that harnesses Position and Shape Priors (PSP) for cross-domain FSMIS.
  arXiv  Detail & Related papers  (2026-06-27T08:10:12Z)
• SCAPO: Self-Supervised Category-Level Articulated Pose Estimation from a Single 3D Observation [54.98864436812187]
  We propose a self-supervised framework that estimates canonical geometry, rigid part segmentation, and joint pivots, axes, and articulation states from a single RGB-D observation.<n> Experiments on synthetic and real articulated-object datasets show that our SCAPO recovers consistent part structure and accurate articulation parameters and outperforms all self-supervised baselines.
  arXiv  Detail & Related papers  (2026-06-01T09:04:49Z)
• Geometry-Guided Self-Supervision for Ultra-Fine-Grained Recognition with Limited Data [73.20753799870282]
  This paper investigates the intrinsic geometrical features of highly similar objects.<n>It introduces a general self-supervised framework called the Geometric Attribute Exploration Network (GAEor)<n>GAEor generates geometric attributes as novel alternative recognition cues.
  arXiv  Detail & Related papers  (2026-04-21T11:27:33Z)
• GeoReFormer: Geometry-Aware Refinement for Lane Segment Detection and Topology Reasoning [0.7769712273444055]
  GeoReFormer is a query-based architecture that embeds geometry- and topology-aware inductive biases directly within the transformer decoder.<n>GeoReFormer achieves state-of-the-art performance with 34.5% mAP on the OpenLane-V2 benchmark.
  arXiv  Detail & Related papers  (2026-03-27T02:26:12Z)
• Contrastive Graph Modeling for Cross-Domain Few-Shot Medical Image Segmentation [58.41482540044918]
  Cross-domain few-shot medical image segmentation (CD-FSMIS) offers a promising and data-efficient solution for medical applications.<n>We present Contrastive Graph Modeling (C-Graph), a framework that leverages the structural consistency of medical images as a reliable domain-transferable prior.
  arXiv  Detail & Related papers  (2025-12-25T14:00:17Z)
• AGENet: Adaptive Edge-aware Geodesic Distance Learning for Few-Shot Medical Image Segmentation [0.0]
  AGENet is a novel framework that incorporates spatial relationships through edge-aware geodesic distance learning.<n>Our key insight is that medical structures follow predictable geometric patterns that can guide prototype extraction.<n>Our method reduces boundary errors compared to existing approaches while maintaining computational efficiency.
  arXiv  Detail & Related papers  (2025-11-11T09:56:35Z)
• Geometry-Editable and Appearance-Preserving Object Compositon [67.98806888489385]
  General object composition (GOC) aims to seamlessly integrate a target object into a background scene with desired geometric properties.<n>Recent approaches derive semantic embeddings and integrate them into advanced diffusion models to enable geometry-editable generation.<n>We introduce a Disentangled Geometry-editable and Appearance-preserving Diffusion model that first leverages semantic embeddings to implicitly capture desired geometric transformations.
  arXiv  Detail & Related papers  (2025-05-27T09:05:28Z)
• Teaching AI the Anatomy Behind the Scan: Addressing Anatomical Flaws in Medical Image Segmentation with Learnable Prior [34.54360931760496]
  Key anatomical features, such as the number of organs, their shapes and relative positions, are crucial for building a robust multi-organ segmentation model. We introduce a novel architecture called the Anatomy-Informed Network (AIC-Net) AIC-Net incorporates a learnable input termed "Anatomical Prior", which can be adapted to patient-specific anatomy.
  arXiv  Detail & Related papers  (2024-03-27T10:46:24Z)
• SMC-UDA: Structure-Modal Constraint for Unsupervised Cross-Domain Renal Segmentation [100.86339246424541]
  We propose a novel Structure-Modal Constrained (SMC) UDA framework based on a discriminative paradigm and introduce edge structure as a bridge between domains. With the structure-constrained self-learning and progressive ROI, our methods segment the kidney by locating the 3D spatial structure of the edge. experiments show that our proposed SMC-UDA has a strong generalization and outperforms generative UDA methods.
  arXiv  Detail & Related papers  (2023-06-14T02:57:23Z)
• A Learning-based Framework for Topology-Preserving Segmentation using Quasiconformal Mappings [3.4798343542796593]
  We propose a deformation-based model that can extract objects in an image while maintaining their topological properties. This network generates segmentation masks that have the same topology as the template mask, even when trained with limited data.
  arXiv  Detail & Related papers  (2022-10-07T03:13:35Z)
• Contrastive Semi-supervised Learning for Domain Adaptive Segmentation Across Similar Anatomical Structures [21.54339967787734]
  We propose Contrastive Semi-supervised learning for Cross Anatomy Domain Adaptation. It adapts a model to segment similar structures in a target domain. It requires only limited annotations in the target domain.
  arXiv  Detail & Related papers  (2022-08-18T02:54:04Z)
• Two-Stream Graph Convolutional Network for Intra-oral Scanner Image Segmentation [133.02190910009384]
  We propose a two-stream graph convolutional network (i.e., TSGCN) to handle inter-view confusion between different raw attributes. Our TSGCN significantly outperforms state-of-the-art methods in 3D tooth (surface) segmentation.
  arXiv  Detail & Related papers  (2022-04-19T10:41:09Z)
• Shape-aware Meta-learning for Generalizing Prostate MRI Segmentation to Unseen Domains [68.73614619875814]
  We present a novel shape-aware meta-learning scheme to improve the model generalization in prostate MRI segmentation. Experimental results show that our approach outperforms many state-of-the-art generalization methods consistently across all six settings of unseen domains.
  arXiv  Detail & Related papers  (2020-07-04T07:56:02Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.