Fast Sampling of Diffusion Models via Operator Learning
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion models have found widespread adoption in various areas. However, their sampling process is slow because it requires hundreds to thousands of network evaluations to emulate a continuous process defined by differential equations. In this work, we use neural operators, an efficient method to solve the probability flow differential equations, to accelerate the sampling process of diffusion models. Compared to other fast sampling methods that have a sequential nature, we are the first to propose a parallel decoding method that generates images with only one model forward pass. We propose diffusion model sampling with neural operator (DSNO) that maps the initial condition, i.e., Gaussian distribution, to the continuous-time solution trajectory of the reverse diffusion process. To model the temporal correlations along the trajectory, we introduce temporal convolution layers that are parameterized in the Fourier space into the given diffusion model backbone. We show our method achieves state-of-the-art FID of 3.78 for CIFAR-10 and 7.83 for ImageNet-64 in the one-model-evaluation setting.
Key facts. Cited by 175 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 36 of its 47 references that are in the map. First posted on arXiv 24 Nov 2022.
Builds on
36 of 47 references.
Closest ancestors · 20
Same direction, newest first
- 2022GENIE: Higher-Order Denoising Diffusion Solvers11 Oct
- 2022On Distillation of Guided Diffusion Models6 Oct
- 2022All are Worth Words: A ViT Backbone for Diffusion Models25 Sep
- 2022Estimating the Optimal Covariance with Imperfect Mean in Diffusion Probabilistic Models15 Jun
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2022Fast Sampling of Diffusion Models with Exponential Integrator29 Apr
- 2022Pseudo Numerical Methods for Diffusion Models on Manifolds20 Feb
- 2022Learning Fast Samplers for Diffusion Models by Differentiating Through Sample Quality11 Feb
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models17 Jan
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANs15 Dec
8 more
- 2021Bilateral Denoising Diffusion Models26 Aug
- 2021Score-based Generative Modeling in Latent Space10 Jun
- 2021On Fast Sampling of Diffusion Probabilistic Models31 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Knowledge Distillation in Iterative Generative Models for Improved Sampling Speed7 Jan
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 16
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Generative Adversarial Networkscited by 5,627
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,626
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
- 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 734
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 516
- 2022Diffusion Models for Adversarial Purificationcited by 476
4 more
- 2021Grad-TTS: A Diffusion Probabilistic Model for Text-to-Speechcited by 447
- 2021Neural Operator: Learning Maps Between Function Spacescited by 414
- 2022(Certified!!) Adversarial Robustness for Free!cited by 132
- 2021Seismic wave propagation and inversion with Neural Operatorscited by 51
Built on by
Citing papers per year: 2023: 42; 2024: 71; 2025: 47; 2026: 15.
175 papers.
Sort:
By area:
- 2023Consistency Modelscited by 1,822
- 2023One-step Diffusion with Distribution Matching Distillationcited by 1,019
- 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inferencecited by 846
- 2024Improved Distribution Matching Distillation for Fast Image Synthesiscited by 684
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 459
- 2023Improved Techniques for Training Consistency Modelscited by 439
- 2023DeepCache: Accelerating Diffusion Models for Freecited by 359
- 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Modelscited by 283
- 2023SinSR: Diffusion-Based Image Super-Resolution in a Single Stepcited by 238
- 2023Diff-Instruct: A Universal Approach for Transferring Knowledge From Pre-trained Diffusion Modelscited by 236
- 2024Score identity Distillation: Exponentially Fast Distillation of Pretrained Diffusion Models for One-Step Generationcited by 214
- 2023Fast Training of Diffusion Models with Masked Transformerscited by 176
38 more
- 2024Consistency Models Made Easycited by 144
- 2023TRACT: Denoising Diffusion Models with Transitive Closure Time-Distillationcited by 116
- 2024One-Step Image Translation with Text-to-Image Modelscited by 116
- 2025Inductive Moment Matchingcited by 113
- 2024Is Sora a World Simulator? A Comprehensive Survey on General World Models and Beyondcited by 111
- 2023Residual Corrective Diffusion Modeling for Km-scale Atmospheric Downscalingcited by 110
- 2023Diffusion Models for Image Restoration and Enhancement: A Comprehensive Surveycited by 109
- 2024Flow map matching with stochastic interpolants: A mathematical framework for consistency modelscited by 109
- 2024Improving the Training of Rectified Flowscited by 108
- 2024Consistency Flow Matching: Defining Straight Flows with Velocity Consistencycited by 104
- 2025How to build a consistency model: Learning flow maps via self-distillationcited by 103
- 2024Align Your Steps: Optimizing Sampling Schedules in Diffusion Modelscited by 96
- 2023Compositional Foundation Models for Hierarchical Planningcited by 95
- 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 89
- 2024One-Step Diffusion Policy: Fast Visuomotor Policies via Diffusion Distillationcited by 89
- 2024T2V-Turbo: Breaking the Quality Bottleneck of Video Consistency Model with Mixed Reward Feedbackcited by 87
- 2025Align Your Flow: Scaling Continuous-Time Flow Map Distillationcited by 87
- 2024Distilling Diffusion Models into Conditional GANscited by 78
- 2023A Mathematical Guide to Operator Learningcited by 76
- 2024One-Step Diffusion Distillation through Score Implicit Matchingcited by 71
- 2024Multistep Consistency Modelscited by 67
- 2024EM Distillation for One-step Diffusion Modelscited by 62
- 2023One-Step Diffusion Distillation via Deep Equilibrium Modelscited by 61
- 2023On the Design Fundamentals of Diffusion Models: A Surveycited by 60
- 2023StreamDiffusion: A Pipeline-level Solution for Real-time Interactive Generationcited by 58
- 2024Flash Diffusion: Accelerating Any Conditional Diffusion Model for Few Steps Image Generationcited by 56
- 2024On the Trajectory Regularity of ODE-based Diffusion Samplingcited by 55
- 2024Distillation of Discrete Diffusion through Dimensional Correlationscited by 48
- 2023On Exact Inversion of DPM-Solverscited by 47
- 2024Adversarial Score identity Distillation: Rapidly Surpassing the Teacher in One Stepcited by 47
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithmscited by 47
- 2023A Comprehensive Survey on Knowledge Distillation of Diffusion Modelscited by 45
- 2024Neural Flow Diffusion Models: Learnable Forward Process for Improved Diffusion Modellingcited by 45
- 2025One-step Diffusion Models with f-Divergence Distribution Matchingcited by 43
- 2024Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time Complexitycited by 42
- 2024Equivariant Graph Neural Operator for Modeling 3D Dynamicscited by 38
- 2023Bespoke Solvers for Generative Flow Modelscited by 37
- 2025LBM: Latent Bridge Matching for Fast Image-to-Image Translationcited by 36
50 most cited of 175 All 175 (JSON)
- 2026Data-free Multistep Diffusion Distillation13 Oct 2026
- 2026Iterating Consistency Models: Stability, Error Bounds and Noise Schedules2 Oct 2026
- 2026Uncertainty-Aware Consistency Distillation for Few-Step Video Generation30 Sep 2026
- 2026From Draft to Draft-Free: One-Step Video Object Removal via Privileged Distillation and Fast Planting16 Jul 2026
- 2026FlashDiff: Efficient Regional Execution and Scheduling for Diffusion Model Serving13 Jul 2026
- 2026Mitigating the Contractivity Trap in Diffusion ODEs via Stein Stabilization5 Jun 2026
- 2026Midpoint Generative Models28 May 2026
- 2026Learning to Think in Physics: Breaking Shortcut Learning in Scientific Diffusion via Representation Alignment20 May 2026
- 2026Efficient Diffusion Distillation via Embedding Loss24 Apr 2026
- 2026You Only Erase Once: Erasing Anything without Bringing Unexpected Content29 Mar 2026
- 2026Inverse Neural Operator for ODE Parameter Optimization12 Mar 2026
- 2026Color Matters: Demosaicing-Guided Color Correlation Training for Generalizable AI-Generated Image Detection30 Jan 2026
38 more
- 2026Learning Hamiltonian Flow Maps: Mean Flow Consistency for Large-Timestep Molecular Dynamics29 Jan 2026
- 2026Transition Matching Distillation for Fast Video Generation14 Jan 2026
- 2026Expanding the Chaos: Neural Operator for Stochastic (Partial) Differential Equations3 Jan 2026
- 2025ReDiF: Reinforced Distillation for Few Step Diffusion28 Dec 2025
- 2025Self-Supervised AI-Generated Image Detection: A Camera Metadata Perspective5 Dec 2025
- 2025Flow Map Distillation Without Data24 Nov 2025
- 2025Understanding, Accelerating, and Improving MeanFlow Training24 Nov 2025
- 2025Test-Time Iterative Error Correction for Efficient Diffusion Models9 Nov 2025
- 2025Balanced conic rectified flow29 Oct 2025
- 2025Adaptive Discretization for Consistency Models20 Oct 2025
- 2025Rectified-CFG++ for Flow Based Models9 Oct 2025
- 2025Align Your Tangent: Training Better Consistency Models via Manifold-Aligned Tangents1 Oct 2025
- 2025EVODiff: Entropy-aware Variance Optimized Diffusion Inference30 Sep 2025
- 2025CMT: Mid-Training for Efficient Learning of Consistency, Mean Flow, and Flow Map Models29 Sep 2025
- 2025Modular MeanFlow: Towards Stable and Scalable One-Step Generative Modeling24 Aug 2025
- 2025Learning to See Through Flare19 Aug 2025
- 2025TeEFusion: Blending Text Embeddings to Distill Classifier-Free Guidance24 Jul 2025
- 2025Bridging GANs and Bayesian Neural Networks via Partial Stochasticity1 Jul 2025
- 2025From Cradle to Cane: A Two-Pass Framework for High-Fidelity Lifespan Face Aging26 Jun 2025
- 2025Align Your Flow: Scaling Continuous-Time Flow Map Distillation17 Jun 2025
- 2025DiffPR: Diffusion-Based Phase Reconstruction via Frequency-Decoupled Learning12 Jun 2025
- 2025Principled Approaches for Extending Neural Architectures to Function Spaces for Operator Learning12 Jun 2025
- 2025Generative Algorithms for Wildfire Progression Reconstruction from Multi-Modal Satellite Active Fire Measurements and Terrain Height12 Jun 2025
- 2025Geometric Regularity in Deterministic Sampling Dynamics of Diffusion-based Generative Models11 Jun 2025
- 2025Anomaly Detection and Generation with Diffusion Models: A Survey11 Jun 2025
- 2025Dual-Expert Consistency Model for Efficient and High-Quality Video Generation3 Jun 2025
- 2025Physics-Informed Distillation of Diffusion Models for PDE-Constrained Generation28 May 2025
- 2025One-Way Ticket:Time-Independent Unified Encoder for Distilling Text-to-Image Diffusion Models28 May 2025
- 2025Uni-Instruct: One-step Diffusion Model through Unified Diffusion Divergence Instruction27 May 2025
- 2025Latent Mamba Operator for Partial Differential Equations25 May 2025
- 2025How to build a consistency model: Learning flow maps via self-distillation24 May 2025
- 2025Learning to Integrate Diffusion ODEs by Averaging the Derivatives20 May 2025
- 2025One-Step Offline Distillation of Diffusion-based Models via Koopman Modeling19 May 2025
- 2025Distilling Two-Timed Flow Models by Separately Matching Initial and Terminal Velocities2 May 2025
- 2025Integration Flow Models28 Apr 2025
- 2025Heat Diffusion Models -- Interpixel Attention Mechanism28 Apr 2025
- 2025Fast Autoregressive Models for Continuous Latent Generation24 Apr 2025
- 2025Autoregressive Distillation of Diffusion Transformers15 Apr 2025
50 newest of 175 All 175 (JSON)
- 2023Consistency Modelscited by 1,822
- 2023One-step Diffusion with Distribution Matching Distillationcited by 1,019
- 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inferencecited by 846
- 2024Improved Distribution Matching Distillation for Fast Image Synthesiscited by 684
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 459
- 2023Improved Techniques for Training Consistency Modelscited by 439
- 2023DeepCache: Accelerating Diffusion Models for Freecited by 359
- 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Modelscited by 283
- 2023SinSR: Diffusion-Based Image Super-Resolution in a Single Stepcited by 238
- 2023Diff-Instruct: A Universal Approach for Transferring Knowledge From Pre-trained Diffusion Modelscited by 236
- 2024Score identity Distillation: Exponentially Fast Distillation of Pretrained Diffusion Models for One-Step Generationcited by 214
- 2023Fast Training of Diffusion Models with Masked Transformerscited by 176
12 most cited of 135 All 135 (JSON)
- 2026Data-free Multistep Diffusion Distillation13 Oct 2026
- 2026Iterating Consistency Models: Stability, Error Bounds and Noise Schedules2 Oct 2026
- 2026FlashDiff: Efficient Regional Execution and Scheduling for Diffusion Model Serving13 Jul 2026
- 2026Mitigating the Contractivity Trap in Diffusion ODEs via Stein Stabilization5 Jun 2026
- 2026Midpoint Generative Models28 May 2026
- 2026Learning to Think in Physics: Breaking Shortcut Learning in Scientific Diffusion via Representation Alignment20 May 2026
- 2026Efficient Diffusion Distillation via Embedding Loss24 Apr 2026
- 2026You Only Erase Once: Erasing Anything without Bringing Unexpected Content29 Mar 2026
- 2026Transition Matching Distillation for Fast Video Generation14 Jan 2026
- 2025ReDiF: Reinforced Distillation for Few Step Diffusion28 Dec 2025
- 2025Test-Time Iterative Error Correction for Efficient Diffusion Models9 Nov 2025
- 2025Balanced conic rectified flow29 Oct 2025
12 newest of 135 All 135 (JSON)
- 2025Latent Mamba Operator for Partial Differential Equationscited by 10
- 2025Latent Mamba Operator for Partial Differential Equations25 May 2025
- 2023Compositional Foundation Models for Hierarchical Planningcited by 95
- 2023A Mathematical Guide to Operator Learningcited by 76
- 2023StreamDiffusion: A Pipeline-level Solution for Real-time Interactive Generationcited by 58
- 2024Equivariant Graph Neural Operator for Modeling 3D Dynamicscited by 38
- 2023Bespoke Solvers for Generative Flow Modelscited by 37
- 2025LBM: Latent Bridge Matching for Fast Image-to-Image Translationcited by 36
- 2023CaloClouds II: Ultra-Fast Geometry-Independent Highly-Granular Calorimeter Simulationcited by 20
- 2024Imagination Policy: Using Generative Point Cloud Models for Learning Manipulation Policiescited by 18
- 2025Flow Map Distillation Without Datacited by 14
- 2024DiffFNO: Diffusion Fourier Neural Operatorcited by 13
- 2025Principled Approaches for Extending Neural Architectures to Function Spaces for Operator Learningcited by 12
- 2024StainDiffuser: MultiTask Dual Diffusion Model for Virtual Stainingcited by 11
12 most cited of 37 All 37 (JSON)
- 2026Uncertainty-Aware Consistency Distillation for Few-Step Video Generation30 Sep 2026
- 2026From Draft to Draft-Free: One-Step Video Object Removal via Privileged Distillation and Fast Planting16 Jul 2026
- 2026Inverse Neural Operator for ODE Parameter Optimization12 Mar 2026
- 2026Color Matters: Demosaicing-Guided Color Correlation Training for Generalizable AI-Generated Image Detection30 Jan 2026
- 2026Learning Hamiltonian Flow Maps: Mean Flow Consistency for Large-Timestep Molecular Dynamics29 Jan 2026
- 2026Expanding the Chaos: Neural Operator for Stochastic (Partial) Differential Equations3 Jan 2026
- 2025Self-Supervised AI-Generated Image Detection: A Camera Metadata Perspective5 Dec 2025
- 2025Flow Map Distillation Without Data24 Nov 2025
- 2025Understanding, Accelerating, and Improving MeanFlow Training24 Nov 2025
- 2025Modular MeanFlow: Towards Stable and Scalable One-Step Generative Modeling24 Aug 2025
- 2025Learning to See Through Flare19 Aug 2025
- 2025TeEFusion: Blending Text Embeddings to Distill Classifier-Free Guidance24 Jul 2025
12 newest of 37 All 37 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025TADA: Improved Diffusion Sampling with Training-free Augmented Dynamics
- 2025ODEt(ODEl): Shortcutting the Time and the Length in Diffusion and Flow Models for Faster Sampling
- 2023Accelerating Guided Diffusion Sampling with Splitting Numerical Methods
- 2024An Ordinary Differential Equation Sampler with Stochastic Start for Diffusion Bridge Models
- 2023Fast ODE-based Sampling for Diffusion Models in Around 5 Steps
- 2025Are First-Order Diffusion Samplers Really Slower? A Fast Forward-Value Approach
- 2023Reflected Diffusion Models
- 2022Diffusion Models in Vision: A Survey