mlsanityData to 14 Oct 2026

Breaking AR's Sampling Bottleneck: Provable Acceleration via Diffusion Language Models

arXiv 2505.21400 ↗First posted 27 May 2025cs.LG cs.IT math.ITcited by 23

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Diffusion models have emerged as a powerful paradigm for modern generative modeling, demonstrating strong potential for large language models (LLMs). Unlike conventional autoregressive (AR) models that generate tokens sequentially, diffusion models allow for parallel sampling, offering a promising path to accelerate generation and eliminate the left-to-right generation constraints. Despite their empirical success, theoretical understandings of diffusion language models remain underdeveloped. In this work, we develop convergence guarantees for diffusion language models from an information-theoretic perspective. Our analysis demonstrates that the sampling error, measured by the Kullback-Leibler (KL) divergence, decays inversely with the number of iterations T and scales linearly with the mutual information between tokens in the target text sequence. Crucially, our theory covers the regime T<L, where L is the text sequence length. This justifies that high-quality samples can be generated with fewer iterations than L, thereby breaking the fundamental sampling bottleneck of L steps required by AR models. We further establish matching upper and lower bounds, up to some constant factor, that shows the tightness of our convergence analysis. These results offer novel theoretical insights into the practical effectiveness of diffusion language models.

Key facts. Cited by 23 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 48 of its 54 references that are in the map. First posted on arXiv 27 May 2025.

Builds on

48 of 54 references.

Closest ancestors · 14

Same direction, newest first

  1. 2025Large Language Diffusion Models14 Feb
  2. 2025Theoretical Benefit and Limitation of Diffusion Language Model13 Feb
  3. 2024Beyond Autoregression: Fast LLMs via Self-Distillation Through Time28 Oct
  4. 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models23 Oct
  5. 2024Simple and Effective Masked Diffusion Language Models11 Jun
  6. 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
  7. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
  8. 2023Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning23 Aug
  9. 2023Likelihood-Based Diffusion Language Models30 May
  10. 2022Latent Diffusion for Language Generation19 Dec
  11. 2022Continuous diffusion for categorical data28 Nov
  12. 2022DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models28 Nov
2 more
  1. 2022Self-conditioned Embedding Diffusion for Text Generation8 Nov
  2. 2022SSD-LM: Semi-autoregressive Simplex-based Diffusion Language Model for Text Generation and Modular Control31 Oct

Other references · 34

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  3. 2020Denoising Diffusion Implicit Modelscited by 9,781
  4. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  5. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
  6. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
  7. 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
  8. 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 602
  9. 2022Efficiently Scaling Transformer Inferencecited by 539
  10. 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
  11. 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptionscited by 445
  12. 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 373
22 more
  1. 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
  2. 2023Nearly d-Linear Convergence Bounds for Diffusion Models via Stochastic Localizationcited by 235
  3. 2022Improved Analysis of Score-based Generative Modeling: User-Friendly Bounds under Minimal Smoothness Assumptionscited by 219
  4. 2022Convergence of score-based generative modeling for general data distributionscited by 181
  5. 2022Convergence for score-based generative modeling with polynomial complexitycited by 177
  6. 2023Score Approximation, Estimation and Distribution Recovery of Diffusion Models on Low-Dimensional Datacited by 174
  7. 2022Score-based Continuous-time Discrete Diffusion Modelscited by 169
  8. 2023Diffusion Models are Minimax Optimal Distribution Estimatorscited by 165
  9. 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 148
  10. 2020Generative Modeling with Denoising Auto-Encoders and Langevin Samplingcited by 119
  11. 2024A Sharp Convergence Theory for The Probability Flow ODEs of Diffusion Modelscited by 65
  12. 2024Optimal score estimation via empirical Bayes smoothingcited by 61
  13. 2024O(d/T) Convergence Theory for Diffusion Probabilistic Models under Minimal Assumptionscited by 50
  14. 2024Distillation of Discrete Diffusion through Dimensional Correlationscited by 48
  15. 2024Minimax Optimality of Score-based Diffusion Models: Beyond the Density Lower Bound Assumptionscited by 47
  16. 2024Convergence Analysis of Discrete Diffusion Model: Exact Implementation through Uniformizationcited by 44
  17. 2024Learning general Gaussian mixtures with efficient score matchingcited by 44
  18. 2024Learning Mixtures of Gaussians Using Diffusion Modelscited by 42
  19. 2024From optimal score matching to optimal samplingcited by 28
  20. 2024Provable Acceleration for Diffusion Models under Minimal Assumptionscited by 27
  21. 2025Minimax Optimality of the Probability Flow ODE for Diffusion Modelscited by 23
  22. 2025Dimension-Free Convergence of Diffusion Models for Approximate Gaussian Mixturescited by 23

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 8; 2026: 15.

