arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2509.23355cs.CV

通过变换等价性估计预训练医学图像配准模型的不确定性

Uncertainty Estimation for Pretrained Medical Image Registration Models via Transformation Equivariance

Massachusetts General Hospital(麻省总医院);Harvard Medical School(哈佛医学院);The Hawkes Institute, University College London(霍克斯研究所,伦敦大学学院);AI Laboratory, Massachusetts Institute of Technology(人工智能实验室,麻省理工学院)

Lin Tian, Xiaoling Hu, Juan Eugenio Iglesias

AI总结:

通过变换等价性属性,提出一种无需修改预训练配准模型的不确定性估计方法,提升医学图像配准的安全性和可靠性。

摘要:

准确的图像配准对于许多医学影像应用至关重要,但大多数深度配准网络很少提供预测不可靠的时间和位置的指示。现有的不确定性估计方法,如贝叶斯方法、集成方法或MC-dropout,通常需要架构修改或重新训练,从而排除了它们在预训练配准模型中的应用。我们提出了一种在推理时应用的、对模型无关的不确定性估计框架,可直接应用于任何预训练配准网络。我们的方法基于图像配准的变换等价性属性,即在输入的空间扰动下,底层解剖学映射应保持一致。在三个预训练配准模型和四个解剖结构上的实验表明,所得到的不确定性图一致地与配准误差相关,并突出显示不可靠对齐的区域。该框架使预训练配准网络在测试时成为风险意识工具,使医学图像配准更接近安全的临床和大规模研究部署。

英文摘要:

Accurate image registration is essential in many medical imaging applications, yet most deep registration networks provide little indication of when or where their predictions are unreliable. Existing uncertainty estimation approaches, such as Bayesian methods, ensembles, or MC-dropout, typically require architectural modifications or retraining, precluding their applicability to pretrained registration models. We propose an inference-time, model-agnostic uncertainty estimation framework that applies directly to any pretrained registration network. Our approach is grounded in the transformation equivariance property of image registration, which states that the underlying anatomical mapping should remain consistent under spatial perturbations of the input. Experiments across three pretrained registration models and four anatomical structures show that the resulting uncertainty maps consistently correlate with registration error and highlight unreliably aligned regions. This framework turns pretrained registration networks into risk-aware tools at test time, moving medical image registration closer to safe clinical and large-scale research deployment.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