Vocabulary-size-independent Convergence of Discrete Diffusion Models: adjoint equations induce the right space
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
- 2026Neural Continuous-Time Markov Chain: Discrete Diffusion via Decoupled Jump Timing and Direction17 Apr
- 2026Sharp Convergence Rates for Masked Diffusion Models26 Feb
- 2026Efficient Sampling with Discrete Diffusion Models: Sharp and Adaptive Guarantees16 Feb
- 2025Non-Asymptotic Convergence of Discrete Diffusion Models: Masked and Random Walk dynamics29 Nov
Other references · 2
- 2023Muse: Text-To-Image Generation via Masked Generative Transformerscited by 648
- 2025Optimal Inference Schedules for Masked Diffusion Modelscited by 25
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Uniform Discrete Diffusion Models are Minimax Optimal for Estimating Distributions with Small Effective Support Size
- 2025Discrete Diffusion Models: Novel Analysis and New Sampler Guarantees
- 2026Scaling Beyond Masked Diffusion Language Models
- 2023Towards Faster Non-Asymptotic Convergence for Diffusion-Based Generative Models
- 2025Error Bounds and Optimal Schedules for Masked Diffusions with Factorized Approximations
- 2026Consistent Diffusion Language Models
- 2025Non-asymptotic error bounds for probability flow ODEs under weak log-concavity
- 2026TUBE: Tangent Upper Bound on Evidence for Discrete Diffusion Language Models