NeFut Logo NeFut
Admin Login

[CS.AI] When to Invoke Search? Counterfactual Supervision for Search Routing

Published at: 2026-07-09 22:00 Last updated: 2026-07-10 03:15
#AI #Machine Learning #LLM

Abstract

Search-augmented language models can use external evidence to compensate for limitations in parametric knowledge, but search is not uniformly beneficial: models may call search for questions they can already answer, or rely on noisy evidence when correction, clarification, or abstention would be more appropriate. We formulate this as an instance-level search-routing problem: deciding whether search is needed to improve task success relative to a no-search execution.

To derive supervision, we compare no-search and forced-search outcomes for the same question and construct an oracle over NO SEARCH, SEARCH, and UNSOLVED based on task-specific success. Using this oracle as both an evaluation criterion and a learning signal, we train search-routing policies with supervised fine-tuning and preference optimization, improving routing macro-F1 on oracle-eligible examples from 0.7082 to 0.8235 for Gemma E2B and from 0.7053 to 0.8365 for Qwen3.5-4B.

Further analysis shows that the learned policies reduce model-specific routing failures: Gemma primarily learns no-search restraint, while Qwen further reduces missed search; residual UNSOLVED cases reveal heterogeneous bottlenecks involving model capacity, retrieval budget, evidence use, and policy behavior.

Blogger's Review: This paper enhances the effectiveness of search routing strategies through counterfactual supervision, illustrating how to optimize search invocation decisions in specific tasks. This research provides crucial guidance for future search-augmented language models, particularly in handling complex queries, effectively reducing the occurrence of ineffective searches and improving processing efficiency.

Original Source: https://arxiv.org/abs/2607.05752

[h] Back to Home