Contrastive uncertainty learning for robust multi-structure segmentation in lower-pelvic MR
Read the original on arXiv Machine Learning →The paper introduces Contrastive Uncertainty Learning (CUL), a supervised framework that leverages voxel‑wise predictive uncertainty to guide representation learning for multi‑structure segmentation in lower‑pelvic MRI. Using Monte Carlo Dropout, CUL identifies uncertain voxels and shapes the embedding space so that low‑uncertainty, correctly predicted features become stable while high‑uncertainty or misclassified features are repelled, preventing feature collapse at challenging boundaries. Evaluated on 589 prostate MRI scans with a ViT‑based UperNet decoder, CUL achieves comparable Dice scores to a baseline while significantly reducing surface distance for most structures, and ablation studies confirm the necessity of both uncertainty‑guided sampling and the contrastive objective.
Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.