Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Beyond Outcome Reward: Decoupling Search and Answering Improves LLM Agents

About

Enabling large language models (LLMs) to utilize search tools offers a promising path to overcoming fundamental limitations such as knowledge cutoffs and hallucinations. Recent work has explored reinforcement learning (RL) for training search-augmented agents that interleave reasoning and retrieval before answering. These approaches usually rely on outcome-based rewards (e.g., exact match), implicitly assuming that optimizing for final answers will also yield effective intermediate search behaviors. Our analysis challenges this assumption: we uncover multiple systematic deficiencies in search that arise under outcome-only training and ultimately degrade final answer quality, including failure to invoke tools, invalid queries, and redundant searches. To address these shortcomings, we introduce DeSA (Decoupling Search-and-Answering), a simple two-stage training framework that explicitly separates search optimization from answer generation. In Stage 1, agents are trained to improve search effectiveness with retrieval recall-based rewards. In Stage 2, outcome rewards are employed to optimize final answer generation. Across seven QA benchmarks, DeSA-trained agents consistently improve search behaviors, delivering substantially higher search recall and answer accuracy than outcome-only baselines. Notably, DeSA outperforms single-stage training approaches that simultaneously optimize recall and outcome rewards, underscoring the necessity of explicitly decoupling the two objectives.

Yiding Wang, Zhepei Wei, Xinyu Zhu, Yu Meng• 2025

Related benchmarks

TaskDatasetResultRank
Multi-hop Question AnsweringHotpotQA in-domain
EM42.4
53
General Question AnsweringTriviaQA out-of-domain (val test)
EM63.1
33
Question AnsweringNQ (Natural Questions) in-domain (test)
Exact Match46.8
29
Multi-hop Question Answering2WIKI (out-of-domain)
Accuracy37.4
28
Multi-hop Question AnsweringMuSiQue (out-of-domain)
Exact Match19.7
18
Multi-hop Question AnsweringBamboogle (out-of-domain)
Exact Match39.5
18
Question AnsweringPopQA out-of-domain
Exact Match44
18
Showing 7 of 7 rows

Other info

Follow for update