On the Error-Correcting Effects of Stochasticity in Discrete Diffusion
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Discrete diffusion models achieve strong performance in text and image generation, but their inference remains slow and must inherently balance sampling efficiency and sample quality. In this work, we present a systematic study of how the degree of stochasticity in Markov transitions governs the sampling tradeoff. We show that highly deterministic transitions converge rapidly but suffer from error accumulation, while more stochastic transitions converge more slowly yet can achieve higher final sample quality. Using an information-theoretic analysis, we identify the underlying mechanism as an error-correcting effect induced by redundant transitions that symmetrically exchange mass between states, and show that these transitions can provably contract sampling errors. Motivated by this analysis, we propose Discrete Churn and Restart Sampling (DCRS), a novel inference algorithm that injects controlled stochasticity by alternating between forward and reverse diffusion processes. Experiments on synthetic datasets and large-scale benchmarks show that DCRS improves the speed-quality tradeoff in the low number of function evaluations regime. On image datasets, DCRS achieves up to a 10× reduction in sampling steps compared to standard samplers while maintaining competitive sample quality, whereas on language benchmarks, we observe more nuanced behavior depending on the corruption process and sampling procedure.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 30 of its 34 references that are in the map. First posted on arXiv 26 May 2026.
Builds on
30 of 34 references.
Closest ancestors · 17
Same direction, newest first
- 2025Why Masking Diffusion Works: Condition on the Jump Schedule for Improved Discrete Diffusion10 Jun
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithms1 Feb
- 2024Simple Guidance Mechanisms for Discrete Diffusion Models13 Dec
- 2024Distillation of Discrete Diffusion through Dimensional Correlations11 Oct
- 2024How Discrete and Continuous Diffusion Meet: Comprehensive Analysis of Discrete Diffusion Models via a Stochastic Integral Framework4 Oct
- 2024Informed Correctors for Discrete Diffusion Models30 Jul
- 2023Formulating Discrete Probability Flow Through Optimal Transport7 Nov
- 2023Restart Sampling for Improving Generative Processes26 Jun
- 2023Exploring the Optimal Choice for Generative Processes in Diffusion Models: Ordinary vs Stochastic Differential Equations3 Jun
- 2023The probability flow ODE is provably fast19 May
- 2022Score-based Continuous-time Discrete Diffusion Models30 Nov
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
5 more
- 2022A Continuous Time Framework for Discrete Denoising Models30 May
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesis29 Nov
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
Other references · 13
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 626
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
- 2024Discrete Flow Matchingcited by 317
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
- 2025Remasking Discrete Diffusion Models with Inference-Time Scalingcited by 189
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 148
- 2025The Diffusion Dualitycited by 135
- 2024Beyond Autoregression: Fast LLMs via Self-Distillation Through Timecited by 78
- 2025Generalized Interpolating Discrete Diffusioncited by 70
- 2024Flow Matching with General Discrete Paths: A Kinetic-Optimal Perspectivecited by 57
1 more
- 2025Fine-Tuning Masked Diffusion for Provable Self-Correctioncited by 39
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025On the Collapse Errors Induced by the Deterministic Sampler for Diffusion Models
- 2025Information-Theoretic Proofs for Diffusion Sampling
- 2024textitJump Your Steps: Optimizing Sampling Schedule of Discrete Diffusion Models
- 2025Cross-fluctuation phase transitions reveal sampling dynamics in diffusion models
- 2026Efficient Sampling with Discrete Diffusion Models: Sharp and Adaptive Guarantees
- 2025SynBoost: A Synergistic Framework for Fast Sampling of Diffusion Models
- 2025Effective Test-Time Scaling of Discrete Diffusion through Iterative Refinement
- 2026From Truncation to Commitment: Persistent Context in Uniform Discrete Diffusion