Search papers, labs, and topics across Lattice.
This paper introduces a novel algorithm, \pddim, that efficiently solves linear inverse problems using diffusion priors through a DDIM-type sampler. By modifying the standard DDIM update with lightweight, coordinate-wise adjustments and incorporating the measurement model, the method achieves rigorous theoretical guarantees and improved computational efficiency. Empirical evaluations demonstrate that \pddim outperforms existing diffusion-based posterior samplers in various image restoration tasks, establishing its effectiveness and consistency in posterior sampling.
Achieving provable posterior consistency with a simple, efficient algorithm that outperforms existing methods in image restoration tasks.
Diffusion-based methods have achieved remarkable empirical success in solving inverse problems. However, many existing posterior samplers either lack rigorous theoretical guarantees or incur substantial computational overhead. We propose a simple and efficient algorithm, called \pddim, for solving linear inverse problems with diffusion priors via a DDIM-type sampler. Our method requires only lightweight, coordinate-wise modifications to the standard DDIM update, while explicitly incorporating the measurement model. The key idea is to perform posterior sampling separately along each singular direction of the measurement operator: for each direction, the sampler follows the learned diffusion prior when the observation signal-to-noise ratio (SNR) is below the corresponding diffusion SNR, and switches to a calibrated measurement-based predictor otherwise. We prove that the proposed sampler converges to the Bayesian posterior conditioned on the measurements. Empirical results show that the proposed sampler performs favorably against existing diffusion-based posterior samplers across a range of image restoration tasks, achieving the best performance on the majority of evaluation metrics considered. Overall, our results convert posterior sampling for noisy linear inverse problems to simple coordinate-wise DDIM updates, yielding an efficient, easy-to-implement algorithm with provable posterior consistency.
