Source-linked AI summary

LexiHorizon: Stabilizing Reinforcement Learning for Long-Horizon Deep Search

Zhiqing Nong, Liang Wen, Chao-Hsuan Liu

arXiv:2610.05119v1cs.AI

TL;DR

Query-sensitive retrieval can require repeated reformulation and longer search trajectories, creating a challenge of sustaining exploration while managing accumulated tool outputs. LEXIHORIZON combines expanded context, a reasoning-preserving context window, and outcome-gated search-effort reward. Its 9B model consistently outperforms its base model and MiroThinker-1.7-mini across three benchmarks.

  • Problem

    Query-sensitive retrieval can miss relevant evidence because of query wording, while long search trajectories must manage an expanding history of tool outputs.

    Source: §1 · pp. 1–2
  • Method

    LEXIHORIZON expands trajectory context, preserves reasoning while keeping only recent tool observations in full, and gives a bounded tool-use bonus only to trajectories with nonzero answer reward.

    Source: §1 · p. 2
  • Results

    Across XBench, WebWalkerQA, and BrowseComp-ZH, LEXIHORIZON-9B consistently outperforms Qwen3.5-9B by up to 8.7 percentage points and MiroThinker-1.7-mini by up to 23.8 percentage points.

    Source: §1 · p. 2
  • Takeaways & Limitations

    Controlled ablations indicate that context capacity and context management provide complementary benefits for long search trajectories.

    Source: §5 · p. 9

Abstract

from arXiv · show

Deep search agents tackle complex knowledge tasks through iterative retrieval, multi-hop reasoning, and evidence synthesis across multiple sources. Existing approaches typically assume relatively stable retrieval systems and operate over short-horizon tool interaction. However, when retrieval is sensitive to query formulation, even a semantically appropriate query may fail to surface critical evidence because of mismatched entity names, aliases, or keyword combinations. Recovering from such failures requires repeated query reformulation and longer interaction trajectories. This setting poses a distinct training challenge, as the policy must sustain long-horizon query exploration while managing an expanding volume of retrieved content. We propose LexiHorizon, a framework for training search agents over long horizons that expands the trajectory context budget, manages accumulated retrieval content using a window over recent tool observations while preserving the reasoning history, and introduces an outcome-gated search-effort reward that provides a bounded bonus for tool invocations to trajectories with nonzero answer reward. Experiments on XBench, WebWalkerQA, and BrowseComp-ZH show that the resulting 9B model consistently outperforms both its base model and MiroThinker-1.7-mini, with maximum absolute gains of 8.7 and 23.8 percentage points, respectively. These results suggest that combining an extended context budget with reasoning-preserving context management benefits long-horizon deep search agents.

1 INTRODUCTION

Query-sensitive retrieval can turn appropriate searches into long trajectories, making it difficult to sustain exploration while managing accumulated evidence. LEXIHORIZON addresses this training challenge with expanded context, reasoning-preserving context management, and outcome-gated search effort; its 9B model outperforms both comparison models across three benchmarks.

1 INTRODUCTION

  • Query-sensitive retrieval can miss relevant evidence because of differences in entity names, aliases, terminology, or keyword combinations, requiring repeated query reformulation and producing long trajectories.Long trajectories may exhaust context before sufficient evidence is collected or a final answer is produced, while accumulated tool outputs can interfere with later query formulation.Source: §1 · pp. 1–2
  • LEXIHORIZON expands the trajectory context budget, keeps complete reasoning history while retaining only recent tool observations in full, and gives a bounded search-effort bonus only to trajectories with nonzero answer reward.The policy is optimized with GSPO over the resulting trajectories to learn evidence selection and query adaptation.Source: §1 · p. 2
  • Across XBench, WebWalkerQA, and BrowseComp-ZH, LEXIHORIZON-9B achieves maximum absolute improvements of 8.7 percentage points over Qwen3.5-9B and 23.8 percentage points over MiroThinker-1.7-mini.The model consistently outperforms both comparison models across all three benchmarks.Source: §1 · p. 2
  • The paper frames long-horizon search reinforcement learning as a stability problem involving premature context exhaustion, interference from accumulated retrieval content, and weak incentives for sustained exploration.Its proposed framework combines expanded context, reasoning-preserving context management, and outcome-gated reward to address these coupled challenges.Source: §1 · p. 2

2 RELATED WORK

Prior deep-search research spans proprietary systems, open agents and training resources, and reinforcement learning for increasingly realistic and extended search. The section identifies stable optimization over substantially longer trajectories under real, query-sensitive retrieval as insufficiently understood.

