Masking Stale Observations Helps Search Agents -- Until It Doesn't: A Regime Map and Its Mechanism

Haoxiang Zhang, Qixin Xu, Zhuofeng Li, Lei Zhang, Pengcheng Jiang +2 more
5/29/2026
cs.CLcs.AIcs.IR

Official code: i-DeepSearch/observation-masking (24 stars, last commit 2026-06-04).

Abstract

Long-horizon search agents accumulate large amounts of retrieved content across many tool calls, making context-budget efficiency increasingly important. A minimal intervention is to mask stale observations from the context as the trajectory progresses, but it remains unclear when this form of context management helps and why. We study observation masking through a systematic sweep over various agent backbones (4B to 284B parameters) and three retrievers on offline and live-web agentic search benchmarks. We find that the accuracy gain from masking follows an asymmetric inverted-U shape when plotted against the model's accuracy without context management: a plateau under weak retrievers, a peak when a strong retriever meets a mid-capacity model, and a sharp collapse when the model is saturated. This pattern reflects the interaction between retriever recall and the model's implicit filtering capacity, rather than either factor in isolation. Mechanistically, masking implements a token-for-turn trade-off: it removes observations the model has largely stopped attending to and pages the agent rarely re-opens. The added turns help when they convert failures into successes, but they fail when masking removes evidence the model would otherwise have used. We therefore reframe context management as a regime-dependent intervention and provide a holistic perspective for analyzing context use in agentic deep search. We release our scaffold and trajectories here (https://github.com/i-DeepSearch/observation-masking) to support future research.

View on arXivView PDF

Code Implementations(1)

A parallel native tool caling deep search agentic framework with base window retention observation masking context management.

241Shell, PythonMay 28, 20264 months ago

Similar Papers

Cite this paper

@article{zhang2026masking,
  title  = {Masking Stale Observations Helps Search Agents -- Until It Doesn't: A Regime Map and Its Mechanism},
  author = {Haoxiang Zhang and Qixin Xu and Zhuofeng Li and Lei Zhang and Pengcheng Jiang and Yu Zhang and Julian McAuley},
  year   = {2026},
  eprint = {2606.00408},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.00408v1}
}

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/2606.00408.svg)](https://codeofpaper.com/paper/2606.00408)

Discussion