FAST-DIPS: Adjoint-Free Analytic Steps and Hard-Constrained Likelihood Correction for Diffusion-Prior Inverse Problems
Organizations: Department of Electrical and Computer Engineering, Inha University Incheon, 22212, South Korea
Abstract
Training-free diffusion priors enable inverse-problem solvers without retraining, but for nonlinear forward operators data consistency often relies on repeated derivatives or inner optimization/MCMC loops with conservative step sizes, incurring many iterations and denoiser/score evaluations. We propose a training-free solver that replaces these inner loops with a hard measurement-space feasibility constraint (closed-form projection) and an analytic, model-optimal step size, enabling a small, fixed compute budget per noise level. Anchored at the denoiser prediction, the correction is approximated via an adjoint-free, ADMM-style splitting with projection and a few steepest-descent updates, using one VJP and either one JVP or a forward-difference probe, followed by backtracking and decoupled re-annealing. We prove local model optimality and descent under backtracking for the step-size rule, and derive an explicit KL bound for mode-substitution re-annealing under a local Gaussian conditional surrogate. We also develop a latent variant and a one-parameter pixellatent hybrid schedule. Experiments achieve competitive PSNR/SSIM/LPIPS with up to 19.5 speedup, without hand-coded adjoints or inner MCMC.
Figures & tables
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Algorithm | Parameter | Linear task | Non Linear task | ||||||
| Super Resolution 4× | Inpaint (Box) | Inpaint (Random) | Gaussian deblurring | Motion deblurring | Phase retrieval | Nonlinear deblurring | High dynamic range | ||
| FAST-DIPS | 75 | 75 | 75 | 50 | 50 | 150 | 150 | 150 | |
| 3 | 3 | 3 | 3 | 3 | 2 | 2 | 2 | ||
| 1 | 1 | 1 | 2 | 2 | 5 | 5 | 5 | ||
| 200 | 200 | 200 | 200 | 200 | 200 | 200 | 5 | ||
| 0.05 | 0.05 | 0.05 | 0.05 | 0.05 | 0.05 | 0.05 | 0.05 | ||
Diffusion-Based Inverse Problems
Diffusion-based inverse problems use generative priors to recover images and physical fields from incomplete, noisy, or indirect measurements.
See all 141 papers