mlsanityData to 14 Oct 2026

Vocabulary-size-independent Convergence of Discrete Diffusion Models: adjoint equations induce the right space

arXiv 2605.17232 ↗First posted 17 May 2026cs.LG math.ST stat.MLcited by 3

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Discrete diffusion has become a leading framework for generative modeling in various applications including language, vision, and biology. Existing convergence theory, however, exhibits fundamental limitations. KL-based analyses diverge under singular priors such as the masked distribution, while bounds in total variation (TV) depend on the vocabulary size S and become vacuous for modern language tasks, where vocabularies contain hundreds of thousands of tokens. We develop a unified adjoint-equation-based framework that establishes vocabulary-size-independent convergence guarantees in any integral probability metric (IPM). To the best of our knowledge, our bounds are the first to be entirely free of S and applicable to both masked and uniform priors. Importantly, our results can extend existing step complexity guarantees to any IPM. Also, our theory relies only on a single standard rate-matrix regularity assumption and applies to general priors. Five novel techniques drive our improvements: 1. working in the space of observables via adjoint equations rather than directly with probability measures; 2. a regularity analysis that yields bounds on any IPM; 3. a coupling argument that removes S-dependence under uniform transitions; and 4. score-marginal cancellation and 5. exit-routing techniques that remove S-dependence under masked transitions. Our framework thus sharply departs from prior analyses and avoids the shortcomings of pathspace-KL and existing TV-based approaches. Beyond convergence bounds, our framework provides a versatile toolkit for further theoretical study of discrete diffusion models, including principled choices of loss functions and vocabulary-size-independent step complexity.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 6 of its 61 references that are in the map. First posted on arXiv 17 May 2026.

Builds on

6 of 61 references.

Closest ancestors · 4

Same direction, newest first

  1. 2026Neural Continuous-Time Markov Chain: Discrete Diffusion via Decoupled Jump Timing and Direction17 Apr
  2. 2026Sharp Convergence Rates for Masked Diffusion Models26 Feb
  3. 2026Efficient Sampling with Discrete Diffusion Models: Sharp and Adaptive Guarantees16 Feb
  4. 2025Non-Asymptotic Convergence of Discrete Diffusion Models: Masked and Random Walk dynamics29 Nov

Other references · 2

  1. 2023Muse: Text-To-Image Generation via Masked Generative Transformerscited by 648
  2. 2025Optimal Inference Schedules for Masked Diffusion Modelscited by 25

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Uniform Discrete Diffusion Models are Minimax Optimal for Estimating Distributions with Small Effective Support Size
  2. 2025Discrete Diffusion Models: Novel Analysis and New Sampler Guarantees
  3. 2026Scaling Beyond Masked Diffusion Language Models
  4. 2023Towards Faster Non-Asymptotic Convergence for Diffusion-Based Generative Models
  5. 2025Error Bounds and Optimal Schedules for Masked Diffusions with Factorized Approximations
  6. 2026Consistent Diffusion Language Models
  7. 2025Non-asymptotic error bounds for probability flow ODEs under weak log-concavity
  8. 2026TUBE: Tangent Upper Bound on Evidence for Discrete Diffusion Language Models