FANO: Anchored Forward-Noise Optimization for Diffusion Inverse Problems
Abstract
Pre-trained diffusion models have become a widely adopted zero-shot prior for image inverse problems, among which annealed decoupling methods achieve leading reconstruction performance on nonlinear tasks. We find, however, that annealed methods that directly optimize the denoiser input exhibit *radial drift* in later layers: within-layer optimization drives the noise-component norm of away from its typical value, pushing the denoiser out of its high-performance regime and degrading reconstruction quality. To address this, we propose FANO (Anchored Forward-Noise Optimization). At each annealing layer, FANO fixes the previous denoiser estimate as an anchor and optimizes only the forward noise , whose distribution is known, instead of , thereby preserving the forward-diffusion structure of the denoiser input. The standard Gaussian prior on concentrates its norm near the typical value, and deviations degrade denoiser performance. We accordingly decompose in polar coordinates into a direction and a radial magnitude, and apply a soft potential to regulate the radius, keeping in the high-density region of the forward distribution. On multiple linear and nonlinear inverse problems, FANO matches the strongest current baselines on linear tasks and achieves consistent improvements on nonlinear tasks.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.