DAPS++: Rethinking Diffusion Inverse Problems with Decoupled Posterior Annealing

arXiv stat.ML / 3/23/2026

📰 NewsModels & Research

Key Points

  • The paper shows that diffusion priors in inverse problems mainly act as a warm initializer near the data manifold, while reconstruction is largely driven by measurement consistency.
  • It introduces DAPS++, a method that fully decouples diffusion-based initialization from likelihood-driven refinement to allow the likelihood term to guide inference more directly and stably.
  • DAPS++ achieves higher computational efficiency by requiring fewer function evaluations and measurement-optimization steps while maintaining robust reconstruction across diverse image restoration tasks.
  • The work also provides insight into why unified diffusion trajectories can remain effective in practice despite the decoupling.

Abstract

From a Bayesian perspective, score-based diffusion solves inverse problems through joint inference, embedding the likelihood with the prior to guide the sampling process. However, this formulation fails to explain its practical behavior: the prior offers limited guidance, while reconstruction is largely driven by the measurement-consistency term, leading to an inference process that is effectively decoupled from the diffusion dynamics. We show that the diffusion prior in these solvers functions primarily as a warm initializer that places estimates near the data manifold, while reconstruction is driven almost entirely by measurement consistency. Based on this observation, we introduce \textbf{DAPS++}, which fully decouples diffusion-based initialization from likelihood-driven refinement, allowing the likelihood term to guide inference more directly while maintaining numerical stability and providing insight into why unified diffusion trajectories remain effective in practice. By requiring fewer function evaluations (NFEs) and measurement-optimization steps, \textbf{DAPS++} achieves high computational efficiency and robust reconstruction performance across diverse image restoration tasks.