Plug-and-Play Priors as a Score-Based Method

Chicago Y. Park, Yuyang Hu, Michael T. McCann, Cristina Garcia-Cardona, Brendt Wohlberg +1 more
12/15/2024
eess.IVcs.CV

Official code: wustl-cig/score_pnp (6 stars, BSD-3-Clause, last commit 2025-05-21).

Abstract

Plug-and-play (PnP) methods are extensively used for solving imaging inverse problems by integrating physical measurement models with pre-trained deep denoisers as priors. Score-based diffusion models (SBMs) have recently emerged as a powerful framework for image generation by training deep denoisers to represent the score of the image prior. While both PnP and SBMs use deep denoisers, the score-based nature of PnP is unexplored in the literature due to its distinct origins rooted in proximal optimization. This letter introduces a novel view of PnP as a score-based method, a perspective that enables the re-use of powerful SBMs within classical PnP algorithms without retraining. We present a set of mathematical relationships for adapting popular SBMs as priors within PnP. We show that this approach enables a direct comparison between PnP and SBM-based reconstruction methods using the same neural network as the prior. Code is available at https://github.com/wustl-cig/score_pnp.

View on arXivView PDF

Code Implementations(1)

BSD-3-Clause
62PythonNov 25, 20241 years agoBSD-3-Clause

Similar Papers

Cite this paper

@article{park2024plugandplay,
  title  = {Plug-and-Play Priors as a Score-Based Method},
  author = {Chicago Y. Park and Yuyang Hu and Michael T. McCann and Cristina Garcia-Cardona and Brendt Wohlberg and Ulugbek S. Kamilov},
  year   = {2024},
  eprint = {2412.11108},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.11108v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.11108.svg)](https://codeofpaper.com/paper/2412.11108)

Discussion