AI/ ai · machine-learning · diffusion-models · research

What Discrete Diffusion Models Actually Learn

A new theoretical framework shows that denoisers, score ratios, and bridge predictors are the same object in different coordinates.

A unified theory of discrete diffusion training objectives just made the field's competing loss functions a lot easier to reconcile.

Researchers posted a paper proving what they call the Oracle Distance theorem: the negative ELBO (the standard training objective) is not merely a bound on data entropy but exactly equal to it plus the path KL divergence between the learned reverse process and the ideal one. In plain terms, every discrete diffusion model — regardless of how its loss is written — is trying to learn the same underlying object: the conditional expectation of the true reverse jump rate. The paper derives this rigorously for continuous-time Markov chains, boundary terms included, and verifies every identity numerically on an exactly solvable model. It also recovers MDM, UDM, SEDD, and GIDD as special cases of the same framework.

Why it matters: the field has accumulated a confusing zoo of loss functions, each with its own paper claiming superiority. This work suggests the disagreements are partly a coordinate problem — the same optimizer looks like a denoiser, a score ratio, or a bridge plug-in depending on how you read the network output. One concrete payoff: the paper proves that a denoiser parameterization causes the uniform diffusion ELBO to diverge at initialization, while the bridge plug-in stays finite, which is a practical warning for anyone debugging training instability.

Discrete diffusion has quietly become one of the more competitive alternatives to autoregressive text generation; clarifying what these models are actually learning should help practitioners choose parameterizations more deliberately rather than by trial and error.

TR

The Revision

Written by an AI system from the public sources credited above. How we write →