通过树采样生成细粒度奖励,提升大模型数学推理能力
TreeRPO: Tree Relative Policy Optimization
- 用树采样直接估算每步推理的奖励期望,无需额外奖励模型
- 在数学推理任务上将通过率从19.0%提升至35.5%
- 比GRPO更高效,响应长度减少18.1%,适合强化学习优化推理过程
大型语言模型在基于可验证奖励的强化学习(RLVR)方法下展现出强大的推理能力。然而,现有方法在轨迹级定义奖励,难以有效指导推理过程中的中间步骤优化。为此,我们提出 extbf{TreeRPO},一种利用树采样估计各推理步骤奖励数学期望的新方法。不同于依赖独立步骤奖励模型的先前方法,TreeRPO通过采样过程直接估计奖励。基于GRPO的组相对奖励训练机制,TreeRPO创新性地利用树采样生成的步骤级分组计算奖励。该方法产生细粒度且密集的奖励信号,显著提升大模型的学习效率与性能。实验表明,TreeRPO将Qwen-2.5-Math在测试基准上的平均Pass@1准确率从19.0%提升至35.5%。此外,其性能优于GRPO 2.9%,同时平均响应长度减少18.1%,展现出卓越的有效性与效率。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have shown remarkable reasoning capabilities through Reinforcement Learning with Verifiable Rewards (RLVR) methods. However, a key limitation of existing approaches is that rewards defined at the full trajectory level provide insufficient guidance for optimizing the intermediate steps of a reasoning process. To address this, we introduce \textbf{\name}, a novel method that estimates the mathematical expectations of rewards at various reasoning steps using tree sampling. Unlike prior methods that rely on a separate step reward model, \name directly estimates these rewards through this sampling process. Building on the group-relative reward training mechanism of GRPO, \name innovatively computes rewards based on step-level groups generated during tree sampling. This advancement allows \name to produce fine-grained and dense reward signals, significantly enhancing the learning process and overall performance of LLMs. Experimental results demonstrate that our \name algorithm substantially improves the average Pass@1 accuracy of Qwen-2.5-Math on test benchmarks, increasing it from 19.0\% to 35.5\%. Furthermore, \name significantly outperforms GRPO by 2.9\% in performance while simultaneously reducing the average response length by 18.1\%, showcasing its effectiveness and efficiency. Our code will be available at \href{https://github.com/yangzhch6/TreeRPO}{https://github.com/yangzhch6/TreeRPO}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。