这篇论文提出了一种名为CARD(Calibration via Agreement in Reverse Diffusion)的新方法,用于解决MRI图像分割模型在域偏移场景下的置信度校准问题。传统校准方法依赖终端预测的熵、逻辑模式等代理指标,但这些指标本身会被域偏移破坏。CARD利用分类扩散模型的生成式形状先验,通过比较主分割器与参考模型在反向扩散过程中的分歧来识别错误,并沿轨迹聚合这种分歧以区分持续错误与瞬时偏差。实验表明,该分歧与Dice系数的相关性达0.788,显著优于判别式对照的0.521。在心脏、前列腺和脑部MRI的跨域测试中,CARD在49项对比中有45项降低了校准误差,同时保持分割结果不变,为临床可靠AI辅助诊断提供了有效工具。
| Probability calibration | 概率校准,调整模型预测置信度以匹配实际准确率的过程。 |
| Domain shift | 域偏移,训练和测试数据分布不一致导致模型性能下降的现象。 |
| Categorical diffusion | 分类扩散,一种生成模型,通过逐步去噪生成类别分布。 |
| CARD | 通过反向扩散中的一致性进行校准,一种利用扩散轨迹分歧进行校准的方法。 |
| Temperature field | 温度场,用于调整置信度标度的逐像素参数。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