Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing
Bingliang Zhang, Wenda Chu, Julius Berner, Chenlin Meng, Anima Anandkumar, Yang Song
2025Year
64Top-tier citations
Abstract
anneal to the true posterior as we reduce noise levels. This approach enables the exploration of a larger solution space, improving the success rate for accurate reconstructions. We demonstrate that DAPS significantly improves sample quality and stability across multiple image restoration tasks, particularly in complicated nonlinear inverse problems. Our code is available at the GitHub repository DAPS.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers64
- Guided Diffusion Sampling on Function Spaces with Applications to PDEsJiachen Yao, Abbas Mammadov, Julius Berner, Gavin Kerrigan et al.NeurIPS 2025 · 37 citations
- Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative ModelingMichal Balcerak, Tamaz Amiranashvili, Antonio Terpin, Suprosanna Shit et al.NeurIPS 2025 · 33 citations
- Flower: A Flow-Matching Solver for Inverse ProblemsMehrsa Pourya, Bassam El Rawas, Michael UnserICLR 2026 · 25 citations
- Split Gibbs Discrete Diffusion Posterior SamplingWenda Chu, Zihui Wu, Yifan Chen, Yang Song et al.NeurIPS 2025 · 22 citations
- Solving Inverse Problems with FLAIRJulius Erbach, Dominik Narnhofer, Andreas Dombos, Bernt Schiele et al.NeurIPS 2025 · 20 citations
Builds on30
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Diffusion Models Beat GANs on Image SynthesisPrafulla Dhariwal, Alexander Quinn NicholNeurIPS 2021 · 13,211 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- SDXL: Improving Latent Diffusion Models for High-Resolution Image SynthesisDustin Podell, Zion English, Kyle Lacey, Andreas Blattmann et al.ICLR 2024 · 4,569 citations
- Elucidating the Design Space of Diffusion-Based Generative ModelsTero Karras, Miika Aittala, Timo Aila, Samuli LaineNeurIPS 2022 · 3,959 citations
Related papers
- Consistency Posterior Sampling for Diverse Image SynthesisVishal Purohit, Matthew Repasky, Jianfeng Lu, Qiang Qiu et al.CVPR 2025
- Tomographic Auto-Encoder: Unsupervised Bayesian Recovery of Corrupted DataFrancesco Tonolini, Pablo Garcia Moreno, Andreas C. Damianou, Roderick Murray-SmithICLR 2021 · 1 citation
- DGSolver: Diffusion Generalist Solver with Universal Posterior Sampling for Image RestorationHebaixu Wang, Jing Zhang, Haonan Guo, Di Wang et al.NeurIPS 2025 · 7 citations
- Dual Ascent Diffusion for Inverse ProblemsMinseo Kim, Axel Levy, Gordon WetzsteinCVPR 2026 · 3 citations
- pcaGAN: Improving Posterior-Sampling cGANs via Principal Component RegularizationMatthew C. Bendel, Rizwan Ahmad, Philip SchniterNeurIPS 2024 · 2 citations
