mlsanityData to 7 Oct 2026

BRIDGE: Bilevel Retrieval-Credit-Aware Agentic Reinforcement Learning

arXiv 2609.36505 ↗First posted 29 Sep 2026cs.AI cs.LG math.OCcited by 0

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

  1. 2026Reasoning and Tool-use Compete in Agentic RL:From Quantifying Interference to Disentangled Tuning1 Feb
  2. 2025In-the-Flow Agentic System Optimization for Effective Planning and Tool Use7 Oct
  3. 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Use1 Sep

Other references · 52

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  4. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  5. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  6. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
  7. 2024Qwen2.5 Technical Reportcited by 4,713
  8. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
  9. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
  10. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
  11. 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
  12. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
38 more
  1. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  2. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
  3. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
  4. 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
  5. 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,216
  6. 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
  7. 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,103
  8. 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
  9. 2022Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questionscited by 911
  10. 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
  11. 2023REPLUG: Retrieval-Augmented Black-Box Language Modelscited by 786
  12. 2023In-Context Retrieval-Augmented Language Modelscited by 694
  13. 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 525
  14. 2025Kimi K2: Open Agentic Intelligencecited by 523
  15. 2025Search-o1: Agentic Search-Enhanced Large Reasoning Modelscited by 423
  16. 2024Benchmarking Retrieval-Augmented Generation for Medicinecited by 326
  17. 2025R1-Searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learningcited by 322
  18. 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 303
  19. 2024Large Language Models for Mathematical Reasoning: Progresses and Challengescited by 283
  20. 2024RAFT: Adapting Language Model to Domain Specific RAGcited by 258
  21. 2016Equilibrium Propagation: Bridging the Gap Between Energy-Based Models and Backpropagationcited by 251
  22. 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
  23. 2023Query Rewriting for Retrieval-Augmented Large Language Modelscited by 211
  24. 2023RA-DIT: Retrieval-Augmented Dual Instruction Tuningcited by 194
  25. 2023MedCPT: Contrastive Pre-trained Transformers with Large-scale PubMed Search Logs for Zero-shot Biomedical Information Retrievalcited by 182
  26. 2023A Fully First-Order Method for Stochastic Bilevel Optimizationcited by 97
  27. 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
  28. 2024SEAL: Safety-enhanced Aligned LLM Fine-tuning via Bilevel Data Selectioncited by 64
  29. 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
  30. 2025s3: You Don't Need That Much Data to Train a Search Agent via RLcited by 41
  31. 2025Improving Retrieval-Augmented Generation through Multi-Agent Reinforcement Learningcited by 39
  32. 2022Beyond backpropagation: bilevel optimization through implicit differentiation and equilibrium propagationcited by 26
  33. 2025Automating Expert-Level Medical Reasoning Evaluation of Large Language Modelscited by 21
  34. 2024SmartRAG: Jointly Learn RAG-Related Tasks From the Environment Feedbackcited by 19
  35. 2025DynamicRAG: Leveraging Outputs of Large Language Model as Feedback for Dynamic Reranking in Retrieval-Augmented Generationcited by 14
  36. 2026Agentic-R: Learning to Retrieve for Agentic Searchcited by 10
  37. 2025Direct Retrieval-augmented Optimization: Synergizing Knowledge Selection and Language Modelscited by 9
  38. 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

  1. 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Models
  2. 2026APEX-Searcher: Refining Credit Assignment with Subgoaling for Agentic Retrieval-Augmented Generation
  3. 2025A Comprehensive Survey on Reinforcement Learning-based Agentic Search: Foundations, Roles, Optimizations, Evaluations, and Applications
  4. 2026RL with Learnable Textual Feedback: A Bilevel Approach
  5. 2026Beyond Outcome Rewards: Constructing and Assigning Retrieval Credit for Search Agents
  6. 2026AgenticRag-R1: Agentic Reinforcement Learning with Stack Memory for Multi-Step Reasoning, Retrieval and Memorizing
  7. 2026PiCA: Pivot-Based Credit Assignment for Search Agentic Reinforcement Learning
  8. 2026RAPO: Expanding Exploration for LLM Agents via Retrieval-Augmented Policy Optimization