Provably Accelerated Imaging with Restarted Inertia and Score-based Image Priors
Marien Renaud, Julien Hermant, Deliang Wei, Yu Sun
Abstract
Fast convergence and high-quality image recovery are two essential features of algo- rithms for solving ill-posed imaging inverse problems. Existing methods, such as regularization by denoising (RED), often focus on designing sophisticated image priors to improve reconstruction quality, while leaving convergence acceleration to heuristics. To bridge the gap, we propose Restarted Inertia with Score-based Priors (RISP) as a principled extension of RED. RISP incorporates a restarting inertia for fast convergence, while still allowing score-based image priors for high-quality reconstruction. We prove that RISP attains a faster stationary-point convergence rate than RED, without requiring the convexity of the image prior. We further derive and analyze the associated continuous-time dynamical system, offering insight into the connection between RISP and the heavyball ordinary differential equation (ODE). Experiments across a range of imaging inverse problems demonstrate that RISP enables fast convergence while achieving high-quality reconstructions. † Equal contribution.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on14
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Gradient Step Denoiser for convergent Plug-and-PlaySamuel Hurault, Arthur Leclaire, Nicolas PapadakisICLR 2022 · 154 citations
- Proximal Denoiser for Convergent Plug-and-Play Optimization with Nonconvex RegularizationSamuel Hurault, Arthur Leclaire, Nicolas PapadakisICML 2022 · 121 citations
- Tuning-free Plug-and-Play Proximal Algorithm for Inverse Imaging ProblemsKaixuan Wei, Angelica I. Avilés-Rivero, Jingwei Liang, Ying Fu et al.ICML 2020 · 114 citations
- Plug-and-Play image restoration with Stochastic deNOising REgularizationMarien Renaud, Jean Prost, Arthur Leclaire, Nicolas PapadakisICML 2024 · 19 citations
Related papers
- Async-RED: A Provably Convergent Asynchronous Block Parallel Stochastic Method using Deep Denoising PriorsYu Sun, Jiaming Liu, Yiran Sun, Brendt Wohlberg et al.ICLR 2021 · 5 citations
- RED-PSM: Regularization by Denoising of Partially Separable Models for Dynamic ImagingBerk Iskender, Marc Louis Klasky, Yoram BreslerICCV 2023 · 4 citations
- It Has Potential: Gradient-Driven Denoisers for Convergent Solutions to Inverse ProblemsRegev Cohen, Yochai Blau, Daniel Freedman, Ehud RivlinNeurIPS 2021 · 84 citations
- Recovery Analysis for Plug-and-Play Priors using the Restricted Eigenvalue ConditionJiaming Liu, M. Salman Asif, Brendt Wohlberg, Ulugbek KamilovNeurIPS 2021 · 55 citations
- A Restoration Network as an Implicit PriorYuyang Hu, Mauricio Delbracio, Peyman Milanfar, Ulugbek KamilovICLR 2024 · 17 citations
