English

ParetoBandit: Budget-Paced Adaptive Routing for Non-Stationary LLM Serving

Machine Learning 2026-04-15 v2 Computation and Language

Abstract

Multi-model LLM serving operates in a non-stationary, noisy environment: providers revise pricing, model quality can shift or regress without notice, and new models arrive regularly. More than a dozen recent methods have proposed learned routers to navigate the resulting quality--cost tradeoff across portfolios spanning a \sim530×\times cost range. Despite this activity, two gaps in the current solution space limit routing effectiveness under these conditions: no existing router enforces a dollar-denominated cost ceiling in closed loop over an open-ended request stream, and none provides principled online adaptation to post-deployment shifts in pricing or model quality. We present ParetoBandit, an open-source adaptive router built on cost-aware contextual bandits that addresses both gaps. Its core contributions are: (1) an online primal--dual budget pacer that enforces a per-request cost ceiling without a known horizon, and (2) geometric forgetting on sufficient statistics that gives the bandit bounded memory for tracking quality and cost shifts. A hot-swap model registry further supports runtime model changes with budget-controlled exploration. On 1,824 benchmark prompts with a three-model portfolio, the router maintains budget compliance within 0.4%, adapts to price and quality shifts with up to +0.071 quality lift, and integrates a cold-started model within \sim142 steps.

Keywords

Cite

@article{arxiv.2604.00136,
  title  = {ParetoBandit: Budget-Paced Adaptive Routing for Non-Stationary LLM Serving},
  author = {Annette Taberner-Miller},
  journal= {arXiv preprint arXiv:2604.00136},
  year   = {2026}
}

Comments

27 pages, 15 figures, 13 tables. Code available at https://github.com/ParetoBandit/ParetoBandit