Feed 0% source
AI/ML AI-generated

Harness-1: Reinforcement Learning for Search Agents with State-Externalizing Harnesses

Generated by a local model (nvidia/Gemma-4-26B-A4B-NVFP4) from a scientific paper, claim-checked against the full text. Provenance is open by design.

Instead of forcing an AI to remember everything it sees during a search, researchers built a "harness" that acts like a digital notepad. This notepad handles the boring organization—managing candidate pools, tracking evidence, and deduplicating results. This allows the model to focus purely on making smart decisions about what to look for next.

In the field of agentic search, we typically train models as policies (mathematical functions mapping states to actions) that act over a growing transcript of text. The model must decide how to search. Simultaneously, it must perform the mental heavy lifting of remembering which documents are useful, which constraints remain unmet, and which claims have been verified. This formulation forces reinforcement learning (RL) to optimize both high-level semantic search decisions and routine, recoverable bookkeeping.

The authors of Harness-1 argue that this dual burden makes learning poorly conditioned and inefficient. They propose "stateful cognitive offloading." This involves moving the housekeeping into the environment itself. By separating the "brain" (the policy) from the "notepad" (the harness), they claim they can produce more capable and generalizable search agents.

The Problem

The status quo in retrieval-agent training treats the agent's interaction with tools as an append-only stream of text. As a search episode progresses, the transcript grows. This consumes the model's context window (the amount of text a model can process at once). It also forces the policy to reconstruct the search state from raw, noisy observations.

This approach breaks down in two specific ways. First, the reward signal becomes incredibly sparse and noisy. If a search fails, the RL algorithm struggles to discern the cause. It cannot easily tell if the failure stemmed from a poor search query, forgotten evidence, or a failure to verify a claim. Second, the policy often collapses into suboptimal behaviors. Without an explicit way to manage information, models frequently fall into "search-only" modes. They repeatedly issue queries because they lack a reliable mechanism to curate and organize the evidence they have already found.

How It Works

The core architectural shift in Harness-1 is the introduction of a stateful search harness. This harness maintains a persistent WORKINGMEMORY. Instead of the policy seeing a massive, undifferentiated transcript, it observes a structured, rendered view of the current search state.

As illustrated in, the system operates through a clear division of labor:

Figure 2
Figure 2. Overview of Harness-1. The policy makes semantic decisions over search, inspection, curation, verification, and termination.
  1. State Maintenance: The harness manages several internal data structures. These include a candidate pool ($P_t$), a curated output set ($C_t$) with importance tags, an evidence graph ($G_t$) summarizing entity relationships, and a verification cache ($V_t$).
  2. Semantic Policy Actions: The 20B policy (built on gpt-oss-20b) makes high-level decisions. It doesn't just "search." It uses structured actions to curate (adding, removing, or tagging documents by importance), verify (writing claims to be checked against the text), and review (re-examining previously seen documents).
  3. Derived-State Rendering: To keep the context manageable, the harness performs "cognitive offloading." It uses Sentence-BM25 compression (selecting the most relevant sentences based on keyword overlap) to boil down search results. It also employs a regex-based extractor (using pattern matching) to build a compact evidence graph. This graph highlights "bridge" documents. These are documents connecting multiple entities. This makes connections visible to the policy without a full-text reread.
  4. Capacity Management: The harness enforces a hard cap of 30 documents for the curated set. When the limit is reached, the harness handles the mechanical logic. It evicts the lowest-importance documents based on the tags assigned by the policy.

Numbers

The authors report that Harness-1 (20B) achieves an average curated recall of 0.730 across eight benchmarks. This represents an +11.4 point improvement over the next strongest open-source search subagent .

Figure 1
Figure 1. Averaged performance across eight challenging search benchmarks. Each method reports: Curated Set Recall and Trajectory Recall (documents encountered anywhere in the episode).

