English

RACER: Risk-Aware Calibrated Efficient Routing for Large Language Models

Machine Learning 2026-03-10 v1 Artificial Intelligence Statistics Theory Statistics Theory

Abstract

Efficiently routing queries to the optimal large language model (LLM) is crucial for optimizing the cost-performance trade-off in multi-model systems. However, most existing routers rely on single-model selection, making them susceptible to misrouting. In this work, we formulate LLM routing as the α\alpha-VOR problem to minimize expected set size while controlling the misrouting risk, and propose a novel method -- RACER, extending base routers to output model sets that can be subsequently aggregated for improved output. In particular, RACER constructs nested model sets via augmented scoring and utilizes finite-sample concentration bounds to calibrate a threshold that allows for both variable set sizes and abstention. We theoretically prove that RACER achieves rigorous distribution-free risk control on unseen test data in a post-hoc and model-agnostic manner. Extensive experiments verify our theoretical guarantees and demonstrate that RACER consistently enhances downstream accuracy across a wide range of benchmarks.

Keywords

Cite

@article{arxiv.2603.06616,
  title  = {RACER: Risk-Aware Calibrated Efficient Routing for Large Language Models},
  author = {Sai Hao and Hao Zeng and Hongxin Wei and Bingyi Jing},
  journal= {arXiv preprint arXiv:2603.06616},
  year   = {2026}
}
R2 v1 2026-07-01T11:07:32.837Z