mlsanityData to 14 Oct 2026

CARD: Calibration via Agreement in Reverse Diffusion for Out-of-Domain MRI Segmentation

arXiv 2608.28681 ↗First posted 26 Aug 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Probability calibration aligns model confidence with predictive accuracy, enabling clinicians to identify unreliable segmentation regions. This alignment breaks down under domain shift, where artifacts and unseen protocols produce confident errors. Existing post-hoc methods adapt the correction at test time, conditioning on predictive entropy, the logit pattern, or augmentation response, but each proxy is read from the terminal prediction, the very quantity that shift corrupts. This motivates reliability evidence beyond the terminal prediction, which categorical diffusion provides in two ways. First, a generative shape prior keeps a capacity-limited reference intact when appearance is corrupted, so its disagreement with the primary segmentor highlights primary-model errors. Second, every reverse step yields a class distribution, separating persistent disagreement from transient discrepancy. Aggregated over the trajectory, this disagreement correlates with Dice at 0.788, against 0.521 for a matched discriminative control. We therefore propose CARD (Calibration via Agreement in Reverse Diffusion), which maps the temporal aggregate of this disagreement to a temperature field applied per pixel across all classes, so that confidence changes while the segmentation does not. Across cardiac, prostate and brain MRI shifts, CARD lowers calibration error in 45 of 49 comparisons against the strongest baseline in each setting.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 35 of its 52 references that are in the map. First posted on arXiv 26 Aug 2026.

Builds on

35 of 52 references.

References · 35

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  3. 2020Denoising Diffusion Implicit Modelscited by 9,781
  4. 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
  5. 2017On Calibration of Modern Neural Networkscited by 4,960
  6. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  7. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
  8. 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
  9. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
  10. 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
  11. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
  12. 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
23 more
  1. 2019Measuring Calibration in Deep Learningcited by 446
  2. 2020Calibrating Deep Neural Networks using Focal Losscited by 357
  3. 2019On Mixup Training: Improved Calibration and Predictive Uncertainty for Deep Neural Networkscited by 305
  4. 2024Guiding a Diffusion Model with a Bad Version of Itselfcited by 299
  5. 2018Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networkscited by 280
  6. 2020Mix-n-Match: Ensemble and Compositional Methods for Uncertainty Calibration in Deep Learningcited by 190
  7. 2022MedSegDiff: Medical Image Segmentation with Diffusion Probabilistic Modelcited by 180
  8. 2020TorchIO: A Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learningcited by 177
  9. 2019Confidence Calibration and Predictive Uncertainty Estimation for Deep Medical Image Segmentationcited by 163
  10. 2019PHiSeg: Capturing Uncertainty in Medical Image Segmentationcited by 128
  11. 2023Bidirectional Copy-Paste for Semi-Supervised Medical Image Segmentationcited by 116
  12. 2020Local Temperature Scaling for Probability Calibrationcited by 95
  13. 2021Parameterized Temperature Scaling for Boosting the Expressive Power in Post-Hoc Uncertainty Calibrationcited by 48
  14. 2018Leveraging Uncertainty Estimates for Predicting Segmentation Qualitycited by 44
  15. 2022Categorical SDEs with Simplex Diffusioncited by 42
  16. 2018Bayesian QuickNAT: Model Uncertainty in Deep Whole-Brain Segmentation for Structure-wise Quality Controlcited by 39
  17. 2020Towards Trustworthy Predictions from Deep Neural Networks with Fast Adversarial Calibrationcited by 34
  18. 2021Confidence Calibration for Domain Generalization under Covariate Shiftcited by 27
  19. 2022Adaptive Temperature Scaling for Robust Calibration of Deep Neural Networkscited by 24
  20. 2020Deep Generative Model-based Quality Control for Cardiac MRI Segmentationcited by 22
  21. 2022Improved post-hoc probability calibration for out-of-domain MRI segmentationcited by 5
  22. 2025Average Calibration Losses for Reliable Uncertainty in Medical Image Segmentationcited by 3
  23. 2024Mask-TS Net: Mask Temperature Scaling Uncertainty Calibration for Polyp Segmentationcited by 1

Built on by

Not cited yet.