Accelerating Score-based Generative Models with Preconditioned Diffusion Sampling
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Score-based generative models (SGMs) have recently emerged as a promising class of generative models. However, a fundamental limitation is that their inference is very slow due to a need for many (e.g., 2000) iterations of sequential computations. An intuitive acceleration method is to reduce the sampling iterations which however causes severe performance degradation. We investigate this problem by viewing the diffusion sampling process as a Metropolis adjusted Langevin algorithm, which helps reveal the underlying cause to be ill-conditioned curvature. Under this insight, we propose a model-agnostic preconditioned diffusion sampling (PDS) method that leverages matrix preconditioning to alleviate the aforementioned problem. Crucially, PDS is proven theoretically to converge to the original target distribution of a SGM, no need for retraining. Extensive experiments on three image datasets with a variety of resolutions and diversity validate that PDS consistently accelerates off-the-shelf SGMs whilst maintaining the synthesis quality. In particular, PDS can accelerate by up to 29x on more challenging high resolution (1024x1024) image generation.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 23 of its 39 references that are in the map. First posted on arXiv 5 Jul 2022.
Builds on
23 of 39 references.
Closest ancestors · 14
Same direction, newest first
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models17 Jan
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANs15 Dec
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusion14 Dec
- 2021Score-based Generative Modeling in Latent Space10 Jun
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modeling1 Jun
- 2021Cascaded Diffusion Models for High Fidelity Image Generation30 May
- 2021Gotta Go Fast When Generating Data with Score-Based Models28 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Improved Denoising Diffusion Probabilistic Models18 Feb
- 2021Maximum Likelihood Training of Score-Based Diffusion Models22 Jan
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
2 more
Other references · 9
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2022Generative Adversarial Networkscited by 5,603
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,967
- 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,609
- 2015LSUN: Construction of a Large-scale Image Dataset using Deep Learning with Humans in the Loopcited by 1,350
- 2015Preconditioned Stochastic Gradient Langevin Dynamics for Deep Neural Networkscited by 121
Built on by
Citing papers per year: 2022: 5; 2023: 8; 2024: 3; 2025: 3.
19 papers.
Sort:
By area:
- 2022Diffusion Probabilistic Model Made Slimcited by 121
- 2022Wavelet Diffusion Models are fast and scalable Image Generatorscited by 116
- 2022Diffusion Models Beat GANs on Topology Optimizationcited by 79
- 2022Multiscale Structure Guided Diffusion for Image Deblurringcited by 76
- 2022Conversion Between CT and MRI Images Using Diffusion and Score-Matching Modelscited by 69
- 2023WaveDM: Wavelet-Based Diffusion Models for Image Restorationcited by 46
- 2023Multi-Architecture Multi-Expert Diffusion Modelscited by 34
- 2023ExposureDiffusion: Learning to Expose for Low-light Image Enhancementcited by 29
- 2023HiDiffusion: Unlocking Higher-Resolution Creativity and Efficiency in Pretrained Diffusion Modelscited by 27
- 2023FRDiff : Feature Reuse for Universal Training-free Acceleration of Diffusion Modelscited by 19
- 2023Image generation with shortest path diffusioncited by 8
- 2023Preconditioned Score-based Generative Modelscited by 6
7 more
- 2023Solving Linear Inverse Problems using Higher-Order Annealed Langevin Diffusioncited by 6
- 2024Hitchhiker's guide on the relation of Energy-Based Models with other generative models, sampling and statistical physics: a comprehensive reviewcited by 5
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problemscited by 5
- 2024Efficient Combinatorial Optimization via Heat Diffusioncited by 2
- 2024Inverse design with conditional cascaded diffusion modelscited by 2
- 2025TS-Diff: Two-Stage Diffusion Model for Low-Light RAW Image Enhancementcited by 2
- 2025PID-controlled Langevin Dynamics for Faster Sampling of Generative Modelscited by 0
- 2025PID-controlled Langevin Dynamics for Faster Sampling of Generative Models16 Nov 2025
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problems23 May 2025
- 2025TS-Diff: Two-Stage Diffusion Model for Low-Light RAW Image Enhancement7 May 2025
- 2024Inverse design with conditional cascaded diffusion models16 Aug 2024
- 2024Hitchhiker's guide on the relation of Energy-Based Models with other generative models, sampling and statistical physics: a comprehensive review19 Jun 2024
- 2024Efficient Combinatorial Optimization via Heat Diffusion13 Mar 2024
- 2023FRDiff : Feature Reuse for Universal Training-free Acceleration of Diffusion Models6 Dec 2023
- 2023HiDiffusion: Unlocking Higher-Resolution Creativity and Efficiency in Pretrained Diffusion Models29 Nov 2023
- 2023ExposureDiffusion: Learning to Expose for Low-light Image Enhancement15 Jul 2023
- 2023Multi-Architecture Multi-Expert Diffusion Models8 Jun 2023
- 2023Image generation with shortest path diffusion1 Jun 2023
- 2023WaveDM: Wavelet-Based Diffusion Models for Image Restoration23 May 2023
7 more
- 2023Solving Linear Inverse Problems using Higher-Order Annealed Langevin Diffusion8 May 2023
- 2023Preconditioned Score-based Generative Models13 Feb 2023
- 2022Multiscale Structure Guided Diffusion for Image Deblurring4 Dec 2022
- 2022Wavelet Diffusion Models are fast and scalable Image Generators29 Nov 2022
- 2022Diffusion Probabilistic Model Made Slim27 Nov 2022
- 2022Conversion Between CT and MRI Images Using Diffusion and Score-Matching Models24 Sep 2022
- 2022Diffusion Models Beat GANs on Topology Optimization20 Aug 2022
- 2022Diffusion Probabilistic Model Made Slimcited by 121
- 2022Wavelet Diffusion Models are fast and scalable Image Generatorscited by 116
- 2022Diffusion Models Beat GANs on Topology Optimizationcited by 79
- 2022Multiscale Structure Guided Diffusion for Image Deblurringcited by 76
- 2023WaveDM: Wavelet-Based Diffusion Models for Image Restorationcited by 46
- 2023Multi-Architecture Multi-Expert Diffusion Modelscited by 34
- 2023ExposureDiffusion: Learning to Expose for Low-light Image Enhancementcited by 29
- 2023HiDiffusion: Unlocking Higher-Resolution Creativity and Efficiency in Pretrained Diffusion Modelscited by 27
- 2023FRDiff : Feature Reuse for Universal Training-free Acceleration of Diffusion Modelscited by 19
- 2023Image generation with shortest path diffusioncited by 8
- 2023Preconditioned Score-based Generative Modelscited by 6
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problemscited by 5
12 most cited of 14 All 14 (JSON)
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problems23 May 2025
- 2025TS-Diff: Two-Stage Diffusion Model for Low-Light RAW Image Enhancement7 May 2025
- 2024Inverse design with conditional cascaded diffusion models16 Aug 2024
- 2023FRDiff : Feature Reuse for Universal Training-free Acceleration of Diffusion Models6 Dec 2023
- 2023HiDiffusion: Unlocking Higher-Resolution Creativity and Efficiency in Pretrained Diffusion Models29 Nov 2023
- 2023ExposureDiffusion: Learning to Expose for Low-light Image Enhancement15 Jul 2023
- 2023Multi-Architecture Multi-Expert Diffusion Models8 Jun 2023
- 2023Image generation with shortest path diffusion1 Jun 2023
- 2023WaveDM: Wavelet-Based Diffusion Models for Image Restoration23 May 2023
- 2023Preconditioned Score-based Generative Models13 Feb 2023
- 2022Multiscale Structure Guided Diffusion for Image Deblurring4 Dec 2022
- 2022Wavelet Diffusion Models are fast and scalable Image Generators29 Nov 2022
12 newest of 14 All 14 (JSON)
- 2022Conversion Between CT and MRI Images Using Diffusion and Score-Matching Modelscited by 69
- 2023Solving Linear Inverse Problems using Higher-Order Annealed Langevin Diffusioncited by 6
- 2024Hitchhiker's guide on the relation of Energy-Based Models with other generative models, sampling and statistical physics: a comprehensive reviewcited by 5
- 2024Efficient Combinatorial Optimization via Heat Diffusioncited by 2
- 2025PID-controlled Langevin Dynamics for Faster Sampling of Generative Modelscited by 0
- 2025PID-controlled Langevin Dynamics for Faster Sampling of Generative Models16 Nov 2025
- 2024Hitchhiker's guide on the relation of Energy-Based Models with other generative models, sampling and statistical physics: a comprehensive review19 Jun 2024
- 2024Efficient Combinatorial Optimization via Heat Diffusion13 Mar 2024
- 2023Solving Linear Inverse Problems using Higher-Order Annealed Langevin Diffusion8 May 2023
- 2022Conversion Between CT and MRI Images Using Diffusion and Score-Matching Models24 Sep 2022
Similar, not cited
Close in wording, no citation either way
- 2023A Complete Recipe for Diffusion Generative Models
- 2022Accelerating Score-based Generative Models for High-Resolution Image Synthesis
- 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptions
- 2024Accelerating Convergence of Score-Based Diffusion Models, Provably
- 2024A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models
- 2024A Practical Diffusion Path for Sampling
- 2024Score-based Generative Models with Adaptive Momentum
- 2021Conditional Image Generation with Score-Based Diffusion Models