A Flexible Diffusion Model
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion (score-based) generative models have been widely used for modeling various types of complex data, including images, audios, and point clouds. Recently, the deep connection between forward-backward stochastic differential equations (SDEs) and diffusion-based models has been revealed, and several new variants of SDEs are proposed (e.g., sub-VP, critically-damped Langevin) along this line. Despite the empirical success of the hand-crafted fixed forward SDEs, a great quantity of proper forward SDEs remain unexplored. In this work, we propose a general framework for parameterizing the diffusion model, especially the spatial part of the forward SDE. An abstract formalism is introduced with theoretical guarantees, and its connection with previous diffusion models is leveraged. We demonstrate the theoretical advantage of our method from an optimization perspective. Numerical experiments on synthetic datasets, MINIST and CIFAR10 are also presented to validate the effectiveness of our framework.
Key facts. Cited by 10 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 66 references that are in the map. First posted on arXiv 17 Jun 2022.
Builds on
36 of 66 references.
Closest ancestors · 15
Same direction, newest first
- 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
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusion14 Dec
- 2021Diffusion Normalizing Flow14 Oct
- 2021Variational Diffusion Models1 Jul
- 2021Score-based Generative Modeling in Latent Space10 Jun
- 2021A Variational Perspective on Diffusion-Based Generative Models and Score Matching5 Jun
- 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
3 more
Other references · 21
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2016Density estimation using Real NVPcited by 2,042
- 2018Glow: Generative Flow with Invertible 1x1 Convolutionscited by 1,545
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
- 2021Diffusion Probabilistic Models for 3D Point Cloud Generationcited by 735
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 677
- 2020NVAE: A Deep Hierarchical Variational Autoencodercited by 637
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 594
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 516
- 2019Sliced Score Matching: A Scalable Approach to Density and Score Estimationcited by 384
- 2019Residual Flows for Invertible Generative Modelingcited by 257
9 more
- 2020How to train your neural ODE: the world of Jacobian and kinetic regularizationcited by 228
- 2017A Conceptual Introduction to Hamiltonian Monte Carlocited by 209
- 2020DiffusionNet: Discretization Agnostic Learning on Surfacescited by 164
- 2021Learning Gradient Fields for Molecular Conformation Generationcited by 163
- 2020OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transportcited by 143
- 2021SE(3) Equivariant Graph Neural Networks with Complete Local Framescited by 76
- 2021DiffSVC: A Diffusion Probabilistic Model for Singing Voice Conversioncited by 39
- 2021Learning Disentangled Representation by Exploiting Pretrained Generative Models: A Contrastive Learning Viewcited by 34
- 2021Towards Building A Group-based Unsupervised Representation Disentanglement Frameworkcited by 22
Built on by
Citing papers per year: 2023: 5; 2024: 4; 2025: 1.
10 papers.
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2023Diffusion Models With Learned Adaptive Noisecited by 51
- 2024Neural Flow Diffusion Models: Learnable Forward Process for Improved Diffusion Modellingcited by 45
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 35
- 2023A Group Symmetric Stochastic Differential Equation Model for Molecule Multi-modal Pretrainingcited by 32
- 2023A Complete Recipe for Diffusion Generative Modelscited by 17
- 2023Molecule Joint Auto-Encoding: Trajectory Pretraining with 2D and 3D Diffusioncited by 10
- 2024What's the score? Automated Denoising Score Matching for Nonlinear Diffusionscited by 7
- 2025Flow Along the K-Amplitude for Generative Modelingcited by 3
- 2024Sculpting Molecules in Text-3D Space: A Flexible Substructure Aware Framework for Text-Oriented Molecular Optimizationcited by 0
- 2025Flow Along the K-Amplitude for Generative Modeling27 Apr 2025
- 2024What's the score? Automated Denoising Score Matching for Nonlinear Diffusions10 Jul 2024
- 2024Neural Flow Diffusion Models: Learnable Forward Process for Improved Diffusion Modelling19 Apr 2024
- 2024Sculpting Molecules in Text-3D Space: A Flexible Substructure Aware Framework for Text-Oriented Molecular Optimization6 Mar 2024
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraints28 Feb 2024
- 2023Diffusion Models With Learned Adaptive Noise20 Dec 2023
- 2023Molecule Joint Auto-Encoding: Trajectory Pretraining with 2D and 3D Diffusion6 Dec 2023
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimization4 Oct 2023
- 2023A Group Symmetric Stochastic Differential Equation Model for Molecule Multi-modal Pretraining28 May 2023
- 2023A Complete Recipe for Diffusion Generative Models3 Mar 2023
Similar, not cited
Close in wording, no citation either way
- 2026A Tutorial on Diffusion Theory: From Differential Equations to Diffusion Models
- 2022How Much is Enough? A Study on Diffusion Times in Score-based Generative Models
- 2023Closing the ODE-SDE gap in score-based diffusion models through the Fokker-Planck equation
- 2023Score-based Generative Modeling Through Backward Stochastic Differential Equations: Inversion and Generation
- 2025Generative diffusion models from a PDE perspective
- 2023Reflected Diffusion Models
- 2025Diffusion Models under Alternative Noise: Simplified Analysis and Sensitivity
- 2022An optimal control perspective on diffusion-based generative modeling