Algorithm for Contextual Queueing Bandits with Rate-Optimal Queue Length Regret
Not on the map. Why?
Abstract (arXiv)
Contextual queueing bandits provide a framework for learning to schedule heterogeneous jobs under unknown context-dependent service rates. Under stochastic contexts, existing algorithms achieve widetildeO(T-1/4) queue length regret, defined as the expected difference between the learner's and oracle's queue lengths at horizon T. In this paper, we improve this rate to widetildeO(T-1/2). The key observation is that random exploration is needed only up to a carefully chosen cutoff round, rather than throughout the entire horizon. We propose CQB-η-2, a three-phase algorithm: (i) pure random exploration to construct an initial estimator, (ii) η-random exploration combined with a UCB rule to continue learning while maintaining negative drift, and (iii) pure UCB after the exploration cutoff. Our proof decomposes the queue length regret at the cutoff round. Before the cutoff, negative drift suppresses queue length differences caused by suboptimal choices. After the cutoff, the first two phases provide sufficient random exploration samples, ensuring that UCB decisions incur small departure-rate gaps. Combining these two bounds yields queue length regret of order widetildeO(T-1/2). We further prove a minimax lower bound of order Omega(T-1/2). The proof constructs two hard instances that are statistically indistinguishable up to the final service decision, and uses a queue-specific coupling argument to convert the resulting testing error into queue length regret. Together, our upper and lower bounds characterize the minimax dependence on the horizon T up to logarithmic factors.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 29 of its 48 references that are in the map. First posted on arXiv 8 Jun 2026.
Builds on
29 of 48 references.
References · 29
- 2024RouteLLM: Learning to Route LLMs with Preference Datacited by 633
- 2020Improved Optimistic Algorithms for Logistic Banditscited by 117
- 2024RouterDC: Query-Based Router by Dual Contrastive Learning for Assembling Large Language Modelscited by 84
- 2024Efficient LLM Scheduling by Learning to Rankcited by 83
- 2025MixLLM: Dynamic Routing in Mixed Large Language Modelscited by 56
- 2017MNL-Bandit: A Dynamic Learning Approach to Assortment Selectioncited by 50
- 2024EmbedLLM: Learning Compact Representations of Large Language Modelscited by 46
- 2020Improved Confidence Bounds for the Linear Logistic Model and Applications to Linear Banditscited by 31
- 2024A Unified Confidence Sequence for Generalized Linear Models, with Applications to Banditscited by 27
- 2016Learning Unknown Service Rates in Queues: A Multi-Armed Bandit Approachcited by 18
- 2024A Queueing Theoretic Perspective on Low-Latency LLM Inference with Variable Token Lengthcited by 16
- 2025Generalized Linear Bandits: Almost Optimal Regret with One-Pass Updatecited by 16
17 more
- 2025Cost-Aware Contrastive Routing for LLMscited by 16
- 2025Improved Online Confidence Bounds for Multinomial Logistic Banditscited by 14
- 2021Job Dispatching Policies for Queueing Systems with Unknown Service Ratescited by 11
- 2022Learning While Scheduling in Multi-Server Systems with Unknown Statistics: MaxWeight with Discounted UCBcited by 8
- 2020Stability and Learning in Strategic Queuing Systemscited by 7
- 2021Decentralized Learning in Online Queuing Systemscited by 7
- 2024Efficient Reinforcement Learning for Routing Jobs in Heterogeneous Queueing Systemscited by 7
- 2024Queueing Matching Bandits with Preference Feedbackcited by 6
- 2025LLM Routing with Dueling Feedbackcited by 6
- 2025Neural Logistic Banditscited by 5
- 2025Optimal Scheduling Algorithms for LLM Inference: Theory and Practicecited by 5
- 2024Performance of NPG in Countable State-Space Average-Cost RLcited by 4
- 2024Design and Scheduling of an AI-based Queueing Systemcited by 2
- 2025Minimizing Queue Length Regret for Arbitrarily Varying Channelscited by 2
- 2026Queue Length Regret Bounds for Contextual Queueing Banditscited by 2
- 2026Learning to Route and Schedule LLMs from User Retrials via Contextual Queueing Banditscited by 2
- 2025Bandit-Based Rate Adaptation for a Single-Server Queuecited by 1
Built on by
Not cited yet.