RESEARCH · RESEARCH · #1464
Limits of Confidence in Diffusion
This paper analyzes discrete diffusion samplers (including remasking and uniform-state samplers) and proves that a diffusion step matches the training distribution only when the positions it writes are conditionally independent given already-fixed tokens. The authors show that products of per-position marginals cannot represent dependent groups, that identical per-position marginals can hide different joint dependencies, and validate these phenomena on the ScanAndAdd synthetic task where groups of two-or-more undetermined positions are dependent and the generated distribution exhibits 29× the sampling-noise-floor total variation despite per-sample metrics being 1.0.
KEY POINTS
- This paper analyzes discrete diffusion samplers (including remasking and uniform-state samplers) and proves that a diffusion step matches the training distribution only when the positions it writes are conditionally independent given already-fixed tokens.
- The authors show that products of per-position marginals cannot represent dependent groups, that identical per-position marginals can hide different joint dependencies, and validate these phenomena on the ScanAndAdd synthetic task where groups of two-or-more undetermined positions are dependent and the generated distribution exhibits 29× the sampling-noise-floor total variation despite per-sample metrics being 1.0.
- This matters because many discrete diffusion sampling and confidence-based selection procedures rely on per-position marginals, but those marginals can fail to capture joint dependencies and thus produce distributions that diverge substantially from the training distribution.
WHY IT MATTERS
This matters because many discrete diffusion sampling and confidence-based selection procedures rely on per-position marginals, but those marginals can fail to capture joint dependencies and thus produce distributions that diverge substantially from the training distribution.