Semantic Class Distribution Learning for Debiasing Semi-Supervised Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
Medical image segmentation is critical for computer-aided diagnosis. However, dense pixel-level annotation is time-consuming and costly, and medical datasets often exhibit severe class imbalance. Such an imbalance causes minority structures to be overwhelmed by dominant classes in feature representations, hindering the learning of discriminative features and making reliable segmentation particularly challenging. To address this, we propose the Semantic Class Distribution Learning (SCDL) framework, a plug-and-play module that mitigates supervision and representation biases by learning structured class-conditional feature distributions. SCDL integrates Class Distribution Bidirectional Alignment (CDBA) to align embeddings with learnable class proxies and leverages Semantic Anchor Constraints (SAC) to guide proxies using labeled data. Experiments on the Synapse and AMOS datasets demonstrate that SCDL largely improves segmentation performance across both overall and class-level metrics, with particularly notable gains for several low-frequency organs. Our anonymous code is released at https://anonymous.4open.science/r/SCDL.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 39 references that are in the map. First posted on arXiv 5 Mar 2026.
Builds on
22 of 39 references.
References · 22
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,284
- 2019Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentationcited by 378
- 2022AMOS: A Large-Scale Abdominal Multi-Organ Benchmark for Versatile Medical Image Segmentationcited by 347
- 2021In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learningcited by 301
- 2017No Fuss Distance Metric Learning using Proxiescited by 225
- 2020Proxy Anchor Loss for Deep Metric Learningcited by 195
- 2021SimCVD: Simple Contrastive Voxel-Wise Representation Distillation for Semi-Supervised Medical Image Segmentationcited by 86
- 2018Combo Loss: Handling Input and Output Imbalance in Multi-Organ Segmentationcited by 76
- 2022Enhancing Pseudo Label Quality for Semi-Supervised Domain-Generalized Medical Image Segmentationcited by 51
- 2020Self-Loop Uncertainty: A Novel Pseudo-Label for Semi-Supervised Medical Image Segmentationcited by 42
10 more
- 2023Towards Generic Semi-Supervised Framework for Volumetric Medical Image Segmentationcited by 35
- 2022Calibrating Label Distribution for Class-Imbalanced Barely-Supervised Knee Segmentationcited by 27
- 2023DHC: Dual-debiased Heterogeneous Co-training Framework for Class-imbalanced Semi-supervised Medical Image Segmentationcited by 25
- 2023Co-training with High-Confidence Pseudo Labels for Semi-supervised Medical Image Segmentationcited by 24
- 2022MagicNet: Semi-Supervised Multi-Organ Segmentation via Magic-Cube Partition and Recoverycited by 22
- 2016Bridging Category-level and Instance-level Semantic Image Segmentationcited by 21
- 2023Self-aware and Cross-sample Prototypical Learning for Semi-supervised Medical Image Segmentationcited by 20
- 2022PHTrans: Parallelly Aggregating Global and Local Representations for Medical Image Segmentationcited by 16
- 2022An Embarrassingly Simple Baseline for Imbalanced Semi-Supervised Learningcited by 12
- 2024Neighbor Does Matter: Density-Aware Contrastive Learning for Medical Semi-supervised Segmentationcited by 4
Built on by
Not cited yet.