mlsanityData to 7 Oct 2026

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

arXiv 2603.03078 ↗First posted 3 Mar 2026cs.AIcited by 6

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

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  3. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  4. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
  5. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
  6. 2023Let's Verify Step by Stepcited by 3,598
  7. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  8. 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
  9. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  10. 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
  11. 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,193
  12. 2023GAIA: a benchmark for General AI Assistantscited by 1,084
35 more
  1. 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
  2. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
  3. 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
  4. 2025BrowseComp: A Simple Yet Challenging Benchmark for Browsing Agentscited by 616
  5. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
  6. 2025Group-in-Group Policy Optimization for LLM Agent Trainingcited by 442
  7. 2025ReTool: Reinforcement Learning for Strategic Tool Use in LLMscited by 377
  8. 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRscited by 373
  9. 2023Large Language Models for Information Retrieval: A Surveycited by 331
  10. 2025WebThinker: Empowering Large Reasoning Models with Deep Research Capabilitycited by 320
  11. 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 288
  12. 2025Reasoning with Exploration: An Entropy Perspectivecited by 277
  13. 2025Learning to Reason under Off-Policy Guidancecited by 266
  14. 2025ZeroSearch: Incentivize the Search Capability of LLMs without Searchingcited by 203
  15. 2025WebDancer: Towards Autonomous Information Seeking Agencycited by 180
  16. 2025A Comprehensive Survey of Self-Evolving AI Agents: A New Paradigm Bridging Foundation Models and Lifelong Agentic Systemscited by 177
  17. 2025ReSearch: Learning to Reason with Search for LLMs via Reinforcement Learningcited by 152
  18. 2025SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoningcited by 145
  19. 2025Beyond Ten Turns: Unlocking Long-Horizon Agentic Search with Large-Scale Asynchronous RLcited by 144
  20. 2025ReSum: Unlocking Long-Horizon Search Intelligence via Context Summarizationcited by 132
  21. 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretrainingcited by 123
  22. 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Usecited by 92
  23. 2025On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weightingcited by 88
  24. 2021From Motor Control to Team Play in Simulated Humanoid Footballcited by 87
  25. 2019Distilling Policy Distillationcited by 81
  26. 2019InfoBot: Transfer and Exploration via the Information Bottleneckcited by 78
  27. 2025Agent RL Scaling Law: Agent RL with Spontaneous Code Execution for Mathematical Problem Solvingcited by 62
  28. 2025Improving Data Efficiency for LLM Reinforcement Fine-tuning Through Difficulty-targeted Online Data Selection and Rollout Replaycited by 52
  29. 2025Klear-Reasoner: Advancing Reasoning Capability via Gradient-Preserving Clipping Policy Optimizationcited by 39
  30. 2025Agentic Entropy-Balanced Policy Optimizationcited by 39
  31. 2025Toward Effective Tool-Integrated Reasoning via Self-Evolved Preference Learningcited by 17
  32. 2025From Trial-and-Error to Improvement: A Systematic Analysis of LLM Exploration Mechanisms in RLVRcited by 16
  33. 2020The Variational Bandwidth Bottleneck: Stochastic Evaluation on an Information Budgetcited by 13
  34. 2025Learn the Ropes, Then Trust the Wins: Self-imitation with Progressive Exploration for Agentic Reinforcement Learningcited by 10
  35. 2025Unifying Text Semantics and Graph Structures for Temporal Text-attributed Graphs with Large Language Modelscited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 6.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025A Comprehensive Survey on Reinforcement Learning-based Agentic Search: Foundations, Roles, Optimizations, Evaluations, and Applications
  2. 2025The Landscape of Agentic Reinforcement Learning for LLMs: A Survey
  3. 2025Agentic Reinforced Policy Optimization
  4. 2026Exploratory Memory-Augmented LLM Agent via Hybrid On- and Off-Policy Optimization
  5. 2026Learning Agentic Policy from Action Guidance
  6. 2025Toward Efficient Exploration by Large Language Model Agents
  7. 2025Guiding Exploration in Reinforcement Learning Through LLM-Augmented Observations
  8. 2025Agent-R1: A Unified and Modular Framework for Agentic Reinforcement Learning