用补全能力引导扩散语言模型的强化学习,提升探索效率与性能。
Inpainting-Guided Policy Optimization for Diffusion Large Language Models
- 在采样中插入部分正确推理路径,引导模型探索
- 在三个数学基准上达到新最好结果,样本效率显著提升
- 适合研究扩散模型、强化学习与自动推理的学者
掩码扩散语言模型(dLLMs)正成为自回归大模型的有力替代,兼具优异性能与补全生成等独特能力。本文探索如何利用补全能力改进dLLMs的强化学习算法设计。将大模型与强化学习对齐面临探索困境:稀疏奖励信号和失败样本浪费。尽管此问题普遍存在于大模型中,但dLLMs凭借其补全能力提供了独特机会——我们提出IGPO(Inpainting Guided Policy Optimization),在在线采样中战略性地插入部分真实推理轨迹,而非完整答案,从而引导探索至有希望的路径空间,同时保留自我生成推理过程,实现监督微调与强化学习的衔接。将IGPO应用于分组优化方法如GRPO,可恢复有效梯度并改善样本效率。此外,提出基于合成重写简洁轨迹的监督微调策略,结合熵过滤等技术,最终在GSM8K、Math500和AMC三个数学基准上取得全注意力掩码dLLM的新最佳表现。
原文摘要 · Abstract (English)
Masked diffusion large language models (dLLMs) are emerging as promising alternatives to autoregressive LLMs, offering competitive performance while supporting unique generation capabilities such as inpainting. We explore how inpainting can inform RL algorithm design for dLLMs. Aligning LLMs with reinforcement learning faces an exploration challenge: sparse reward signals and sample waste when models fail to discover correct solutions. While this inefficiency affects LLMs broadly, dLLMs offer a distinctive opportunity--their inpainting ability can guide exploration. We introduce IGPO (Inpainting Guided Policy Optimization), an RL framework that strategically inserts partial ground-truth reasoning traces during online sampling. Unlike providing full solutions, inpainting steers exploration toward promising trajectory spaces while preserving self-generated reasoning, bridging supervised fine-tuning and reinforcement learning. We apply IGPO to group-based optimization methods such as GRPO, where exploration failures cause zero advantages and gradients. IGPO restores meaningful gradients while improving sample efficiency. We also propose supervised fine-tuning on synthetically rewritten concise traces that better align with dLLM generation patterns. With additional techniques including entropy-based filtering, our training recipe yields substantial gains across three mathematical benchmarks--GSM8K, Math500, and AMC--achieving new state-of-the-art results for full-attention masked dLLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。