Provably adaptive sampling with uniform and remasking discrete diffusion models

arXiv:2608.23554 2026 Sampling 2 ideas extracted · analyzed Aug 29, 2026

What the math gives to ML

The paper identifies a concrete failure mode of parallel discrete-diffusion sampling: independently applying stale denoising predictions cannot correct coordinates that were updated incorrectly within the same large time step. Its leave-one-out denoiser and first-order auxiliary sampler provide a principled correction mechanism while retaining parallel coordinate updates. The key transferable asset is an information-theoretic sampling bound whose discretization complexity depends on the target distribution's dependence, measured by dual total correlation, rather than directly on sequence length. This suggests both a drop-in sampler for masked or uniform diffusion and a dependence-aware timestep controller for reducing the number of denoising evaluations.

Ideas from this paper

Mechanism works 2026

Leave-One-Out Corrective Parallel Sampler

Replace standard tau-leaping in discrete diffusion generation with a first-order sampler whose per-coordinate transition is conditioned on all other current coordinates and excludes the coordinate being updated. After a parallel proposal, use the same leave-one-out conditionals to correct coordinates whose newly sampled values are inconsistent with the rest of the state, allowing large timesteps without permanently propagating simultaneous denoising errors.

Useful8/10
Difficulty5/10
Novelty7/10
Paper: Provably adaptive sampling with uniform and remasking discrete diffusion models arXiv:2608.23554
Mechanism failed 2026

Dependence-Adaptive Timestep Controller

Use an online estimate of the target distribution's remaining coordinate dependence to choose diffusion step sizes instead of using a uniform timestep grid. Take large jumps when the current state is nearly coordinate-factorized and shorten the step when the denoiser detects strong cross-coordinate dependence, targeting the paper's dependence-adaptive discretization scaling.

Useful7/10
Difficulty4/10
Novelty8/10
Paper: Provably adaptive sampling with uniform and remasking discrete diffusion models arXiv:2608.23554