Uncertainty Estimation for Computed Tomography with a Linearised Deep Image Prior
Existing deep-learning based tomographic image reconstruction methods do not provide accurate estimates of reconstruction uncertainty, hindering their real-world deployment. This paper develops a method, termed as the linearised deep image prior (DIP), to estimate the uncertainty associated with reconstructions produced by the DIP with total variation regularisation (TV). Specifically, we endow the DIP with conjugate Gaussian-linear model type error-bars computed from a local linearisation of the neural network around its optimised parameters. To preserve conjugacy, we approximate the TV regulariser with a Gaussian surrogate. This approach provides pixel-wise uncertainty estimates and a marginal likelihood objective for hyperparameter optimisation. We demonstrate the method on synthetic data and real-measured high-resolution 2D $\mu$CT data, and show that it provides superior calibration of uncertainty estimates relative to previous probabilistic formulations of the DIP. Our code is available at https://github.com/educating-dip/bayes_dip.
Code (2)
Tasks
Image ReconstructionSimilar Papers 제목 키워드 기반
Bayesian Experimental Design for Computed Tomography with the Linearised Deep Image Prior
We investigate adaptive design based on a single sparse pilot scan for generating effective scanning strategies for computed tomography reconstruction. We propose a novel approach using the linearised deep image prior. I…
Experimental DesignConformal Risk Control for Semantic Uncertainty Quantification in Computed Tomography
Uncertainty quantification is necessary for developers, physicians, and regulatory agencies to build trust in machine learning predictors and improve patient care. Beyond measuring uncertainty, it is crucial to express i…
AnatomyComputed Tomography (CT)Uncertainty QuantificationvalidScalable Bayesian Inference in the Era of Deep Learning: From Gaussian Processes to Deep Neural Networks
Large neural networks trained on large datasets have become the dominant paradigm in machine learning. These systems rely on maximum likelihood point estimates of their parameters, precluding them from expressing model u…
Bayesian InferenceGaussian ProcessesSequential Decision MakingTomographic ReconstructionsPrincipled Confidence Estimation for Deep Computed Tomography
We present a principled framework for confidence estimation in computed tomography (CT) reconstruction. Based on the sequential likelihood mixing framework (Kirschner et al., 2025), we establish confidence regions with t…
Sampling-based inference for large linear models, with application to linearised Laplace
Large-scale linear models are ubiquitous throughout machine learning, with contemporary application as surrogate models for neural network uncertainty quantification; that is, the linearised Laplace method. Alas, the com…
Bayesian InferenceUncertainty Quantification