BRIDGE: Bilevel Retrieval-Credit-Aware Agentic Reinforcement Learning
On the map: LLM agents and tool use → Reinforcement learning for LLM agents
Abstract (arXiv)
Agentic reinforcement learning (ARL) with verifiable rewards improves the ability of large language models (LLMs) to tackle knowledge-intensive tasks by learning to interleave search and reasoning. However, most existing ARL methods optimize only LLM-generated tokens and treat retrieved evidence as environment observations. This creates an information-credit gap: failures caused by missing or misleading evidence are attributed to the LLM policy rather than to the retriever, which motivates training the LLM and the retriever jointly. In this paper, we show that retrieval and LLM policy learning are order-sensitive: adapting the retriever before optimizing the policy yields a larger reward gain than the reverse order. To preserve this hierarchy while allowing both components to co-adapt, we formulate retrieval-augmented agentic RL as a bilevel optimization problem. To solve it efficiently, we introduce BRIDGE, a memory-efficient first-order bilevel method motivated by a loss-landscape analysis of the RL and retrieval objectives. Across seven open-domain QA benchmarks, BRIDGE achieves the highest average accuracy with both 3B and 7B backbones, improving the multi-hop average over the strongest baseline by 9.6 and 3.4 EM points, respectively. It also achieves the best averaged answer accuracy and reasoning quality across medical QA benchmarks.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 55 of its 94 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
55 of 94 references.
Closest ancestors · 3
Same direction, newest first
- 2026Reasoning and Tool-use Compete in Agentic RL:From Quantifying Interference to Disentangled Tuning1 Feb
- 2025In-the-Flow Agentic System Optimization for Effective Planning and Tool Use7 Oct
- 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Use1 Sep
Other references · 52
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
38 more
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
- 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,216
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,103
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
- 2022Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questionscited by 911
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2023REPLUG: Retrieval-Augmented Black-Box Language Modelscited by 786
- 2023In-Context Retrieval-Augmented Language Modelscited by 694
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 525
- 2025Kimi K2: Open Agentic Intelligencecited by 523
- 2025Search-o1: Agentic Search-Enhanced Large Reasoning Modelscited by 423
- 2024Benchmarking Retrieval-Augmented Generation for Medicinecited by 326
- 2025R1-Searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learningcited by 322
- 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 303
- 2024Large Language Models for Mathematical Reasoning: Progresses and Challengescited by 283
- 2024RAFT: Adapting Language Model to Domain Specific RAGcited by 258
- 2016Equilibrium Propagation: Bridging the Gap Between Energy-Based Models and Backpropagationcited by 251
- 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
- 2023Query Rewriting for Retrieval-Augmented Large Language Modelscited by 211
- 2023RA-DIT: Retrieval-Augmented Dual Instruction Tuningcited by 194
- 2023MedCPT: Contrastive Pre-trained Transformers with Large-scale PubMed Search Logs for Zero-shot Biomedical Information Retrievalcited by 182
- 2023A Fully First-Order Method for Stochastic Bilevel Optimizationcited by 97
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2024SEAL: Safety-enhanced Aligned LLM Fine-tuning via Bilevel Data Selectioncited by 64
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
- 2025s3: You Don't Need That Much Data to Train a Search Agent via RLcited by 41
- 2025Improving Retrieval-Augmented Generation through Multi-Agent Reinforcement Learningcited by 39
- 2022Beyond backpropagation: bilevel optimization through implicit differentiation and equilibrium propagationcited by 26
- 2025Automating Expert-Level Medical Reasoning Evaluation of Large Language Modelscited by 21
- 2024SmartRAG: Jointly Learn RAG-Related Tasks From the Environment Feedbackcited by 19
- 2025DynamicRAG: Leveraging Outputs of Large Language Model as Feedback for Dynamic Reranking in Retrieval-Augmented Generationcited by 14
- 2026Agentic-R: Learning to Retrieve for Agentic Searchcited by 10
- 2025Direct Retrieval-augmented Optimization: Synergizing Knowledge Selection and Language Modelscited by 9
- 2025Beyond Value Functions: Single-Loop Bilevel Optimization under Flatness Conditionscited by 9
50 most cited of 52 All 52 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Models
- 2026APEX-Searcher: Refining Credit Assignment with Subgoaling for Agentic Retrieval-Augmented Generation
- 2025A Comprehensive Survey on Reinforcement Learning-based Agentic Search: Foundations, Roles, Optimizations, Evaluations, and Applications
- 2026RL with Learnable Textual Feedback: A Bilevel Approach
- 2026Beyond Outcome Rewards: Constructing and Assigning Retrieval Credit for Search Agents
- 2026AgenticRag-R1: Agentic Reinforcement Learning with Stack Memory for Multi-Step Reasoning, Retrieval and Memorizing
- 2026PiCA: Pivot-Based Credit Assignment for Search Agentic Reinforcement Learning
- 2026RAPO: Expanding Exploration for LLM Agents via Retrieval-Augmented Policy Optimization