Predictable performance is suboptimal, and what it means for model routing
Organizations: Max Planck Institute for Intelligent Systems, Tübingen · Tübingen AI Center
Abstract
Large language models (LLMs) differ in capabilities and costs, motivating routing methods that adaptively choose a model for each query. These methods rely on performance prediction: given a query, estimate whether a model will answer it correctly. Most routing methods train separate predictors of candidate models' performance to guide model selection. Our work shows that performance prediction only beats a simple baseline when the candidate model is suboptimal on the task or the performance predictor adds significant capacity relative to the candidate model. Specifically, we take a model's self-confidence, the probability it assigns to its chosen answer, as a baseline and study how much a performance predictor can improve on it. First, we show that any performance predictor that outperforms a model's self-confidence can be used to improve the model's loss: the predictor's advantage transfers exactly in log loss through recalibration. Second, accounting for the predictor's additional parameters, we bound its advantage by the model's suboptimality plus the gain from those parameters. Third, we extend our theoretical results to two-model cascades, which route queries to a more capable model when the first model is predicted to fail. Experiments with 16 language models on four benchmarks show that predictors gain a larger advantage over self-confidence on more suboptimal models. This advantage shrinks as models improve through finetuning. Cascading exhibits the same two patterns: predictor-based routing offers larger gains over confidence-based routing for more suboptimal models, and these gains diminish after finetuning.
Figures & tables
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Base release | Instruction-tuned release |
|---|---|
| Qwen3.5-9B-Base | Qwen3.5-9B |
| Gemma-3-12B-PT | Gemma-3-12B-IT |
| Llama-3.1-8B | Llama-3.1-8B-Instruct |
| Apertus-8B-2509 | Apertus-8B-Instruct-2509 |
| OLMo-3-1025-7B | OLMo-3-7B-Instruct |
| Falcon3-7B-Base | Falcon3-7B-Instruct |
| Development split | |||||
|---|---|---|---|---|---|
| Task | Finetuning | Development | Fit (85%) | Select (15%) | Test |
| WinoGrande | 32,313 | 8,080 | 6,868 | 1,212 | 1,267 |
| CommonsenseQA | 6,394 | 3,342 | 2,840 | 502 | 1,221 |
| GSM8K | 5,088 | 2,241 | 1,904 | 337 | 1,319 |
| MMLU-Pro | 5,233 | 3,000 | 2,550 | 450 | 3,032 |
| Task | Loss (nats) | Finetuned model attaining the reference |
|---|---|---|
| WinoGrande | 0.3229 | Mistral-NeMo-Minitron-8B-Base |
| CommonsenseQA | 0.3390 | Qwen3.5-9B-Base |
| GSM8K | 0.1378 | Qwen3.5-9B |
| MMLU-Pro | 0.7247 | Qwen3.5-9B-Base |
| Task | Steps | Token limit | Checkpoint selection |
|---|---|---|---|
| WinoGrande | 2,000 | 256 | Lowest development answer loss |
| CommonsenseQA | 1,200 | 512 | Lowest development answer loss |
| GSM8K | 1,500 | 4,096 | Highest development greedy accuracy |
| MMLU-Pro | 1,200 | 6,144 | Highest development greedy accuracy |
| WinoGrande | CommonsenseQA | GSM8K | MMLU-Pro | |||||
|---|---|---|---|---|---|---|---|---|
| Model | Rel. | FT | Rel. | FT | Rel. | FT | Rel. | FT |
| Qwen3.5-9B-Base | 88.6 | 93.3 | 81.5 | 83.7 | 94.3 | 96.0 | 66.8 | 76.7 |
| Qwen3.5-9B | 81.9 | 93.7 | 81.5 | 83.6 | 95.9 | 96.1 | 78.9 † | 77.3 † |
| Gemma-3-12B-PT | 89.5 | 94.2 | 76.5 | 82.4 | 87.1 | 92.7 | 53.5 | 63.0 |
| Gemma-3-12B-IT | 82.1 | 93.1 | 76.2 | 80.4 | 93.4 | 95.4 | 64.6 † | 62.6 † |
| Llama-3.1-8B | 88.3 | 93.4 | 73.1 | 78.8 | 70.6 | 82.7 | 41.6 | 50.2 |
| Comparison | Task | Pearson | Spearman | ||
| vs. | WinoGrande | 0.98 | 0.73 | ||
| CommonsenseQA | 0.98 | 0.94 | |||
| GSM8K | 0.93 | 0.94 | |||
| MMLU-Pro | 0.65 | 0.74 | |||
| vs. | WinoGrande | 0.99 | 0.90 | ||
| CommonsenseQA | 0.99 | 0.97 |
| Comparison | Task | Pearson | Spearman | ||
|---|---|---|---|---|---|
| vs. (released) | WinoGrande | 1.00 | 0.92 | ||
| CommonsenseQA | 0.98 | 0.97 | |||
| GSM8K | 0.89 | 0.90 | |||
| MMLU-Pro | 0.49 | 0.045 | 0.60 | 0.015 | |
| vs. (finetuned) | WinoGrande | 0.14 | 0.301 | -0.03 | 0.547 |
| CommonsenseQA | 0.65 | 0.003 | 0.64 | 0.004 |
| Task | Models | Weak-error term | Strong-error term | |
|---|---|---|---|---|
| WinoGrande | Released | 1.38 | 1.80 | 0.42 |
| Finetuned | 0.18 | 0.21 | 0.03 | |
| CommonsenseQA | Released | 1.09 | 1.26 | 0.17 |
| Finetuned | 0.18 | 0.17 | -0.01 | |
| GSM8K | Released | 0.52 | 0.59 | 0.07 |
| Finetuned | 0.20 | 0.24 | 0.03 |
LLM Routing
Large language model routing selects models, tools, experts, or inference strategies to balance answer quality, cost, latency, and deployment constraints.
See all 351 papers