2 RELATED WORK

  • Proprietary deep-search systems retrieve from heterogeneous sources and synthesize research reports, but their models, training procedures, and extended-trajectory mechanisms remain largely inaccessible.Source: §2 · p. 2
  • Open research agents explore question decomposition, parallel search, search-augmented reasoning, and iterative search-and-draft workflows, while related work builds training data and trajectories for broader research workflows.Source: §2 · p. 2
  • Reinforcement learning research progresses from outcome-based search behavior to denser process rewards and realistic web interaction, while longer trajectories raise optimization-stability and context-management challenges.The section states that scaling to substantially longer search trajectories while maintaining stable optimization under real retrieval feedback remains insufficiently understood.Source: §2 · p. 3

3 METHOD

The method treats long-horizon search training as a coupled context-and-reward stability problem. It combines a larger context budget, a reasoning-preserving observation window, and an outcome-gated search-effort reward.

3.1 OVERVIEW AND PROBLEM FORMULATION

  • The agent follows ReAct, interleaving reasoning with tool calls or a final answer, while query reformulation is learned jointly with search planning under finite interaction and context budgets.The formulation considers lexically sensitive retrieval, where semantically similar queries can return substantially different results.Source: §3.1 · p. 3

3.2 LONG-HORIZON SEARCH TRAINING

  • LEXIHORIZON pairs an extended context budget with a reasoning-preserving context window and a bounded search-effort bonus gated on nonzero answer reward.Trajectory reset is an auxiliary mechanism for pathological long-tail trajectories.Source: §3.2 · p. 4

3.2.1 REASONING-PRESERVING CONTEXT WINDOW

  • The window keeps reasoning and tool actions intact, retains full observations from the K most recent interactions, and replaces earlier raw observations with markers.This preserves information already incorporated into reasoning while reducing redundant historical retrieval content in the active context.Source: §3.2.1 · p. 4

3.2.2 OUTCOME-GATED SEARCH-EFFORT REWARD

  • The search-effort bonus scales with answer reward, saturates after 30 tool calls, and is paired with format and repetition penalties.Zero-answer-reward trajectories receive no positive tool bonus; penalties address malformed or repetitive trajectories.Source: §3.2.2 · pp. 4–5

3.2.3 TRAINING OBJECTIVE

  • Training uses live tool-interaction rollouts and GSPO, computes group-normalized advantages, and applies policy loss only to policy-generated tokens.Tool observations provide conditioning context but are excluded from the loss.Source: §3.2.3 · pp. 5–6
  • Benchmark accuracy is averaged over eight independently sampled trajectories per question, while reliability and efficiency are reported under matched evaluation configurations.Policy length excludes tool-observation tokens; throughput is measured in policy-generated tokens per second.Source: Table 1 · p. 6, Table 2 · p. 6

4 EXPERIMENTS

Across three benchmarks, LEXIHORIZON-9B ranks first or second and surpasses MiroThinker-1.7-mini; controlled ablations show that extended context and a reasoning-preserving window are associated with better long-horizon search and training dynamics.

4.1 EXPERIMENTAL SETUP

  • The evaluation compares Qwen3.5-9B-based LEXIHORIZON-9B with its base model, a larger model, and a specialized search agent on XBench, WebWalkerQA, and BrowseComp-ZH.All models use the same search engine, tools, interaction budget, decoding configuration, and answer evaluator.Source: §4.1 · p. 6

4.2 MAIN RESULTS

  • LEXIHORIZON-9B ranks first or second across three benchmarks, improving accuracy over MiroThinker-1.7-mini by 23.8, 4.6, and 6.5 percentage points on XBench, WebWalkerQA, and BrowseComp-ZH, respectively.It also outperforms Qwen3.5-27B on XBench and BrowseComp-ZH, though that larger model scores highest on WebWalkerQA.Source: §4.2 · pp. 6–7
  • Compared with Qwen3.5-9B, LEXIHORIZON-9B reduces tool-call failures from 12.80% to 2.00%, raises valid answers from 87.60% to 99.88%, and reduces policy-generated tokens by 38.4%.Under the same hardware allocation and evaluation configuration, it reaches 1,069 policy-generated tokens per second and 0.169 completed trajectories per second.Source: §4.2 · p. 7

4.3 ABLATION STUDIES

  • Reducing the context budget from 128K to 64K lowers the valid-answer rate from 99.88% to 66.97%, while removing the context window also reduces accuracy across all three benchmarks.The ablations compare the 128K windowed full configuration with variants that remove either the extended horizon or the window, holding other settings fixed.Source: §4.3 · p. 7

4.3.1 EFFECT OF THE SEARCH HORIZON

  • Under 64K, persistent length clipping precedes sharp declines in response length and tool use; 128K sustains longer responses and gradually increasing, stabilizing tool use.The 64K policy's reward and actor entropy also decline near zero, whereas 128K reward generally improves without the same abrupt entropy decline.Source: §4.3.1 · p. 7, Figure 1 · p. 8, Figure 2 · p. 8

