Search papers, labs, and topics across Lattice.
This paper introduces MAP-RPS and LMAP-RPS, novel stage-wise frameworks leveraging diffusion models to navigate the distortion-perception (D-P) tradeoff in zero-shot inverse problems. The approach combines a MAP estimation stage for low-distortion initialization with a re-noised posterior sampling stage to enhance perceptual quality. Theoretical analysis and experiments demonstrate the effectiveness of these methods in traversing the D-P tradeoff and solving real-world inverse problems.
Achieve superior control over the distortion-perception tradeoff in diffusion-based inverse problems by decoupling MAP estimation and posterior sampling into distinct stages.
The distortion-perception (D-P) tradeoff is a fundamental phenomenon of Bayesian inverse problems, which characterizes the inherent tension between distortion performance and perceptual quality. Enabling flexible traversal of the D-P tradeoff at inference time is crucial for practical applications. Despite the recent success of diffusion models in zero-shot inverse problem solving, efficient and principled strategies for D-P traversal in diffusion-based inverse algorithms remain inadequately characterized. In this paper, we propose a stage-wise framework for realizing D-P traversal using a single diffusion model in zero-shot inverse problems. Our proposed method, termed MAP-RPS, starts with an MAP estimation stage that approximates the MMSE solution and provides a low-distortion initialization, followed by a re-noised posterior sampling stage that progressively improves perceptual quality. We provide theoretical analyses for both stages, establishing the validity and effectiveness of the proposed design. Furthermore, we extend MAP-RPS to the latent space, yielding LMAP-RPS, which enjoys broader applicability by leveraging large-scale pre-trained latent diffusion backbones. Extensive experiments demonstrate that MAP-RPS and LMAP-RPS enable more effective D-P traversal on various tasks, while also exhibiting strong performance as efficient solvers for real-world inverse problems.
