Rethinking Predictive Modeling for LLM Routing: When Simple kNN Beats Complex Learned Routers
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
As large language models (LLMs) grow in scale and specialization, routing--selecting the best model for a given input--has become essential for efficient and effective deployment. While recent methods rely on complex learned routing strategies, their dependence on disparate training data and evaluation setups makes comparison and generalization difficult. In this work, we revisit LLM routing through the lens of simplicity. We show that a well-tuned k-Nearest Neighbors (kNN) approach not only matches but often outperforms state-of-the-art learned routers across diverse tasks. To support systematic evaluation, we introduce a suite of standardized routing benchmarks spanning instruction-following, question-answering, and reasoning tasks, as well as the first multi-modal routing dataset involving visual inputs. Our findings reveal that the locality properties of model performance in embedding space enable simple non-parametric methods to achieve strong routing decisions with lower sample complexity than parametric approaches. This challenges the prevailing trend toward sophisticated architectures and highlights the importance of thoroughly evaluating simple baselines before investing in complex solutions. To support reproducibility and further exploration, we will release all benchmarks and code upon publication.
Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 35 of its 47 references that are in the map. First posted on arXiv 19 May 2025.
Builds on
35 of 47 references.
References · 35
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2024Length-Controlled AlpacaEval: A Simple Way to Debias Automatic Evaluatorscited by 897
- 2023FrugalGPT: How to Use Large Language Models While Reducing Cost and Improving Performancecited by 853
- 2017Spectrally-normalized margin bounds for neural networkscited by 686
- 2024RouteLLM: Learning to Route LLMs with Preference Datacited by 682
- 2016Error bounds for approximations with deep ReLU networkscited by 574
- 2023LLM-Blender: Ensembling Large Language Models with Pairwise Ranking and Generative Fusioncited by 564
- 2024Hybrid LLM: Cost-Efficient and Quality-Aware Query Routingcited by 372
- 2017Size-Independent Sample Complexity of Neural Networkscited by 290
23 more
- 2018Estimating the intrinsic dimension of datasets by a minimal neighborhood informationcited by 250
- 2024RouterBench: A Benchmark for Multi-LLM Routing Systemcited by 238
- 2024VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Taskscited by 228
- 2023Large Language Model Routing with Benchmark Datasetscited by 172
- 2024GraphRouter: A Graph-based Router for LLM Selectionscited by 165
- 2023Routing to the Expert: Efficient Reward-guided Ensemble of Large Language Modelscited by 145
- 2023AutoMix: Automatically Mixing Language Modelscited by 125
- 2023Fly-Swat or Cannon? Cost-Effective Language Model Choice via Meta-Modelingcited by 95
- 2024RouterDC: Query-Based Router by Dual Contrastive Learning for Assembling Large Language Modelscited by 88
- 2025Harnessing Multiple Large Language Models: A Survey on LLM Ensemblecited by 86
- 2024Exploiting LLM Quantizationcited by 65
- 2024TensorOpera Router: A Multi-Model Router for Efficient LLM Inferencecited by 60
- 2024VHELM: A Holistic Evaluation of Vision Language Modelscited by 53
- 2023Fusing Models with Complementary Expertisecited by 46
- 2024MetaLLM: A High-performant and Cost-efficient Dynamic Framework for Wrapping LLMscited by 41
- 2024Eagle: Efficient Training-Free Router for Multi-LLM Inferencecited by 35
- 2024Routoo: Learning to Route to Large Language Models Effectivelycited by 34
- 2025LLM Bandit: Cost-Efficient LLM Generation via Preference-Conditioned Dynamic Routingcited by 34
- 2024Optimising Calls to Large Language Models with Uncertainty-Based Two-Tier Selectioncited by 33
- 2023Tryage: Real-time, intelligent Routing of User Prompts to Large Language Modelscited by 30
- 2024Blending Is All You Need: Cheaper, Better Alternative to Trillion-Parameters LLMcited by 27
- 2025Doing More with Less: A Survey on Routing Strategies for Resource Optimisation in Large Language Model-Based Systemscited by 20
- 2014Reducing Offline Evaluation Bias in Recommendation Systemscited by 1
Built on by
Citing papers per year: 2025: 1; 2026: 10.
11 papers.
- 2025ProxRouter: Proximity-Weighted LLM Query Routing for Improved Robustness to Outlierscited by 6
- 2026LLM Router: Rethinking Routing with Prefill Activationscited by 5
- 2026LLMRouter: Unified Infrastructure for Developing, Evaluating, and Deploying LLM Routerscited by 4
- 2026Federate the Router: Learning Language Model Routers with Sparse and Decentralized Evaluationscited by 3
- 2026From Sampled Outcomes to Capability Distributions: Rethinking Supervision for LLM Routingcited by 3
- 2026GraphRAG-Router: Learning Cost-Efficient Routing over GraphRAGs and LLMs with Reinforcement Learningcited by 1
- 2026The Routing Plateau: Understanding the Accuracy Limits of LLM Routerscited by 1
- 2026Balance of Benchmarks: Semantic Density Reweighting for Task-Conditioned Model Comparisoncited by 0
- 2026TasteRoute: Personalized Routing for Video Generationcited by 0
- 2026Decoupled Multi-Agent Orchestrationcited by 0
- 2026Embedding Geometry Shapes LLM Routing: Tracing Failure Modes and Improving Generalizationcited by 0
- 2026Embedding Geometry Shapes LLM Routing: Tracing Failure Modes and Improving Generalization13 Oct 2026
- 2026Decoupled Multi-Agent Orchestration6 Oct 2026
- 2026TasteRoute: Personalized Routing for Video Generation5 Oct 2026
- 2026Balance of Benchmarks: Semantic Density Reweighting for Task-Conditioned Model Comparison30 Aug 2026
- 2026LLMRouter: Unified Infrastructure for Developing, Evaluating, and Deploying LLM Routers7 Aug 2026
- 2026From Sampled Outcomes to Capability Distributions: Rethinking Supervision for LLM Routing5 Jun 2026
- 2026The Routing Plateau: Understanding the Accuracy Limits of LLM Routers27 May 2026
- 2026GraphRAG-Router: Learning Cost-Efficient Routing over GraphRAGs and LLMs with Reinforcement Learning31 Mar 2026
- 2026LLM Router: Rethinking Routing with Prefill Activations21 Mar 2026
- 2026Federate the Router: Learning Language Model Routers with Sparse and Decentralized Evaluations29 Jan 2026
- 2025ProxRouter: Proximity-Weighted LLM Query Routing for Improved Robustness to Outliers10 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2025Universal Model Routing for Efficient LLM Inference
- 2025Multilingual Routing in Mixture-of-Experts
- 2026Dynamic Model Routing and Cascading for Efficient LLM Inference: A Survey
- 2025Lookahead Routing for Large Language Models
- 2026RouteWeaver: Weaving Mode Selection and Execution into Unified LLM Routing
- 2024Performance Characterization of Expert Router for Scalable LLM Inference
- 2026Do Dynamic Routers Need Memory? HeRo: History-Aware Routing for Efficient LLM Inference
- 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Models