arXiv:2510.06672cs.LG2025-10被引 17

通过探索与利用的动态平衡,提升大模型强化学习的推理能力。

XRPO: Pushing the limits of GRPO with Targeted Exploration and Exploitation

  • 根据不确定度降低潜力自适应分配采样数量,增强对难题的探索。
  • 在零奖励提示中注入精选示例,引导模型走向更复杂的推理路径。
  • 通过新颖性感知的优势增强机制,放大低概率但正确的回复,适合强化学习研究者。

强化学习算法如GRPO推动了大语言模型(LLM)推理能力的发展。尽管增加采样数量可稳定训练,但现有方法在复杂提示上探索不足,且未充分利用有价值的反馈信号,原因在于对所有提示采用固定的采样分配(如每提示生成16次采样)并依赖稀疏奖励。本文提出XRPO(eXplore - eXploit GRPO),一个统一框架,从采样探索-利用的原理出发重构策略优化。为增强探索,XRPO引入数学基础的采样分配器,自适应优先处理有望减少不确定性的提示;并通过上下文种子策略注入精选示例,缓解零奖励提示中的训练停滞。为强化利用,设计组内相对、新颖性感知的优势锐化机制,利用序列似然放大低概率但正确的响应,从而扩展策略在稀疏奖励下的表现。在涵盖数学与编码任务的多个基准测试中,包括推理与非推理模型,XRPO相比现有方法(如GRPO和GSPO)最高实现pass@1提升4%、cons@32提升6%,同时训练收敛速度最快提升2.7倍。

原文摘要 · Abstract (English)

Reinforcement learning algorithms such as GRPO have driven recent advances in large language model (LLM) reasoning. While scaling the number of rollouts stabilizes training, existing approaches suffer from limited exploration on challenging prompts and leave informative feedback signals underexploited, due to context-independent rollout allocation across prompts (e.g., generating 16 rollouts per prompt) and relying heavily on sparse rewards. This paper presents XRPO(eXplore - eXploit GRPO), a unified framework that recasts policy optimization through the principled lens of rollout exploration-exploitation. To enhance exploration, XRPO introduces a mathematically grounded rollout allocator that adaptively prioritizes prompts with higher potential for uncertainty reduction. It further addresses stagnation on zero-reward prompts through an in-context seeding strategy that injects curated exemplars, steering the model into more difficult reasoning trajectories. To strengthen exploitation, XRPO develops a group-relative, novelty-aware advantage sharpening mechanism that leverages sequence likelihoods to amplify low-probability yet correct responses, thereby extending the policy's reach beyond sparse rewards. Experiments across diverse math and coding benchmarks on both reasoning and non-reasoning models demonstrate that XRPO outperforms existing advances (e.g., GRPO and GSPO) up to 4% pass@1 and 6% cons@32, while accelerating training convergence by up to 2.7X.

强化学习大模型推理策略优化探索利用

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