Score-Based Turbo Message Passing for Plug-and-Play Compressive Image Recovery
This addresses the challenge of producing satisfactory results in largely underdetermined compressive imaging scenarios, representing an incremental improvement over existing methods.
The paper tackled the problem of insufficient accuracy in compressive image recovery by integrating a score-based MMSE denoiser into a message passing framework, achieving a significantly better performance-complexity tradeoff with less than 20 neural function evaluations to converge.
Message passing algorithms have been tailored for compressive imaging applications by plugging in different types of off-the-shelf image denoisers. These off-the-shelf denoisers mostly rely on some generic or hand-crafted priors for denoising. Due to their insufficient accuracy in capturing the true image prior, these methods often fail to produce satisfactory results, especially in largely underdetermined scenarios. On the other hand, score-based generative modeling offers a promising way to accurately characterize the sophisticated image distribution. In this paper, by exploiting the close relation between score-based modeling and empirical Bayes-optimal denoising, we devise a message passing framework that integrates a score-based minimum mean squared error (MMSE) denoiser for compressive image recovery. This framework is firmly rooted in Bayesian formalism, in which state evolution (SE) equations accurately predict its asymptotic performance. Experiments on the FFHQ dataset demonstrate that our method strikes a significantly better performance-complexity tradeoff than conventional message passing, regularized linear regression, and score-based posterior sampling baselines. Remarkably, our method typically requires less than 20 neural function evaluations (NFEs) to converge.