用现有数据引导大模型推理,提升强化学习的探索与稳定性
Think Outside the Policy: In-Context Steered Policy Optimization
- 利用大模型上下文学习能力,无需强专家模型即可生成指导轨迹
- 在数学推理任务上显著提升性能与训练稳定性,优于传统方法
- 适合需要高效优化大模型推理能力的研究者与开发者
现有的基于可验证奖励的强化学习(RLVR)方法,如组相对策略优化(GRPO),在提升大推理模型(LRMs)的推理能力方面取得了显著进展。然而,这些方法依赖于当前策略分布内的在线策略采样,导致探索范围受限,轨迹多样性不足。近期方法尝试通过引入更强专家模型生成的轨迹来扩大策略覆盖范围,但增加了计算成本,且先进模型往往难以获取。为此,本文提出上下文引导策略优化(ICPO),一个统一框架,利用大推理模型固有的上下文学习能力,通过现有数据提供专家指导。ICPO引入混合策略GRPO与隐式专家强制机制,可在不依赖高级模型轨迹的情况下扩展探索范围。为进一步稳定优化过程,ICPO结合专家区域拒绝采样以过滤不可靠的离线轨迹,并采用渐进式专家奖励塑造,平衡早期专家引导与后期自主改进。实验表明,ICPO在数学推理基准测试中持续提升RLVR性能与训练稳定性,揭示了一种可扩展、高效的LRM强化学习范式。代码已公开于 https://github.com/Celine-hxy/ICPO。
原文摘要 · Abstract (English)
Existing Reinforcement Learning from Verifiable Rewards (RLVR) methods, such as Group Relative Policy Optimization (GRPO), have achieved remarkable progress in improving the reasoning capabilities of Large Reasoning Models (LRMs). However, they exhibit limited exploration due to reliance on on-policy rollouts which are confined to the current policy's distribution, resulting in narrow trajectory diversity. Recent approaches attempt to expand policy coverage by incorporating trajectories generated from stronger expert models, yet this reliance increases computational cost and such advanced models are often inaccessible. To address these issues, we propose In-Context Steered Policy Optimization (ICPO), a unified framework that leverages the inherent in-context learning capability of LRMs to provide expert guidance using existing datasets. ICPO introduces mixed-policy GRPO with implicit expert forcing, which expands exploration beyond the current policy distribution without requiring advanced LRM trajectories. To further stabilize optimization, ICPO integrates expert region reject sampling to filter unreliable off-policy trajectories and annealed expert-bonus reward shaping to balance early expert guidance with later autonomous improvement. Results demonstrate that ICPO consistently enhances RLVR performance and training stability on mathematical reasoning benchmarks, revealing a scalable and effective RLVR paradigm for LRMs. Our code is available at https://github.com/Celine-hxy/ICPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。