Task-Driven Uncertainty Quantification in Inverse Problems via Conformal Prediction
In imaging inverse problems, one seeks to recover an image from missing/corrupted measurements. Because such problems are ill-posed, there is great motivation to quantify the uncertainty induced by the measurement-and-recovery process. Motivated by applications where the recovered image is used for a downstream task, such as soft-output classification, we propose a task-centered approach to uncertainty quantification. In particular, we use conformal prediction to construct an interval that is guaranteed to contain the task output from the true image up to a user-specified probability, and we use the width of that interval to quantify the uncertainty contributed by measurement-and-recovery. For posterior-sampling-based image recovery, we construct locally adaptive prediction intervals. Furthermore, we propose to collect measurements over multiple rounds, stopping as soon as the task uncertainty falls below an acceptable level. We demonstrate our methodology on accelerated magnetic resonance imaging (MRI): https://github.com/jwen307/TaskUQ.
Code (1)
Tasks
Conformal PredictionPrediction IntervalsUncertainty QuantificationSimilar Papers 제목 키워드 기반
Goal-oriented Uncertainty Quantification for Inverse Problems via Variational Encoder-Decoder Networks
In this work, we describe a new approach that uses variational encoder-decoder (VED) networks for efficient goal-oriented uncertainty quantification for inverse problems. Contrary to standard inverse problems, these appr…
DecoderUncertainty QuantificationUncertainty Quantification in PINNs for Turbulent Flows: Bayesian Inference and Repulsive Ensembles
Physics-informed neural networks (PINNs) have emerged as a promising framework for solving inverse problems governed by partial differential equations (PDEs), including the reconstruction of turbulent flow fields from sp…
Bayesian InferenceBayesian BiLO: Bilevel Local Operator Learning for Efficient Uncertainty Quantification of Bayesian PDE Inverse Problems with Low-Rank Adaptation
Uncertainty quantification in PDE inverse problems is essential in many applications. Scientific machine learning and AI enable data-driven learning of model components while preserving physical structure, and provide th…
Bayesian InferenceCycle Consistency-based Uncertainty Quantification of Neural Networks in Inverse Imaging Problems
Uncertainty estimation is critical for numerous applications of deep neural networks and draws growing attention from researchers. Here, we demonstrate an uncertainty quantification approach for deep neural networks used…
DeblurringImage DeblurringSuper-ResolutionUncertainty QuantificationUncertainty Quantification with Generative Models
We develop a generative model-based approach to Bayesian inverse problems, such as image reconstruction from noisy and incomplete images. Our framework addresses two common challenges of Bayesian reconstructions: 1) It m…
Image ReconstructionUncertainty Quantification