arXivDaily arXiv每日学术速递 周一至周五更新
看起来今天可以划水了
arXiv 2007.06516cs.CV
收藏

Uncertain-DeepSSM:从图像到概率形状模型

Uncertain-DeepSSM: From Images to Probabilistic Shape Models

University of Utah(犹他大学)

Jadie Adams, Riddhish Bhalodia, Shireen Elhabian

摘要:

统计形状建模(SSM)最近利用了深度学习的进步,以减轻对解剖分割、形状配准和优化群体级形状表示的耗时且专家驱动的工作流的需求。DeepSSM是一种端到端的深度学习方法,它直接从未分割的图像中提取统计形状表示,几乎没有手动开销。它使用最先进的形状建模方法进行建模,以估计后续下游任务可行的形态。尽管如此,DeepSSM对形状的估计过于自信,不能盲目地认为是准确的。因此,通过量化不确定性的粒度估计来传达DeepSSM所不知道的内容,对于其作为按需诊断工具的直接临床应用至关重要,以确定模型输出的可信度。在这里,我们提出了Uncertain-DeepSSM作为一个统一的模型,它通过调整网络来预测内在输入方差来量化数据相关的任意不确定性,以及通过蒙特卡洛丢弃抽样来近似网络参数的变分分布来量化模型相关的认知不确定性。实验表明,与DeepSSM相比,精度有所提高,同时保持了端到端的相同优势,几乎没有预处理。

英文摘要:

Statistical shape modeling (SSM) has recently taken advantage of advances in deep learning to alleviate the need for a time-consuming and expert-driven workflow of anatomy segmentation, shape registration, and the optimization of population-level shape representations. DeepSSM is an end-to-end deep learning approach that extracts statistical shape representation directly from unsegmented images with little manual overhead. It performs comparably with state-of-the-art shape modeling methods for estimating morphologies that are viable for subsequent downstream tasks. Nonetheless, DeepSSM produces an overconfident estimate of shape that cannot be blindly assumed to be accurate. Hence, conveying what DeepSSM does not know, via quantifying granular estimates of uncertainty, is critical for its direct clinical application as an on-demand diagnostic tool to determine how trustworthy the model output is. Here, we propose Uncertain-DeepSSM as a unified model that quantifies both, data-dependent aleatoric uncertainty by adapting the network to predict intrinsic input variance, and model-dependent epistemic uncertainty via a Monte Carlo dropout sampling to approximate a variational distribution over the network parameters. Experiments show an accuracy improvement over DeepSSM while maintaining the same benefits of being end-to-end with little pre-processing.

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

↑