Beta Diffusion
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
We introduce beta diffusion, a novel generative modeling method that integrates demasking and denoising to generate data within bounded ranges. Using scaled and shifted beta distributions, beta diffusion utilizes multiplicative transitions over time to create both forward and reverse diffusion processes, maintaining beta distributions in both the forward marginals and the reverse conditionals, given the data at any point in time. Unlike traditional diffusion-based generative models relying on additive Gaussian noise and reweighted evidence lower bounds (ELBOs), beta diffusion is multiplicative and optimized with KL-divergence upper bounds (KLUBs) derived from the convexity of the KL divergence. We demonstrate that the proposed KLUBs are more effective for optimizing beta diffusion compared to negative ELBOs, which can also be derived as the KLUBs of the same KL divergence with its two arguments swapped. The loss function of beta diffusion, expressed in terms of Bregman divergence, further supports the efficacy of KLUBs for optimization. Experimental results on both synthetic data and natural images demonstrate the unique capabilities of beta diffusion in generative modeling of range-bounded data and validate the effectiveness of KLUBs in optimizing diffusion models, thereby making them valuable additions to the family of diffusion-based generative models and the optimization techniques used to train them.
Key facts. Cited by 17 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 60 of its 81 references that are in the map. First posted on arXiv 14 Sep 2023.
Builds on
60 of 81 references.
Closest ancestors · 34
Same direction, newest first
- 2023Fast Diffusion Model12 Jun
- 2023Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Models25 Apr
- 2023Diffusion Models for Constrained Domains11 Apr
- 2023Reflected Diffusion Models10 Apr
- 2023Efficient Diffusion Training via Min-SNR Weighting Strategy16 Mar
- 2023Optimizing DDPM Sampling with Shortcut Fine-Tuning31 Jan
- 2022Fast Sampling of Diffusion Models via Operator Learning24 Nov
- 2022Blurring Diffusion Models12 Sep
- 2022Soft Diffusion: Score Matching for General Corruptions12 Sep
- 2022Cold Diffusion: Inverting Arbitrary Image Transforms Without Noise19 Aug
- 2022Classifier-Free Diffusion Guidance26 Jul
- 2022CARD: Classification and Regression Diffusion Models15 Jun
22 more
- 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
- 2022A Continuous Time Framework for Discrete Denoising Models30 May
- 2022Fast Sampling of Diffusion Models with Exponential Integrator29 Apr
- 2022Pseudo Numerical Methods for Diffusion Models on Manifolds20 Feb
- 2022Riemannian Score-Based Generative Modelling6 Feb
- 2022DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents2 Jan
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANs15 Dec
- 2021Global Context with Discrete Diffusion in Vector Quantised Modelling for Image Generation3 Dec
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesis29 Nov
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul
- 2021Variational Diffusion Models1 Jul
- 2021Score-based Generative Modeling in Latent Space10 Jun
- 2021Cascaded Diffusion Models for High Fidelity Image Generation30 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Improved Denoising Diffusion Probabilistic Models18 Feb
- 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
- 2020Improved Techniques for Training Score-Based Generative Models16 Jun
Other references · 26
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2022DreamFusion: Text-to-3D using 2D Diffusioncited by 3,025
- 2016Variational Inference: A Review for Statisticianscited by 1,472
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
- 2020WaveGrad: Estimating Gradients for Waveform Generationcited by 622
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 562
14 more
- 2022Torsional Diffusion for Molecular Conformer Generationcited by 263
- 2023Imitating Human Behaviour with Diffusion Modelscited by 263
- 2022Diffsound: Discrete Diffusion Model for Text-to-sound Generationcited by 257
- 2022Diffusion-GAN: Training GANs with Diffusioncited by 205
- 2022Scalable Adaptive Computation for Iterative Generationcited by 195
- 2022Generative Modelling With Inverse Heat Dissipationcited by 167
- 2021Learning Gradient Fields for Molecular Conformation Generationcited by 163
- 2023Re-imagine the Negative Prompt Algorithm: Transform 2D Diffusion into 3D, alleviate Janus problem and Beyondcited by 139
- 2023In-Context Learning Unlocked for Diffusion Modelscited by 104
- 2023DR2: Diffusion-based Robust Degradation Remover for Blind Face Restorationcited by 79
- 2023Dirichlet Diffusion Score Model for Biological Sequence Generationcited by 75
- 2018Semi-Implicit Variational Inferencecited by 55
- 2020Exploiting Chain Rule and Bayes' Theorem to Compare Probability Distributionscited by 22
- 2023Learning to Jump: Thinning and Thickening Latent Counts for Generative Modelingcited by 18
Built on by
Citing papers per year: 2023: 2; 2024: 7; 2025: 6; 2026: 2.
17 papers.
Sort:
By area:
- 2024Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Designcited by 79
- 2025Continuously Augmented Discrete Diffusion model for Categorical Generative Modelingcited by 34
- 2024Diffusion-RPO: Aligning Diffusion Models through Relative Preference Optimizationcited by 32
- 2024Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Modelscited by 23
- 2025Variational Control for Guidance in Diffusion Modelscited by 22
- 2023Learning Stackable and Skippable LEGO Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modelingcited by 13
- 2024Guided Score identity Distillation for Data-Free One-Step Text-to-Image Generationcited by 13
- 2024Advancing Graph Generation through Beta Diffusioncited by 13
- 2023Improving In-Context Learning in Diffusion Models with Visual Context-Modulated Promptscited by 10
- 2025Score Distillation of Flow Matching Modelscited by 8
- 2025ItDPDM: Information-Theoretic Discrete Poisson Diffusion Modelcited by 6
- 2025PowerGrow: Feasible Co-Growth of Structures and Dynamics for Power Grid Synthesiscited by 5
5 more
- 2024ParamReL: Learning Parameter Space Representation via Progressively Encoding Bayesian Flow Networkscited by 2
- 2024Marked Temporal Bayesian Flow Point Processescited by 0
- 2025Closed-Form Beta Distribution Estimation from Sparse Statistics with Random Forest Implicit Regularizationcited by 0
- 2026Operationally Feasible Synthetic Power-Grid Scenarios via Learning the AC-Operable Joint Distributioncited by 0
- 2026Diffusion-2BC: Hybrid Diffusion and Regression Training for Offline Behavior Cloning in Autonomous Drivingcited by 0
- 2026Diffusion-2BC: Hybrid Diffusion and Regression Training for Offline Behavior Cloning in Autonomous Driving29 Sep 2026
- 2026Operationally Feasible Synthetic Power-Grid Scenarios via Learning the AC-Operable Joint Distribution4 Aug 2026
- 2025Continuously Augmented Discrete Diffusion model for Categorical Generative Modeling1 Oct 2025
- 2025Score Distillation of Flow Matching Models29 Sep 2025
- 2025PowerGrow: Feasible Co-Growth of Structures and Dynamics for Power Grid Synthesis29 Aug 2025
- 2025Closed-Form Beta Distribution Estimation from Sparse Statistics with Random Forest Implicit Regularization31 Jul 2025
- 2025ItDPDM: Information-Theoretic Discrete Poisson Diffusion Model8 May 2025
- 2025Variational Control for Guidance in Diffusion Models6 Feb 2025
- 2024Marked Temporal Bayesian Flow Point Processes25 Oct 2024
- 2024Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Design17 Oct 2024
- 2024Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Models17 Sep 2024
- 2024Advancing Graph Generation through Beta Diffusion13 Jun 2024
5 more
- 2024Diffusion-RPO: Aligning Diffusion Models through Relative Preference Optimization10 Jun 2024
- 2024Guided Score identity Distillation for Data-Free One-Step Text-to-Image Generation3 Jun 2024
- 2024ParamReL: Learning Parameter Space Representation via Progressively Encoding Bayesian Flow Networks24 May 2024
- 2023Improving In-Context Learning in Diffusion Models with Visual Context-Modulated Prompts3 Dec 2023
- 2023Learning Stackable and Skippable LEGO Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modeling10 Oct 2023
- 2024Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Designcited by 79
- 2025Continuously Augmented Discrete Diffusion model for Categorical Generative Modelingcited by 34
- 2024Diffusion-RPO: Aligning Diffusion Models through Relative Preference Optimizationcited by 32
- 2024Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Modelscited by 23
- 2025Variational Control for Guidance in Diffusion Modelscited by 22
- 2023Learning Stackable and Skippable LEGO Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modelingcited by 13
- 2024Guided Score identity Distillation for Data-Free One-Step Text-to-Image Generationcited by 13
- 2024Advancing Graph Generation through Beta Diffusioncited by 13
- 2023Improving In-Context Learning in Diffusion Models with Visual Context-Modulated Promptscited by 10
- 2025Score Distillation of Flow Matching Modelscited by 8
- 2025ItDPDM: Information-Theoretic Discrete Poisson Diffusion Modelcited by 6
- 2025Continuously Augmented Discrete Diffusion model for Categorical Generative Modeling1 Oct 2025
- 2025Score Distillation of Flow Matching Models29 Sep 2025
- 2025ItDPDM: Information-Theoretic Discrete Poisson Diffusion Model8 May 2025
- 2025Variational Control for Guidance in Diffusion Models6 Feb 2025
- 2024Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Design17 Oct 2024
- 2024Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Models17 Sep 2024
- 2024Advancing Graph Generation through Beta Diffusion13 Jun 2024
- 2024Diffusion-RPO: Aligning Diffusion Models through Relative Preference Optimization10 Jun 2024
- 2024Guided Score identity Distillation for Data-Free One-Step Text-to-Image Generation3 Jun 2024
- 2023Improving In-Context Learning in Diffusion Models with Visual Context-Modulated Prompts3 Dec 2023
- 2023Learning Stackable and Skippable LEGO Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modeling10 Oct 2023
- 2025PowerGrow: Feasible Co-Growth of Structures and Dynamics for Power Grid Synthesiscited by 5
- 2024ParamReL: Learning Parameter Space Representation via Progressively Encoding Bayesian Flow Networkscited by 2
- 2024Marked Temporal Bayesian Flow Point Processescited by 0
- 2025Closed-Form Beta Distribution Estimation from Sparse Statistics with Random Forest Implicit Regularizationcited by 0
- 2026Operationally Feasible Synthetic Power-Grid Scenarios via Learning the AC-Operable Joint Distributioncited by 0
- 2026Operationally Feasible Synthetic Power-Grid Scenarios via Learning the AC-Operable Joint Distribution4 Aug 2026
- 2025PowerGrow: Feasible Co-Growth of Structures and Dynamics for Power Grid Synthesis29 Aug 2025
- 2025Closed-Form Beta Distribution Estimation from Sparse Statistics with Random Forest Implicit Regularization31 Jul 2025
- 2024Marked Temporal Bayesian Flow Point Processes25 Oct 2024
- 2024ParamReL: Learning Parameter Space Representation via Progressively Encoding Bayesian Flow Networks24 May 2024
Similar, not cited
Close in wording, no citation either way
- 2025Diffusion Models: A Mathematical Introduction
- 2024Beta Sampling is All You Need: Efficient Image Generation Strategy for Diffusion Models using Stepwise Spectral Analysis
- 2025Improving Diffusion Generative Models via Truncated Karhunen--Lo`eve Expansion
- 2025Beyond Fixed Horizons: A Theoretical Framework for Adaptive Denoising Diffusions
- 2023Towards Faster Non-Asymptotic Convergence for Diffusion-Based Generative Models
- 2025Low-dimensional adaptation of diffusion models: Convergence in total variation
- 2022Diffusion models as plug-and-play priors
- 2025A Sharp KL-Convergence Analysis for Diffusion Models under Minimal Assumptions