arXiv:2504.12216cs.CLcs.LG2025-04NeurIPS被引 174

让扩散语言模型学会复杂推理,用强化学习提升思维能力。

d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learning

  • 用监督微调和无评分手法的强化学习,让扩散模型学会自我改进。
  • 在数学与规划任务中,性能超越现有最先进扩散模型。
  • 适合想用扩散架构做推理的科研人员和工程师。

近期的大语言模型(LLM)通过在线强化学习展现出强大的推理能力,主要基于从左到右的自回归生成范式。相比之下,基于扩散的非自回归模型以粗到精的方式生成文本。尽管最近的扩散语言模型(dLLMs)在语言建模性能上已接近自回归模型,但其是否能利用最新进展实现推理能力仍不清楚。为此,我们提出d1框架,通过监督微调(SFT)和强化学习(RL)将预训练的掩码dLLMs转化为推理模型。具体地,我们开发并扩展了两项技术:(a) 使用掩码SFT方法,直接从现有数据集蒸馏知识并引入自我改进行为;(b) 提出一种新颖的无评分手动策略梯度强化学习算法diffu-GRPO,首次将策略梯度方法应用于掩码dLLMs。通过实证研究,我们考察了多种后训练方案在多个数学与规划基准上的表现。结果表明,d1取得最佳性能,并显著提升当前最先进dLLM的推理能力。代码已公开于https://dllm-reasoning.github.io/。

原文摘要 · Abstract (English)

Recent large language models (LLMs) have demonstrated strong reasoning capabilities that benefits from online reinforcement learning (RL). These capabilities have primarily been demonstrated within the left-to-right autoregressive (AR) generation paradigm. In contrast, non-autoregressive paradigms based on diffusion generate text in a coarse-to-fine manner. Although recent diffusion-based large language models (dLLMs) have achieved competitive language modeling performance compared to their AR counterparts, it remains unclear if dLLMs can also leverage recent advances in LLM reasoning. To this end, we propose d1, a framework to adapt pre-trained masked dLLMs into reasoning models via a combination of supervised finetuning (SFT) and RL. Specifically, we develop and extend techniques to improve reasoning in pretrained dLLMs: (a) we utilize a masked SFT technique to distill knowledge and instill self-improvement behavior directly from existing datasets, and (b) we introduce a novel critic-free, policy-gradient based RL algorithm called diffu-GRPO, the first integration of policy gradient methods to masked dLLMs. Through empirical studies, we investigate the performance of different post-training recipes on multiple mathematical and planning benchmarks. We find that d1 yields the best performance and significantly improves performance of a state-of-the-art dLLM. Our code is released at https://dllm-reasoning.github.io/.

扩散模型强化学习推理能力大模型

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