Skip to content

Limits of Confidence in Diffusion

Sep 2026 · 1 citation · 22 references
Computer Science

TL;DR

It is shown that a step matches the training distribution only when the positions it writes are conditionally independent given the tokens already fixed, that no product of per-position distributions can match a dependent group, and that per-position distributions do not determine whether a group is dependent.

Abstract

Discrete diffusion, including remasking and uniform-state samplers, generate a sequence by writing multiple token positions per step, drawing each from a per-position distribution and choosing which positions to write from those same distributions. For domains of general interest (pixels, phonemes, or words) there are inherent dependencies between tokens. We show that a step matches the training distribution only when the positions it writes are conditionally independent given the tokens already fixed, that no product of per-position distributions can match a dependent group, and that per-position distributions do not determine whether a group is dependent: two joint distributions can have identical per-position marginals while differing in which combinations of values occur. On ScanAndAdd, a synthetic task whose joint distribution is available in closed form, we verify that every group of two or more undetermined positions a confidence ranking writes is dependent, and measure the generated distribution to be $29\times$ the sampling-noise floor total variation while per-sample metrics are $1.0$.

View source

Similar papers

#machine learning Preprint Sep 2026

Distribution Matching Distillation for Continuous Diffusion Language Models

Continuous diffusion language models generate all tokens in parallel, yet high-quality generation can still require hundreds of network evaluations (NFEs). We study how distributional distillation can reduce this cost by exploiting the student's probabilistic token outputs. Our unified formulation connects the student'...

Paul Le Van Kiem, Dario Shariatian, Umut Simsekli et al. · 0 citations
#machine learning Preprint Oct 2026

Two-Sample Testing via Generative Processes

Deciding whether two samples come from the same distribution is a classical problem in statistics, and generative transport offers a new way to approach it. We build a stochastic interpolant directly between the two samples and observe that, for a symmetric schedule, its law is invariant under the time reflection $t \m...

Eshant English, Kenji Fukumizu, Taiji Suzuki · 0 citations
#machine learning Preprint Sep 2026

CoDR: Training-Free Confidence-Drift Remasking for Diffusion Language Models

Masked diffusion language models (MDLMs) decode by repeatedly committing tokens to masked positions, but these commitments are usually irreversible. A token chosen under sparse, partial context is kept fixed, even when later context no longer supports it. Existing samplers mainly decide when to commit a token, but rare...

Yue Wu, Qing-He Zhang, Yu Zhang et al. · 0 citations
#machine learning Preprint Sep 2026

Low-Confidence Remasking Traps Flexibility: Realizing Arbitrary-Order Potential for Diverse Rollouts in Diffusion LLMs

Masked diffusion language models support arbitrary-order generation, suggesting a natural way to produce diverse outputs. However, recent work argues that this flexibility reduces diversity by delaying high-uncertainty tokens that can lead to different generation paths. We trace this diversity loss not to arbitrary-ord...

Moongyu Jeon, Dongjae Jeon, Bumjun Kim et al. · 0 citations
#artificial intelligence Preprint Sep 2026

Grab a Coffee: Future-Aware Guidance for Discrete Diffusion with Compiled Objectives

By making objectives available to inference rather than only evaluation, COFFEE brings joint conditioning, completion-weighted guidance, and optimization-based constraints into pretrained neural generation, showing the potential of neural-symbolic methods in diffusion guidance.

Hua-Qiu Xu, Dong-Xin Li, Gwen Yidou-Weng et al. · 0 citations
#machine learning Preprint Oct 2026

SOL: Measuring Gaps between Text Distributions by Double Sliced Wasserstein Metrics

Evaluating text generation requires measuring how well the generated distribution matches the data distribution. For autoregressive models, this is done by the perplexity. Diffusion and flow-based language models can only provide a likelihood bound, whose tightness differs between model families. Sample-based substitut...

Gregor Kornhardt, Moritz Piening, J. Chemseddine et al. · 0 citations

Related blog posts

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.