From Sampled Outcomes to Capability Distributions: Rethinking Supervision for LLM Routing

Guannan Lai, Haoran Hu, Long Chen, Zhenguo Li, Han-Jia Ye


Abstract
Existing LLM routing methods often construct supervision from a single sampled response for each query–model pair. Because LLM generation is stochastic, however, such an observation can be an unstable estimate of model capability: semantically equivalent query formulations and repeated decoding may yield different scores and even different model preferences. We show that this instability can further propagate from routing labels to learned routing policies. To address this issue, we propose **DARS** (Distribution-Aware Routing Supervision), which estimates query-level model capability from repeated observations spanning semantics-preserving query rewrites and stochastic decoding. DARS summarizes expected quality, expected cost, and performance variability to construct risk-aware supervision without changing the downstream router architecture. Experiments across diverse tasks and routing methods show that DARS generally improves routing utility and cost–quality trade-offs over single-shot supervision. Further analyses show that its benefits persist under moderate sampling budgets and different decoding temperatures. These results suggest that reliable LLM routing should move beyond individual sampled outcomes and instead model query-level capability distributions.
Anthology ID:
2026.emnlp-main.333
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
7667–7688
Language:
URL:
https://aclanthology.org/2026.emnlp-main.333/
DOI:
Bibkey:
Cite (ACL):
Guannan Lai, Haoran Hu, Long Chen, Zhenguo Li, and Han-Jia Ye. 2026. From Sampled Outcomes to Capability Distributions: Rethinking Supervision for LLM Routing. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 7667–7688, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
From Sampled Outcomes to Capability Distributions: Rethinking Supervision for LLM Routing (Lai et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.333.pdf