🤖 AI Summary
This work addresses the challenge of handling diverse data-driven priors in Bayesian inverse problems by proposing a unified framework based on score functions. The framework seamlessly incorporates various prior models—including Regularization by Denoising (RED), normalizing flows, score-based generative models, and convex ridge regularizers—into a common modeling paradigm. It is naturally integrated into efficient posterior sampling algorithms, significantly enhancing both the generality and computational efficiency of inference. Extensive experiments on image inpainting, single-image super-resolution, and real-world geological image restoration demonstrate state-of-the-art performance, confirming the broad applicability and effectiveness of the proposed approach.
📝 Abstract
This paper reviews how a diverse set of popular data-driven priors commonly used in Bayesian inverse problems can be unified through their respective score functions. By framing these priors under this common perspective, we show that they can benefit from their straightfoward and effective integration into a recently proposed sampling algorithm. The applicability of this common framework is illustrated by considering several data-driven priors, namely regularization-by-denoising, normalizing flow-based priors, score-based generative models, and convex-ridge regularizers. For these four particular priors, the performance of the method is evaluated when conducting image inpainting and single image super-resolution. These results, as well as those obtained when restoring real images acquired in a geological context, demonstrate the efficiency of the method. This unified framework proves versatile enough to handle any posterior distribution defined by a broad class of score function-based priors, beyond the specific cases considered in this paper.