4.3.2 EFFECT OF THE TOOL CONTEXT WINDOW

  • The window retains recent tool observations in full while preserving complete reasoning and action history, limiting accumulation of verbose historical observations.The reported analysis suggests this reduces interference and is consistent with more sustained tool interaction and a shorter response-length upper tail.Source: §4.3.2 · pp. 8–9
  • The context window activates in more than 80% of evaluation cases and reduces overall active context by 79.1%; reasoning retains relevant evidence in 44 of 55 evidence-bearing removed observations.All 11 retention failures occur in trajectories with more than 40 tool calls.Source: §4.3.2 · p. 8, Table 4 · p. 9
  • With the window, test-time tool calls rise from 29.5 to 54.9 while the 95th-percentile active context falls from 120,818.9 to 28,324.5 tokens.During training, window-enabled reward rises steadily while reward without the window remains largely flat; the two configurations use the same outcome-gated search-effort reward.Source: §4.3.2 · pp. 8–9, Table 5 · p. 9

5 CONCLUSION

LEXIHORIZON combines extended context, a reasoning-preserving context window, and an outcome-gated search-effort reward for long-horizon deep search. Its 9B policy outperforms its base model and MiroThinker-1.7-mini across three benchmarks.

  • Across XBench, WebWalkerQA, and BrowseComp-ZH, the resulting 9B policy consistently outperforms its base model and MiroThinker-1.7-mini.Controlled ablations indicate that context capacity and context management provide complementary benefits for long search trajectories.Source: §5 · p. 9
  • LEXIHORIZON combines an extended context budget, a reasoning-preserving context window, and an outcome-gated search-effort reward.Source: §5 · p. 9
  • The findings underscore balancing context capacity and composition to support deeper search without a proportional increase in active context length.Source: §5 · p. 9

A.1 CONTEXT WINDOW ACTIVATION METRIC

The section defines an activation indicator for the context window and an activation rate over a set of trajectories.

  • The context-window activation metric records whether the window has been activated for each rollout.Source: §A.1 · p. 12
  • The corresponding window activation rate is defined over a set of N trajectories.Source: §A.1 · p. 12

A.2 ANSWER CORRECTNESS JUDGE PROMPT

Figure 4 presents the prompt for judging answer correctness. The judge compares extracted final answers with reference answers and is instructed to ignore instructions embedded in model answers.

  • The answer-correctness judge receives the original question, reference answer, and extracted final answer.Source: §A.2 · p. 12
  • The judge assigns an ordinal score from {0, 0.25, 0.5, 0.75, 1} based on semantic agreement between final and reference answers.Source: §A.2 · p. 12
  • The prompt treats evaluation fields as untrusted data and tells the judge to ignore instructions embedded in model answers.Source: §A.2 · p. 12, Figure 4 · p. 13

A.3.1 RESET MECHANISM

The reset mechanism addresses excessively long trajectories that can result from repeated query reformulation. It abandons the active search path, while the context window instead controls what information remains within that path.

  • Repeated query reformulation after retrieval failures can produce excessively long trajectories, especially when the agent cannot tell whether failure reflects its search hypothesis or query wording.Source: §A.3.1 · p. 13
  • A length-triggered reset mechanism is introduced to handle pathological long-tail trajectories.Source: §A.3.1 · p. 13
  • Discarded trajectory segments contribute to rollout cost but are excluded from the active trajectory used for policy optimization.Source: §A.3.1 · p. 14
  • Reset abandons the current search path, whereas the context window controls retained information within an active trajectory.Source: §A.3.1 · p. 14

A.3.2 EXPERIMENT

Trajectory reset has mixed benchmark effects. During training, it reduces average active response length and rollout time per token, and the reset-trained policy has a lighter test-time response-length upper tail.

A.3.2 EXPERIMENT

  • Reset improves BrowseComp-ZH by 1.9 percentage points, leaves WebWalkerQA unchanged, and decreases XBench by 2.3 percentage points.The experiment uses an active trajectory threshold of 32,768 response tokens and allows at most three resets per rollout.Source: Table 6 · p. 14, §A.3.2 · p. 14
  • With reset disabled during evaluation for both policies, the reset-trained policy produces a lighter upper tail, with extremely long responses less frequent.This difference suggests reset affects learned search behavior rather than merely truncating evaluation trajectories.Source: §A.3.2 · p. 14, Figure 5 · p. 14
  • Trajectory reset reduces average active response length throughout training by restarting trajectories that exceed the predefined threshold.This limits the response content retained for policy optimization.Source: §A.3.2 · p. 15
  • Under the same hardware and batch settings, reset training requires less wall-clock time per token, despite generation costs for discarded segments.Reset stops an unproductive active path from continuing indefinitely and reduces observed rollout-generation cost per token.Source: §A.3.2 · p. 15
Loading 2610.05119v1…