SLMs as Multi-Agent Routers: A Progressive SFT and Reinforcement Learning Approach

arXiv cs.CL Papers

Summary

This paper proposes training small language models as multi-agent routers via progressive supervised fine-tuning and reinforcement learning, achieving better retrieval relevance and lower latency compared to LLM baselines that route on intent alone.

arXiv:2608.00030v1 Announce Type: new Abstract: Specialised retrieval agents typically surface higher quality results than general-purpose search, but selecting the optimal agent for a given query remains an open problem. Current approaches route queries based on inferred topic or intent, however intent-based selection is fundamentally limited: it does not incorporate signal from retrieved content, and cannot detect when a topically aligned agent produces low-relevance results. We address this by training a small language model via supervised fine-tuning followed by reinforcement learning to jointly perform agent selection and structured parameter generation for downstream tool calls, using a hierarchical reward function grounded in retrieval relevance along with query-agent topic alignment. This enables the model to learn task-dependent agent suitability from retrieval performance: which agents reliably yield high-relevance results for which query distributions, and when to redirect queries away from specialised agents despite surface-level topical overlap. On a targeted subset of such agent-query mismatches, the trained model achieves an NDCG@10 of 0.918 compared to 0.539 and 0.490 for two LLM baselines (Amazon Nova Lite and Claude Haiku 4.5) that route on intent alone. Overall, it achieves a mean NDCG@10 of 0.771 (+0.177 over Nova Lite, +0.219 over Haiku) with a mean selection latency of 120.1ms, an 82.4% reduction over Nova Lite.
Original Article
View Cached Full Text

Cached at: 08/04/26, 07:37 AM

