提出高效方法,精准刻画高维图像配准的不确定性。
Structured SIR: Efficient and Expressive Importance-Weighted Inference for High-Dimensional Image Registration
- 用低秩与稀疏结构分解协方差,降低计算复杂度
- 在脑部MRI配准中实现更校准的不确定性估计
- 适合需要可信置信度的医学图像分析场景
图像配准是病态的密集视觉任务,存在多个损失值相近的解,需通过概率推断捕捉分布。以往变分推断受限于后验假设,易导致表征不足、过度自信及生成质量差。高维3D图像配准所需的密集协方差矩阵常使灵活后验难以实现。本文提出结构化SIR方法,采用采样重要性重采样算法,并设计新型内存高效的高维协方差参数化:由低秩协方差与稀疏空间结构化的乔列斯基精度因子之和构成。该结构能捕捉复杂空间相关性且保持可计算性。在脑部MRI三维密集配准任务(高维问题)上验证,所提方法生成的不确定性估计显著优于变分方法,校准性更好,精度相当或更高。关键成果为生成高度结构化的多模态后验分布,支持高效可靠的不确定性量化。
原文摘要 · Abstract (English)
Image registration is an ill-posed dense vision task, where multiple solutions achieve similar loss values, motivating probabilistic inference. Variational inference has previously been employed to capture these distributions, however restrictive assumptions about the posterior form can lead to poor characterisation, overconfidence and low-quality samples. More flexible posteriors are typically bottlenecked by the complexity of high-dimensional covariance matrices required for dense 3D image registration. In this work, we present a memory and computationally efficient inference method, Structured SIR, that enables expressive, multi-modal, characterisation of uncertainty with high quality samples. We propose the use of a Sampled Importance Resampling (SIR) algorithm with a novel memory-efficient high-dimensional covariance parameterisation as the sum of a low-rank covariance and a sparse, spatially structured Cholesky precision factor. This structure enables capturing complex spatial correlations while remaining computationally tractable. We evaluate the efficacy of this approach in 3D dense image registration of brain MRI data, which is a very high-dimensional problem. We demonstrate that our proposed method produces uncertainty estimates that are significantly better calibrated than those produced by variational methods, achieving equivalent or better accuracy. Crucially, we show that the model yields highly structured multi-modal posterior distributions, enable effective and efficient uncertainty quantification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。