Coupled Data and Measurement Space Dynamics for Enhanced Diffusion Posterior Sampling
By: Shayan Mohajer Hamidi, En-Hui Yang, Ben Liang
Potential Business Impact:
Makes blurry pictures clear from bad data.
Inverse problems, where the goal is to recover an unknown signal from noisy or incomplete measurements, are central to applications in medical imaging, remote sensing, and computational biology. Diffusion models have recently emerged as powerful priors for solving such problems. However, existing methods either rely on projection-based techniques that enforce measurement consistency through heuristic updates, or they approximate the likelihood $p(\boldsymbol{y} \mid \boldsymbol{x})$, often resulting in artifacts and instability under complex or high-noise conditions. To address these limitations, we propose a novel framework called \emph{coupled data and measurement space diffusion posterior sampling} (C-DPS), which eliminates the need for constraint tuning or likelihood approximation. C-DPS introduces a forward stochastic process in the measurement space $\{\boldsymbol{y}_t\}$, evolving in parallel with the data-space diffusion $\{\boldsymbol{x}_t\}$, which enables the derivation of a closed-form posterior $p(\boldsymbol{x}_{t-1} \mid \boldsymbol{x}_t, \boldsymbol{y}_{t-1})$. This coupling allows for accurate and recursive sampling based on a well-defined posterior distribution. Empirical results demonstrate that C-DPS consistently outperforms existing baselines, both qualitatively and quantitatively, across multiple inverse problem benchmarks.
Similar Papers
Robust Posterior Diffusion-based Sampling via Adaptive Guidance Scale
CV and Pattern Recognition
Makes blurry pictures sharp and clear.
Injecting Measurement Information Yields a Fast and Noise-Robust Diffusion-Based Inverse Problem Solver
Machine Learning (CS)
Makes blurry pictures clear using smart guessing.
DAPS++: Rethinking Diffusion Inverse Problems with Decoupled Posterior Annealing
Artificial Intelligence
Improves image cleanup by separating steps.