arXiv:2412.18279cs.AI2024-12被引 20

用密集奖励提升大模型多步推理能力,训练更稳定。

Improving Multi-Step Reasoning Abilities of Large Language Models with Direct Advantage Policy Optimization

  • 每一步预测推理正确率,生成密集反馈信号
  • 独立训练策略与评估网络,避免传统强化学习不稳定
  • 在数学和代码任务上显著提升SFT与强化学习模型表现

强化学习在提升大语言模型推理能力方面作用日益突出。尽管取得进展,但面临奖励稀疏和训练不稳定的挑战。为此,我们提出直接优势策略优化(DAPO),一种新型的步骤级离线强化学习算法。与仅依赖最终结果奖励的对齐方法(如DPO)不同,DAPO通过评估网络在每一步预测推理正确率,生成密集信号以优化生成策略。此外,DAPO中策略网络与评价网络独立训练,避免了标准演员-评论家方法(如PPO)中的共训练不稳定性。我们在数学和代码查询数据集上训练DAPO,并在多个基准测试中评估其性能。结果表明,DAPO能有效提升SFT模型和强化学习模型在数学与代码任务上的表现,验证了该方法的有效性。

原文摘要 · Abstract (English)

The role of reinforcement learning (RL) in enhancing the reasoning of large language models (LLMs) is becoming increasingly significant. Despite the success of RL in many scenarios, there are still many challenges in improving the reasoning of LLMs. One challenge is the sparse reward, which makes optimization difficult for RL and necessitates a large amount of data samples. Another challenge stems from the inherent instability of RL, particularly when using Actor-Critic (AC) methods to derive optimal policies, which often leads to unstable training processes. To address these issues, we introduce Direct Advantage Policy Optimization (DAPO), an novel step-level offline RL algorithm. Unlike standard alignment that rely solely outcome rewards to optimize policies (such as DPO), DAPO employs a critic function to predict the reasoning accuracy at each step, thereby generating dense signals to refine the generation strategy. Additionally, the Actor and Critic components in DAPO are trained independently, avoiding the co-training instability observed in standard AC algorithms like PPO. We train DAPO on mathematical and code query datasets and then evaluate its performance on multiple benchmarks. Our results show that DAPO can effectively enhance the mathematical and code capabilities on both SFT models and RL models, demonstrating the effectiveness of DAPO.

强化学习推理增强大模型

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