Learning Unknown Service Rates in Queues 2: Q-UCB Has Queue-Regret at Most 6K v_u(t) log² t / t + (24.004 + UK)/(6t²) in the Late StageResearch Paper
Motivation
Many service systems must match jobs to servers whose quality is not known in advance: crowdsourcing platforms routing tasks to worker types, wireless schedulers assigning users to channels, and switches in data networks. The scheduler learns server quality only by trying servers, and every poor assignment lets work pile up. The natural performance measure is therefore not the cumulative reward loss of classical bandits but the queue-regret: the expected excess backlog compared with a genie that knows the best assignment.
Krishnasamy, Sen, Johari and Shakkottai, Learning Unknown Service Rates in Queues: A Multiarmed Bandit Approach (Operations Research, 2021; arXiv:1604.06377), show that queue-regret has two regimes. In an early stage it grows like the regret of a bandit. In a late stage, once the queue regularly empties, past mistakes are forgotten at every empty instant and the queue-regret decays. This mission formalizes their upper bound for the late stage: a UCB scheduler with forced exploration, Q-UCB, has queue-regret of order .
Setting
A discrete-time switch has queues and servers. In every slot , queue receives an arrival and link offers a potential service , all independent and i.i.d. over slots. The scheduler picks a matching , an injective assignment of a server to every queue, and queue evolves as
The scheduler observes only past schedules and the services offered on scheduled links.
Write , for its maximizer, , and . Assumption 1: each queue has a unique best server and the best servers are distinct, so is the unique optimal matching. Assumption 2: for all . The genie queue always uses . Assumption 3: the system starts from the stationary law of . The queue-regret is
Q-UCB (Algorithm 1) explores in slot with probability , scheduling a uniformly random member of a fixed family of matchings that together cover all links. Otherwise it exploits: for each queue it picks the server with the largest index , with the empirical mean and the number of past assignments of link , and schedules a matching closest in Hamming distance to these choices. Fix and set
Formalization targets
Goal: Theorem 5 for Q-UCB (p. 16)
For every with , and ,
The constants are the paper's, stated exactly.
Milestones (§8.2, pp. 24–34)
- Lemma 10: the number of forced-exploration slots in a window exceeds with probability at most . Over it exceeds with probability at most .
- Lemma 11: with probability at least , no exploit slot in schedules a sub-optimal link (event ).
- Lemma 12: pathwise, is at most the number of sub-optimal schedules since queue was last empty, at time .
- Lemmas 13–17: on , a coarse queue bound (13), a first bound on the regenerative cycle (14), a maximal inequality for the arrival-minus-service walk (15), a refined polylogarithmic queue bound (16), and the refined cycle bound (17), each failing with probability .
Significance
The theorem shows that a simple learning scheduler loses only in expected backlog in the late stage. Together with the paper's lower bound for consistent policies (Theorem 3), this places the optimal late-stage queue-regret within a polylogarithmic factor. The analysis method has two parts: a high-probability statement that the learner has locked onto the optimal matching, and a regenerative argument that converts it into a bound on queue length. It applies to other learning-in-queues problems in the same way.
All results are proved in the paper (§8.2). None of them has a machine-checked proof. The mission produces a formal model of the learning switch, including the algorithm with arbitrary tie-breaking, the coupled genie and the stationary start, and checks the explicit constants. These constants rely on numerical facts stated in the proofs, for example a positivity check for in Lemma 10(b) and the absorption of lower-order terms into , which a formal proof must verify.
Difficulty
The obvious argument says that UCB eventually identifies the best server, so the queue behaves like the genie's. This fails for two reasons. First, the regret is an expectation of a queue length, and a queue remembers every early mistake until it next empties. So one must bound the current regenerative cycle, and that bound itself depends on the queue length, which forces the recursion of Lemmas 13, 14, 16 and 17. Second, UCB's standard logarithmic-regret analysis bounds the expected number of sub-optimal pulls. It says nothing about whether a sub-optimal pull happens in a specific late window. Forced exploration at rate is what makes all indices concentrate at once, and Lemma 11 requires a union bound over every link, slot and sample count with explicit exponents.
Formalization scope
- Probability space. One canonical space: the initial queue vector together with an i.i.d. sequence of slot seeds, namely uniforms for arrivals, links and the explore coin, and a uniform explore index.
- Arrivals and services are threshold events of the uniforms.
- has the product geometric law with , the stationary law of the genie chain under the printed recursion.
- Indexing and coupling. Queues and servers are
Fin UandFin K, and slots are . The genie shares arrivals, services and with the system. This coupling is implicit in the paper's proofs and does not change . - Algorithm. Q-UCB is formalized for every covering family and every tie-breaking rule, deterministic in the slot and history, for the UCB argmax and the Hamming projection.
- counts slots .
- An unsampled link has index .
- when queue never emptied on .
- Disclosed additions.
- in every statement; is trivial, and the constants fail there.
- in Lemma 10(a).
- in Lemma 10(b); the printed statement is false without it, and its proof uses it.
- Lemmas 14 and 16 take the "arbitrary function" natural-valued.
- Excluded trivializations. Theorem 5 also asserts that is integrable, so cannot be a default value of a non-integrable integral. Every probability bound concerns the actual events, and every hypothesis set is satisfiable. For example, with , , , , , the range of Theorem 5 contains .
- Out of scope. Theorem 5 and Lemma 11 for Q-ThS (Algorithm 2) are not formalized, and neither is Corollary 6.
- Infrastructure needed and welcome. Chernoff bounds for sums of independent non-identically distributed Bernoulli variables (Lemma 10), Hoeffding's inequality and a maximal suffix-sum bound (Lemmas 14, 15), measurability of the algorithm's trajectory in the seeds, and regenerative-cycle arguments for the reflected random walk. The Chernoff and stationary-queue lemmas are reusable well beyond this mission.
The source is the arXiv v4 preprint; the paper's proofs are in its §8 and §9.
Selected references
- S. Krishnasamy, R. Sen, R. Johari, S. Shakkottai, Learning Unknown Service Rates in Queues: A Multiarmed Bandit Approach, Operations Research 69(1), 2021. Preprint arXiv:1604.06377v4. https://arxiv.org/abs/1604.06377 ; https://doi.org/10.1287/opre.2020.1995
- P. Auer, N. Cesa-Bianchi, P. Fischer, Finite-time Analysis of the Multiarmed Bandit Problem, Machine Learning 47, 2002. https://doi.org/10.1023/A:1013689704352
- M. J. Neely, Stochastic Network Optimization with Application to Communication and Queueing Systems, Morgan & Claypool, 2010. https://doi.org/10.2200/S00271ED1V01Y201006CNT007