Score-Based Turbo Message Passing for Plug-and-Play Compressive Image Recovery

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Cai, Chang, Yuan, Xiaojun, Zhang, Ying-Jun Angela
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866908710107873280
author Cai, Chang
Yuan, Xiaojun
Zhang, Ying-Jun Angela
author_facet Cai, Chang
Yuan, Xiaojun
Zhang, Ying-Jun Angela
contents 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 highly 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. 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 converges in fewer than 20 neural function evaluations (NFEs).
format Preprint
id arxiv_https___arxiv_org_abs_2503_22140
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Score-Based Turbo Message Passing for Plug-and-Play Compressive Image Recovery
Cai, Chang
Yuan, Xiaojun
Zhang, Ying-Jun Angela
Image and Video Processing
Computer Vision and Pattern Recognition
Signal Processing
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 highly 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. 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 converges in fewer than 20 neural function evaluations (NFEs).
title Score-Based Turbo Message Passing for Plug-and-Play Compressive Image Recovery
topic Image and Video Processing
Computer Vision and Pattern Recognition
Signal Processing
url https://arxiv.org/abs/2503.22140