Stratified Hazard Sampling: Minimal-Variance Event Scheduling for CTMC/DTMC Discrete Diffusion and Flow Models
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Uniform-noise discrete diffusion and flow models generate sequences non-autoregressively through iterative, context-dependent token replacements. However, these models are typically formulated as time-inhomogeneous continuous- or discrete-time Markov chains (CTMC/DTMC), sampled using independent Bernoulli change decisions per discretization step. This induces Poisson-binomial variance in per-position jump counts that grows with the number of required edits, leading to the common under-editing (residual noise) and over-editing (cascading substitutions) failure modes that degrade sample quality. We identify this sampler-induced variance as an orthogonal source of degradation, distinct from model-side errors and addressable purely at inference time. We propose Systematic Hazard Sampling (SHS), a training-free, drop-in, and hyperparameter-free inference principle for any sampler that admits a stay-vs.-replace decomposition. SHS models per-token edits as events driven by cumulative hazard (CTMC) or jump mass (DTMC) and triggers an edit whenever this quantity exceeds unit-spaced thresholds with a single random phase per position. For any fixed cumulative mass, this preserves the expected jump count while achieving the minimum conditional variance possible among unbiased integer estimators (at most 1/4), without altering per-jump destination sampling. Experiments on four uniform-noise discrete diffusion and flow language models spanning ~110M to ~3B parameters show that SHS consistently improves sample quality across numbers of function evaluations (NFE). Code is available at https://github.com/Jang-seunghwan/Systematic-Hazard-Sampling.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 24 of its 38 references that are in the map. First posted on arXiv 6 Jan 2026.
Builds on
24 of 38 references.
Closest ancestors · 7
Same direction, newest first
- 2025Deterministic Discrete Denoising25 Sep
- 2025Why Masking Diffusion Works: Condition on the Jump Schedule for Improved Discrete Diffusion10 Jun
- 2024Simple Guidance Mechanisms for Discrete Diffusion Models13 Dec
- 2024textitJump Your Steps: Optimizing Sampling Schedule of Discrete Diffusion Models10 Oct
- 2024Informed Correctors for Discrete Diffusion Models30 Jul
- 2022A Continuous Time Framework for Discrete Denoising Models30 May
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul
Other references · 17
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2018Hierarchical Neural Story Generationcited by 1,182
- 2024Simple and Effective Masked Diffusion Language Modelscited by 800
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 606
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 371
- 2024Discrete Flow Matchingcited by 301
- 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusionscited by 208
- 2024Masked Diffusion Models are Secretly Time-Agnostic Masked Models and Exploit Inaccurate Categorical Samplingcited by 204
- 2025Remasking Discrete Diffusion Models with Inference-Time Scalingcited by 181
- 2025The Diffusion Dualitycited by 124
5 more
- 2025Path Planning for Masked Diffusion Model Samplingcited by 79
- 2025Generalized Interpolating Discrete Diffusioncited by 68
- 2024Think While You Generate: Discrete Diffusion with Planned Denoisingcited by 61
- 2025Scaling Behavior of Discrete Diffusion Language Modelscited by 32
- 2025FS-DFM: Fast and Accurate Long Text Generation with Few-Step Diffusion Language Modelscited by 16
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Corrected Samplers for Discrete Flow Models
- 2026Stochastic Transition-Map Distillation for Fast Probabilistic Inference
- 2026Efficient Sampling with Discrete Diffusion Models: Sharp and Adaptive Guarantees
- 2025Error Bounds and Optimal Schedules for Masked Diffusions with Factorized Approximations
- 2026On the Error-Correcting Effects of Stochasticity in Discrete Diffusion
- 2025Edit-Based Flow Matching for Temporal Point Processes
- 2026Provably adaptive sampling with uniform and remasking discrete diffusion models
- 2023User-defined Event Sampling and Uncertainty Quantification in Diffusion Models for Physical Dynamical Systems