为脑部MRI配准提出分层不确定性估计方法,提升结果可信度与下游任务性能。
Hierarchical Uncertainty Estimation for Learning-based Registration in Neuroimaging
- 基于空间位置的不确定性建模,通过高斯分布传播至全局变换模型
- 不确定性估计与真实误差相关性显著优于传统蒙特卡洛丢弃法,提升配准精度
- 适用于需要可靠配准结果的神经影像研究,尤其关注结果可信度的场景
近年来,基于深度学习的图像配准在多个领域(包括磁共振成像的神经影像)取得了显著进展。然而,现有不确定性估计方法多采用通用技术(如蒙特卡洛丢弃),未能利用问题领域的特性,特别是空间建模能力。本文提出一种系统性的方法,将局部空间位置的不确定性(认知型或偶然型)逐级传播至全局变换模型,并进一步影响下游任务。我们证明了局部不确定性建模采用高斯分布的合理性,并设计了一个依赖变换模型选择的分层传播框架。在公开数据集上的实验表明,蒙特卡洛丢弃与参考配准误差的相关性极低,而本文方法的相关性显著更高。关键的是,引入不确定性感知的变换拟合可提升脑部MRI配准的准确性。最后,我们展示了从变换后验分布中采样,如何有效将不确定性传递至下游神经影像任务。代码已开源:https://github.com/HuXiaoling/Regre4Regis。
原文摘要 · Abstract (English)
Over recent years, deep learning based image registration has achieved impressive accuracy in many domains, including medical imaging and, specifically, human neuroimaging with magnetic resonance imaging (MRI). However, the uncertainty estimation associated with these methods has been largely limited to the application of generic techniques (e.g., Monte Carlo dropout) that do not exploit the peculiarities of the problem domain, particularly spatial modeling. Here, we propose a principled way to propagate uncertainties (epistemic or aleatoric) estimated at the level of spatial location by these methods, to the level of global transformation models, and further to downstream tasks. Specifically, we justify the choice of a Gaussian distribution for the local uncertainty modeling, and then propose a framework where uncertainties spread across hierarchical levels, depending on the choice of transformation model. Experiments on publicly available data sets show that Monte Carlo dropout correlates very poorly with the reference registration error, whereas our uncertainty estimates correlate much better. Crucially, the results also show that uncertainty-aware fitting of transformations improves the registration accuracy of brain MRI scans. Finally, we illustrate how sampling from the posterior distribution of the transformations can be used to propagate uncertainties to downstream neuroimaging tasks. Code is available at: https://github.com/HuXiaoling/Regre4Regis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。