arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.10847eess.IVcs.CVcs.LG
收藏

UncertaINR:面向计算机断层扫描的端到端隐式神经表征的不确定性量化

UncertaINR: Uncertainty Quantification of End-to-End Implicit Neural Representations for Computed Tomography

University of California, Berkeley(加州大学伯克利分校);University of Oxford(牛津大学);Massachusetts Institute of Technology(麻省理工学院)

Francisca Vasconcelos, Bobby He, Nalini Singh, Yee Whye Teh

AI总结:

针对高风险场景下隐式神经表征推理的不确定性量化需求,提出贝叶斯重构方法UncertaINR,在CT重建中精度可比肩主流方法,且用少量验证图像就能实现更优校准。

摘要:

隐式神经表征(INRs)在场景重建和计算机图形学领域已取得亮眼成果,其性能主要通过重建精度来评估。随着INRs逐步应用于模型预测会支撑高风险决策的其他领域,INR推理的不确定性量化变得至关重要。为此,我们在计算机断层扫描场景下研究了INR的贝叶斯重构方法UncertaINR,并从精度和校准度两方面评估了多种贝叶斯深度学习实现方案。我们发现这些方案既能实现校准良好的不确定性,又能保持与其他经典方法、基于INR的方法以及基于CNN的重建技术相当的精度。与贝叶斯深度学习文献中的普遍认知相反,我们发现INR采用计算高效的蒙特卡洛随机失活(Monte Carlo dropout)能获得最佳校准效果,性能优于哈密顿蒙特卡洛(Hamiltonian Monte Carlo)和深度集成(deep ensembles)方法。此外,与性能最优的现有方法不同,UncertaINR不需要大型训练数据集,仅需少量验证图像即可。

英文摘要:

Implicit neural representations (INRs) have achieved impressive results for scene reconstruction and computer graphics, where their performance has primarily been assessed on reconstruction accuracy. As INRs make their way into other domains, where model predictions inform high-stakes decision-making, uncertainty quantification of INR inference is becoming critical. To that end, we study a Bayesian reformulation of INRs, UncertaINR, in the context of computed tomography, and evaluate several Bayesian deep learning implementations in terms of accuracy and calibration. We find that they achieve well-calibrated uncertainty, while retaining accuracy competitive with other classical, INR-based, and CNN-based reconstruction techniques. Contrary to common intuition in the Bayesian deep learning literature, we find that INRs obtain the best calibration with computationally efficient Monte Carlo dropout, outperforming Hamiltonian Monte Carlo and deep ensembles. Moreover, in contrast to the best-performing prior approaches, UncertaINR does not require a large training dataset, but only a handful of validation images.

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

↑