paper-with-me

홈 › Papers

Algorithm for Contextual Queueing Bandits with Rate-Optimal Queue Length Regret

2026-06-08 · Seoungbin Bae, Dabeen Lee 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 $\widetilde{\mathcal{O}}(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 $\widetilde{\mathcal{O}}(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 $\widetilde{\mathcal{O}}(T^{-1/2})$. We further prove a minimax lower bound of order $Ω(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.

📄 PDF Abstract BibTeX arXiv:2606.09668

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Queue Length Regret Bounds for Contextual Queueing Bandits

2026-01-27 · Seoungbin Bae, Garyeong Kang, Dabeen Lee arxiv

We introduce contextual queueing bandits, a new context-aware framework for scheduling while simultaneously learning unknown service rates. Individual jobs carry heterogeneous contextual features, based on which the agen…

Learning to Route and Schedule LLMs from User Retrials via Contextual Queueing Bandits

2026-02-02 · Seoungbin Bae, Junyoung Son, Dabeen Lee arxiv

Explosive demands for LLMs often cause user queries to accumulate in server queues, requiring efficient routing (query-LLM matching) and scheduling (query prioritization) mechanisms. Several online algorithms are being d…

Contrastive Learning

Beyond UCB: Optimal and Efficient Contextual Bandits with Regression Oracles

2020-02-12 · ICML 2020 1 · Dylan J. Foster, Alexander Rakhlin

A fundamental challenge in contextual bandits is to develop flexible, general-purpose algorithms with computational requirements no worse than classical supervised learning tasks such as classification and regression. Al…

Multi-Armed Banditsregression

A conversion theorem and minimax optimality for continuum contextual bandits

2024-06-09 · Arya Akhavan, Karim Lounici, Massimiliano Pontil, Alexandre B. Tsybakov

We study the contextual continuum bandits problem, where the learner sequentially receives a side information vector and has to choose an action in a convex set, minimizing a function associated with the context. The goa…

Multi-Armed Bandits

Regret of Queueing Bandits

2016-12-01 · NeurIPS 2016 12 · Subhashini Krishnasamy, Rajat Sen, Ramesh Johari, Sanjay Shakkottai

We consider a variant of the multiarmed bandit problem where jobs queue for service, and service rates of different servers may be unknown. We study algorithms that minimize queue-regret: the (expected) difference betwe…