AIAny
Icon for item

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

Analyzes when masking stale observations improves long-horizon search agents and why, identifying an asymmetric inverted-U relationship between masking benefit, retriever quality, and model capacity; explains a token-for-turn trade-off and releases evaluation scaffolds and trajectories.

Introduction

Long-horizon search agents accumulate many retrieved observations across repeated tool calls, forcing a trade-off between keeping more context tokens versus enabling additional interaction turns. This paper reframes simple observation masking as a regime-dependent intervention: masking helps in some retriever×model regimes but actively harms in others, producing an asymmetric inverted‑U of gains.

Key Findings
  • Regime map: the accuracy gain from masking is not monotonic — it plateaus with weak retrievers, peaks when a strong retriever meets a mid-capacity model, and collapses when the model is saturated. So what? You can't assume masking always helps; its value depends on both retriever recall and model implicit filtering.
  • Mechanism (token-for-turn trade-off): masking frees token budget for additional turns at the cost of removing potentially useful evidence. So what? Gains arise when extra turns convert failures into successes; losses occur when removed evidence was decisive.
  • Robust sweep: results hold across multiple agent backbones (4B–284B parameters), three retrievers, and both offline and live-web search benchmarks. So what? The effect is broad, not an artifact of a single model or dataset.
  • Practical artifact: authors release their scaffold and trajectories to enable reproducible follow-ups. So what? You can reproduce regime maps and test alternative context-management heuristics.
Who it's for and tradeoffs

Great fit if you design or evaluate agentic retrieval systems and need principled guidance on context management — especially when tuning retrievers and choosing model sizes for long-horizon search. Look elsewhere if your model is already saturated (very large model + very high-recall retriever) or your retriever is too weak; in those regimes masking is unlikely to help and can reduce accuracy. Also note masking is a lightweight, heuristic intervention — it informs when to prune but does not replace improvements to retrieval or model reasoning.

Where it fits

This work situates context masking alongside RAG-style retrieval and other memory/pruning strategies: instead of proposing a new retriever or learning-to-write memory, it provides an empirical and mechanistic guide for when a minimal masking heuristic is beneficial. Use it to decide whether to invest effort in smarter retrievers, larger models, or context-management policies for a given application.

Information

  • Websitearxiv.org
  • AuthorsHaoxiang Zhang, Qixin Xu, Zhuofeng Li, Lei Zhang, Pengcheng Jiang, Yu Zhang, Julian McAuley
  • Published date2026/05/29

More Items

Turns each research paper into a training environment to generate verifiable research plans by synthesizing questions from goals/background and deriving evaluation criteria from methods/experiments. Key features: four-stage extraction that reduces criterion leakage to 3.7%, a two-stage rubric-centered training (self-distillation then GRPO), and the PaperGym-20k corpus with two held-out benchmarks.

Analyzes on-policy distillation for LLM fine-tuning, shows teacher token-level supervision is often noisy and not the main driver of gains, and introduces OPSA, a supervision-free, entropy-adaptive method that suppresses low-probability tokens to improve downstream accuracy.

Trains LLM agents to proactively edit and manage their working context for long-horizon tasks using an expanded toolset (planning, long-term memory, soft offloading) and a fine-grained RL algorithm that identifies critical edits and assigns action-level credit. Improves accuracy while keeping contexts compact on long-context QA and deep search.