RAPO: Expanding Exploration for LLM Agents via Retrieval-Augmented Policy Optimization
On the map: Retrieval-augmented generation → Agentic RAG, agents and memory
also in 1 more
Abstract (arXiv)
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.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Placed in Agentic RAG, agents and memory, one of 15 directions in Retrieval-augmented generation. Builds on 47 of its 66 references that are in the map. First posted on arXiv 3 Mar 2026.
Builds on
47 of 66 references.
References · 47
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2023Let's Verify Step by Stepcited by 3,598
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,193
- 2023GAIA: a benchmark for General AI Assistantscited by 1,084
35 more
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2025BrowseComp: A Simple Yet Challenging Benchmark for Browsing Agentscited by 616
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
- 2025Group-in-Group Policy Optimization for LLM Agent Trainingcited by 442
- 2025ReTool: Reinforcement Learning for Strategic Tool Use in LLMscited by 377
- 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRscited by 373
- 2023Large Language Models for Information Retrieval: A Surveycited by 331
- 2025WebThinker: Empowering Large Reasoning Models with Deep Research Capabilitycited by 320
- 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 288
- 2025Reasoning with Exploration: An Entropy Perspectivecited by 277
- 2025Learning to Reason under Off-Policy Guidancecited by 266
- 2025ZeroSearch: Incentivize the Search Capability of LLMs without Searchingcited by 203
- 2025WebDancer: Towards Autonomous Information Seeking Agencycited by 180
- 2025A Comprehensive Survey of Self-Evolving AI Agents: A New Paradigm Bridging Foundation Models and Lifelong Agentic Systemscited by 177
- 2025ReSearch: Learning to Reason with Search for LLMs via Reinforcement Learningcited by 152
- 2025SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoningcited by 145
- 2025Beyond Ten Turns: Unlocking Long-Horizon Agentic Search with Large-Scale Asynchronous RLcited by 144
- 2025ReSum: Unlocking Long-Horizon Search Intelligence via Context Summarizationcited by 132
- 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretrainingcited by 123
- 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Usecited by 92
- 2025On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weightingcited by 88
- 2021From Motor Control to Team Play in Simulated Humanoid Footballcited by 87
- 2019Distilling Policy Distillationcited by 81
- 2019InfoBot: Transfer and Exploration via the Information Bottleneckcited by 78
- 2025Agent RL Scaling Law: Agent RL with Spontaneous Code Execution for Mathematical Problem Solvingcited by 62
- 2025Improving Data Efficiency for LLM Reinforcement Fine-tuning Through Difficulty-targeted Online Data Selection and Rollout Replaycited by 52
- 2025Klear-Reasoner: Advancing Reasoning Capability via Gradient-Preserving Clipping Policy Optimizationcited by 39
- 2025Agentic Entropy-Balanced Policy Optimizationcited by 39
- 2025Toward Effective Tool-Integrated Reasoning via Self-Evolved Preference Learningcited by 17
- 2025From Trial-and-Error to Improvement: A Systematic Analysis of LLM Exploration Mechanisms in RLVRcited by 16
- 2020The Variational Bandwidth Bottleneck: Stochastic Evaluation on an Information Budgetcited by 13
- 2025Learn the Ropes, Then Trust the Wins: Self-imitation with Progressive Exploration for Agentic Reinforcement Learningcited by 10
- 2025Unifying Text Semantics and Graph Structures for Temporal Text-attributed Graphs with Large Language Modelscited by 6
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026SIRI: Self-Internalizing Reinforcement Learning with Intrinsic Skills for LLM Agent Trainingcited by 5
- 2026When Denser Credit Is Not Enough: Evidence-Calibrated Policy Optimization for Long-Horizon LLM Agent Trainingcited by 4
- 2026EDGE: Experience-Distillation for Guided Exploration in Agentic Reinforcement Learningcited by 2
- 2026RAT: RunAnyThing via Fully Automated Environment Configurationcited by 1
- 2026PTCG-Bench: Can LLM Agents Master Pok'emon Trading Card Game?cited by 0
- 2026UnifiedPlayers: Enhance Tool-Integrated Reasoning in Agentic Reinforcement Learningcited by 0
- 2026UnifiedPlayers: Enhance Tool-Integrated Reasoning in Agentic Reinforcement Learning17 Sep 2026
- 2026EDGE: Experience-Distillation for Guided Exploration in Agentic Reinforcement Learning22 Aug 2026
- 2026When Denser Credit Is Not Enough: Evidence-Calibrated Policy Optimization for Long-Horizon LLM Agent Training4 Jun 2026
- 2026SIRI: Self-Internalizing Reinforcement Learning with Intrinsic Skills for LLM Agent Training1 Jun 2026
- 2026PTCG-Bench: Can LLM Agents Master Pok'emon Trading Card Game?28 May 2026
- 2026RAT: RunAnyThing via Fully Automated Environment Configuration25 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2025A Comprehensive Survey on Reinforcement Learning-based Agentic Search: Foundations, Roles, Optimizations, Evaluations, and Applications
- 2025The Landscape of Agentic Reinforcement Learning for LLMs: A Survey
- 2025Agentic Reinforced Policy Optimization
- 2026Exploratory Memory-Augmented LLM Agent via Hybrid On- and Off-Policy Optimization
- 2026Learning Agentic Policy from Action Guidance
- 2025Toward Efficient Exploration by Large Language Model Agents
- 2025Guiding Exploration in Reinforcement Learning Through LLM-Augmented Observations
- 2025Agent-R1: A Unified and Modular Framework for Agentic Reinforcement Learning