mlsanityData to 7 Oct 2026

Outcome-Aware Tool Selection for Semantic Routers: Latency-Constrained Learning Without LLM Inference

arXiv 2603.13426 ↗First posted 13 Mar 2026cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Semantic routers in LLM inference gateways select tools in the critical request path, where every millisecond of added latency compounds across millions of requests. We propose Outcome-Aware Tool Selection (OATS), which interpolates tool embeddings toward the centroid of queries where they historically succeed -- an offline process that adds no parameters, latency, or GPU cost at serving time. On MetaTool (199~tools, 4,287~queries), this improves NDCG@5 from 0.869 to 0.940; on ToolBench (2,413~APIs), from 0.834 to 0.848. We also evaluate two learned extensions: a 2,625-parameter MLP re-ranker and a 197K-parameter contrastive adapter. The MLP re-ranker hurts or matches baseline when outcome data is sparse relative to the tool set; the contrastive adapter provides comparable gains on MetaTool (NDCG@5: 0.931). All methods are evaluated on the same held-out 30% test split. The practical takeaway is to start with the zero-cost refinement and add learned components only when data density warrants it. All mechanisms run within single-digit millisecond CPU budgets.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 16 of its 33 references that are in the map. First posted on arXiv 13 Mar 2026.

Builds on

16 of 33 references.

References · 16

  1. 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,049
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  3. 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
  4. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
  5. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
  6. 2023ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIscited by 1,880
  7. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  8. 2025ToolRL: Reward is All Tool Learning Needscited by 342
  9. 2023MetaTool Benchmark for Large Language Models: Deciding Whether to Use Tools and Which to Usecited by 209
  10. 2025ToRL: Scaling Tool-Integrated RLcited by 166
  11. 2023CRAFT: Customizing LLMs by Creating and Retrieving from Specialized Toolsetscited by 122
  12. 2025Acting Less is Reasoning More! Teaching Model to Act Efficientlycited by 80
4 more
  1. 2024GTA: A Benchmark for General Tool Agentscited by 77
  2. 2025ToolOrchestra: Elevating Intelligence via Efficient Model and Tool Orchestrationcited by 44
  3. 2024ToolRerank: Adaptive and Hierarchy-Aware Reranking for Tool Retrievalcited by 40
  4. 2025AppSelectBench: Application-Level Tool Selection Benchmarkcited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.