Targeting the Exceptions: Online Active Learning for Retrieval-Based LLM Routers

Clovis Varangot-Reille, Antoine Gourru, Christophe Bouvard, Baptiste Jeudy


Abstract
Large language Models (LLM) routing reduces inference cost by selecting the most cost-effective model per query, but production deployment requires a router that adapts to an evolving model pool while adding negligible per-query overhead. Unsupervised retrieval-based routers satisfy both, but their support corpus must be built online from streaming traffic, and labeling each query requires running every candidate plus an LLM judge: prohibitively expensive at scale. We frame online corpus construction as stream-based active learning, exploiting a structural property of routing benchmarks: one or two models are globally near-optimal on the majority of queries, so the labeling budget should concentrate on the rare exception regions where selecting a specialist matters. We formalize this with a performance sparsity score, defined as the Kullback-Leibler divergence of a query’s per-candidate performance vector from uniform: high when performance concentrates on a few models and low when all candidates are equivalent. Since this score depends on performances unavailable at inference time, we predict it online from query embeddings with a lightweight regressor. Across four benchmarks, our method SparseNN approaches full-annotation routing performance with up to 44% fewer annotations in the stationary setting and 73% under topic shift. The results transfer to a fully held-out benchmark. A partial model annotation variant achieves higher cost savings, especially on large pools. Finally, both the corpus and the online estimator prove robust to model-pool changes.
Anthology ID:
2026.findings-emnlp.2428
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
52117–52135
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.2428/
DOI:
10.18653/2026.findings-emnlp.2428
Bibkey:
Cite (ACL):
Clovis Varangot-Reille, Antoine Gourru, Christophe Bouvard, and Baptiste Jeudy. 2026. Targeting the Exceptions: Online Active Learning for Retrieval-Based LLM Routers. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 52117–52135, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Targeting the Exceptions: Online Active Learning for Retrieval-Based LLM Routers (Varangot-Reille et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.2428.pdf