Improved Techniques for Training Score-Based Generative Models
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Score-based generative models can produce high quality image samples comparable to GANs, without requiring adversarial optimization. However, existing training procedures are limited to images of low resolution (typically below 32x32), and can be unstable under some settings. We provide a new theoretical analysis of learning and sampling from score models in high dimensional spaces, explaining existing failure modes and motivating new solutions that generalize across datasets. To enhance stability, we also propose to maintain an exponential moving average of model weights. With these improvements, we can effortlessly scale score-based generative models to images with unprecedented resolutions ranging from 64x64 to 256x256. Our score-based models can generate high-fidelity samples that rival best-in-class GANs on various image datasets, including CelebA, FFHQ, and multiple LSUN categories.
Key facts. Cited by 1,036 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 20 of its 32 references that are in the map. First posted on arXiv 16 Jun 2020.
Builds on
20 of 32 references.
References · 20
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,967
- 2017Improved Training of Wasserstein GANscited by 2,223
- 2018Spectral Normalization for Generative Adversarial Networkscited by 1,618
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,376
- 2015LSUN: Construction of a Large-scale Image Dataset using Deep Learning with Humans in the Loopcited by 1,350
- 2015Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)cited by 1,061
- 2016Conditional Image Generation with PixelCNN Decoderscited by 859
- 2019On the Convergence of Adam and Beyondcited by 834
8 more
- 2016RefineNet: Multi-Path Refinement Networks for High-Resolution Semantic Segmentationcited by 463
- 2018Assessing Generative Models via Precision and Recallcited by 401
- 2019Sliced Score Matching: A Scalable Approach to Density and Score Estimationcited by 381
- 2018A Note on the Inception Scorecited by 225
- 2017BEGAN: Boundary Equilibrium Generative Adversarial Networkscited by 121
- 2019HYPE: A Benchmark for Human eYe Perceptual Evaluation of Generative Modelscited by 59
- 2019Neural Empirical Bayescited by 46
- 2019Bridging the Gap Between f-GANs and Wasserstein GANscited by 23
Built on by
Citing papers per year: 2020: 3; 2021: 69; 2022: 134; 2023: 294; 2024: 268; 2025: 187; 2026: 81.
1,036 papers.
Sort:
By area:
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasetscited by 2,526
- 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
- 2023Consistency Modelscited by 1,778
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2021Image Super-Resolution via Iterative Refinementcited by 1,506
38 more
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2021Palette: Image-to-Image Diffusion Modelscited by 1,296
- 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,155
- 2021Variational Diffusion Modelscited by 1,145
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applicationscited by 1,107
- 2022Human Motion Diffusion Modelcited by 1,062
- 2022Diffusion Models in Vision: A Surveycited by 1,008
- 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 921
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 835
- 2022Pseudo Numerical Methods for Diffusion Models on Manifoldscited by 695
- 2022Point-E: A System for Generating 3D Point Clouds from Complex Promptscited by 694
- 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 670
- 2021Score-based Generative Modeling in Latent Spacecited by 634
- 2021ILVR: Conditioning Method for Denoising Diffusion Probabilistic Modelscited by 634
- 2022Score Jacobian Chaining: Lifting Pretrained 2D Diffusion Models for 3D Generationcited by 597
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 577
- 2023Erasing Concepts from Diffusion Modelscited by 530
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANscited by 515
- 2021Solving Inverse Problems in Medical Imaging with Score-Based Generative Modelscited by 514
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 511
- 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 498
- 2021Label-Efficient Semantic Segmentation with Diffusion Modelscited by 495
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 448
- 2021Grad-TTS: A Diffusion Probabilistic Model for Text-to-Speechcited by 445
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
- 2023Improved Techniques for Training Consistency Modelscited by 430
- 2023InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generationcited by 387
- 2022TabDDPM: Modelling Tabular Data with Diffusion Modelscited by 373
- 2023Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Modelscited by 371
- 2024SV3D: Novel Multi-view Synthesis and 3D Generation from a Single Image using Latent Video Diffusioncited by 364
- 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 362
- 2022DiffusionDet: Diffusion Model for Object Detectioncited by 354
- 2024Retrieval-Augmented Generation for AI-Generated Content: A Surveycited by 352
- 2021Knowledge Distillation in Iterative Generative Models for Improved Sampling Speedcited by 344
- 2022Understanding Diffusion Models: A Unified Perspectivecited by 337
- 2021Autoregressive Denoising Diffusion Models for Multivariate Probabilistic Time Series Forecastingcited by 332
- 2021Score-based diffusion models for accelerated MRIcited by 308
50 most cited of 1,036 All 1,036 (JSON)
- 2026Distilling Diffusion Score Discrepancy for Efficient Training Data Attribution30 Sep 2026
- 2026Sharp Convergence and Sampling Trade-offs for Riemannian Diffusion under Nonnegative Ricci Curvature29 Sep 2026
- 2026One-Step Next-Latent Prediction Is Not a World Model28 Sep 2026
- 2026FluxLite: Inference-Time Proposal Control for Discrete Diffusion Models28 Sep 2026
- 2026Mean Velocity Matching: Rethinking Generative Dynamics in Diffusion Models21 Sep 2026
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2026DySCo: Dynamically consistent data-driven downscaling of extremes in climate projections22 Aug 2026
- 2026SlimDiffuSE: Towards Efficient Diffusion-Based Speech Enhancement using Slimmable Networks21 Aug 2026
- 2026Generative Learning of Separatrices13 Aug 2026
- 2026A Shortcut to Statistically Steady-State Turbulence with Flow Matching14 Jul 2026
- 2026Robust Face Super-Resolution and Recognition Through Multi-Feature Aggregation in Diffusion Models6 Jul 2026
- 2026Spatial Attention: Adapting Execution Horizons for Diffusion Policies via Observation Sensitivity6 Jul 2026
38 more
- 2026Reflected Schr"odinger Bridge Matching3 Jul 2026
- 2026SwiftAudio: Data-Efficient Caption-Only Distillation for One-Step Text-to-Audio Diffusion-based Generation30 Jun 2026
- 2026Stochastic Optimal Control Sampling for Diffusion Inverse Problems27 Jun 2026
- 2026Improving Richardson--Lucy Deconvolution with Diffusion Priors for Fluorescence Microscopy24 Jun 2026
- 2026MGI: Member vs Generated Inference22 Jun 2026
- 2026VT-DUDA: Visual Token Conditioning for Diffusion-guided Unsupervised Domain Adaptation19 Jun 2026
- 2026Omega: Operator-based Mixture Ensemble for Generative Assimilation18 Jun 2026
- 2026Kolmogorov Regression for Robust Diffusion Policies16 Jun 2026
- 2026CaricHarmony: Contrastive Diffusion Paths for Identity-Preserving Caricature Synthesis11 Jun 2026
- 2026Recursively Trained Diffusion Models: Limiting Collapse Distribution and Spectral Characterization11 Jun 2026
- 2026Mitigating the Contractivity Trap in Diffusion ODEs via Stein Stabilization5 Jun 2026
- 2026Fast and Lightweight Novel View Synthesis with Differentiable Multiplane Image1 Jun 2026
- 2026Latent Diffusion Pretraining for Crystal Property Prediction30 May 2026
- 2026Zeroth-Order Non-Log-Concave Sampling with Variance Reduction and Applications to Inverse Problems28 May 2026
- 2026Self-Intersection-Aware 3D Human Motion Generation Using an Efficient Human Sphere Proxy26 May 2026
- 2026Theoretical guidelines for annealed Langevin dynamics in compositional simulation-based inference20 May 2026
- 2026Generative Recursive Reasoning19 May 2026
- 2026Fine-tuning Pocket-Aware Diffusion Models via Denoising Policy Optimization17 May 2026
- 2026Accelerating Redshift-Conditioned Galaxy Image Synthesis with One-step Generative Modeling17 May 2026
- 2026StAD: Stein Amortized Divergence for Fast Likelihoods with Diffusion and Flow15 May 2026
- 2026Dimension-Uniform Discretization Analysis of Preconditioned Annealed Langevin Dynamics for Multimodal Gaussian Mixtures15 May 2026
- 2026Bridging Domain Gaps with Target-Aligned Generation for Offline Reinforcement Learning13 May 2026
- 2026Is Monotonic Sampling Necessary in Diffusion Models?12 May 2026
- 2026Product-of-Gaussian-Mixture Diffusion Models for Joint Nonlinear MRI Reconstruction11 May 2026
- 2026Consistency Regularised Gradient Flows for Inverse Problems8 May 2026
- 2026A Unified Measure-Theoretic View of Diffusion, Score-Based, and Flow Matching Generative Models7 May 2026
- 2026Tempered Guided Diffusion5 May 2026
- 2026Phase-map synthesis from magnitude-only MR images using conditional score-based diffusion models with application in training of accelerated MRI reconstruction models2 May 2026
- 2026Beyond Continuity: Simulation-free Reconstruction of Discrete Branching Dynamics from Single-cell Snapshots1 May 2026
- 2026Exploring Time Conditioning in Diffusion Generative Models from Disjoint Noisy Data Manifolds28 Apr 2026
- 2026Diffusion Model as a Generalist Segmentation Learner27 Apr 2026
- 2026On the Memorization of Consistency Distillation for Diffusion Models26 Apr 2026
- 2026One Scale at a Time: Scale-Autoregressive Modeling for Fluid Flow Distributions13 Apr 2026
- 2026Score Shocks: The Burgers Equation Structure of Diffusion Generative Models8 Apr 2026
- 2026Diffusion Processes on Implicit Manifolds8 Apr 2026
- 2026Visual Prototype Conditioned Focal Region Generation for UAV-Based Object Detection3 Apr 2026
- 2026DiffAttn: Diffusion-Based Drivers' Visual Attention Prediction with LLM-Enhanced Semantic Reasoning30 Mar 2026
- 2026CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Think19 Mar 2026
50 newest of 1,036 All 1,036 (JSON)
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
- 2023Consistency Modelscited by 1,778
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2021Palette: Image-to-Image Diffusion Modelscited by 1,296
- 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,155
- 2021Variational Diffusion Modelscited by 1,145
12 most cited of 688 All 688 (JSON)
- 2026Distilling Diffusion Score Discrepancy for Efficient Training Data Attribution30 Sep 2026
- 2026Sharp Convergence and Sampling Trade-offs for Riemannian Diffusion under Nonnegative Ricci Curvature29 Sep 2026
- 2026FluxLite: Inference-Time Proposal Control for Discrete Diffusion Models28 Sep 2026
- 2026Mean Velocity Matching: Rethinking Generative Dynamics in Diffusion Models21 Sep 2026
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2026SlimDiffuSE: Towards Efficient Diffusion-Based Speech Enhancement using Slimmable Networks21 Aug 2026
- 2026A Shortcut to Statistically Steady-State Turbulence with Flow Matching14 Jul 2026
- 2026Robust Face Super-Resolution and Recognition Through Multi-Feature Aggregation in Diffusion Models6 Jul 2026
- 2026Spatial Attention: Adapting Execution Horizons for Diffusion Policies via Observation Sensitivity6 Jul 2026
- 2026Reflected Schr"odinger Bridge Matching3 Jul 2026
- 2026SwiftAudio: Data-Efficient Caption-Only Distillation for One-Step Text-to-Audio Diffusion-based Generation30 Jun 2026
- 2026Stochastic Optimal Control Sampling for Diffusion Inverse Problems27 Jun 2026
12 newest of 688 All 688 (JSON)
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasetscited by 2,526
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2022Learning to Generate Realistic LiDAR Point Cloudscited by 91
- 2023LiDAR Data Synthesis with Denoising Diffusion Probabilistic Modelscited by 53
- 2024Fast LiDAR Data Generation with Rectified Flowscited by 13
- 2025EDELINE: Enhancing Memory in Diffusion-based World Models via Linear-Time Sequence Modelingcited by 3
- 2024Simultaneous Diffusion Sampling for Conditional LiDAR Generationcited by 1
- 2025LiDARDraft: Generating LiDAR Point Cloud from Versatile Inputscited by 1
- 2025LaGen: Towards Autoregressive LiDAR Scene Generationcited by 0
- 2025LiDARDraft: Generating LiDAR Point Cloud from Versatile Inputs23 Dec 2025
- 2025LaGen: Towards Autoregressive LiDAR Scene Generation26 Nov 2025
- 2025EDELINE: Enhancing Memory in Diffusion-based World Models via Linear-Time Sequence Modeling1 Feb 2025
- 2024Fast LiDAR Data Generation with Rectified Flows3 Dec 2024
- 2024Simultaneous Diffusion Sampling for Conditional LiDAR Generation15 Oct 2024
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasets25 Nov 2023
- 2023LiDAR Data Synthesis with Denoising Diffusion Probabilistic Models17 Sep 2023
- 2022Learning to Generate Realistic LiDAR Point Clouds8 Sep 2022
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul 2021
- 2022Know Your Boundaries: The Necessity of Explicit Behavioral Cloning in Offline RLcited by 21
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matchingcited by 17
- 2024Q-Distribution guided Q-learning for offline reinforcement learning: Uncertainty penalized Q-value via consistency modelcited by 4
- 2025Noise-conditioned Energy-based Annealed Rewards (NEAR): A Generative Framework for Imitation Learning from Observationcited by 1
- 2026Bridging Domain Gaps with Target-Aligned Generation for Offline Reinforcement Learningcited by 0
- 2026Bridging Domain Gaps with Target-Aligned Generation for Offline Reinforcement Learning13 May 2026
- 2025Noise-conditioned Energy-based Annealed Rewards (NEAR): A Generative Framework for Imitation Learning from Observation24 Jan 2025
- 2024Q-Distribution guided Q-learning for offline reinforcement learning: Uncertainty penalized Q-value via consistency model27 Oct 2024
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directions21 Feb 2024
- 2023Fighting Uncertainty with Gradients: Offline Reinforcement Learning via Diffusion Score Matching24 Jun 2023
- 2022Know Your Boundaries: The Necessity of Explicit Behavioral Cloning in Offline RL1 Jun 2022
- 2022Rickrolling the Artist: Injecting Backdoors into Text Encoders for Text-to-Image Synthesiscited by 59
- 2024Machine Unlearning for Image-to-Image Generative Modelscited by 47
- 2025Radioactive Watermarks in Diffusion and Autoregressive Image Generative Modelscited by 6
- 2024Controllable Unlearning for Image-to-Image Generative Models via varepsilon-Constrained Optimizationcited by 5
- 2026ALIEN: Analytic Latent Watermarking for Controllable Generationcited by 0
- 2026MGI: Member vs Generated Inferencecited by 0
- 2026MGI: Member vs Generated Inference22 Jun 2026
- 2026ALIEN: Analytic Latent Watermarking for Controllable Generation5 Feb 2026
- 2025Radioactive Watermarks in Diffusion and Autoregressive Image Generative Models30 Jun 2025
- 2024Controllable Unlearning for Image-to-Image Generative Models via varepsilon-Constrained Optimization3 Aug 2024
- 2024Machine Unlearning for Image-to-Image Generative Models1 Feb 2024
- 2022Rickrolling the Artist: Injecting Backdoors into Text Encoders for Text-to-Image Synthesis4 Nov 2022
- 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
- 2021Image Super-Resolution via Iterative Refinementcited by 1,506
- 2023RealFusion: 360deg Reconstruction of Any Object from a Single Imagecited by 305
- 2023Composer: Creative and Controllable Image Synthesis with Composable Conditionscited by 303
- 2021Robust Compressed Sensing MRI with Deep Generative Priorscited by 284
- 2021How to Train Your Energy-Based Modelscited by 271
- 2023Text2Room: Extracting Textured 3D Meshes from 2D Text-to-Image Modelscited by 245
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 216
- 2023MVDiffusion: Enabling Holistic Multi-view Image Generation with Correspondence-Aware Diffusioncited by 210
- 2021SNIPS: Solving Noisy Inverse Problems Stochasticallycited by 191
- 2023Image Restoration with Mean-Reverting Stochastic Differential Equationscited by 182
- 2023Scaling Robot Learning with Semantically Imagined Experiencecited by 175
12 most cited of 321 All 321 (JSON)
- 2026One-Step Next-Latent Prediction Is Not a World Model28 Sep 2026
- 2026DySCo: Dynamically consistent data-driven downscaling of extremes in climate projections22 Aug 2026
- 2026Generative Learning of Separatrices13 Aug 2026
- 2026VT-DUDA: Visual Token Conditioning for Diffusion-guided Unsupervised Domain Adaptation19 Jun 2026
- 2026Omega: Operator-based Mixture Ensemble for Generative Assimilation18 Jun 2026
- 2026Fast and Lightweight Novel View Synthesis with Differentiable Multiplane Image1 Jun 2026
- 2026Zeroth-Order Non-Log-Concave Sampling with Variance Reduction and Applications to Inverse Problems28 May 2026
- 2026Self-Intersection-Aware 3D Human Motion Generation Using an Efficient Human Sphere Proxy26 May 2026
- 2026Theoretical guidelines for annealed Langevin dynamics in compositional simulation-based inference20 May 2026
- 2026Generative Recursive Reasoning19 May 2026
- 2026StAD: Stein Amortized Divergence for Fast Likelihoods with Diffusion and Flow15 May 2026
- 2026Beyond Continuity: Simulation-free Reconstruction of Discrete Branching Dynamics from Single-cell Snapshots1 May 2026
12 newest of 321 All 321 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2021Score-Based Generative Classifiers
- 2023Score Mismatching for Generative Modeling
- 2023Assessing Robustness via Score-Based Adversarial Image Generation
- 2024Score Neural Operator: A Generative Model for Learning and Generalizing Across Multiple Probability Distributions
- 2024A Good Score Does not Lead to A Good Generative Model
- 2023SMaRt: Improving GANs with Score Matching Regularity
- 2024Score-based generative models break the curse of dimensionality in learning a family of sub-Gaussian probability distributions
- 2026Non-asymptotic Convergence of Stochastic Gradient Descent in Score-based Generative Models