# SLMs as Multi-Agent Routers: A Progressive SFT and Reinforcement Learning Approach
Source: [https://arxiv.org/html/2608.00030](https://arxiv.org/html/2608.00030)
\(15 April 2026\)

###### Abstract\.

Specialised retrieval agents typically surface higher quality results than general\-purpose search, but selecting the optimal agent for a given query remains an open problem\. Current approaches route queries based on inferred topic or intent, however intent\-based selection is fundamentally limited: it does not incorporate signal from retrieved content, and cannot detect when a topically aligned agent produces low\-relevance results\. We address this by training a small language model via supervised fine\-tuning followed by reinforcement learning to jointly perform agent selection and structured parameter generation for downstream tool calls, using a hierarchical reward function grounded in retrieval relevance along with query\-agent topic alignment\. This enables the model to learn task\-dependent agent suitability from retrieval performance: which agents reliably yield high\-relevance results for which query distributions, and when to redirect queries away from specialised agents despite surface\-level topical overlap\. On a targeted subset of such agent\-query mismatches, the trained model achieves an NDCG@10 of 0\.918 compared to 0\.539 and 0\.490 for two LLM baselines \(Amazon Nova Lite and Claude Haiku 4\.5\) that route on intent alone\. Overall, it achieves a mean NDCG@10 of 0\.771 \(\+0\.177 over Nova Lite, \+0\.219 over Haiku\) with a mean selection latency of 120\.1ms, an 82\.4% reduction over Nova Lite\.

Multi\-Agent Retrieval, Query Routing, Agent Selection, Reinforcement Learning, Small Language Models, Progressive Post\-Training

††copyright:none††journalyear:2026††conference:The First Workshop on Indexing, Retrieval, and Ranking of AI Agents \(AgentSearch\) at SIGIR 2026; July 24, 2026; Melbourne, Australia††ccs:Information systems Retrieval models and ranking††ccs:Computing methodologies Reinforcement learning††ccs:Information systems Query representation## 1\.Introduction

As AI systems increasingly rely on specialised retrieval agents to answer complex queries, the task of selecting the most suitable agent for a given information need has become a critical determinant of retrieval quality\. Topic\-specific retrieval agents, each tailored to a particular domain or data source, typically surface higher quality results than broad, general\-purpose search systems\(Sugiura and Etzioni,[2000](https://arxiv.org/html/2608.00030#bib.bib29)\)\. However, this specialisation creates a selection problem: given a heterogeneous pool of agents with overlapping but distinct capabilities, how should a system determine which agent, or combination of agents, will best serve a particular query? This problem, the selection and ranking of retrieval agents based on task\-dependent suitability, is a first\-class information retrieval challenge that sits at the intersection of query understanding and agent evaluation\.

Selecting the right agent is difficult because queries are short, often ambiguous, and frequently fail to fully reflect user information needs\(Broderet al\.,[2007](https://arxiv.org/html/2608.00030#bib.bib36); Montazeralghaemet al\.,[2020](https://arxiv.org/html/2608.00030#bib.bib20)\)\. A query that superficially matches a specialised agent’s domain may not be well served by that agent if the query touches the boundary of its expertise\. For example, a query at the intersection of genomics and general biomedical literature may be poorly served by a dedicated genomics agent despite topical overlap\. Intent\-based routing, where the agent is selected solely on the basis of the query’s topic, cannot detect these mismatches because it never observes what agents actually retrieve\. This limitation applies whether the selector is a dedicated classifier, a rule\-based system, or a large language model prompted to infer query intent\(Anandet al\.,[2023](https://arxiv.org/html/2608.00030#bib.bib2)\)\.

We argue that agent selection should be grounded not only in query intent but also in downstream retrieval performance\. By training a selector that receives feedback from the quality of retrieved results, the system can learn task\-dependent agent suitability: which agents reliably produce high\-quality results for which types of queries, and, crucially, when a specialised agent is a poor fit despite apparent topical alignment\. This principle connects to recent work framing retriever selection as a learning\-to\-rank problem grounded in downstream utility\(Kim and Diaz,[2025](https://arxiv.org/html/2608.00030#bib.bib37)\), and extends the classical resource selection problem in federated search\(Shokouhi and Si,[2011](https://arxiv.org/html/2608.00030#bib.bib39); Wanget al\.,[2025b](https://arxiv.org/html/2608.00030#bib.bib38)\)into the setting of agentic retrieval systems where agents are invoked via structured tool calls with domain\-specific parameters\.

Supervised fine\-tuning alone can teach a model to select agents based on intrinsic query features such as topic, keywords, and temporal signals\. However, this is fundamentally limited by the same problem as intent\-only routing: the model never observes what agents actually retrieve, so it cannot learn when a topically appropriate agent produces poor results\. Reinforcement learning addresses this directly by using downstream retrieval quality as a reward signal, allowing the model to discover which agents reliably perform well for which query types, including cases where surface\-level topic alignment is misleading\. The two stages are complementary: supervised fine\-tuning provides a strong foundation of query understanding for reinforcement learning to explore from, while reinforcement learning refines selection decisions based on what actually works downstream\. This progressive approach enables even a small language model to exceed the selection quality of a much larger one, because the selection is grounded in empirical retrieval performance rather than parametric knowledge of domains\.

In production agentic systems, every query passes through the selection layer, making the cost and latency of agent selection a scaling bottleneck\. LLM\-based selectors incur substantial per\-query inference costs that compound across millions of requests, and their latency is further amplified in multi\-step reasoning chains where agents invoke multiple tool calls sequentially\. Beyond the selector itself, inaccurate selection imposes downstream costs: routing a query to a poorly suited agent wastes retrieval compute, consumes tokens on irrelevant results, and may trigger costly fallback or retry mechanisms\. A selector must therefore be both accurate and operationally efficient\. Small language models satisfy both requirements: with fewer parameters they deliver substantially lower inference latency and cost per query\(Subramanianet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib21); Belcaket al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib12)\), and when combined with progressive post\-training, they can match or exceed the selection quality of larger models within the latency and cost budget that real\-time agentic retrieval demands\.

Our main contributions are as follows: \(1\) We demonstrate that grounding agent selection in retrieval quality signal through reinforcement learning produces superior retrieval outcomes, with the trained router achieving a mean NDCG@10 of 0\.771 compared to 0\.594 for Nova Lite and 0\.552 for Haiku\. \(2\) We show that the RL\-trained router identifies agent\-query mismatches that are invisible to intent\-only routing, achieving an NDCG@10 of 0\.918 on redirected queries compared to 0\.539 for Nova Lite and 0\.490 for Haiku\. \(3\) We achieve a mean selection latency of 120\.1ms, an 82\.4% reduction over Nova Lite, enabling sub\-150ms end\-to\-end routing at scale\.

## 2\.Related work

In this section, we review related work on agent and resource selection, learning from retrieval quality, query understanding for agent selection, and progressive post\-training, as they pertain to our multi\-agent router design\.

### 2\.1\.Agent and resource selection

The problem of selecting an appropriate source for a given query has a long history in information retrieval\. In federated search, resource selection determines which search engines to query from a heterogeneous pool of uncooperative sources, with the goal of maximising result quality while minimising unnecessary queries\(Shokouhi and Si,[2011](https://arxiv.org/html/2608.00030#bib.bib39)\)\. Classical methods rely on collection statistics, sampled documents, or query\-resource features to estimate source relevance\(Sugiura and Etzioni,[2000](https://arxiv.org/html/2608.00030#bib.bib29)\)\. More recently, Wang et al\.\(Wanget al\.,[2025b](https://arxiv.org/html/2608.00030#bib.bib38)\)demonstrated that large language models can perform resource selection in zero\-shot and fine\-tuned settings by scoring each resource’s suitability using logit probabilities, achieving competitive performance on the TREC FedWeb collections without human\-generated labels\. However, LLM\-based selection incurs substantial per\-query inference cost, making it impractical when the selection layer sits on the critical path of every retrieval request\.111Dhasade et al\.\(Dhasadeet al\.,[2026](https://arxiv.org/html/2608.00030#bib.bib43)\)address this with RAGRoute, a lightweight neural classifier for dynamic source selection in federated RAG, reducing query overhead by up to 77\.5% while maintaining retrieval quality\. The use of smaller language models to route queries has also been shown to be effective\. Palumbo et al\.\(Palumboet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib4)\)proposed a parallel fusion architecture where a large teacher model generates routing decisions that are distilled into a smaller student model\. Beyond selecting a single source, Ding et al\.\(Dinget al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib31)\)demonstrate in BEST\-Route that adaptive routing across a heterogeneous pool outperforms binary routing approaches, highlighting that queries are not always best served by a single source\. Our work extends this principle to the retrieval domain, supporting the selection of multiple specialised agents where query intent spans more than one domain\. Kim and Diaz\(Kim and Diaz,[2025](https://arxiv.org/html/2608.00030#bib.bib37)\)frame retriever selection as a learning\-to\-rank problem in LTRR, training models to rank retrievers by their expected utility gain to downstream LLM performance\. Their work demonstrates that utility\-grounded selection outperforms heuristic approaches, a principle we share\. However, LTRR ranks different retrieval strategies \(sparse, dense, reranked\) over the same corpus using feature\-engineered models, whereas our work selects across domain\-specialised agents backed by different data sources, using a language model trained via reinforcement learning that jointly generates structured tool call parameters\.

### 2\.2\.Learning from retrieval quality

A key limitation of intent\-based agent selection is that query text alone is an unreliable basis for determining which agent will perform well\. Search queries frequently fail to fully reflect user information needs\(Broderet al\.,[2007](https://arxiv.org/html/2608.00030#bib.bib36); Montazeralghaemet al\.,[2020](https://arxiv.org/html/2608.00030#bib.bib20)\), and retrieval agents may vary in specific strengths that are not apparent from query topic alone\. Mu et al\.\(Muet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib30)\)introduce an unsupervised method for query routing for RAG in which every query is first issued to a single search engine, then an upper bound is established using all engines, and evaluation metrics determine which engine is best suited for each query\. While this demonstrates the value of using retrieved content as a routing signal, it is computationally expensive as it requires querying all engines to establish the upper bound\. In Deep Retrieval, Jiang et al\.\(Jianget al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib6)\)use reinforcement learning to generate augmented queries that maximise retrieval performance using a small language model with only 3 billion parameters and retrieval metrics as the reward\. Their work demonstrates that RL with retrieval quality as a reward can improve retrieval outcomes through a small model, a principle we extend from query augmentation to agent selection: rather than learning to rewrite queries for a single engine, we learn to select which agent to route them to\. Our approach is distinguished by using reinforcement learning to ground agent selection decisions in end\-to\-end retrieval quality\. Rather than optimising query text or selecting agents from collection statistics, the RL reward signal teaches the model which agents reliably produce high\-quality results for which types of queries, including cases where a specialised agent is a poor fit despite superficial topical alignment\.

### 2\.3\.Query understanding for agent selection

Effective agent selection requires not only choosing the right agent but also providing it with well\-structured input\. Our router jointly performs agent selection, keyword extraction, and temporal inference, each of which draws on distinct lines of prior work\. Given a query and a predefined taxonomy, query classification assigns the query to ranked categories which may be defined by topic or intent\(Caoet al\.,[2009](https://arxiv.org/html/2608.00030#bib.bib35)\)\. The challenge is that queries are short and their internal information limited\(Broderet al\.,[2007](https://arxiv.org/html/2608.00030#bib.bib36)\)\. LLMs have shown promise in deciphering query intent even when it is not explicitly stated\(Anandet al\.,[2023](https://arxiv.org/html/2608.00030#bib.bib2)\), however they impose significant latency\. Small language models have demonstrated the ability to perform well on classification tasks at lower cost\(Lepagnolet al\.,[2024](https://arxiv.org/html/2608.00030#bib.bib10)\)\. Historically, search results have been used to improve query classification, compensating for the limited information content of short queries\(Broderet al\.,[2007](https://arxiv.org/html/2608.00030#bib.bib36)\)\. We adopt a similar principle, using retrieval quality as a reinforcement learning reward signal rather than explicit feature enrichment\. On the parsing side, poorly derived attributes from a query can lead to insufficient context for tool calls and higher retrieval costs\(Song and Zheng,[2026](https://arxiv.org/html/2608.00030#bib.bib33); Chuanget al\.,[2023](https://arxiv.org/html/2608.00030#bib.bib34)\)\. Traditional methods use named entity recognition to extract explicit attributes, however NER often struggles with implicit attributes not directly mentioned in the query\. Luo et al\.\(Luoet al\.,[2023](https://arxiv.org/html/2608.00030#bib.bib1)\)employ knowledge graphs to determine implicit query attributes\. LLMs and smaller language models have shown the ability to extract features such as entities and temporal signals, improving the structural parameters of tool calls\(Palumboet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib4)\)\. We aim to perform this extraction jointly with agent selection within a single small language model, so that both tasks benefit from the same training signal\.

### 2\.4\.Progressive post\-training for agent selection

Although supervised fine\-tuning has limited generalisation\(Wuet al\.,[2026](https://arxiv.org/html/2608.00030#bib.bib14); Chuet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib7)\), it allows models to learn in\-domain data, which is particularly helpful given that small language models have little parametric knowledge of multiple and varied specialised domains\. Wei et al\.\(Weiet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib15)\)present a comprehensive study on enhancing reasoning through a two\-stage approach: supervised fine\-tuning as a cold start with structured reasoning patterns, followed by reinforcement learning to further refine these capabilities\. Their experiments show this combined approach consistently outperforms both SFT\-only and RL\-only methods, and they demonstrate that supervised fine\-tuning provides a strong foundation for RL exploration and scaling\. Motivated by these findings, we adopt the same two\-stage strategy, using supervised fine\-tuning to ground the model in domain\-specific routing behaviour before reinforcement learning refines its decisions using downstream retrieval quality as a reward signal\.

## 3\.Methodology

This section defines the agent selection problem, describes the progressive training approach that grounds selection in downstream retrieval quality, and introduces the hierarchical reward function that enables reinforcement learning to refine selection beyond what supervised fine\-tuning alone achieves\.

### 3\.1\.Problem definition

Given a queryqq, the model produces a structured attribute tuple that determines both which agent is selected and how its tool call is parameterised:

\(1\)ϕ​\(q\)=\(δ​\(q\),𝒦​\(q\),ℐ​\(q\)\)\\phi\(q\)=\\bigl\(\\delta\(q\),\\;\\mathcal\{K\}\(q\),\\;\\mathcal\{I\}\(q\)\\bigr\)
whereδ​\(q\)∈𝒟\\delta\(q\)\\in\\mathcal\{D\}is the predicted domain that determines which agent is selected,𝒦​\(q\)⊆𝒱\\mathcal\{K\}\(q\)\\subseteq\\mathcal\{V\}is a set of salient keywords passed to the agent’s retrieval function, andℐ​\(q\)=\[tstart,tend\]\\mathcal\{I\}\(q\)=\[t\_\{\\text\{start\}\},t\_\{\\text\{end\}\}\]is a temporal interval inferred fromqqthat constrains the retrieval scope\. Agent selection is encoded as a bitmask over the set of available domains, supporting the selection of multiple agents for queries whose intent spans more than one domain\. Figure[1](https://arxiv.org/html/2608.00030#S3.F1)illustrates the full pipeline, showing how the extracted attributes flow downstream, informing both agent selection and retrieval parametrisation\.

![Refer to caption](https://arxiv.org/html/2608.00030v1/x1.png)Multi\-agent query routing pipeline showing a user query passing through a router, diverging to eleven specialised agents, producing ranked results, which are evaluated by a reward oracle that feeds back into the router via reinforcement learning\.

Figure 1\.Multi\-agent query routing pipeline with reinforcement learning feedback over ranked retrieval results
### 3\.2\.Progressive training for agent selection

The model is trained in two stages: supervised fine\-tuning followed by reinforcement learning\. Each stage addresses a distinct aspect of the agent selection problem\.

#### 3\.2\.1\.Supervised fine\-tuning

The first stage trains the model on labelled query\-output pairs where the output is the structured tuple defined in Section 3\.1: a bitmask encoding the selected agent\(s\), extracted keywords, and temporal intervals\. Training data is constructed from search logs and synthetic queries \(Section 3\.3\), with labels derived from LLM\-as\-judge consensus \(Section 3\.4\)\. After this stage, the model can select agents and extract query attributes from intrinsic query features\. However, its selection behaviour is bounded by the patterns in the labelled data: like all intent\-based approaches, the SFT\-trained model has no information about which agent will actually retrieve well for a given query\.

#### 3\.2\.2\.Reinforcement learning

The second stage refines selection decisions using reinforcement learning, starting from the SFT checkpoint\. For each training query, the model generates an agent selection and structured parameters\. The selected agent executes retrieval, and the hierarchical reward function \(Section 3\.5\) evaluates both the routing decision and the quality of retrieved results\. The reward signal provides information absent from SFT training data: whether the selected agent actually retrieves relevant content\. A KL divergence penalty constrains the policy to remain close to the SFT checkpoint, preserving learned query understanding while allowing selection refinements that improve retrieval quality\.

The progressive structure is deliberate: RL without SFT initialisation would require the model to simultaneously learn query understanding and agent suitability from reward alone, which is unstable for a small model with limited parametric knowledge\(Weiet al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib15)\)\. SFT provides the foundation, and RL refines at the margin where intent\-based selection is insufficient\.

### 3\.3\.Query generation

Synthetic queries were generated using three prompting techniques with both OpenAI GPT 4\-Turbo and Gemini Flash 2\.0: minimal prompting for creative exploration, constraint heavy prompting\(Khan,[2025](https://arxiv.org/html/2608.00030#bib.bib25)\)for control over length and topic, and few shot prompting\(Tanget al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib23)\)to replicate patterns from real search logs\. Queries were sampled equally from all three techniques to ensure diversity\. For document\-based query generation, the information regularisation technique of Wang et al\.\(Wanget al\.,[2025a](https://arxiv.org/html/2608.00030#bib.bib5)\)was followed, masking 30 percent of document keywords to minimise phrase overlap between source documents and generated queries, followed by instruction generalisation and query regularisation\.

### 3\.4\.Labelling

A two\-stage labelling process was used\. Queries were first labelled using three large language models \(GPT\-5, Claude Sonnet 4\.6 and Gemini 2\.5 Pro\), with 80 percent classified by majority vote\. A random sample of a hundred queries was then manually evaluated to determine which LLM classified queries best based on temporal understanding and intent resolution\. Claude Sonnet 4\.6 was found to be the strongest judge and was used to resolve the remaining 20 percent\.

### 3\.5\.Reward function

Supervised fine\-tuning learns agent selection from query features but does not observe retrieval outcomes\. The reward function introduces this signal, enabling reinforcement learning to refine selection based on both routing decisions and the quality of results actually retrieved\.

\(2\)R=λsrc​Rsrc\+λcon​Rcon,s\.t\.​λsrc\+λcon=1R=\\lambda\_\{\\text\{src\}\}\\,R\_\{\\text\{src\}\}\+\\lambda\_\{\\text\{con\}\}\\,R\_\{\\text\{con\}\},\\quad\\text\{s\.t\. \}\\lambda\_\{\\text\{src\}\}\+\\lambda\_\{\\text\{con\}\}=1λsrc\\lambda\_\{\\text\{src\}\}is set higher thanλcon\\lambda\_\{\\text\{con\}\}to preserve the intent\-based selection behaviour established during SFT as the primary signal, while allowing retrieval relevance to refine selection where intent alone is insufficient\. Content Quality\.

\(3\)Rcon=α⋅Rel\+\(1−α\)⋅TrustR\_\{\\text\{con\}\}=\\alpha\\cdot\\text\{Rel\}\+\(1\-\\alpha\)\\cdot\\text\{Trust\}\(4\)Rel=wd​rdirect\+wc​rcomplete\+ws​rsemantic\+wf​rfactual,∑iwi=1\\text\{Rel\}=w\_\{d\}\\,r\_\{\\text\{direct\}\}\+w\_\{c\}\\,r\_\{\\text\{complete\}\}\+w\_\{s\}\\,r\_\{\\text\{semantic\}\}\+w\_\{f\}\\,r\_\{\\text\{factual\}\},\\quad\\sum\_\{i\}w\_\{i\}=1Relevance measures how well the retrieved content answers the query, decomposed into direct answerability, completeness, semantic alignment, and factual grounding\. Trust captures source credibility\. Both are scored by LLM\-as\-judge on a normalised scale\. Query\-intrinsic Routing Quality\.

\(5\)Rsrc=β​P\+γ​D\+\(1−β−γ\)​KR\_\{\\text\{src\}\}=\\beta\\,P\+\\gamma\\,D\+\(1\-\\beta\-\\gamma\)\\,KLetℛ\+\\mathcal\{R\}^\{\+\}denote the set of correctly selected specialised routes andℛ−\\mathcal\{R\}^\{\-\}the set of incorrectly selected specialised routes

\(6\)P=clip​\(b\+α\+​\|ℛ\+\|−α−​\|ℛ−\|\+δmiss,0,1\)P=\\text\{clip\}\\Big\(b\+\\alpha\_\{\+\}\|\\mathcal\{R\}^\{\+\}\|\-\\alpha\_\{\-\}\|\\mathcal\{R\}^\{\-\}\|\+\\delta\_\{\\text\{miss\}\},\\;0,1\\Big\)\(7\)δmiss=\{−ρmif specialised routing is required but none selected0otherwise\\delta\_\{\\text\{miss\}\}=\\begin\{cases\}\-\\rho\_\{m\}&\\text\{if specialised routing is required but none selected\}\\\\ 0&\\text\{otherwise\}\\end\{cases\}\\quadwhereα−\>α\+\\alpha\_\{\-\}\>\\alpha\_\{\+\}discourages over\-selection of routes, and penalty terms enforce recall of specialised sources\. Date and keyword precision score how well the temporal interval and extracted keywords match the query\. Both penalise overly narrow constraints that risk missing relevant results and overly broad constraints that dilute retrieval focus:

\(8\)D=fdate​\(q\),K=fkeyword​\(q\),D,K∈\[0,1\]D=f\_\{\\text\{date\}\}\(q\),\\quad K=f\_\{\\text\{keyword\}\}\(q\),\\quad D,K\\in\[0,1\]Discussion\.All coefficients are treated as tunable hyperparameters\. The hierarchical structure enables the model to learn agent suitability from two complementary signals:RsrcR\_\{\\text\{src\}\}ensures agents are selected in alignment with query intent, whileRconR\_\{\\text\{con\}\}favours agents that produce relevant results\. This combination is what enables the detection of agent\-query mismatches, where a specialised agent is topically aligned but produces low\-relevance results, triggering redirection to a more suitable agent\.

## 4\.Experimental setup

The experimental evaluation assesses three aspects of the trained router: retrieval quality, routing robustness on queries where specialised agents are mismatched in expertise, and selection latency\. Retrieval quality is measured using NDCG@10\. Robustness is evaluated on a targeted subset of queries whose ground\-truth agent, determined by LLM\-as\-judge, is a specialised agent but whose content falls outside that agent’s specific expertise; the evaluation assesses whether the router correctly redirects these to a more suitable agent\. Selection latency is measured end\-to\-end on the trained router served via vLLM\.

### 4\.1\.Baselines

Two large language models serve as baselines: Amazon Nova Lite and Claude Haiku 4\.5\. Both are lightweight LLMs designed for low\-latency inference, representing the class of models that would realistically be considered for agent selection in production systems where per\-query cost and latency are constrained\. Selecting models from two different providers \(Amazon and Anthropic\) reduces provider\-specific bias in the comparison\. Each is prompted with an elaborate but similar system prompt that defines every retrieval agent along with examples of queries appropriate for each agent\. Both receive the same query as the trained router and produce an agent selection based on their interpretation of query intent, with no access to retrieval quality signal\.

### 4\.2\.Choosing the base SLM

To select a base model, we benchmarked three small language models, each with fewer than 3 billion parameters to satisfy strict latency constraints, on a hundred labelled queries without fine\-tuning\. The results are shown in Table[1](https://arxiv.org/html/2608.00030#S4.T1)\. Qwen3\-0\.6B achieved the strongest performance, correctly detecting temporal recency in queries with an accuracy of 70%\.

Table 1\.Benchmarking Small Language Models without Fine\-tuning
### 4\.3\.Supervised finetuning

#### 4\.3\.1\.Dataset

We utilised a total of 56 thousand queries sourced from production search logs spanning the 11 domains covered by the agent pool to perform supervised fine tuning, upsampling sparse sources where required\.

#### 4\.3\.2\.Prompt

We prompted the small language model using a minimal prompt\.

System promptOutput format:sources\_bitmask \| keywords \| start\_date \| end\_date Sources bitmask\(sum values for multiple sources\):web=1, academic=2, finance=4, health=8, legal=16, patent=32,politics=64, transportation=128, genomics=256, chemistry=512, physics=1024 Keywords:Comma\-separated \(maximum 2 keywords\)Dates:YYYYMMDDor0for null

#### 4\.3\.3\.Hyperparameters

We fine\-tuned Qwen3\-0\.6B using Unsloth inbf16precision without quantisation\. We applied LoRA with rank 16 and alpha 32 across all attention and MLP projection layers \(q\_proj,k\_proj,v\_proj,o\_proj,gate\_proj,up\_proj,down\_proj\)\. Training ran for 2 epochs with a batch size of 8 and gradient accumulation steps of 2, yielding an effective batch size of 16\. We used a peak learning rate of2×10−42\\times 10^\{\-4\}with cosine decay and 5% warmup, and a maximum sequence length of 512\. The dataset was partitioned into 70%, 15%, and 15% splits for training, validation, and testing respectively\.

### 4\.4\.Reinforcement learning

#### 4\.4\.1\.Data

We utilised 11,000 queries sourced from production search logs spanning the 11 domains covered by the agent pool for REINFORCE\+\+ training\.

#### 4\.4\.2\.Prompt

The system prompt used for supervised fine\-tuning was retained unchanged for REINFORCE\+\+\.

#### 4\.4\.3\.Hyperparameters

We set the sampling temperature to 1\.0 to produce diverse outputs, ensuring that the advantages the model learns from are meaningful\. To discourage excessive drift from the SFT checkpoint while preserving exploratory behaviour, we set the KL divergence penalty to 0\.15\. We used an actor learning rate of5×10−65\\times 10^\{\-6\}and a training batch size of 32\.

### 4\.5\.Evaluation criteria

#### 4\.5\.1\.NDCG score

To evaluate end\-to\-end retrieval quality, NDCG is adopted as the primary ranking metric, measuring how well each router’s agent selection causes the search API to surface and order relevant results\. NDCG is advantageous because it allows for graded relevance and involves a discount function over the rank, while many other measures uniformly weight all positions\(Wanget al\.,[2013](https://arxiv.org/html/2608.00030#bib.bib17)\)\. Relevance is calculated using the concept of LLM as a judge\(Zhenget al\.,[2023](https://arxiv.org/html/2608.00030#bib.bib18)\)across three dimensions\.Direct answer matchmeasures how well the retrieved content answers the query\.Source qualityevaluates the quality of the sources used to answer the query, which is important when there is a wide variety of datasets available to answer a query\.Completenessmeasures whether the response fully covers all necessary aspects of the question\. Each router produces an agent selection per query, which is passed directly to the search API to yield a ranked result list\. Rank order is preserved throughout; no re\-ranking is applied at any stage, as the objective is to isolate retrieval quality as delivered by the search API under each routing decision\.

Relevance judgments are obtained from Claude Sonnet 4\.6 acting as a blind judge\. For each query, all results are pooled into a single inference call and shuffled using a deterministic hash of the query string as the random seed, so the judge has no knowledge of which system produced which result\. Each result is scored on a four\-point integer scale: 3 \(highly relevant – directly answers the query\), 2 \(relevant – on\-topic and useful\), 1 \(marginally relevant\), and 0 \(not relevant\)\. After scoring, results are un\-shuffled back to each system’s original rank order for metric computation\.

DCG@10 is computed over the original search API rank order:

\(9\)DCG@10=∑i=110relilog2⁡\(i\+1\)\\text\{DCG@10\}=\\sum\_\{i=1\}^\{10\}\\frac\{\\text\{rel\}\_\{i\}\}\{\\log\_\{2\}\(i\+1\)\}wherereli∈\{0,1,2,3\}\\text\{rel\}\_\{i\}\\in\\\{0,1,2,3\\\}is the relevance score of the result at rankii\. The ideal DCG \(IDCG@10\) is computed by sorting the same 10 relevance scores in descending order and applying the same formula, representing the maximum achievable DCG given the retrieved set:

\(10\)IDCG@10=∑i=110rel\(i\)log2⁡\(i\+1\)\\text\{IDCG@10\}=\\sum\_\{i=1\}^\{10\}\\frac\{\\text\{rel\}\_\{\(i\)\}\}\{\\log\_\{2\}\(i\+1\)\}NDCG@10 is thenDCG@10/IDCG@10\\text\{DCG@10\}/\\text\{IDCG@10\}, bounded in\[0,1\]\[0,1\]\. Queries whereIDCG=0\\text\{IDCG\}=0\(all results irrelevant\) are assignedNDCG=0\\text\{NDCG\}=0; queries returning fewer than 10 results sum only over the returned set without padding\.

#### 4\.5\.2\.Robustness evaluation

To evaluate routing robustness, a targeted subset of queries is selected where the ground\-truth agent, as determined by LLM\-as\-judge, is a specialised agent but the query content falls outside that agent’s specific area of expertise\. For example, a query at the boundary of genomics and general biomedical literature may be assigned to a genomics agent by topic but retrieve poorly from that agent due to the nature of the content\. The evaluation measures whether the SFT\+RL trained router correctly identifies these mismatches and redirects such queries to a more suitable agent, compared to Nova Lite and Claude Haiku 4\.5 which select agents purely on query intent\.

#### 4\.5\.3\.Query\-intrinsic metrics

To understand how much the choice of agents and attributes are anchored in the query, routes and attributes are determined solely from the query using LLM as a judge\. Considering this as the ground truth, route accuracy is then measured\. To measure the accuracy of keywords against the ground truth, cosine similarity is computed\. This is because keywords are often ordered randomly, or may be semantically similar but different\. After embedding the keywords using Alibaba\-NLP/gte\-base, keywords are defined as a match if the cosine similarity is greater than 60%\. Exact date accuracy is also measured against the ground truth dates\.

## 5\.Results

This section evaluates the two\-stage trained multi\-agent router across retrieval quality, routing robustness, selection latency, and supervised fine\-tuning performance\. Results are compared against Amazon Nova Lite and Claude Haiku 4\.5 throughout; both are referred to as Nova Lite and Haiku hereafter\.

### 5\.1\.Retrieval quality

Table[2](https://arxiv.org/html/2608.00030#S5.T2)presents the overall NDCG@10 comparison\. The SLM \(SFT\+RL\) router outperforms both Nova Lite and Haiku on retrieval quality, achieving a mean NDCG@10 of 0\.771 against 0\.594 for Nova Lite and 0\.552 for Haiku\. The median scores are closer \(0\.960 vs\. 0\.926 vs\. 0\.929\), suggesting that all three systems perform well on straightforward queries where routing is unambiguous, and that the mean gap is driven primarily by harder queries where precise agent selection and query expansion\(Daltonet al\.,[2014](https://arxiv.org/html/2608.00030#bib.bib32)\)matter more\. The SLM also exhibits lower variance \(±\\pm0\.372\) than both Nova Lite \(±\\pm0\.491\) and Haiku, indicating that RL fine\-tuning produces more consistent routing behaviour across query types\. The higher NDCG score reconfirms the advantage of reinforcement learning, stated by Montazeralghaem et al\.\(Montazeralghaemet al\.,[2020](https://arxiv.org/html/2608.00030#bib.bib20)\), that it can be optimised for NDCG gains\.

That the SLM outperforms both LLMs despite having orders of magnitude fewer parameters supports the central argument of this work: agent selection quality is determined by the training signal, not model size\. Both Nova Lite and Haiku have access to substantially more parametric knowledge than the 0\.6B SLM, yet this does not compensate for the absence of retrieval quality signal in their routing decisions\.

Table 2\.Overall NDCG@10 comparison
### 5\.2\.Supervised fine\-tuning results

Table[3](https://arxiv.org/html/2608.00030#S5.T3)presents the performance of the model after supervised fine\-tuning alone\. A source exact match of 89\.4% indicates reliable agent selection from intrinsic query features, correctly identifying the target domain in the vast majority of cases\. However, SFT\-only selection is not informed by retrieval relevance, which is the limitation that reinforcement learning addresses in the next stage\.

Table 3\.Evaluation metrics for the SFT\-only router \(56K training examples\)
### 5\.3\.Routing robustness

Having established that SFT produces reliable intent\-based selection, we now examine where the RL stage adds value\. Routing robustness is defined as the router’s ability to recognise when a query requires expertise that a specialised agent was not designed to provide, and redirect it to a more suitable agent\. For example, a query touching the boundary of genomics and general biomedical literature may not be well served by a dedicated genomics agent despite superficial topical overlap\.

A targeted subset of such queries was selected where specialised agents are mismatched in expertise\. Table[4](https://arxiv.org/html/2608.00030#S5.T4)shows that for this subset, queries routed by the SLM \(SFT\+RL\) router yield substantially superior NDCG scores compared to both Nova Lite and Haiku, which select agents solely on the basis of query content\.

Table 4\.NDCG@10 on queries where the router redirects away from specialised agentsThe delta over Nova Lite \(\+0\.379\) and over Haiku \(\+0\.428\) on this subset is substantially larger than the overall deltas, indicating that the retrieval quality advantage is concentrated in precisely the cases where intent\-only routing fails\. Nova Lite and Haiku, selecting purely on topic, route these queries to a specialised agent that is topically aligned but poorly suited to the actual content\. The RL\-trained router, having received retrieval quality signal during training, has learned to detect these mismatches and redirect to a general agent that retrieves more effectively\.

This result demonstrates that reinforcement learning, grounded in retrieval quality, enables the router to learn dimensions of agent suitability that are invisible to intent\-based selection regardless of model size\. It also explains why the median NDCG scores in Table[2](https://arxiv.org/html/2608.00030#S5.T2)are close across all three systems while the means diverge: all systems perform well on unambiguous queries, but the RL\-trained router significantly outperforms on the hard cases where agent\-query fit is non\-obvious\. Table[5](https://arxiv.org/html/2608.00030#S5.T5)illustrates these routing differences on representative queries\.

Table 5\.Example routing decisions across query types
### 5\.4\.Selection latency

Beyond retrieval quality and robustness, the SLM \(SFT\+RL\) router also delivers substantial latency and cost advantages\. Agent selection sits on the critical path of every retrieval request, and in agentic systems where multi\-step reasoning chains invoke multiple tool calls sequentially, selection latency compounds\. Table[6](https://arxiv.org/html/2608.00030#S5.T6)shows that the trained router achieves a mean selection latency of 120\.1 ms when served via vLLM on a NVIDIA L4 with 48GB RAM, an 82\.4% reduction over Nova Lite \(683\.6 ms\) and a 95\.1% reduction over Haiku \(2,457 ms\)\. The tail latency is especially significant for production deployment: at P99, the SLM achieves 179\.4 ms compared to 4,659 ms for Nova Lite and 6,985 ms for Haiku, meaning worst\-case selection time remains under 200ms\.

This is notable because the latency reduction comes at no cost to selection quality\. The SLM simultaneously outperforms both LLMs on NDCG@10 and operates at a fraction of the latency, eliminating the quality\-latency tradeoff that typically constrains the choice between large and small models for agent selection\. At scale, this translates directly to lower per\-query inference cost and enables sub\-150ms end\-to\-end routing, making the approach viable for latency\-sensitive production agentic systems\(Belcaket al\.,[2025](https://arxiv.org/html/2608.00030#bib.bib12)\)\.

Table 6\.Inference latency comparison

## 6\.Conclusion

This work shows that agent selection quality is determined by the training signal, not model size\. A 0\.6B parameter model trained progressively via supervised fine\-tuning and reinforcement learning outperforms two substantially larger LLMs on retrieval quality, produces more robust routing decisions on queries where intent\-based selection fails, and operates at a fraction of the inference latency and cost\. The core insight is that supervised fine\-tuning alone, like any intent\-based approach, cannot learn when a topically appropriate agent produces poor results because it never observes retrieval outcomes\. Reinforcement learning closes this gap by grounding selection in downstream retrieval quality, enabling the model to detect agent\-query mismatches that are invisible to intent\-only routing regardless of model scale\.

The approach is evaluated against prompted LLMs on a fixed pool of 11 domain agents; comparison against learned selection methods and evaluation across different agent configurations would strengthen the empirical picture, as would complementary human relevance assessments alongside the LLM\-as\-judge protocol used throughout the pipeline\. Looking ahead, the most pressing extension is to dynamic agent pools where new agents can be registered and the routing policy updated incrementally without destabilising existing behaviour\. A promising avenue here is conditioning the router on explicit agent capability representations, allowing informed selection for newly registered agents without requiring RL interaction history\. This would bridge the implicit suitability learned through retrieval quality feedback in this work with the explicit agent profiling and discovery that the broader agent search research agenda demands\.

## References

- A\. Anand, V\. V, A\. Anand, and V\. Setty \(2023\)Query understanding in the age of large language models\.External Links:2306\.16004,[Link](https://arxiv.org/abs/2306.16004)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p2.1),[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- P\. Belcak, G\. Heinrich, S\. Diao, Y\. Fu, X\. Dong, S\. Muralidharan, Y\. C\. Lin, and P\. Molchanov \(2025\)Small language models are the future of agentic ai\.External Links:2506\.02153,[Link](https://arxiv.org/abs/2506.02153)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p5.1),[§5\.4](https://arxiv.org/html/2608.00030#S5.SS4.p2.1)\.
- A\. Z\. Broder, M\. Fontoura, E\. Gabrilovich, A\. Joshi, V\. Josifovski, and T\. Zhang \(2007\)Robust classification of rare queries using web knowledge\.InProceedings of the 30th Annual International ACM SIGIR Conference on Research and Development in Information Retrieval,SIGIR ’07,New York, NY, USA,pp\. 231–238\.External Links:ISBN 9781595935977,[Link](https://doi.org/10.1145/1277741.1277783),[Document](https://dx.doi.org/10.1145/1277741.1277783)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p2.1),[§2\.2](https://arxiv.org/html/2608.00030#S2.SS2.p1.1),[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- H\. Cao, D\. H\. Hu, D\. Shen, D\. Jiang, J\. Sun, E\. Chen, and Q\. Yang \(2009\)Context\-aware query classification\.InProceedings of the 32nd International ACM SIGIR Conference on Research and Development in Information Retrieval,SIGIR ’09,New York, NY, USA,pp\. 3–10\.External Links:ISBN 9781605584836,[Link](https://doi.org/10.1145/1571941.1571945),[Document](https://dx.doi.org/10.1145/1571941.1571945)Cited by:[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- T\. Chu, Y\. Zhai, J\. Yang, S\. Tong, S\. Xie, D\. Schuurmans, Q\. V\. Le, S\. Levine, and Y\. Ma \(2025\)SFT memorizes, rl generalizes: a comparative study of foundation model post\-training\.CoRRabs/2501\.17161,pp\. 1–14\.External Links:[Link](https://arxiv.org/abs/2501.17161)Cited by:[§2\.4](https://arxiv.org/html/2608.00030#S2.SS4.p1.1)\.
- Y\. Chuang, W\. Fang, S\. Li, W\. Yih, and J\. Glass \(2023\)Expand, rerank, and retrieve: query reranking for open\-domain question answering\.External Links:2305\.17080,[Link](https://arxiv.org/abs/2305.17080)Cited by:[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- J\. Dalton, L\. Dietz, and J\. Allan \(2014\)Entity query feature expansion using knowledge base links\.InProceedings of the 37th International ACM SIGIR Conference on Research & Development in Information Retrieval,SIGIR ’14,New York, NY, USA,pp\. 365–374\.External Links:ISBN 9781450322577,[Link](https://doi.org/10.1145/2600428.2609628),[Document](https://dx.doi.org/10.1145/2600428.2609628)Cited by:[§5\.1](https://arxiv.org/html/2608.00030#S5.SS1.p1.2)\.
- A\. Dhasade, R\. Guerraoui, A\. Kermarrec, D\. Petrescu, R\. Pires, M\. Randl, and M\. de Vos \(2026\)Efficient federated search for retrieval\-augmented generation using lightweight routing\.External Links:2502\.19280,[Link](https://arxiv.org/abs/2502.19280)Cited by:[footnote 1](https://arxiv.org/html/2608.00030#footnote1)\.
- D\. Ding, A\. Mallick, S\. Zhang, C\. Wang, D\. Madrigal, M\. D\. C\. H\. Garcia, M\. Xia, L\. V\. S\. Lakshmanan, Q\. Wu, and V\. Rühle \(2025\)BEST\-route: adaptive llm routing with test\-time optimal compute\.External Links:2506\.22716,[Link](https://arxiv.org/abs/2506.22716)Cited by:[§2\.1](https://arxiv.org/html/2608.00030#S2.SS1.p1.1)\.
- P\. Jiang, J\. Lin, L\. Cao, R\. Tian, S\. Kang, Z\. Wang, J\. Sun, and J\. Han \(2025\)DeepRetrieval: hacking real search engines and retrievers with large language models via reinforcement learning\.External Links:2503\.00223,[Link](https://arxiv.org/abs/2503.00223)Cited by:[§2\.2](https://arxiv.org/html/2608.00030#S2.SS2.p1.1)\.
- I\. Khan \(2025\)You don’t need prompt engineering anymore: the prompting inversion\.External Links:2510\.22251,[Link](https://arxiv.org/abs/2510.22251)Cited by:[§3\.3](https://arxiv.org/html/2608.00030#S3.SS3.p1.1)\.
- T\. E\. Kim and F\. Diaz \(2025\)LTRR: learning to rank retrievers for llms\.External Links:2506\.13743Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p3.1),[§2\.1](https://arxiv.org/html/2608.00030#S2.SS1.p1.1)\.
- P\. Lepagnol, T\. Gerald, S\. Ghannay, C\. Servan, and S\. Rosset \(2024\)Small language models are good too: an empirical study of zero\-shot classification\.CoRRabs/2404\.11122,pp\. 1–3\.External Links:[Link](https://arxiv.org/abs/2404.11122)Cited by:[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- C\. Luo, R\. Goutam, H\. Zhang, C\. Zhang, Y\. Song, and B\. Yin \(2023\)Implicit query parsing at amazon product search\.InProceedings of the 46th International ACM SIGIR Conference on Research and Development in Information Retrieval,New York, NY, USA,pp\. 3380–3384\.External Links:[Document](https://dx.doi.org/10.1145/3539618.3591858)Cited by:[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- A\. Montazeralghaem, H\. Zamani, and J\. Allan \(2020\)A reinforcement learning framework for relevance feedback\.InProceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval,SIGIR ’20,New York, NY, USA,pp\. 59–68\.External Links:ISBN 9781450380164,[Link](https://doi.org/10.1145/3397271.3401099),[Document](https://dx.doi.org/10.1145/3397271.3401099)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p2.1),[§2\.2](https://arxiv.org/html/2608.00030#S2.SS2.p1.1),[§5\.1](https://arxiv.org/html/2608.00030#S5.SS1.p1.2)\.
- F\. Mu, L\. Zhang, Y\. Jiang, W\. Li, Z\. Zhang, P\. Xie, and F\. Huang \(2025\)Unsupervised query routing for retrieval augmented generation\.External Links:2501\.07793,[Link](https://arxiv.org/abs/2501.07793)Cited by:[§2\.2](https://arxiv.org/html/2608.00030#S2.SS2.p1.1)\.
- E\. Palumbo, M\. Isaksson, A\. Tamborrino, M\. Movin, C\. Dincu, A\. Vardasbi, L\. Nikeshkin, O\. Gorobets, A\. Nyman, P\. Newdick, H\. Bouchard, P\. Bennett, M\. Lalmas, D\. Doro, C\. Doig Cardet, and Z\. Sultan \(2025\)You say search, i say recs: a scalable agentic approach to query understanding and exploratory search at spotify\.InProceedings of the Nineteenth ACM Conference on Recommender Systems,RecSys ’25,New York, NY, USA,pp\. 1117–1121\.External Links:ISBN 9798400713644,[Link](https://doi.org/10.1145/3705328.3748127),[Document](https://dx.doi.org/10.1145/3705328.3748127)Cited by:[§2\.1](https://arxiv.org/html/2608.00030#S2.SS1.p1.1),[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- M\. Shokouhi and L\. Si \(2011\)Federated search\.Foundations and Trends in Information Retrieval5\(1\),pp\. 1–102\.Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p3.1),[§2\.1](https://arxiv.org/html/2608.00030#S2.SS1.p1.1)\.
- M\. Song and M\. Zheng \(2026\)A survey of query optimization in large language models\.External Links:2412\.17558,[Link](https://arxiv.org/abs/2412.17558)Cited by:[§2\.3](https://arxiv.org/html/2608.00030#S2.SS3.p1.1)\.
- S\. Subramanian, V\. Elango, and M\. Gungor \(2025\)Small language models \(slms\) can still pack a punch: a survey\.External Links:2501\.05465,[Link](https://arxiv.org/abs/2501.05465)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p5.1)\.
- A\. Sugiura and O\. Etzioni \(2000\)Query routing for web search engines: architecture and experiments\.Comput\. Netw\.33\(1\),pp\. 417–429\.External Links:ISSN 1389\-1286,[Link](https://doi.org/10.1016/S1389-1286(00)00059-1),[Document](https://dx.doi.org/10.1016/S1389-1286%2800%2900059-1)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p1.1),[§2\.1](https://arxiv.org/html/2608.00030#S2.SS1.p1.1)\.
- Y\. Tang, R\. Hasan, and T\. Runkler \(2025\)FsPONER: few\-shot prompt optimization for named entity recognition in domain\-specific scenarios\.External Links:2407\.08035,[Link](https://arxiv.org/abs/2407.08035)Cited by:[§3\.3](https://arxiv.org/html/2608.00030#S3.SS3.p1.1)\.
- J\. Wang, K\. Wang, X\. Wang, W\. Cao, R\. Paturi, and L\. Bergen \(2025a\)IR2: information regularization for information retrieval\.External Links:2402\.16200,[Link](https://arxiv.org/abs/2402.16200)Cited by:[§3\.3](https://arxiv.org/html/2608.00030#S3.SS3.p1.1)\.
- S\. Wang, S\. Zhuang, B\. Koopman, and G\. Zuccon \(2025b\)ReSLLM: large language models are strong resource selectors for federated search\.InCompanion Proceedings of the ACM on Web Conference 2025,WWW ’25,New York, NY, USA,pp\. 1360–1364\.External Links:ISBN 9798400713316,[Link](https://doi.org/10.1145/3701716.3715595),[Document](https://dx.doi.org/10.1145/3701716.3715595)Cited by:[§1](https://arxiv.org/html/2608.00030#S1.p3.1),[§2\.1](https://arxiv.org/html/2608.00030#S2.SS1.p1.1)\.
- Y\. Wang, L\. Wang, Y\. Li, D\. He, T\. Liu, and W\. Chen \(2013\)A theoretical analysis of ndcg type ranking measures\.External Links:1304\.6480,[Link](https://arxiv.org/abs/1304.6480)Cited by:[§4\.5\.1](https://arxiv.org/html/2608.00030#S4.SS5.SSS1.p1.1)\.
- L\. Wei, Y\. Li, K\. Zheng, C\. Wang, Y\. Wang, L\. Kong, L\. Sun, and W\. Huang \(2025\)Advancing multimodal reasoning via reinforcement learning with cold start\.External Links:2505\.22334,[Link](https://arxiv.org/abs/2505.22334)Cited by:[§2\.4](https://arxiv.org/html/2608.00030#S2.SS4.p1.1),[§3\.2\.2](https://arxiv.org/html/2608.00030#S3.SS2.SSS2.p2.1)\.
- Y\. Wu, Y\. Zhou, Z\. Ziheng, Y\. Peng, X\. Ye, X\. Hu, W\. Zhu, L\. Qi, M\. Yang, and X\. Yang \(2026\)On the generalization of sft: a reinforcement learning perspective with reward rectification\.External Links:2508\.05629,[Link](https://arxiv.org/abs/2508.05629)Cited by:[§2\.4](https://arxiv.org/html/2608.00030#S2.SS4.p1.1)\.
- L\. Zheng, W\. Chiang, Y\. Sheng, S\. Zhuang, Z\. Wu, Y\. Zhuang, Z\. Lin, Z\. Li, D\. Li, E\. P\. Xing, H\. Zhang, J\. E\. Gonzalez, and I\. Stoica \(2023\)Judging llm\-as\-a\-judge with mt\-bench and chatbot arena\.External Links:2306\.05685,[Link](https://arxiv.org/abs/2306.05685)Cited by:[§4\.5\.1](https://arxiv.org/html/2608.00030#S4.SS5.SSS1.p1.1)\.

Similar Articles

Learning Agent Routing From Early Experience

arXiv cs.CL

This paper introduces BoundaryRouter, a training-free framework that optimizes LLM agent usage by routing queries to either lightweight inference or full agent execution based on early experience. It also presents RouteBench, a benchmark for evaluating routing performance, showing significant improvements in speed and accuracy.

Dynamic Latent Routing

Hugging Face Daily Papers

Dynamic Latent Routing (DLR) lets LLMs learn their own inner monologue by composing sub-policies via search, inspired by language compositionality. In low-data fine-tuning, DLR matches or outperforms standard supervised fine-tuning.