PofoliaShared via Pofolia

Nature Communications· 2026Q1

Supervising the search process produces reliable and generalizable information-seeking agents

Guangzhi Xiong, Qiao Jin, Xiao Wang, Fang Yin et al.

Short summary

A new framework, RAG-Gym, shifts supervision for AI search agents from final answers to the intermediate search steps, leading to agents that are more reliable and generalize better to new tasks.

AI-generated from the title and abstract; the full text is not read.

Key points

  • RAG-Gym shifts AI search agent supervision from final answers to intermediate search steps.
  • This process supervision improves agent reliability and generalization, especially for out-of-domain tasks.
  • The Re² Search++ agent, built on RAG-Gym, shows substantial gains on multi-hop information-seeking benchmarks.
  • Performance improvements are driven by higher-quality search queries, not just answer optimization.
  • Learned search critics from RAG-Gym can be transferred across different LLMs, including proprietary ones.

AI-generated from the title and abstract; the full text is not read.

Abstract

Abstract Large language models (LLMs) are transforming web search by shifting from document ranking to synthesizing answers, and are increasingly deployed as autonomous agentic search systems that iteratively interact with external knowledge sources. Despite this progress, building effective search agents remains challenging because high-quality intermediate search steps are difficult to generate. Previous approaches have primarily relied on outcome supervision, rewarding agents only for producing correct final answers. This often leads to reward hacking and excessive dependence on parametric memory, limiting generalization to out-of-domain tasks. To address these limitations, we introduce RAG-Gym, a framework that shifts supervision from final answers to the search process itself. With RAG-Gym, we systematically investigate architecture design, parameter optimization, and action evaluation, identifying reasoning reflection as a critical capability for search agents. Building on this insight, we propose Re 2 Search++, a process-supervised agent that achieves substantial improvements on multi-hop information-seeking benchmarks, especially in out-of-domain settings. Performance gains are driven primarily by higher-quality search queries rather than answer optimization alone, and the learned search critics transfer across models, including proprietary LLMs. These findings show that supervising the search process produces more reliable and generalizable information-seeking agents.

The authors' abstract, as published at the source. Nature Communications, 2026 · DOI ↗

TakeawaysPremium
Ask the paperFree account

Continue with a free account

Ask the paper: 3 free questions a day about this paper; save it, get its citation, new summaries every day for your field. Takeaways are Premium.

Continue free on the web

Sign in with Google or Apple; no card needed. You come back to this paper.

On your phone:

Field: Information Systems

Information SystemsComputer Science