arXiv:2410.01679cs.LGcs.CL2024-10ICML被引 108

改进大模型强化学习中的信用分配,提升推理任务表现。

VinePPO: Refining Credit Assignment in RL Training of LLMs

  • 利用语言环境特性构建无偏蒙特卡洛估计,替代传统价值网络。
  • 在MATH和GSM8K上训练速度最快提升3倍,测试准确率更高。
  • 适合关注大模型强化学习优化与推理能力提升的研究者。

大型语言模型(LLMs)越来越多地用于需要多步复杂推理才能获得奖励的复杂任务,正确分配每一步的信用对提升性能至关重要。虽然近似策略优化(PPO)通常依赖价值网络解决信用分配问题,但近期方法无需价值网络也能取得良好效果,引发对其实际有效性的质疑。本文系统评估发现,价值网络在推理密集型任务中常产生偏差的期望回报估计,比较不同步骤时表现仅略优于随机基线。为此,我们提出VinePPO,一种利用语言环境灵活性计算无偏蒙特卡洛估计的简单方法。该方法在MATH和GSM8K数据集上均显著优于PPO及其他基线,且训练时间最多快3.0倍;更重要的是,在相同训练准确率下实现更高测试准确率,表明其能捕捉更强的泛化信号。结果凸显了准确信用分配在大模型强化学习中的关键作用。

原文摘要 · Abstract (English)

Large language models (LLMs) are increasingly applied to complex reasoning tasks that require executing several complex steps before receiving any reward. Properly assigning credit to these steps is essential for enhancing model performance. Proximal Policy Optimization (PPO), a common reinforcement learning (RL) algorithm used for LLM finetuning, employs value networks to tackle credit assignment. However, recent approaches achieve strong results without it, raising questions about the efficacy of value networks in practice. In this work, we systematically evaluate the efficacy of value networks and reveal their significant shortcomings in reasoning-heavy LLM tasks, showing that they often produce poor estimate of expected return and barely outperform a random baseline when comparing alternative steps. This motivates our key question: Can improved credit assignment enhance RL training for LLMs? To address this, we propose VinePPO, a straightforward approach that leverages the flexibility of language environments to compute unbiased Monte Carlo-based estimates. Our method consistently outperforms PPO and other baselines across MATH and GSM8K datasets in less wall-clock time (up to 3.0x). Crucially, it achieves higher test accuracy for a given training accuracy, capturing more generalization signal per sample. These results emphasize the importance of accurate credit assignment in RL training of LLM.

强化学习大模型信用分配PPO

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