Beyond Outcome Rewards: Constructing and Assigning Retrieval Credit for Search Agents
cs.CL, cs.AI, cs.LG
Submitted: 2026-10-07
Updated: 2026-10-07
License: http://creativecommons.org/licenses/by/4.0/
The gist: Search agents enable Large Language Models (LLMs) to iteratively retrieve and use information for complex multi-hop questions.
Abstract
Search agents enable Large Language Models (LLMs) to iteratively retrieve and use information for complex multi-hop questions. Reinforcement Learning with Verifiable Rewards (RLVR) offers a promising approach for post-training such agents, but its reliance on sparse, outcome-based supervision can make credit assignment difficult and limit learning efficiency. In this paper, we systematically investigate how intermediate supervision can improve reinforcement learning for search agents. We study a range of reward-shaping and credit-assignment strategies that provide learning signals from intermediate retrieval steps. Building on these insights, we develop a training framework that combines intermediate signals with final outcome rewards to improve learning from multi-step search trajectories. Experiments across multiple benchmarks under matched training conditions demonstrate improvements in aggregate search-agent performance and show that both the choice of intermediate signal and where its credit is assigned affect training behaviour. These findings show that reward design and credit assignment are important design dimensions for training effective search agents.
Sources
- BiCAA: Bidirectional Credit Assignment for Search-Augmented Agent
- An Empirical Study on Reinforcement Learning for Reasoning-Search Interleaved LLM Agents
- R3-RAG: Learning Step-by-Step Reasoning and Retrieval for LLMs via Reinforcement Learning
- Beyond Trajectory Rewards: Step-level Credit Assignment for Agentic Search via Graph Modeling
- Qwen3 Technical Report
- Text Embeddings by Weakly-Supervised Contrastive Pre-training
- STAMP: Provenance-Guided Credit Assignment for Deep Search Agents
- DeepResearcher: Scaling Deep Research via Reinforcement Learning in Real-world Environments
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering