arXiv:2605.29398cs.LGcs.AI2026-05被引 1

用自教师指导的去噪器自蒸馏,提升扩散语言模型的强化学习效果。

GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models

论文配图:GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models
图 1 · 摘自论文原文
  • 设计无似然项的自蒸馏框架,直接对齐去噪器输出分布。
  • 在数学、编程等任务上提升最高19.6%准确率,训练更稳定。
  • 适合追求稳定强化学习训练的扩散语言模型研究者。

强化学习可优化扩散大语言模型(dLLMs)的策略(去噪器),但受限于策略似然不可计算。现有主流方法用随机掩码序列估计的证据下界(ELBO)替代似然,虽与预训练对齐,却因训练-推理不一致引入偏差,影响性能。本文提出引导式去噪器自蒸馏(GDSD),通过反KL正则化强化学习的闭式最优解构建自教师,直接将dLLM去噪器的logits匹配教师输出,采用无归一化目标,使强化学习退化为无需似然的自蒸馏,彻底规避了训练-推理不一致偏差。基于不同蒸馏散度的已有ELBO方法可被统一解释,但存在可诊断缺陷,而GDSD避免了这些问题。在LLaDA-8B和Dream-7B模型上,针对规划、数学与编码任务,GDSD持续超越当前最优的ELBO方法,测试准确率提升最高达+19.6%,且训练奖励动态更稳定。结果表明,不依赖ELBO似然代理的直接自蒸馏,能为dLLMs提供更稳定有效的强化学习路径。代码已开源。

原文摘要 · Abstract (English)

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 through training--inference mismatch by using the ELBO as a likelihood surrogate, which can degrade performance. 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 an ELBO likelihood surrogate, can provide a more stable and effective RL procedure for dLLMs. Code is available at https://github.com/GaryBall/GDSD.

扩散模型强化学习自蒸馏语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。