Learning to Retrieve from Agent Trajectories

Yuqi Zhou, Sunhao Dai, Changle Qu, Liang Pang, Jun Xu +1 more
3/30/2026
cs.IRcs.AIcs.CL

No official code found. Best implementation: Yuqi-Zhou/LRAT (64 stars, Apache-2.0, last commit 2026-04-08). 1 more below.

Abstract

Information retrieval (IR) systems have traditionally been designed and trained for human users, with learning-to-rank methods relying heavily on large-scale human interaction logs such as clicks and dwell time. With the rapid emergence of large language model (LLM) powered search agents, however, retrieval is increasingly consumed by agents rather than human beings, and is embedded as a core component within multi-turn reasoning and action loops. In this setting, retrieval models trained under human-centric assumptions exhibit a fundamental mismatch with the way agents issue queries and consume results. In this work, we argue that retrieval models for agentic search should be trained directly from agent interaction data. We introduce learning to retrieve from agent trajectories as a new training paradigm, where supervision is derived from multi-step agent interactions. Through a systematic analysis of search agent trajectories, we identify key behavioral signals that reveal document utility, including browsing actions, unbrowsed rejections, and post-browse reasoning traces. Guided by these insights, we propose LRAT, a simple yet effective framework that mines high-quality retrieval supervision from agent trajectories and incorporates relevance intensity through weighted optimization. Extensive experiments on both in-domain and out-of-domain deep research benchmarks demonstrate that retrievers trained with LRAT consistently improve evidence recall, end-to-end task success, and execution efficiency across diverse agent architectures and scales. Our results highlight agent trajectories as a practical and scalable supervision source, pointing to a promising direction for retrieval in the era of agentic search.

View on arXivView PDF

Code Implementations(2)

Yuqi-Zhou/LRATCommunity77%
Apache-2.0

The implementation for SIGIR 2026: Learning to Retrieve from Agent Trajectories.

643C++, Shell6 months agoApache-2.0
agentagenticllmsearch

Public project homepage for Learning to Retrieve from Agent Trajectories

10CSS, HTML6 months ago

Similar Papers

Cite this paper

@article{zhou2026learning,
  title  = {Learning to Retrieve from Agent Trajectories},
  author = {Yuqi Zhou and Sunhao Dai and Changle Qu and Liang Pang and Jun Xu and Ji-Rong Wen},
  year   = {2026},
  eprint = {2604.04949},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2604.04949v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2604.04949.svg)](https://codeofpaper.com/paper/2604.04949)

Discussion