This margin is significant for engineers looking for high-performing open-weights models. The model also remains competitive with much larger frontier-model searchers. It outperforms models like GPT-5.4 and Sonnet-4.6 in average curated recall under this protocol.

The most striking result is the model's ability to generalize. The paper finds that Harness-1's gains are significantly higher on held-out transfer benchmarks. These are tasks not seen during SFT (supervised fine-tuning) or RL. Specifically, the authors report a mean gain of +17.0 points on transfer tasks. This is much higher than the +7.9 points seen on source-family tasks .

Figure 3
Figure 3. Transfer pattern. Harness-1 gains more on held-out transfer benchmarks (+17.0 pts mean) than on source-family benchmarks (+7.9 pts), a 2.2× gap over Context-1. Transfer is the clearest signature of the mechanism. Absolute recall is useful, but the structure of the gains is more diagnostic.

This suggests the agent learns to manipulate an explicit search state. It is not just memorizing domain-specific patterns.

The training efficiency is also notable. Despite using relatively small amounts of data, the agent achieves high performance. It uses roughly 900 filtered SFT trajectories and 3,500 RL queries .

Figure 4
Figure 4. Training-data scale. Paired bars show reported unique task/query units used for SFT and RL. Data scale. The transfer pattern is not explained by simply using more training data.

The harness provides a stable, structured interface. This allows the RL signal to act more effectively.

What's Missing

While the results are impressive, there are several gaps for practitioners:

  • Brittle Extraction Logic: The evidence graph relies on a lightweight regex-based extractor for entities and dates. In highly technical domains, this could create blind spots. The harness might fail to surface the connections the policy needs.
  • Proxy Verifier Reliability: The verify action uses an LLM-based proxy to perform entailment checks (determining if one text logically follows another). The authors admit this may struggle with highly technical or ambiguous claims. If the "notepad" provides incorrect verification results, the policy will learn to rely on faulty evidence.
  • Missing Latency and Cost Data: The paper focuses on recall and accuracy. It does not report inference-time latency or throughput. It also does not specify the hardware costs of maintaining the stateful harness in production. For real-time RAG (Retrieval-Augmented Generation) systems, these costs matter.

Should You Prototype This

Yes, but with caveats. If you are building a high-precision search agent for specialized domains like finance or law, the principle is sound. Moving bookkeeping out of the context window saves optimization budget.

However, do not expect results by simply wrapping an LLM in a Python dictionary. The success of Harness-1 relies on tight integration. The policy's structured actions must work with the harness's algorithmic enforcement. If you prototype this, focus on the interface. Design a harness that provides a compact, actionable summary of the search state. Also, ensure your reward function explicitly incentivizes tool diversity. This prevents the policy from collapsing into a simple, repetitive search loop.

Code is reportedly available at https://github.com/pat-jj/harness-1.

Figures from the paper

Figure 5
Figure 5. Training dynamics. (a) Recall during RL training. (b) Mean tool diversity. Without the diversity bonus, recall plateaus and tool use collapses; with the bonus, diversity stabilises and final recall is higher. Discovery and selection.
Figure 6
Figure 6. First, Closed-Book and Naive RAG are weak on the hardest displayed benchmark (BC+), confirming that these tasks require deep agentic search; parametric knowledge and single-hop retrieval are insufficient.
Novelty
0.0/10
Overall
0.0/10
#reinforcement learning#retrieval-augmented generation#search agents#agentic workflows
How this was made
Generation

Model: nvidia/Gemma-4-26B-A4B-NVFP4
Persona: habr_engineer
Refinement: 0
Pipeline: forge-1.0

Verification

Evaluator: nvidia/Gemma-4-26B-A4B-NVFP4
Score: 97% (passed)
Claims verified: 15 / 15

Translation

Model: nvidia/Gemma-4-26B-A4B-NVFP4

Hardware & cost

NVIDIA GB10 · 128 GB unified · NVFP4 · 100% local · $0 cloud
Tokens: 157,444
Wall-time: 510.8s
Tokens/s: 308.2