通过前瞻树策略提升强化学习轨迹多样性,加速模型推理能力进化
Lookahead Tree-Based Rollouts for Enhanced Trajectory-Level Exploration in Reinforcement Learning with Verifiable Rewards
- 引入前瞻树滚动机制,在高不确定性步骤主动分叉生成不同路径
- 相比随机采样,平均提速131%,在多个任务上提升4.2%的准确率
- 适合需要高质量推理路径探索的大型语言模型强化学习场景
基于可验证奖励的强化学习(RLVR),特别是组相对策略优化(GRPO)算法,已证明能有效提升大语言模型的推理能力。然而,当前流程中群体滚动采样轨迹的多样性有限,同质化轨迹及其奖励会削弱策略更新的信号,阻碍有效学习。这一问题主要源于逐标记的随机采样,局部差异易坍缩为几乎相同的推理路径。为此,本文提出前瞻树基滚动(LATR)策略,通过强制在候选标记上分叉以促进轨迹层面的多样性。LATR迭代执行三个阶段:(1) 在高不确定性生成步骤进行分叉;(2) 对每个新分支执行前瞻模拟;(3) 剪枝在模拟中持续相似的分支。相较于随机采样,LATR在不同推理任务上使GRPO与动态采样策略优化(DAPO)算法的政策学习速度平均提升131%,最终pass@1性能提升4.2%。代码与数据已公开于https://github.com/starreeze/latr。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR), particularly with algorithms like Group Relative Policy Optimization (GRPO), has proven highly effective in enhancing the reasoning capabilities of large language models. However, a critical bottleneck in current pipelines lies in the limited diversity of sampled trajectories during group rollouts. Homogeneous trajectories and their associated rewards would diminish the return signals for policy updates, thereby hindering effective policy learning. This lack of diversity stems primarily from token-level stochastic sampling, where local variations are likely to collapse into near-identical reasoning paths. To address this limitation, we propose Lookahead Tree-Based Rollouts (LATR), a novel rollout strategy designed to explicitly promotes trajectory-level diversity by enforcing branching into different candidate tokens likely to yield distinct continuations. Specifically, LATR iteratively operates in three stages: (1) branching at high-uncertainty generation steps, (2) performing lookahead simulation for each new branch, and (3) pruning branches that exhibits prolonged similarity during simulation. Compared with stochastic Sampling, LATR accelerates policy learning by 131% on average and improves final pass@1 performance by 4.2% on both GRPO and Dynamic sAmpling Policy Optimization (DAPO) algorithms across different reasoning tasks. Our code and data are publicly available at https://github.com/starreeze/latr.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。