GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models
Xiaohang Tang ⋅ Keyue Jiang ⋅ Che Liu ⋅ Zhao ⋅ Xiaoxiao Xu ⋅ Sangwoong Yoon ⋅ Ilija Bogunovic
Abstract
Reinforcement learning (RL) can be used to improve the policy (denoiser) of diffusion large language models (dLLMs), while being hindered by the intractability of the policy likelihood. A dominant and efficient family of methods replaces the likelihood in standard RL with its evidence lower bound (ELBO), estimated from randomly masked sequences. Despite being well-aligned with pre-training, these approaches introduce bias from training-inference mismatch by applying the likelihood surrogate ELBO, which can potentially cause training collapse. In this work, we propose **Guided Denoiser Self-Distillation (GDSD)** to directly distill the denoiser of dLLMs from an advantage-guided self-teacher, derived from the closed-form optimum of reverse-KL regularized RL. GDSD matches the dLLM's denoiser logits to the teacher's via a normalization-free objective, which reduces RL to likelihood-free self-distillation and thus bypasses the TIM biases. Recent ELBO-based methods emerge as instances of applying different distillation divergences, but with diagnosable pathologies that GDSD avoids. On planning, math, and coding benchmarks with LLaDA-8B and Dream-7B, GDSD consistently outperforms prior state-of-the-art ELBO-based methods with a more stable training reward dynamics, achieving test-accuracy improvements of up to $+19.6\\%$. These results suggest that direct denoiser self-distillation, without relying on ELBO likelihood surrogate, can provide a more stable and effective RL procedure for dLLMs.
Chat is not available.
Successful Page Load