Search papers, labs, and topics across Lattice.
This paper introduces Variation-Conditioned Distributional Proxy Learning (VCDP), a novel training-only regularization module designed to enhance semi-supervised 3D medical image segmentation by better organizing feature spaces for anatomically complex structures. By representing each class with a learnable Gaussian distribution and utilizing multiple variation prototypes, VCDP effectively addresses the challenges posed by small organs and ambiguous boundaries with significant intra-class variations. Experimental results demonstrate that VCDP significantly outperforms existing methods, particularly in the segmentation of small and highly variable organs, without incurring additional inference costs.
VCDP transforms semi-supervised medical image segmentation by aligning voxel embeddings with both global organ identities and local anatomical variations, leading to superior performance on challenging segmentation tasks.
Semi-supervised 3D medical image segmentation reduces the need for dense voxel-level annotations by exploiting unlabeled volumes. Although existing methods such as consistency regularization, pseudo-labeling, and co-training improve prediction-level robustness, they often provide insufficient feature-space organization for anatomically complex structures, especially small organs and ambiguous boundary regions with large intra-class variations. To address this issue, we propose Variation-Conditioned Distributional Proxy Learning (VCDP), a plug-and-play training-only regularization module for semi-supervised 3D medical image segmentation. VCDP represents each class with a learnable Gaussian distribution for shared class semantics and multiple variation prototypes for fine-grained intra-class patterns. A unified variation-conditioned compatibility score is further formulated to fuse distributional similarity and soft variation aggregation, guiding voxel embeddings to align with both global organ identity and local anatomical variations. VCDP is attached to decoder features during training and removed during inference, introducing no additional inference cost. Experiments on multi-organ segmentation benchmarks show that VCDP improves most evaluated baselines, particularly for small, ambiguous, and highly variable organs. Our anonymous code is released at https://anonymous.4open.science/r/VCDP_code-41ED.
