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
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
✗ Mechanism failed
2026
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