改进大模型推理的信用分配机制,提升数学解题能力。
GRPO-$λ$: Credit Assignment improves LLM Reasoning
- 用令牌级概率重构优势迹,实现更细粒度的奖励回溯。
- 在4个数学数据集上训练1.5B至7B模型,性能提升3-4.5点。
- 无需判别器,适用于LLaMA-3.1和Qwen-2.5等主流架构。
大型语言模型(LLMs)在需要复杂推理的任务中日益广泛应用,促使人们通过后训练方法提升其推理能力。基于强化学习的方案,如最先进的GRPO,已证明在后训练阶段显著改善推理行为。然而,缺乏显式奖励或判别模型限制了GRPO在词元序列上的精细信用分配能力。本文提出GRPO-$λ$,一种对GRPO的新扩展,通过重新表述使用生成后词元级对数概率的资格迹,并引入一种新的无判别器时序差分误差近似,增强在复杂推理任务中的信用分配。我们设计了多种$λ$-回报权重策略,应用于资格迹,所有变体均显著优于GRPO。我们在4个不同数学推理数据集上,从1.5B到7B参数规模的模型进行训练,结果表明,在LLaMA-3.1和Qwen-2.5架构上,强化学习训练期间性能提升30%-40%。最终,采用GRPO-$λ$后,模型在AIME24、Math500、OlympiadMath、MinervaMath和AMC上的平均表现超过GRPO超过3分,7B模型更是提升4.5分。
原文摘要 · Abstract (English)
Large language models (LLMs) are increasingly deployed for tasks requiring complex reasoning, prompting significant interest in improving their reasoning abilities through post-training. Especially RL based methods using verifiable reward, like the state-of-the-art GRPO, have shown to tremendously improve reasoning behaviors when applied as post-training methods. However, the lack of an explicit reward or critic model limits GRPO's ability to assign fine-grained credit across token sequences. In this work, we present GRPO-$λ$, a novel extension to GRPO that enhances credit assignment in RL finetuning of LLMs for complex reasoning tasks. We approximate learning from $λ$-return with a reformulation of eligibility traces using token-level log-probabilities applied after each sequence generation, and a novel critic-free approximation of the temporal-difference error. We introduce a few variations for the weighting of the $λ$-return, and their applications to the eligibility-trace, where all the variations provide significant gains over GRPO. We compare GRPO-$λ$ against GRPO by training models from 1.5B to 7B parameters on $4$ different math reasoning datasets. The training plots demonstrate 30-40% improved performance during RL training on both LLaMA-3.1 and Qwen-2.5 architectures. Finally, we show that with GRPO-$λ$, the resulting average performance on AIME24, Math500, OlympiadMath, MinervaMath, and AMC improves over GRPO by over $3$ points and a $4.5$ points improvement on the 7B model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。