arXiv:2512.09675cs.CL2025-12ACL被引 14

提升扩散语言模型的强化学习可靠性,解决奖励稀疏与概率估计偏差问题。

d-TreeRPO: Towards More Reliable Policy Optimization for Diffusion Language Models

  • 采用树状滚动和自下而上优势计算,实现可验证的逐步奖励信号。
  • 在Sudoku等任务上相比基线模型提升最高86.2%。
  • 适合研究扩散模型强化学习与高可信推理系统的设计者。

强化学习(RL)对提升扩散大语言模型(dLLMs)的推理能力至关重要。然而,现有dLLM策略优化方法存在两个关键可靠性瓶颈:(1)奖励稀疏性,源于粗粒度或不可验证的信号,阻碍准确的优势计算;(2)概率估计未考虑所有解码顺序的无偏期望差距,而该值难以计算。为此,我们提出d-TreeRPO,一种面向dLLMs的可靠强化学习框架,通过树状结构的轨迹滚动与基于可验证结果奖励的自下而上优势计算,提供细粒度且可验证的逐步奖励信号。此外,我们给出了理论证明,表明提高预测置信度能有效缩小无偏期望概率与单步前向传播估计之间的差距。基于此分析,我们在训练后期引入时间调度的自蒸馏损失,增强预测置信度,从而实现更精确的概率估计与更优性能。实验表明,d-TreeRPO优于现有基线,在多个推理基准上取得显著提升:Sudoku上提升86.2%,Countdown上提升51.6%,GSM8K上提升4.5%,Math500上提升5.3%。

原文摘要 · Abstract (English)

Reinforcement learning (RL) is pivotal for enhancing the reasoning capabilities of diffusion large language models (dLLMs). However, existing dLLM policy optimization methods suffer from two critical reliability bottlenecks: (1) reward sparsity, arising from coarse or unverifiable signals that impede accurate advantage calculation; and (2) their probability estimates do not account for the gap to the unbiased expectation over all decoding orders, which are intractable to compute. To mitigate these issues, we propose d-TreeRPO, a reliable RL framework for dLLMs that leverages tree-structured rollouts and bottom-up advantage computation based on verifiable outcome rewards to provide fine-grained and verifiable step-wise reward signals. Furthermore, we provide a theoretical proof demonstrating that increasing prediction confidence effectively minimizes the gap between unbiased expected prediction probabilities and its single-step forward pass estimate. Guided by this analysis, we introduce a time-scheduled self-distillation loss during training that enhances prediction confidence in later training stages, thereby enabling more accurate probability estimation and better performance. Experiments demonstrate that d-TreeRPO outperforms existing baselines and achieves significant improvements across multiple reasoning benchmarks. Specifically, it achieves +86.2% on Sudoku, +51.6% on Countdown, +4.5% on GSM8K, and +5.3% on Math500 compared to the base model.

强化学习扩散模型推理优化

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