23 papers.

Sort:

By area:

  1. 2025Optimal Inference Schedules for Masked Diffusion Modelscited by 25
  2. 2026Confidence-Based Decoding is Provably Efficient for Diffusion Language Modelscited by 18
  3. 2025Optimal Convergence Analysis of DDPM for General Distributionscited by 16
  4. 2025Error Bounds and Optimal Schedules for Masked Diffusions with Factorized Approximationscited by 14
  5. 2026Adaptation to Intrinsic Dependence in Diffusion Language Modelscited by 10
  6. 2025Latent Refinement Decoding: Enhancing Diffusion-Based Language Models by Refining Belief Statescited by 9
  7. 2025Beyond Confidence: Adaptive and Coherent Decoding for Diffusion Language Modelscited by 9
  8. 2026Sharp Convergence Rates for Masked Diffusion Modelscited by 9
  9. 2026The data geometry of masking diffusion: Certified-optimal schedules via unmasking growth complexitycited by 9
  10. 2025Learnable Sampler Distillation for Discrete Diffusion Modelscited by 8
  11. 2025On the Reasoning Abilities of Masked Diffusion Language Modelscited by 7
  12. 2025Decoding Large Language Diffusion Models with Foreseeing Movementcited by 6
11 more
  1. 2026Discrete-WAM: Unified Discrete Vision-Action Token Editing for World-Policy Learningcited by 4
  2. 2026Provably adaptive sampling with uniform and remasking discrete diffusion modelscited by 4
  3. 2026Diffusion Models Are Statistically Optimal for Learning Low-Dimensional Multi-Modal Distributionscited by 3
  4. 2026ConvergeFlow: Language Flow with Provable Convergence to Token Embeddingscited by 2
  5. 2026Infinite Mask Diffusion for Few-Step Distillationcited by 1
  6. 2026Differences in Text Generated by Diffusion and Autoregressive Language Modelscited by 1
  7. 2026Diffusion Models Adapt to Low-Dimensional Structure Under Flexible Coefficient Choicescited by 1
  8. 2026Extracting Training Data from Diffusion Language Models via Infillingcited by 0
  9. 2026One-step lowest-variance selection in a Gaussian random-field model motivated by masked diffusion: Total correlation and a square root collision thresholdcited by 0
  10. 2026Schedule optimization for tau-leaping in masked discrete diffusioncited by 0
  11. 2026Sample complexity bounds for categorical Markov random fields via Discrete Diffusionscited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2025Diffusion Language Models are Provably Optimal Parallel Samplers
  2. 2026LESS Is More: Mutual-Stability Sampling for Diffusion Language Models
  3. 2024Energy-Based Diffusion Language Models for Text Generation
  4. 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMs
  5. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models
  6. 2026A Tale of Two Temperatures: Simple, Efficient, and Diverse Sampling from Diffusion Language Models
  7. 2026Simple Diffusion Language Models Are More Effective Few-Step Generators Than Reported
  8. 2025Diffusion Beats Autoregressive in Data-Constrained Settings