arXiv:2608.28681cs.CV2026-08

用反向扩散的共识校准医学影像分割置信度,应对域外数据误差。

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

论文配图:CARD: Calibration via Agreement in Reverse Diffusion for Out-of-Domain MRI Segmentation
图 1 · 摘自论文原文
  • 利用反向扩散过程中的分类一致性发现主模型错误
  • 在心、前列腺和脑部MRI跨域场景中,校准误差降低45/49次
  • 按像素调整置信度,不改变分割结果,适合临床可靠性判断

概率校准将模型置信度与预测准确性对齐,帮助临床识别不可靠分割区域。但在域偏移下,伪影和未见扫描协议会导致高置信度错误。现有后处理方法依赖终端预测的熵、logit模式或增强响应进行校准,但这些代理量本身受域偏移污染。为此,我们提出利用分类扩散的双重优势:其一,生成形状先验在外观被破坏时仍保持有限容量参考,其分歧可揭示主模型错误;其二,每步反向迭代产生类别分布,能区分持续分歧与瞬时差异。对轨迹聚合后,该分歧与Dice相关性达0.788,优于匹配判别模型的0.521。因此,我们提出CARD(Calibration via Agreement in Reverse Diffusion),将此分歧的时间累积映射为每像素的温度场,仅调整置信度而不改变分割结果。在心脏、前列腺和脑部MRI的域偏移测试中,CARD在49次对比中有45次优于各场景最强基线。

原文摘要 · Abstract (English)

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.

医学影像校准扩散模型域外泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。