Factorization-Error-Free Discrete Diffusion Language Model via Speculative Decoding
For practitioners of discrete diffusion language models, this method provides a principled fix for a known approximation error, yielding both better accuracy and faster inference.
FeF-DLLM eliminates factorization errors in discrete diffusion language models by using exact prefix-conditioned clean posterior, improving accuracy by 5.04% on average and achieving 3.86× inference speedup via speculative decoding.
Discrete diffusion language models improve generation efficiency through parallel token prediction, but standard $X_0$ prediction methods introduce factorization errors by approximating the clean token posterior with independent token-wise distributions. This paper proposes Factorization-Error-Free Discrete Diffusion Language Modeling (FeF-DLLM), which replaces independent clean-token prediction with an exact prefix-conditioned factorization of the clean posterior to better preserve token dependencies. To reduce the sequential cost introduced by prefix conditioning, FeF-DLLM further incorporates speculative decoding within diffusion denoising, accelerating inference while maintaining the parallel prediction and re-masking properties of DLLMs. Theoretically, we prove that FeF-DLLM generates from the true joint distribution and derive its expected acceleration ratio. Experiments on GSM8K, MATH, HumanEval, and MBPP demonstrate that our method improves accuracy by an average of 5.04 percentage points while achieving an average inference speedup of $3.86\times$.