arXiv:2603.03078cs.AI2026-03KDD被引 4

让大模型代理通过检索外部经验拓展思维边界,提升复杂任务解决能力。

RAPO: Expanding Exploration for LLM Agents via Retrieval-Augmented Policy Optimization

  • 引入检索机制,让代理在推理时可参考外部非自身生成的步骤级经验。
  • 在14个数据集上平均提升5.0%,训练速度加快1.2倍。
  • 适合需要长期规划与多步推理的智能体系统开发者。

基于大语言模型的智能体强化学习(Agentic RL)在多步、工具融合推理任务中展现出巨大潜力。然而,现有方法依赖纯在线策略探索,仅能基于自身输出进行尝试,难以发现新推理路径。尽管部分工作引入离线信号增强探索,但通常使用完整离线轨迹进行策略估计,忽略了智能体演进过程中细粒度的步骤级探索动态。本文提出检索增强型策略优化(RAPO),将训练分为两个阶段:(i) 混合策略智能体采样,(ii) 检索感知策略优化。前者允许代理持续利用检索到的离线步骤级轨迹进行推理,动态扩展其推理感知范围;后者通过检索奖励与重要性重塑校准策略梯度,稳定训练并优先鼓励具有启发性的检索探索。大量实验表明,RAPO在三个智能体推理任务的14个数据集上实现平均+5.0%性能提升,同时训练效率提高1.2倍。

原文摘要 · Abstract (English)

Agentic Reinforcement Learning (Agentic RL) has shown remarkable potential in large language model-based (LLM) agents. These works can empower LLM agents to tackle complex tasks via multi-step, tool-integrated reasoning. However, an inherent limitation of existing Agentic RL methods is their reliance on a pure on-policy paradigm for exploration, restricting exploration to the agent's self-generated outputs and preventing the discovery of new reasoning perspectives for further improvement. While recent efforts incorporate auxiliary off-policy signals to enhance exploration, they typically utilize full off-policy trajectories for trajectory-level policy estimation, overlooking the necessity for the fine-grained, step-level exploratory dynamics within agentic rollout. In this paper, we revisit exploration in Agentic RL and propose Retrieval-Augmented Policy Optimization (RAPO), a novel RL framework that introduces retrieval to explicitly expand exploration during training. To achieve this, we decompose the Agentic RL training process into two phases: (i) Hybrid-policy Agentic Rollout, and (ii) Retrieval-aware Policy Optimization. Specifically, we propose a Hybrid-policy Agentic Rollout strategy, which allows the agents to continuously reason over the retrieved off-policy step-level traces. It dynamically extends the reasoning receptive field of agents, enabling broader exploration conditioned on external behaviors. Subsequently, we introduce the Retrieval-aware Policy Optimization mechanism, which calibrates the policy gradient estimation with retrieval reward and importance shaping, stabilizing training and prioritizing retrieval-illuminating exploration. Extensive experiments show that RAPO achieves an +5.0% average gain on fourteen datasets across three agentic reasoning tasks, while delivering 1.2x faster training efficiency.

强化学习大模型代理检索增强策略优化

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