LLM Router: Rethinking Routing with Prefill Activations
Organizations: NVIDIA
Abstract
Existing routers rely on semantic query features or handcrafted features, which often fail to capture model-specific failures or intrinsic task difficulty. We instead route using internal LLM activations, specifically the residual stream. Our key idea, Encoder-Target Decoupling, separates the model that produces the predictive signal (the Encoder) from the model whose correctness is being estimated (the Target), allowing open-weight encoders to predict the performance of closed-source target models. We evaluate layerwise geometric probes, finding that Fisher Separability () effectively identifies informative layers, supported by Effective Dimensionality () diagnostics. We then utilize a SharedTrunkNet, a joint multi-output MLP that predicts simultaneous correctness probabilities across candidate models using concatenated prefill features. In our experiments, SharedTrunkNet consistently outperforms semantic baselines. At its best, SharedTrunkNet closes 45.58% of the gap between the strongest standalone model and the oracle while achieving 74.31% cost savings relative to the most expensive model. These results demonstrate that prefill activations provide a robust routing signal, establishing activation-based routing as a high-performance alternative to purely semantic selection.
Figures & tables
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Router | STEM-trained | IID | (pp) |
| SharedTrunkNet (Fisher) | 88.15 | 88.90 | |
| Per-Model Multitask | 86.53 | 86.93 | |
| Unified Multitask | 87.24 | 88.50 | |
| Matrix Factorization | 81.87 | 85.98 | |
| kNN voting ( ) | 85.36 | 85.78 | |
| GraphRouter | 85.61 | 91.29 |
| Router | HLE held-out accuracy | IID accuracy | (pp) | Held-out macro AUC |
| SharedTrunkNet (Fisher) | 32.90 | 44.33 | 0.6367 | |
| Per-Model Multitask | 12.82 | 39.02 | 0.5541 | |
| Unified Multitask | 30.86 | 42.94 | 0.5478 | |
| Matrix Factorization | 19.02 | 41.63 | 0.5413 | |
| kNN voting ( ) | 23.43 | 29.80 | 0.5419 | |
| GraphRouter | 32.90 | 49.96 | 0.5479 |
| Model | Nemo Nano 30B | gpt oss 20b | Qwen 3.5 35B | Qwen 3.5 122B | gpt oss 120b | Nemo Super 120B |
| DeepHermes-3-8B | 0.7744 | 0.7550 | 0.7926 | 0.7928 | 0.7527 | 0.7823 |
| DS-R1-Qwen3-8B | 0.8047 | 0.7853 | 0.8290 | 0.8260 | 0.7848 | 0.8140 |
| DS-R1D-Qwen-7B | 0.7817 | 0.7563 | 0.8002 | 0.7991 | 0.7552 | 0.7918 |
| Fin-R1 | 0.8143 | 0.7981 | 0.8277 | 0.8292 | 0.7950 | 0.8215 |
| GLM-Z1-9B | 0.7923 | 0.7706 | 0.8141 | 0.8113 | 0.7619 | 0.8020 |
| Intern-S1-mini | 0.7776 | 0.7610 | 0.8036 | 0.8012 | 0.7594 | 0.7917 |
| Pool | Step ( ) | N | Q1 | Q2 | Q3 | Q4 |
| Mixed | 101 | 3 | 4 | 18 | 76 | |
| Mixed | 747 | 24 | 45 | 170 | 508 | |
| Mixed | 2198 | 239 | 397 | 728 | 834 | |
| Mixed | 3988 | 1243 | 910 | 940 | 895 | |
| Frontier | 101 | 8 | 10 | 22 | 61 | |
| Frontier | 605 | 71 | 98 | 179 | 257 |
LLM Routing
Large language model routing selects models, tools, experts, or inference strategies to balance answer quality, cost, latency, and deployment constraints.
See all 349 papers