mlsanityData to 14 Oct 2026

Rethinking Predictive Modeling for LLM Routing: When Simple kNN Beats Complex Learned Routers

arXiv 2505.12601 ↗First posted 19 May 2025cs.LGcited by 11

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

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  3. 2024Qwen2.5 Technical Reportcited by 4,942
  4. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
  5. 2024Length-Controlled AlpacaEval: A Simple Way to Debias Automatic Evaluatorscited by 897
  6. 2023FrugalGPT: How to Use Large Language Models While Reducing Cost and Improving Performancecited by 853
  7. 2017Spectrally-normalized margin bounds for neural networkscited by 686
  8. 2024RouteLLM: Learning to Route LLMs with Preference Datacited by 682
  9. 2016Error bounds for approximations with deep ReLU networkscited by 574
  10. 2023LLM-Blender: Ensembling Large Language Models with Pairwise Ranking and Generative Fusioncited by 564
  11. 2024Hybrid LLM: Cost-Efficient and Quality-Aware Query Routingcited by 372
  12. 2017Size-Independent Sample Complexity of Neural Networkscited by 290
23 more
  1. 2018Estimating the intrinsic dimension of datasets by a minimal neighborhood informationcited by 250
  2. 2024RouterBench: A Benchmark for Multi-LLM Routing Systemcited by 238
  3. 2024VLM2Vec: Training Vision-Language Models for Massive Multimodal Embedding Taskscited by 228
  4. 2023Large Language Model Routing with Benchmark Datasetscited by 172
  5. 2024GraphRouter: A Graph-based Router for LLM Selectionscited by 165
  6. 2023Routing to the Expert: Efficient Reward-guided Ensemble of Large Language Modelscited by 145
  7. 2023AutoMix: Automatically Mixing Language Modelscited by 125
  8. 2023Fly-Swat or Cannon? Cost-Effective Language Model Choice via Meta-Modelingcited by 95
  9. 2024RouterDC: Query-Based Router by Dual Contrastive Learning for Assembling Large Language Modelscited by 88
  10. 2025Harnessing Multiple Large Language Models: A Survey on LLM Ensemblecited by 86
  11. 2024Exploiting LLM Quantizationcited by 65
  12. 2024TensorOpera Router: A Multi-Model Router for Efficient LLM Inferencecited by 60
  13. 2024VHELM: A Holistic Evaluation of Vision Language Modelscited by 53
  14. 2023Fusing Models with Complementary Expertisecited by 46
  15. 2024MetaLLM: A High-performant and Cost-efficient Dynamic Framework for Wrapping LLMscited by 41
  16. 2024Eagle: Efficient Training-Free Router for Multi-LLM Inferencecited by 35
  17. 2024Routoo: Learning to Route to Large Language Models Effectivelycited by 34
  18. 2025LLM Bandit: Cost-Efficient LLM Generation via Preference-Conditioned Dynamic Routingcited by 34
  19. 2024Optimising Calls to Large Language Models with Uncertainty-Based Two-Tier Selectioncited by 33
  20. 2023Tryage: Real-time, intelligent Routing of User Prompts to Large Language Modelscited by 30
  21. 2024Blending Is All You Need: Cheaper, Better Alternative to Trillion-Parameters LLMcited by 27
  22. 2025Doing More with Less: A Survey on Routing Strategies for Resource Optimisation in Large Language Model-Based Systemscited by 20
  23. 2014Reducing Offline Evaluation Bias in Recommendation Systemscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 10.

11 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Universal Model Routing for Efficient LLM Inference
  2. 2025Multilingual Routing in Mixture-of-Experts
  3. 2026Dynamic Model Routing and Cascading for Efficient LLM Inference: A Survey
  4. 2025Lookahead Routing for Large Language Models
  5. 2026RouteWeaver: Weaving Mode Selection and Execution into Unified LLM Routing
  6. 2024Performance Characterization of Expert Router for Scalable LLM Inference
  7. 2026Do Dynamic Routers Need Memory? HeRo: History-Aware Routing for Efficient LLM Inference
  8. 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Models