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

从配准不确定性到分割不确定性

From Registration Uncertainty to Segmentation Uncertainty

Johns Hopkins University(约翰斯·霍普金斯大学)

Junyu Chen, Yihao Liu, Shuwen Wei, Zhangxing Bian, Aaron Carass, Yong Du

AI总结:

针对现有配准不确定性无法有效估计标签传播误差的问题,提出同时估计认知与偶然分割不确定性的框架,经公开数据集验证,其估计结果与标签传播误差相关性好且配准性能更优。

摘要:

理解基于深度学习的图像配准模型固有的不确定性一直是一个持续的研究领域。现有方法已被开发用于量化与配准过程相关的变换不确定性和外观不确定性,阐明模型在生成的形变方面可能表现出歧义的区域。然而,我们的研究表明,当配准模型用于标签传播时,这两种不确定性都无法有效估计潜在误差。在此,我们提出了一个新颖的框架,用于同时估计图像配准的认知分割不确定性和偶然分割不确定性。为此,我们实现了一个紧凑型深度神经网络(DNN),该网络旨在通过最小化负对数似然损失函数,将配准中的外观差异转化为偶然分割不确定性。此外,我们将标签传播过程中的认知分割不确定性表示为传播标签的熵。通过引入分割不确定性以及现有的配准不确定性估计方法,我们为图像配准不同阶段的潜在不确定性提供了重要见解。我们使用公开可用的数据集验证了所提出的框架,结果证明,用所提方法估计的分割不确定性与标签传播中的误差具有良好的相关性,同时还实现了更优的配准性能。

英文摘要:

Understanding the uncertainty inherent in deep learning-based image registration models has been an ongoing area of research. Existing methods have been developed to quantify both transformation and appearance uncertainties related to the registration process, elucidating areas where the model may exhibit ambiguity regarding the generated deformation. However, our study reveals that neither uncertainty effectively estimates the potential errors when the registration model is used for label propagation. Here, we propose a novel framework to concurrently estimate both the epistemic and aleatoric segmentation uncertainties for image registration. To this end, we implement a compact deep neural network (DNN) designed to transform the appearance discrepancy in the warping into aleatoric segmentation uncertainty by minimizing a negative log-likelihood loss function. Furthermore, we present epistemic segmentation uncertainty within the label propagation process as the entropy of the propagated labels. By introducing segmentation uncertainty along with existing methods for estimating registration uncertainty, we offer vital insights into the potential uncertainties at different stages of image registration. We validated our proposed framework using publicly available datasets, and the results prove that the segmentation uncertainties estimated with the proposed method correlate well with errors in label propagation, all while achieving superior registration performance.

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

↑