Prove2Me
Navigate
DiscoverCollectionsFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

Bandit Algorithms

125 missions · 55 completed

Missions

Open70Completed55All125
🏆Completed
Machine LearningOperations ResearchProbability·Captain: mikedeng1

Analysis of Thompson Sampling for the Multi-armed Bandit Problem 2: Logarithmic Regret for N ArmsResearch Paper

Motivation

Thompson Sampling is the oldest heuristic for the multi-armed bandit problem: proposed by Thompson in 1933, it plays each arm with the posterior probability that the arm is the best one. It is simple to implement, performs well empirically (Chapelle and Li, NIPS 2011), and has been used in production systems such as click-through-rate prediction for search advertising. For a long time, however, no finite-time regret guarantee was known for it: the analyses available before 2012 gave only o(T)o(T)o(T) regret.

Agrawal and Goyal (arXiv:1111.1797, COLT 2012) gave the first logarithmic bounds on the expected regret of Thompson Sampling. This mission formalizes their bound for the general case of NNN arms (their Theorem 2). A companion mission of the same series formalizes their two-armed bound (Theorem 1), whose proof is independent.

Timeline. Lai and Robbins (1985) proved that every consistent algorithm has regret at least of order ∑iΔiD(μi∥μ1)ln⁡T\sum_i \frac{\Delta_i}{D(\mu_i\|\mu_1)}\ln T∑i​D(μi​∥μ1​)Δi​​lnT. Auer, Cesa-Bianchi and Fischer (2002) showed that UCB1 achieves O(∑iln⁡T/Δi)O(\sum_i \ln T/\Delta_i)O(∑i​lnT/Δi​) in finite time. Agrawal and Goyal (2012) proved O((∑a1/Δa2)2ln⁡T)O((\sum_a 1/\Delta_a^2)^2\ln T)O((∑a​1/Δa2​)2lnT) for Thompson Sampling with NNN arms; Kaufmann, Korda and Munos (2012) and Agrawal and Goyal (2013) later proved the asymptotically optimal constant for Bernoulli rewards.

Setting

A stochastic NNN-armed bandit has arms 1,…,N1,\dots,N1,…,N. Arm iii, when played, yields a random reward drawn from a fixed distribution νi\nu_iνi​ supported in [0,1][0,1][0,1], with mean μi\mu_iμi​; rewards of an arm are i.i.d. and independent of the other arms. Arm 111 is assumed to be the unique optimal arm, μ1>μi\mu_1>\mu_iμ1​>μi​ for i≠1i\ne1i=1, and Δi=μ1−μi>0\Delta_i=\mu_1-\mu_i>0Δi​=μ1​−μi​>0 is the gap of arm iii.

Thompson Sampling for general stochastic bandits (Algorithm 2 of the paper) keeps, for each arm iii, a count SiS_iSi​ of successes and FiF_iFi​ of failures, both starting at 000. In each round ttt it draws θi(t)∼Beta(Si+1,Fi+1)\theta_i(t)\sim\mathrm{Beta}(S_i+1,F_i+1)θi​(t)∼Beta(Si​+1,Fi​+1) independently for every arm, plays i(t)=arg⁡max⁡iθi(t)i(t)=\arg\max_i\theta_i(t)i(t)=argmaxi​θi​(t), observes a reward r~t∼νi(t)\tilde r_t\sim\nu_{i(t)}r~t​∼νi(t)​, performs a Bernoulli trial with success probability r~t\tilde r_tr~t​, and increments Si(t)S_{i(t)}Si(t)​ on success and Fi(t)F_{i(t)}Fi(t)​ on failure.

The expected regret in time TTT is

E[R(T)]=E[∑t=1T(μ∗−μi(t))],μ∗=max⁡iμi,\mathbb E[\mathcal R(T)]=\mathbb E\Big[\sum_{t=1}^T(\mu^*-\mu_{i(t)})\Big],\qquad \mu^*=\max_i\mu_i,E[R(T)]=E[t=1∑T​(μ∗−μi(t)​)],μ∗=imax​μi​,

the expectation being over the rewards, the Bernoulli trials and the posterior samples.

The proof works with the reward stacks Zi,mZ_{i,m}Zi,m​: the outcome of the mmm-th Bernoulli trial of arm iii, all independent. Then s(j)=∑m≤jZ1,ms(j)=\sum_{m\le j}Z_{1,m}s(j)=∑m≤j​Z1,m​, the number of successes in the first jjj plays of arm 111, is a Binomial(j,μ1)\mathrm{Binomial}(j,\mu_1)Binomial(j,μ1​) random variable. The other objects of the proof are the threshold Li=24ln⁡T/Δi2L_i=24\ln T/\Delta_i^2Li​=24lnT/Δi2​, the saturated set C(t)C(t)C(t) of suboptimal arms with at least LiL_iLi​ plays before round ttt, the intervals IjI_jIj​ between the jjj-th and (j+1)(j+1)(j+1)-th plays of arm 111, and the counts γj\gamma_jγj​ and Vjℓ,aV_j^{\ell,a}Vjℓ,a​ defined in §4.

Formalization targets

Goal: Theorem 2

There is an absolute constant C>0C>0C>0 such that for every N≥2N\ge2N≥2, every instance as above and every horizon T≥2T\ge2T≥2,

E[R(T)]≤C(∑a=2N1Δa2)2ln⁡T.\mathbb E[\mathcal R(T)]\le C\Big(\sum_{a=2}^N\frac{1}{\Delta_a^2}\Big)^2\ln T .E[R(T)]≤C(a=2∑N​Δa2​1​)2lnT.

CCC does not depend on NNN, on the reward distributions or on TTT.

Milestones

  1. Lemma 4: with E(t)E(t)E(t) the event that every saturated arm's sample lies within Δi/2\Delta_i/2Δi​/2 of its mean, Pr⁡(E(t))≥1−4(N−1)/T2\Pr(E(t))\ge1-4(N-1)/T^2Pr(E(t))≥1−4(N−1)/T2, also conditionally on s(j)=ss(j)=ss(j)=s.
  2. Lemma 5 (Eq. (7)): the expected regret from saturated arms inside IjI_jIj​ is at most E[E[γj+1∣s(j)]∑aΔaE[min⁡{X(j,s(j),μa+Δa/2),T}∣s(j)]]\mathbb E\big[\mathbb E[\gamma_j+1\mid s(j)]\sum_a\Delta_a\mathbb E[\min\{X(j,s(j),\mu_a+\Delta_a/2),T\}\mid s(j)]\big]E[E[γj​+1∣s(j)]∑a​Δa​E[min{X(j,s(j),μa​+Δa​/2),T}∣s(j)]].
  3. Lemma 1: E[X(j,s,y)]=1/Fj+1,yB(s)−1\mathbb E[X(j,s,y)]=1/F^B_{j+1,y}(s)-1E[X(j,s,y)]=1/Fj+1,yB​(s)−1, where X(j,s,y)X(j,s,y)X(j,s,y) counts the trials before an independent Beta(s+1,j−s+1)\mathrm{Beta}(s+1,j-s+1)Beta(s+1,j−s+1) sample exceeds yyy.
  4. Lemma 3: a three-case bound on E[E[min⁡{X(j,s(j),y),T}∣s(j)]]\mathbb E[\mathbb E[\min\{X(j,s(j),y),T\}\mid s(j)]]E[E[min{X(j,s(j),y),T}∣s(j)]] in terms of the Bernoulli KL divergence DDD between yyy and μ1\mu_1μ1​.

Significance

The result. Theorem 2 shows that Thompson Sampling, a randomized Bayesian heuristic, achieves regret logarithmic in the horizon for any number of arms with bounded rewards, matching the order in TTT of the Lai–Robbins lower bound. Its dependence on the gaps, (∑aΔa−2)2(\sum_a\Delta_a^{-2})^2(∑a​Δa−2​)2, is worse than UCB1's; the paper's own Remark 1 and later work improve it. The proof introduced the device of bounding the waiting time between plays of the optimal arm through geometric variables with Beta-cdf parameters (Lemmas 1 and 3), which reappears in later analyses of Thompson Sampling.

Formalizing it. The theorem is proved on paper; it has not been machine-checked. Bandit theory in Lean (bandit environments, regret, UCB-type analyses) is still young, and no Beta–Bernoulli Thompson Sampling result is formalized. The mission produces a Lean model of Algorithm 2 for general [0,1][0,1][0,1] rewards with the paper's stack coupling, the §4 bookkeeping of saturated arms and intervals, and the paper's lemmas as separate targets.

Difficulty

Two difficulties are specific to the NNN-armed analysis. First, the arm that competes with arm 111 changes over time: the set of saturated arms grows, and which saturated arm is "best" depends on the history, so the waiting time between plays of arm 111 cannot be compared with a single geometric variable as in the two-armed case. Second, the number γj\gamma_jγj​ of rounds at which arm 111's sample is large but arm 111 is not played is not independent of the counts Vjℓ,aV_j^{\ell,a}Vjℓ,a​: both depend on the same posterior samples, and Lemma 5 needs a careful conditioning on the history to separate them. The obvious union bound over arms, treating each suboptimal arm as in the two-armed proof, fails because it ignores the interruptions by unsaturated arms, whose number is the source of the squared sum in the bound.

Formalization scope

  • Probability space. Algorithm 2 is realized on a product of three independent i.i.d. tables: Beta draws indexed by (arm, round, successes, failures), rewards indexed by (arm, round) and uniform variables indexed by (arm, round); the Bernoulli trial of a round succeeds when the played arm's uniform variable is below its reward. The law of the run is that of Algorithm 2, which runs for every round t=1,2,…t=1,2,\dotst=1,2,…. s(j)s(j)s(j) is the number of successful trials among the first jjj plays of arm 111 in this infinite run (possibly after the horizon TTT), so it is a Binomial(j,μ1)\mathrm{Binomial}(j,\mu_1)Binomial(j,μ1​) random variable for every jjj, as the paper's independent Z1,mZ_{1,m}Z1,m​ make it. Ties in the arg max (probability 000) go to the smallest index.
  • Indexing. Arms are Fin N, and Lean arm 0 is the paper's arm 111. Rounds are 0,…,T−10,\dots,T-10,…,T−1; Lean round ttt is the paper's round t+1t+1t+1. Sums over a=2,…,Na=2,\dots,Na=2,…,N are sums over a≠0a\ne0a=0.
  • Expectations are lower Lebesgue integrals in [0,∞][0,\infty][0,∞], which has no junk value for non-integrable functions. Conditional expectations given s(j)s(j)s(j) are written as finite sums over the values of s(j)s(j)s(j).
  • The O(⋅)O(\cdot)O(⋅). The paper writes O(⋅)O(\cdot)O(⋅) in the sense of its footnote 1 (f(n)≤c g(n)f(n)\le c\,g(n)f(n)≤cg(n) for n≥n0n\ge n_0n≥n0​). The goal states it with one universal constant CCC, quantified before NNN, the instance and TTT, for every T≥2T\ge2T≥2. The explicit constants printed in App. D are not formalized: expanding the paper's Eq. (21) gives terms 288(N−1)(ln⁡T)∑aΔa−2288(N-1)(\ln T)\sum_a\Delta_a^{-2}288(N−1)(lnT)∑a​Δa−2​ and 48(N−1)248(N-1)^248(N−1)2 where the paper prints 288(ln⁡T)∑iΔi−2288(\ln T)\sum_i\Delta_i^{-2}288(lnT)∑i​Δi−2​, and Eq. (22) drops a factor ln⁡T\ln TlnT in its 192/Δa2192/\Delta_a^2192/Δa2​ term. The O(⋅)O(\cdot)O(⋅) claim does not depend on these slips; a statement pinned to the printed numerals might be false.
  • Ruled out. A constant depending on NNN, on the means or on TTT; a fixed number of arms; Bernoulli rewards only; or any algorithm other than Algorithm 2 would each make the goal a different and weaker theorem. The statement quantifies over all N≥2N\ge2N≥2 and all reward distributions on [0,1][0,1][0,1].
  • Not included. Eq. (8), the bound ∑jE[γj∣s(j)]≤∑uLu+4(N−1)\sum_{j}\mathbb E[\gamma_j\mid s(j)]\le\sum_uL_u+4(N-1)∑j​E[γj​∣s(j)]≤∑u​Lu​+4(N−1) "for all instantiations", is not a milestone: each term is conditioned on a different s(j)s(j)s(j), and the pointwise reading does not follow from the argument given. Remark 1 (an alternate bound) and App. A (several optimal arms) are not part of this mission.
  • Contributions welcome: Beta–Binomial identities, geometric waiting times, Hoeffding bounds for binomial cdfs, and the stopping-time arguments behind Lemma 5. Lemma 1 and Lemma 3 are shared with the two-armed mission of this series.

Selected references

  • S. Agrawal and N. Goyal, Analysis of Thompson Sampling for the Multi-armed Bandit Problem, COLT 2012; arXiv:1111.1797v3, 2012. https://arxiv.org/abs/1111.1797
  • W. R. Thompson, On the likelihood that one unknown probability exceeds another in view of the evidence of two samples, Biometrika 25, 1933. https://doi.org/10.1093/biomet/25.3-4.285
  • T. L. Lai and H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6, 1985. https://doi.org/10.1016/0196-8858(85)90002-8
  • P. Auer, N. Cesa-Bianchi and P. Fischer, Finite-time analysis of the multiarmed bandit problem, Machine Learning 47, 2002. https://doi.org/10.1023/A:1013689704352
  • O. Chapelle and L. Li, An empirical evaluation of Thompson Sampling, NIPS 2011. https://papers.nips.cc/paper/4321-an-empirical-evaluation-of-thompson-sampling
  • E. Kaufmann, N. Korda and R. Munos, Thompson Sampling: an asymptotically optimal finite-time analysis, ALT 2012. https://arxiv.org/abs/1205.4217
  • S. Agrawal and N. Goyal, Further optimal regret bounds for Thompson Sampling, AISTATS 2013. https://arxiv.org/abs/1209.3353
11 thms3 active usersReviewed
🏆Completed
Machine LearningOperations ResearchProbability·Captain: mikedeng1

Online Network Revenue Management Using Thompson Sampling: Bayesian Regret of TS-fixedResearch Paper

Motivation

A retailer who sells several products from shared, non-replenishable inventory over a finite season must set prices without knowing how demand responds to them. Every price posted is both a sale and an experiment. This is the network revenue management problem with demand learning, and it sits between two literatures: dynamic pricing with inventory, where demand is known and the fluid linear program of Gallego and van Ryzin (1997) is the standard benchmark, and multi-armed bandits, where learning is the whole problem but there are no resource constraints.

Ferreira, Simchi-Levi and Wang (Oper. Res. 2018) combine Thompson sampling with a linear-programming step: sample a demand model from the posterior, solve the fluid LP for that model, and randomize prices according to its solution. The same paper extends the scheme to continuous price sets, contextual pricing and bandits with knapsacks.

Timeline of the relevant results:

  • 1997: Gallego and van Ryzin introduce the fluid LP upper bound for network revenue management with known demand.
  • 2012: Besbes and Zeevi give a non-Bayesian network pricing algorithm with worst-case regret O(K5/3T2/3log⁡T)O(K^{5/3}T^{2/3}\sqrt{\log T})O(K5/3T2/3logT​).
  • 2013: Badanidiyuru, Kleinberg and Slivkins (bandits with knapsacks) give worst-case regret O(KTlog⁡T)O(\sqrt{KT\log T})O(KTlogT​).
  • 2013–2014: Bubeck and Liu and Russo and Van Roy give prior-free Bayesian regret bounds for Thompson sampling in unconstrained bandits.
  • 2018: Ferreira, Simchi-Levi and Wang prove the O(TKlog⁡K)O(\sqrt{TK\log K})O(TKlogK​) Bayesian regret bound for TS-fixed (Theorem 1), the target of this mission.

Setting

There are NNN products and MMM resources. One unit of product iii consumes aij≥0a_{ij}\ge0aij​≥0 units of resource jjj, and resource jjj starts with inventory Ij≥0I_j\ge0Ij​≥0 that is never replenished. The season has TTT periods. In each period the retailer posts one of KKK price vectors pk=(p1k,…,pNk)p_k=(p_{1k},\dots,p_{Nk})pk​=(p1k​,…,pNk​) or a shut-off price p∞p_\inftyp∞​ under which demand is zero.

Given the posted price pkp_kpk​, the demand vector D(t)∈R+ND(t)\in\mathbb R^N_+D(t)∈R+N​ has law F(⋅ ;pk,θ)F(\cdot\,;p_k,\theta)F(⋅;pk​,θ), where θ∈Θ\theta\in\Thetaθ∈Θ is unknown and drawn from a known, arbitrary prior μ0\mu_0μ0​. Demand is independent of the past given the posted price and θ\thetaθ, and is bounded: Di(t)∈[0,dˉi]D_i(t)\in[0,\bar d_i]Di​(t)∈[0,dˉi​]. Write dik(ρ)d_{ik}(\rho)dik​(ρ) for the mean demand of product iii under pkp_kpk​ and parameter ρ\rhoρ, and d=d(θ)d=d(\theta)d=d(θ).

When inventory covers all demand, all demand is sold. Otherwise the satisfied demand D~(t)\tilde D(t)D~(t) satisfies 0≤D~i(t)≤Di(t)0\le\tilde D_i(t)\le D_i(t)0≤D~i​(t)≤Di​(t), leaves every inventory nonnegative, and leaves at least one resource at zero; no other rule is imposed. Revenue is Rev(T)=∑t∑iD~i(t)Pi(t)\mathrm{Rev}(T)=\sum_t\sum_i\tilde D_i(t)P_i(t)Rev(T)=∑t​∑i​D~i​(t)Pi​(t).

For a mean-demand matrix ddd and capacities cj=Ij/Tc_j=I_j/Tcj​=Ij​/T, the linear program LP(d)\mathrm{LP}(d)LP(d) is

max⁡x≥0 ∑k=1K(∑i=1Npikdik)xks.t.∑k=1K(∑i=1Naijdik)xk≤cj  ∀j,∑k=1Kxk≤1,\max_{x\ge0}\ \sum_{k=1}^K\Bigl(\sum_{i=1}^N p_{ik}d_{ik}\Bigr)x_k\quad\text{s.t.}\quad\sum_{k=1}^K\Bigl(\sum_{i=1}^N a_{ij}d_{ik}\Bigr)x_k\le c_j\ \ \forall j,\qquad\sum_{k=1}^K x_k\le1,x≥0max​ k=1∑K​(i=1∑N​pik​dik​)xk​s.t.k=1∑K​(i=1∑N​aij​dik​)xk​≤cj​  ∀j,k=1∑K​xk​≤1,

with optimal value OPT(d)\mathrm{OPT}(d)OPT(d).

TS-fixed (Algorithm 1): in each period, sample θ(t)\theta(t)θ(t) from the posterior of θ\thetaθ given the history of posted prices and observed demands; let x(t)x(t)x(t) be an optimal solution of LP(d(θ(t)))\mathrm{LP}(d(\theta(t)))LP(d(θ(t))); post pkp_kpk​ with probability xk(t)x_k(t)xk​(t) and p∞p_\inftyp∞​ with the remaining probability; observe demand and update the posterior.

Finally pmax⁡=max⁡k∑ipikdˉip_{\max}=\max_k\sum_ip_{ik}\bar d_ipmax​=maxk​∑i​pik​dˉi​ and pmax⁡j=max⁡i:aij≠0, kpik/aijp^j_{\max}=\max_{i:a_{ij}\neq0,\,k}p_{ik}/a_{ij}pmaxj​=maxi:aij​=0,k​pik​/aij​.

Formalization targets

Goal: Theorem 1 against the LP benchmark

For K≥2K\ge2K≥2, T≥1T\ge1T≥1, every prior, every bounded demand family, every admissible fulfilment rule and every run of TS-fixed,

E[OPT(d)]⋅T−E[Rev(T)] ≤ (18 pmax⁡+37∑i=1N∑j=1Mpmax⁡jaijdˉi)TKlog⁡K.\mathbb E\bigl[\mathrm{OPT}(d)\bigr]\cdot T-\mathbb E\bigl[\mathrm{Rev}(T)\bigr]\ \le\ \Bigl(18\,p_{\max}+37\sum_{i=1}^N\sum_{j=1}^M p^j_{\max}a_{ij}\bar d_i\Bigr)\sqrt{TK\log K}.E[OPT(d)]⋅T−E[Rev(T)] ≤ (18pmax​+37i=1∑N​j=1∑M​pmaxj​aij​dˉi​)TKlogK​.

The paper prints this bound for BayesRegret(T)=E[Rev∗(T)]−E[Rev(T)]\mathrm{BayesRegret}(T)=\mathbb E[\mathrm{Rev}^*(T)]-\mathbb E[\mathrm{Rev}(T)]BayesRegret(T)=E[Rev∗(T)]−E[Rev(T)], where Rev∗\mathrm{Rev}^*Rev∗ is the revenue of the optimal policy that knows θ\thetaθ; see Formalization scope for why the LP benchmark is stated instead.

Milestones

The article states Theorem 1 and says that its proof is in the online appendix (Supplemental Material at the DOI). The article itself contains no numbered lemma. The milestone list is therefore empty; the appendix's lemmas will be added as milestones once the appendix is held.

Significance

The bound is prior-free and has explicit constants that depend only on prices, consumption rates and demand bounds. Its dependence on TTT matches the Ω(KT)\Omega(\sqrt{KT})Ω(KT​) lower bound for Bayesian regret in unconstrained bandits with rewards in [0,1][0,1][0,1], a special case of the model with no inventory constraints (Bubeck and Cesa-Bianchi 2012, Theorem 3.5). It shows that the posterior-sampling principle survives the addition of resource constraints, lost sales and randomized LP-based pricing, and it is the template for the paper's later results (TS-update, contextual pricing, bandits with knapsacks).

The theorem is proved on paper but, as far as a platform search shows, not formalized anywhere. The platform has a formal proof of the unconstrained Bayesian Thompson sampling bound knlog⁡k/2\sqrt{kn\log k/2}knlogk/2​ (BanditAlgorithm.thompson_sampling_bayesian_regret, Lattimore–Szepesvári Theorem 36.5) and an open single-product deterministic upper bound in revenue management (RevenueManagement.deterministic_upper_bound). Neither has inventory, an LP subroutine, or lost sales. A formal proof here would supply the first machine-checked analysis of Thompson sampling under resource constraints and would check the paper's constants.

Difficulty

In an unconstrained bandit, Thompson sampling's regret reduces to a sum of per-period gaps between an upper confidence bound and the sampled reward, because the sampled optimal arm and the true optimal arm are identically distributed given the history. Here the action is a randomized mixture x(t)x(t)x(t) from an LP, the reward is not additive in the prices chosen, and revenue is lost when inventory runs out. Two quantities must be controlled: the revenue the algorithm would collect if all demand could be served, and the revenue lost to stock-outs. The second depends on the random time at which each resource is exhausted under a pricing rule that was optimized for a sampled, not the true, demand, and on an arbitrary fulfilment rule once some resource is empty. Standard bandit arguments do not bound such lost sales, which are a nonlinear function of the whole trajectory.

Formalization scope

Lean representation. Products, resources and price vectors are indexed by Fin N, Fin M, Fin K; the posted price is an Option (Fin K) with none the shut-off price. Periods are 0-based (t=0,…,T−1t=0,\dots,T-1t=0,…,T−1 stands for the paper's 1,…,T1,\dots,T1,…,T). Θ\ThetaΘ is a standard Borel space with a probability measure μ0\mu_0μ0​; demand is a Markov kernel FFF from Θ×\Theta\timesΘ×Fin K to RN\mathbb R^NRN, bounded in [0,dˉi][0,\bar d_i][0,dˉi​] for every parameter. A run of TS-fixed is a family of random variables on a probability space satisfying, almost surely and via conditional expectations: θ∼μ0\theta\sim\mu_0θ∼μ0​; the posterior-sampling property of θ(t)\theta(t)θ(t) given everything before period ttt; the price draw with probabilities x(θ(t))x(\theta(t))x(θ(t)) for a measurable optimal LP selection xxx; the demand law given the past, θ(t)\theta(t)θ(t) and the posted price; and fulfilment rules (a)/(b). The logarithm is natural. Prices, consumption and inventory are nonnegative (implicit in the paper). OPT(d)\mathrm{OPT}(d)OPT(d) is a supremum over a nonempty bounded feasible set, so it has no junk value.

Corrections to the printed statement.

  1. K≥2K\ge2K≥2 is added. At K=1K=1K=1 the printed right-hand side is 000, yet on a one-price instance with Bernoulli(0.8)(0.8)(0.8) demand, I=T/2I=T/2I=T/2 and a point-mass prior, TS-fixed loses about 0.2pT0.2p\sqrt T0.2pT​ in expectation.
  2. The LP benchmark replaces E[Rev∗(T)]\mathbb E[\mathrm{Rev}^*(T)]E[Rev∗(T)]. Section 3.1.1 bounds E[Rev∗(T)∣d]\mathbb E[\mathrm{Rev}^*(T)\mid d]E[Rev∗(T)∣d] by OPT(d)⋅T\mathrm{OPT}(d)\cdot TOPT(d)⋅T, citing Gallego–van Ryzin. Under the paper's fulfilment rule this fails when products use disjoint resources: with two products, I=(T,1)I=(T,1)I=(T,1), p1=(1,0)p_1=(1,0)p1​=(1,0), p2=(1/2,0)p_2=(1/2,0)p2​=(1/2,0) and deterministic demand (1,1)(1,1)(1,1), the known-θ\thetaθ policy earns at least TTT while OPT(d)⋅T=1\mathrm{OPT}(d)\cdot T=1OPT(d)⋅T=1. The paper states that its proof bounds the gap to "the LP benchmark defined in Section 3.1.1" (p. 1594), and the last display of Section 3.1.1 bounds BayesRegret(T)\mathrm{BayesRegret}(T)BayesRegret(T) by exactly E[OPT(d)]⋅T−E[Rev(T)]\mathbb E[\mathrm{OPT}(d)]\cdot T-\mathbb E[\mathrm{Rev}(T)]E[OPT(d)]⋅T−E[Rev(T)]. Wherever the Gallego–van Ryzin bound holds, the corrected goal implies the printed one.

Ruled out. A bound for the "ideal" revenue ∑iDi(t)Pi(t)\sum_iD_i(t)P_i(t)∑i​Di​(t)Pi​(t) instead of the satisfied revenue, or for an arbitrary policy whose prices are merely close to the LP solution, is not Theorem 1; the goal carries the full TS-fixed run and the lost-sales accounting.

Infrastructure needed. Posterior-sampling identities for general (standard Borel) priors, a Hoeffding/Azuma-type concentration for bounded demand along the price-selection process, LP sensitivity with respect to the mean-demand matrix, and a pathwise bound on lost sales under an arbitrary fulfilment rule. The LP and fluid-benchmark definitions are reusable for later missions on TS-update (Theorem 2), contextual pricing (Theorem 4) and bandits with knapsacks (Theorem 5). Contributions welcome: proofs of the goal, and formal statements of the online appendix's lemmas.

Selected references

  • K. J. Ferreira, D. Simchi-Levi, H. Wang, Online Network Revenue Management Using Thompson Sampling, Operations Research 66(6):1586–1602, 2018. https://doi.org/10.1287/opre.2018.1755
  • G. Gallego, G. van Ryzin, A Multiproduct Dynamic Pricing Problem and Its Applications to Network Yield Management, Operations Research 45(1):24–41, 1997. https://doi.org/10.1287/opre.45.1.24
  • O. Besbes, A. Zeevi, Blind Network Revenue Management, Operations Research 60(6):1537–1550, 2012. https://doi.org/10.1287/opre.1120.1057
  • A. Badanidiyuru, R. Kleinberg, A. Slivkins, Bandits with Knapsacks, FOCS 2013. https://arxiv.org/abs/1305.2545
  • S. Bubeck, C.-Y. Liu, Prior-free and Prior-dependent Regret Bounds for Thompson Sampling, NeurIPS 2013. https://arxiv.org/abs/1311.0466
  • D. Russo, B. Van Roy, Learning to Optimize via Posterior Sampling, Mathematics of Operations Research 39(4):1221–1243, 2014. https://doi.org/10.1287/moor.2014.0650
  • S. Bubeck, N. Cesa-Bianchi, Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems, Foundations and Trends in Machine Learning 5(1), 2012. https://arxiv.org/abs/1204.5721
  • T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020, Chapter 36. https://doi.org/10.1017/9781108571401
31 thms3 active usersReviewed
🏆Completed
Operations ResearchOptimizationProbability+1·Captain: naimengye

Multi-armed Bandit Allocation Indices VI: Bandit Sampling Processes, Favourable Priors and Invariance of the IndexTextbook

Motivation

The bandit processes that motivated the index theorem are sampling processes: an arm is a population from which one draws i.i.d. observations whose distribution has an unknown parameter, and each draw both earns something and teaches something. Chapter 7 of Gittins, Glazebrook and Weber, Multi-armed Bandit Allocation Indices (2nd ed., doi:10.1002/9780470980033), develops the theory of such processes in the Bayesian setting: the state of the process is the current posterior for the parameter, continuing it samples the next value from the predictive distribution and moves to the new posterior. When the observations are themselves the rewards one has a reward process, the classical Bayesian multi-armed bandit; when the aim is to find as quickly as possible an individual whose measurement reaches a target TTT (a compound active enough to warrant further testing, in the drug-screening problem from which the index theorem came) one has a target process, which is a job that completes when the target is reached. Two questions organize the chapter. When can the index be written down without any optimization, and when do symmetries of the model reduce the index to a function of fewer variables? The first is answered by the notion of a favourable prior (Section 7.3): if no run of observations below the target can raise the current probability of success, then the index is that probability, exactly, by Proposition 2.7. The second is answered by the invariance theorems of Section 7.4: a location parameter with a conjugate prior gives ν(xˉ,n)=xˉ+ν(0,n)\nu(\bar x, n) = \bar x + \nu(0, n)ν(xˉ,n)=xˉ+ν(0,n), a scale parameter gives ν(xˉ,n)=xˉ ν(1,n)\nu(\bar x, n) = \bar x\,\nu(1, n)ν(xˉ,n)=xˉν(1,n), and for target processes the target can be absorbed into the state, ν(xˉ,n,T)=ν(xˉ−T,n,0)\nu(\bar x, n, T) = \nu(\bar x - T, n, 0)ν(xˉ,n,T)=ν(xˉ−T,n,0). These identities are what make the tables of Chapter 8 one-dimensional.

Setting

A sampling model consists of a likelihood f(⋅∣θ)f(\cdot \mid \theta)f(⋅∣θ), a family of priors π(⋅∣p)\pi(\cdot \mid p)π(⋅∣p) on the parameter indexed by the parameters ppp of a conjugate family, and the Bayes update p↦pxp \mapsto p_xp↦px​ of those parameters after observing xxx; the family is conjugate if the posterior of π(⋅∣p)\pi(\cdot \mid p)π(⋅∣p) given X=xX = xX=x is π(⋅∣px)\pi(\cdot \mid p_x)π(⋅∣px​). The predictive distribution is f(⋅∣p)=∫f(⋅∣θ)π(dθ∣p)f(\cdot \mid p) = \int f(\cdot \mid \theta)\pi(d\theta \mid p)f(⋅∣p)=∫f(⋅∣θ)π(dθ∣p). The reward process moves from ppp to pxp_xpx​ with x∼f(⋅∣p)x \sim f(\cdot \mid p)x∼f(⋅∣p) and earns r(p)=∫xf(x∣p)dxr(p) = \int x f(x \mid p)dxr(p)=∫xf(x∣p)dx. The target process with target TTT moves to the completion state CCC if x≥Tx \ge Tx≥T and to pxp_xpx​ otherwise, earning the current probability of success r(p)=f([T,∞)∣p)r(p) = f([T, \infty) \mid p)r(p)=f([T,∞)∣p), and 000 in CCC. A state ppp is favourable if r(px1⋯xm)≤r(p)r(p_{x_1 \cdots x_m}) \le r(p)r(px1​⋯xm​​)≤r(p) for every finite sequence of observations xi<Tx_i < Txi​<T. For the invariance theorems the parameters are (xˉ,n)(\bar x, n)(xˉ,n) with the update ((nxˉ+x)/(n+1),n+1)((n\bar x + x)/(n+1), n+1)((nxˉ+x)/(n+1),n+1); μ\muμ is a location parameter of the likelihood if f(⋅∣μ+c)f(\cdot \mid \mu + c)f(⋅∣μ+c) is f(⋅∣μ)f(\cdot \mid \mu)f(⋅∣μ) shifted by ccc, and xˉ\bar xxˉ is a location parameter of the prior family if π(⋅∣xˉ+c,n)\pi(\cdot \mid \bar x + c, n)π(⋅∣xˉ+c,n) is π(⋅∣xˉ,n)\pi(\cdot \mid \bar x, n)π(⋅∣xˉ,n) shifted by ccc; scale parameters are defined with x↦bxx \mapsto bxx↦bx, b>0b > 0b>0. The Gittins index is that of the Bandit Algorithms model on these chains.

Formalization targets

Goal: Theorem 7.9 (in the form of Corollary 7.10)

If μ\muμ is a location parameter of a reward process with a conjugate prior family in which xˉ\bar xxˉ is a location parameter and the parameters update as the sample mean and count, then for every n>0n > 0n>0

r(xˉ+c,n)=r(xˉ,n)+candν(xˉ,n)=xˉ+ν(0,n),r(\bar x + c, n) = r(\bar x, n) + c \quad\text{and}\quad \nu(\bar x, n) = \bar x + \nu(0, n),r(xˉ+c,n)=r(xˉ,n)+candν(xˉ,n)=xˉ+ν(0,n),

under the standing assumptions that the observations have a mean and the discounted rewards of the chain are integrable.

Milestones

Proposition 7.4 (favourable state: ν=r\nu = rν=r); Example 7.5 (Bernoulli target process, ν(α,β)=α/(α+β)\nu(\alpha, \beta) = \alpha/(\alpha + \beta)ν(α,β)=α/(α+β)); Example 7.6 (normal target process with known variance, ν(xˉ,n)=Φ(xˉ(1+n−1)−1/2)\nu(\bar x, n) = \Phi(\bar x (1 + n^{-1})^{-1/2})ν(xˉ,n)=Φ(xˉ(1+n−1)−1/2) for xˉ≥0\bar x \ge 0xˉ≥0); Theorem 7.11 (scale parameter: ν(xˉ,n)=xˉ ν(1,n)\nu(\bar x, n) = \bar x\,\nu(1, n)ν(xˉ,n)=xˉν(1,n)); Theorem 7.17 (target process with a location parameter: ν(xˉ,n,T)=ν(xˉ−T,n,0)\nu(\bar x, n, T) = \nu(\bar x - T, n, 0)ν(xˉ,n,T)=ν(xˉ−T,n,0)).

Significance

Theorem 7.9 and its companions are the reason the Gittins index of the normal reward process is tabulated as a function of nnn alone and that of the exponential process as a function of nnn and one ratio; every computational method of Chapter 8 starts by reducing the state space with them. Proposition 7.4 is the source of every closed-form index in the book: it identifies the states in which sampling for information is worthless, so that the index collapses to the immediate expected reward, and Examples 7.5 and 7.6 show that for the Bernoulli target process this is every state and for the normal target process every state with a nonnegative posterior mean. The formalization gives the platform its first Bayesian sampling-process model, in which the state is a posterior and conjugacy is stated through the posterior kernel of the likelihood, and its first index identities on unbounded-reward chains, which is where the integrability assumptions of the Bandit Algorithms model do real work.

None of this is machine-checked. The invariance theorems are stated in the proper-prior form of the corollaries, with the model's symmetry as hypotheses, so that they apply to any conjugate family with the stated structure rather than to a particular density.

Difficulty

The invariance theorems require showing that the chain of parameters from the shifted (scaled) state is the image of the chain from the original state under the shift (scaling) of trajectories, which is an equivariance of the Ionescu–Tulcea construction with respect to a measurable bijection commuting with the kernel; that stopping times are carried to stopping times; that the discounted reward of a stopping time shifts by ccc times the discounted time; and that the supremum of a nonempty bounded set of reals shifts and scales accordingly. Boundedness of the set of ratios is where the integrability assumption enters. Proposition 7.4 is the chain-level statement that all rewards along every trajectory from a favourable state are at most r(p)r(p)r(p), which needs an induction on the trajectory law of the target chain, followed by the argument of Proposition 2.7. Example 7.6 needs the monotonicity of xˉm(1+1/(n+m))−1/2\bar x_m (1 + 1/(n+m))^{-1/2}xˉm​(1+1/(n+m))−1/2 in the observations below the target, a small inequality, plus the Gaussian probability of a half-line as the current probability of success; Example 7.5 needs only that α/(α+β+m)\alpha/(\alpha + \beta + m)α/(α+β+m) decreases.

Formalization scope

The sampling model is a structure with Markov likelihood and prior kernels and a jointly measurable update; the predictive distribution is the kernel composition; conjugacy is an almost-everywhere identity between Mathlib's posterior of the likelihood with respect to the prior and the prior at the updated parameters, and is carried as a hypothesis of the invariance theorems and of Proposition 7.4 so that their subject is the Bayesian process. For the parameters (xˉ,n)(\bar x, n)(xˉ,n) it is required on n>0n > 0n>0 only (IsConjugateOn): a proper prior has n>0n > 0n>0, and conjugacy at every (xˉ,n)∈R2(\bar x, n) \in \mathbb{R}^2(xˉ,n)∈R2 is impossible with a location parameter, since at n=−1n = -1n=−1 the update divides by zero and sends every observation to one state, which made the first draft's location theorems vacuous. The chains are built with Kernel.map of product kernels, so their measurability is structural, and the target process lives on P ⊕ Unit with the completion state absorbing. The book's improper priors are replaced by proper conjugate families with the location or scale structure of Corollaries 7.10 and 7.12, as those corollaries do; the discrete-time correction factor of Section 2.8 is not applied since it cancels in every identity stated. The two examples are built directly from a uniform or Gaussian seed with the transition probabilities the book computes (the beta and normal posterior computations of Exercise 7.1 are not formalized). Hypotheses: a∈(0,1)a \in (0, 1)a∈(0,1); integrable observations and L&S Assumption 35.6 for the reward processes; n>0n > 0n>0 for the invariance theorems and xˉ>0\bar x > 0xˉ>0 for the scale theorem; α,β>0\alpha, \beta > 0α,β>0; xˉ≥0\bar x \ge 0xˉ≥0 and n>0n > 0n>0 for the normal example.

Trivializing readings are excluded: the indices are the genuine suprema of the Bandit Algorithms definition with integrable rewards, the update rule is the book's and not a free parameter, and the favourability condition ranges over all finite observation sequences. Welcome contributions: the equivariance of the trajectory measure under a state bijection commuting with the kernel, the transport of stopping times, and the reward bound along the target chain from a favourable state.

Selected references

  • J. Gittins, K. Glazebrook, R. Weber, Multi-armed Bandit Allocation Indices, 2nd ed., Wiley, 2011, Chapter 7. doi:10.1002/9780470980033
  • J. C. Gittins, D. M. Jones, A dynamic allocation index for the sequential design of experiments, in Progress in Statistics (J. Gani, ed.), North-Holland, 1974.
  • D. M. Jones, Search Procedures for Industrial Chemical Research, PhD thesis, University of Wales, 1975.
  • H. Raiffa, R. Schlaifer, Applied Statistical Decision Theory, Harvard University Press, 1961.
  • T. S. Ferguson, Mathematical Statistics: A Decision Theoretic Approach, Academic Press, 1967.
  • T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020, Chapters 34–35. doi:10.1017/9781108571401
9 thms3 active usersReviewed
🏆Completed
Dynamic ProgrammingOperations ResearchOptimization+1·Captain: naimengye

Multi-armed Bandit Allocation Indices V: Restless Bandits, Indexability and Whittle Indices for Monotone ModelsTextbook

Motivation

Every proof of the index theorem in Gittins, Glazebrook and Weber, Multi-armed Bandit Allocation Indices (2nd ed., doi:10.1002/9780470980033), uses the fact that a bandit not being processed is frozen. Chapter 6 drops that: Whittle's restless bandits evolve under the passive action too, by a different law, and mmm of nnn must be active at every time. The problem is PSPACE-hard in general, so Whittle proposed a heuristic built from a Lagrangian relaxation: replace the hard constraint by a subsidy WWW paid whenever a bandit is passive, solve the resulting single-bandit average-reward problem, and read off, for each state, the least subsidy W(x)W(x)W(x) at which the passive action becomes optimal. When the set of states where passivity is optimal grows monotonically with WWW, the bandit is indexable and W(x)W(x)W(x) is its Whittle index; the Whittle index policy activates the mmm bandits of largest index. It reduces to the Gittins index policy when the passive action freezes, it is asymptotically optimal as nnn grows under a fluid-stability condition (Weber and Weiss), and it has become the standard heuristic for sensor management, opportunistic channel access, maintenance and queueing control. The price is that indexability must be established model by model. Section 6.5 shows how easy this is when the single-bandit problem is solved by a monotone policy, on two bi-directional models: the spinning plates asset, which improves under investment and deteriorates when neglected, and the vigour bandit of Whittle's Ehrenfest project, which tires when worked and recovers when rested.

Setting

A restless bandit is a Markov decision process with two actions, active (u=1u = 1u=1) and passive (u=0u = 0u=0), each with its own transition kernel and reward. Under a deterministic stationary Markov policy ggg with passive subsidy WWW the reward in state xxx is r(x,g(x))+W(1−g(x))r(x, g(x)) + W(1 - g(x))r(x,g(x))+W(1−g(x)), and the average reward from xxx is the Cesàro limit of the expected rewards. The optimal average reward g(W)g(W)g(W) is the supremum over such policies and initial states; a policy is optimal if it attains g(W)g(W)g(W) from every initial state; E0(W)E_0(W)E0​(W) is the set of states in which some optimal policy is passive; the bandit is indexable if E0(W)E_0(W)E0​(W) is nondecreasing in WWW; and W(x)=inf⁡{W:x∈E0(W)}W(x) = \inf\{W : x \in E_0(W)\}W(x)=inf{W:x∈E0​(W)}.

The spinning plates asset lives on {1,…,k}\{1, \dots, k\}{1,…,k}: active moves x→x+1x \to x + 1x→x+1 at rate λ(x)\lambda(x)λ(x), passive moves x→x−1x \to x - 1x→x−1 at rate μ(x)\mu(x)μ(x), λ(k)=μ(1)=0\lambda(k) = \mu(1) = 0λ(k)=μ(1)=0, and r(x)r(x)r(x) is earned under both actions, rrr increasing. Uniformized so that rates are at most one, it is a discrete-time bandit whose kernels move with the rate's probability and otherwise stay. The monotone policy (y)(y)(y) is passive exactly on {x≥y}\{x \ge y\}{x≥y}; under it the asset alternates between y−1y - 1y−1 and yyy, spending the fraction ϕ(y)=λ(y−1)/(λ(y−1)+μ(y))\phi(y) = \lambda(y-1)/(\lambda(y-1) + \mu(y))ϕ(y)=λ(y−1)/(λ(y−1)+μ(y)) of its time at yyy, so its average reward is Wϕ(y)+R(y)W\phi(y) + R(y)Wϕ(y)+R(y) with R(y)=r(y)ϕ(y)+r(y−1)(1−ϕ(y))R(y) = r(y)\phi(y) + r(y-1)(1 - \phi(y))R(y)=r(y)ϕ(y)+r(y−1)(1−ϕ(y)), and W∗(x)=(R(x+1)−R(x))/(ϕ(x)−ϕ(x+1))W^*(x) = (R(x+1) - R(x))/(\phi(x) - \phi(x+1))W∗(x)=(R(x+1)−R(x))/(ϕ(x)−ϕ(x+1)). The vigour bandit is the mirror image: active moves down at rate ν(x)\nu(x)ν(x) and earns r(x)r(x)r(x), passive moves up at rate ρ(x)\rho(x)ρ(x) and earns nothing, ψ(y)=ν(y)/(ν(y)+ρ(y−1))\psi(y) = \nu(y)/(\nu(y) + \rho(y-1))ψ(y)=ν(y)/(ν(y)+ρ(y−1)), and W∗∗(x)=(r(x)(1−ψ(x))−r(x+1)(1−ψ(x+1)))/(ψ(x+1)−ψ(x))W^{**}(x) = (r(x)(1 - \psi(x)) - r(x+1)(1 - \psi(x+1)))/(\psi(x+1) - \psi(x))W∗∗(x)=(r(x)(1−ψ(x))−r(x+1)(1−ψ(x+1)))/(ψ(x+1)−ψ(x)).

Formalization targets

Goal: Theorem 6.4

For the spinning plates asset: (i) if ϕ\phiϕ is strictly decreasing over the thresholds 1≤y≤k+11 \le y \le k + 11≤y≤k+1, the asset is indexable; (ii) if additionally W∗W^*W∗ is strictly decreasing over the states, the Whittle index is

W(x)=W∗(x)=R(x+1)−R(x)ϕ(x)−ϕ(x+1),1≤x≤k.W(x) = W^*(x) = \frac{R(x+1) - R(x)}{\phi(x) - \phi(x+1)}, \qquad 1 \le x \le k.W(x)=W∗(x)=ϕ(x)−ϕ(x+1)R(x+1)−R(x)​,1≤x≤k.

Milestones

Eqs. (6.9)–(6.10): the monotone policy (y)(y)(y) earns Wϕ(y)+R(y)W\phi(y) + R(y)Wϕ(y)+R(y) from every initial state and g(W)=max⁡y[Wϕ(y)+R(y)]g(W) = \max_y [W\phi(y) + R(y)]g(W)=maxy​[Wϕ(y)+R(y)], because a monotone policy always achieves g(W)g(W)g(W); Theorem 6.5, the same two statements for the vigour bandit with ψ\psiψ increasing and W∗∗W^{**}W∗∗ increasing.

Significance

Theorem 6.4 is the chapter's template for proving indexability: the single-bandit value g(W)g(W)g(W) is the upper envelope of finitely many lines Wϕ(y)+R(y)W\phi(y) + R(y)Wϕ(y)+R(y) whose slopes decrease in the threshold, so the optimal threshold moves monotonically with the subsidy and the hinge points of the envelope are the indices. The same argument gives Theorem 6.5, the admission-control indices of Section 6.7, and the marginal productivity indices of Niño-Mora; it is the reason Whittle indices are computable in closed form for bi-directional models. Its formalization establishes, on the platform, the first restless-bandit model with a proved index, and the general notions of passive set, indexability and Whittle index that every later restless-bandit statement will use.

None of this is machine-checked. The average-reward optimality notion is stated without the DP equation (6.6), through optimality from every initial state, which is what the equation's solution encodes on a finite state space and avoids the relative value function altogether.

Difficulty

The proof in the book is two paragraphs, but it stands on the reduction to monotone policies, which is only sketched: every deterministic stationary policy, from every initial state, drives the asset into an absorbing endpoint or a two-state cycle {z−1,z}\{z - 1, z\}{z−1,z} whose average reward is that of the monotone policy (z)(z)(z), so no policy beats the best monotone one and the passive set under an optimal-from-everywhere policy is exactly {x≥x(W)}\{x \ge x(W)\}{x≥x(W)} for the smallest maximizing threshold. Formalizing this needs the average reward of a finite Markov chain as a limit determined by the stationary distribution of the recurrent class reached, for the two-point kernels of the model, and a case analysis of policies as {0,1}\{0,1\}{0,1}-strings. The envelope argument then needs that the smallest maximizer of max⁡y[Wϕ(y)+R(y)]\max_y [W\phi(y) + R(y)]maxy​[Wϕ(y)+R(y)] is nonincreasing in WWW when ϕ\phiϕ is strictly decreasing, and that with W∗W^*W∗ strictly decreasing the maximizer is ≤x\le x≤x exactly when W≥W∗(x)W \ge W^*(x)W≥W∗(x). Theorem 6.5 is the same with the roles of up and down exchanged. Nothing in Mathlib computes Cesàro limits of finite Markov chains.

Formalization scope

Restless bandits are the two-action DecisionProcesses of the superprocess module; average reward is a real limsup of Cesàro means of Bochner integrals over the chain law of the Bandit Algorithms model under the stationary kernel; the optimal average reward is a supremum over the finite type of deterministic stationary Markov policies and the finite state space, bounded by the reward bound. Both models are on Fin k with the book's states shifted down by one, kernels driftKernel p f that move to f x with probability p x, and the boundary conventions of ϕ\phiϕ and ψ\psiψ (the book's "convenient positive values") replaced by their values 1,01, 01,0 and 0,10, 10,1 at the two extreme thresholds; the model assumptions λ(k)=μ(1)=0\lambda(k) = \mu(1) = 0λ(k)=μ(1)=0, ν(1)=ρ(k)=0\nu(1) = \rho(k) = 0ν(1)=ρ(k)=0, rates in [0,1][0, 1][0,1], and rrr increasing and nonnegative are hypotheses. Theorem 6.5's "increasing" is read as strictly increasing, as in Theorem 6.4, since a nonstrict ψ\psiψ admits zero interior rates for which the monotone reduction fails. The milestone (6.9) requires k≥1k \ge 1k≥1 and positive interior rates, which Theorem 6.4's hypothesis (i) implies.

Trivializing readings are excluded: indexability is monotonicity of the passive set over all real subsidies, the passive set is defined through policies optimal from every initial state, and the index identity is for every state. Welcome contributions: the average reward of a two-state cycle, the reduction of an arbitrary {0,1}\{0,1\}{0,1}-policy to a monotone one, and the envelope lemma for lines with decreasing slopes.

Selected references

  • J. Gittins, K. Glazebrook, R. Weber, Multi-armed Bandit Allocation Indices, 2nd ed., Wiley, 2011, Chapter 6. doi:10.1002/9780470980033
  • P. Whittle, Restless bandits: activity allocation in a changing world, Journal of Applied Probability 25(A), 1988. doi:10.2307/3214163
  • R. R. Weber, G. Weiss, On an index policy for restless bandits, Journal of Applied Probability 27(3), 1990. doi:10.2307/3214547
  • K. D. Glazebrook, C. Kirkbride, D. Ruiz-Hernandez, Spinning plates and squad systems: policies for bi-directional restless bandits, Advances in Applied Probability 38(1), 2006. doi:10.1239/aap/1143936141
  • J. Niño-Mora, Restless bandits, partial conservation laws and indexability, Advances in Applied Probability 33(1), 2001. doi:10.1017/S0001867800010661
  • C. H. Papadimitriou, J. N. Tsitsiklis, The complexity of optimal queueing network control, Mathematics of Operations Research 24(2), 1999. doi:10.1287/moor.24.2.293
7 thms3 active usersReviewed
🏆Completed
Linear OptimizationOperations ResearchOptimization+1·Captain: naimengye

Multi-armed Bandit Allocation Indices IV: The Achievable Region, Generalized Conservation Laws and the Adaptive Greedy AlgorithmTextbook

Motivation

Chapter 5 of Gittins, Glazebrook and Weber, Multi-armed Bandit Allocation Indices (2nd ed., doi:10.1002/9780470980033), presents the achievable region methodology of Tsoucas, Bertsimas and Niño-Mora, Glazebrook and Garbe, and Dacre, Glazebrook and Niño-Mora: instead of arguing about policies, one argues about the set of performance vectors they can produce. For a multi-armed bandit the natural performance of a policy is the vector of discounted numbers of times each state is continued; the expected return is linear in it; and the set of achievable performances turns out to be a polytope cut out by conservation laws, one inequality per subset of states, with equality exactly for the priority policies that put that subset last. Optimizing a linear objective over a polytope is a linear program, its dual is solved by an adaptive greedy algorithm, and the primal solution is the performance of a priority policy whose priorities are the algorithm's outputs, the Gittins indices. This gives yet another proof of the index theorem (Section 5.3) and, more importantly, a definition, generalized conservation laws (Section 5.4), of the class of systems for which the same argument works: branching bandits, multi-class queues, job scheduling with discounted rewards, systems with imposed priority classes. The chapter's main result, Theorem 5.5, is the statement that every such system is solved by an index policy.

Setting

There are NNN job types E={1,…,N}E = \{1, \dots, N\}E={1,…,N}. A policy π\piπ has a performance xπ∈R+Nx^\pi \in \mathbb{R}^N_+xπ∈R+N​, a vector of expectations; a permutation σ\sigmaσ of EEE defines the permutation policy giving σN\sigma_NσN​ highest and σ1\sigma_1σ1​ lowest priority, and Sk={σ1,…,σk}S_k = \{\sigma_1, \dots, \sigma_k\}Sk​={σ1​,…,σk​} is the set of the kkk lowest-priority types. The system satisfies GCL(1) if there are a base function b:2E→R+b : 2^E \to \mathbb{R}_+b:2E→R+​ and a matrix A=(AiS)A = (A_i^S)A=(AiS​), positive on SSS and zero off it, such that for every policy

∑i∈SAiSxiπ≥b(S)(S⊆E),∑i∈EAiExiπ=b(E),\sum_{i \in S} A_i^S x_i^\pi \ge b(S) \quad (S \subseteq E), \qquad \sum_{i \in E} A_i^E x_i^\pi = b(E),i∈S∑​AiS​xiπ​≥b(S)(S⊆E),i∈E∑​AiE​xiπ​=b(E),

with equality in the first for every permutation policy whose ∣S∣|S|∣S∣ lowest-priority types are SSS. GCL(2) reverses the inequality. The adaptive greedy algorithm AG(A,r)AG(A, r)AG(A,r) picks iNi_NiN​ maximizing ri/AiEr_i/A_i^Eri​/AiE​, sets yˉE\bar y_Eyˉ​E​ to the maximum, removes iNi_NiN​, and repeats with the adjusted rewards ri−∑j≥kAiSjyˉSjr_i - \sum_{j \ge k} A_i^{S_j}\bar y_{S_j}ri​−∑j≥k​AiSj​​yˉ​Sj​​ divided by AiSk−1A_i^{S_{k-1}}AiSk−1​​; its outputs are the order i1,…,iNi_1, \dots, i_Ni1​,…,iN​, the dual variables yˉSk\bar y_{S_k}yˉ​Sk​​ and the indices νik=∑j≥kyˉSj\nu_{i_k} = \sum_{j \ge k} \bar y_{S_j}νik​​=∑j≥k​yˉ​Sj​​.

For the SFABP of Section 5.3, nnn identical bandit processes on EEE with kernel PPP and discount factor aaa in the model of the Bandit Algorithms series, xiπ=Eπ∑tatIi(t)x_i^\pi = \mathbb{E}^\pi \sum_t a^t I_i(t)xiπ​=Eπ∑t​atIi​(t) is the discounted number of continuations of a bandit in state iii, AiS=E[1+a+⋯+aTiS−1]A_i^S = \mathbb{E}[1 + a + \cdots + a^{T_i^S - 1}]AiS​=E[1+a+⋯+aTiS​−1] is the discounted return time to SSS from i∈Si \in Si∈S, and b(S)b(S)b(S) is the minimal cost ∑i∈SAiSxiπ\sum_{i \in S} A_i^S x_i^\pi∑i∈S​AiS​xiπ​, namely (1−a)−1E[aτ](1-a)^{-1}\mathbb{E}[a^\tau](1−a)−1E[aτ] with τ\tauτ the number of continuations needed to bring every bandit into SSS.

Formalization targets

Goal: Theorem 5.5

For a GCL(1) system whose achievable region is convex, and any reward vector rrr: the achievable region is the polytope

P(A,b)={x∈R+N:∑i∈SAiSxi≥b(S), S⊂E, ∑i∈EAiExi=b(E)};P(A, b) = \Big\{x \in \mathbb{R}_+^N : \sum_{i \in S} A_i^S x_i \ge b(S),\ S \subset E,\ \sum_{i \in E} A_i^E x_i = b(E)\Big\};P(A,b)={x∈R+N​:i∈S∑​AiS​xi​≥b(S), S⊂E, i∈E∑​AiE​xi​=b(E)};

its extreme points are performances of permutation policies; AG(A,r)AG(A, r)AG(A,r) has an output; and for every output the permutation policy in the order it finds, the Gittins index policy, maximizes ∑irixiπ\sum_i r_i x_i^\pi∑i​ri​xiπ​ over all policies.

Milestones

Lemma 5.1 (the SFABP satisfies the conservation laws, with equality for policies giving priority to states outside SSS); the identification on p. 123 of the adaptive greedy indices of a SFABP with the Gittins indices, together with their monotonicity along the order found; Theorem 5.10, the GCL(2) counterpart of the goal for cost minimization.

Significance

Theorem 5.5 is the index theorem in its most general form of this kind: it says nothing about Markov chains, only that performances are expectations, objectives are linear and conservation laws hold, and it delivers both the optimal policy and the algorithm that computes its priorities in polynomial time in the number of job types. It is the theorem behind the index results for branching bandits and Klimov's multi-class queue and behind the suboptimality bounds of Sections 5.5 and 5.7, all of which are calculations on the polytope. Lemma 5.1 and the p. 123 identification are what tie the abstract theorem to the Gittins index: they show that the multi-armed bandit is a GCL(1) system and that the priorities the algorithm produces are the same indices as Chapters 2 to 4 define through stopping times.

None of these is machine-checked. Formalizing Theorem 5.5 puts an LP-duality index theorem on the platform in a form any system can instantiate by verifying its conservation laws; formalizing Lemma 5.1 relates the Bandit Algorithms run law to the single-chain return times, which is the first conservation law on that model; and the p. 123 theorem gives an algorithmic characterization of the Gittins index on finite chains, distinct from the restart and largest-remaining-index characterizations of Chapter 2.

Difficulty

The goal's optimality clause is weak LP duality once one shows that the greedy dual variables are nonpositive except yˉE\bar y_Eyˉ​E​ and satisfy the dual constraints with equality, which is a finite induction on the stages; the extreme-point clause needs that every vertex of a polyhedron is the unique maximizer of some linear functional, and the region clause that a compact convex set is the convex hull of its extreme points (Krein–Milman in finite dimension, or the polyhedral fact directly). None of this is in Mathlib in the required form. Lemma 5.1 is probabilistic: the lower bound requires the strong Markov property of the continued bandit under an arbitrary past-measurable policy, a pathwise accounting of the discounted periods paid for by each continuation from SSS, and the observation that at most τ\tauτ slots can be spent on bandits that have never been in SSS; the equality for priority policies requires that these policies use exactly those slots first and then tile the future with return excursions, and the product form of b(S)b(S)b(S) requires independence of the bandits' process-time trajectories under the run law, which is built decision time by decision time rather than as a product. The p. 123 theorem is the computation (5.13) to (5.14) combined with the optimal-stopping characterization of Chapter 2 for the stop sets {i1,…,ik−2}\{i_1, \dots, i_{k-2}\}{i1​,…,ik−2​}, which lie between {ν<ν(ik−1)}\{\nu < \nu(i_{k-1})\}{ν<ν(ik−1​)} and {ν≤ν(ik−1)}\{\nu \le \nu(i_{k-1})\}{ν≤ν(ik−1​)}; ties make the induction delicate, and the statement is claimed for every tie-breaking.

Formalization scope

GCL(1) and GCL(2) systems are structures over an arbitrary policy type: performance, base function, matrix, permutation policies and the three laws are fields, so the theorems are statements about finite-dimensional data and the platform's proof needs no probability. The adaptive greedy algorithm is specified relationally, as the set of its possible outputs with arbitrary tie-breaking, and the conclusion holds for each of them; existence of an output is asserted separately. The optimality clause is stated as a comparison with every policy rather than as a real supremum. The hypothesis that the achievable region is convex is explicit: the book's argument from extreme points to the whole polytope uses randomization of policies, and without it the region of a system with only its permutation policies is finite. The SFABP items use nnn identical bandits on Fin N in the Bandit Algorithms model, the coefficients AiSA_i^SAiS​ through Mission I's stoppedTime at the return time, and b(S)b(S)b(S) in the product form (1−a)−1∏j:kj∉SE[aTkjS](1-a)^{-1}\prod_{j : k_j \notin S}\mathbb{E}[a^{T^S_{k_j}}](1−a)−1∏j:kj​∈/S​E[aTkj​S​], which is the minimal cost the argument on p. 120 establishes; the book prints a sum, which is 000 when all bandits start in SSS where the minimal cost is 1/(1−a)1/(1-a)1/(1−a). Discount factors are in (0,1)(0, 1)(0,1) throughout.

Trivializing readings are excluded: AiS>0A_i^S > 0AiS​>0 for i∈Si \in Si∈S is part of the structure and of Lemma 5.1's conclusion, the polytope equations are over all subsets, and the index clause quantifies over every greedy output. Welcome contributions: the nonpositivity and dual feasibility of the greedy variables, the vertex-exposure lemma for polyhedra, and the product decomposition of the run law of identical bandits.

Selected references

  • J. Gittins, K. Glazebrook, R. Weber, Multi-armed Bandit Allocation Indices, 2nd ed., Wiley, 2011, Chapter 5. doi:10.1002/9780470980033
  • D. Bertsimas, J. Niño-Mora, Conservation laws, extended polymatroids and multiarmed bandit problems; a polyhedral approach to indexable systems, Mathematics of Operations Research 21(2), 1996. doi:10.1287/moor.21.2.257
  • P. Tsoucas, The region of achievable performance in a model of Klimov, IBM Research Report RC16543, 1991.
  • E. G. Coffman, I. Mitrani, A characterization of waiting time performance realizable by single-server queues, Operations Research 28(3), 1980. doi:10.1287/opre.28.3.810
  • K. D. Glazebrook, R. Garbe, Almost optimal policies for stochastic systems which almost satisfy conservation laws, Annals of Operations Research 92, 1999. doi:10.1023/A:1018992306696
  • T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020, Chapter 35. doi:10.1017/9781108571401
8 thms3 active usersReviewed
🏆Completed
Algorithmic Game TheoryMachine LearningOperations Research+1·Captain: naimengye

Introduction to Multi-Armed Bandits XI: Bandits and Agents, Incentivized Exploration via Hidden ExplorationTextbook

Motivation

A recommendation system learns from the users it serves: the diner who tries a restaurant produces the review the next diner reads. Each user would rather exploit what is already known than explore for the benefit of those who come later, so a population of self-interested agents under-explores, and an alternative that looks bad on sparse early evidence may never be tried again even when it is the best. Chapter 11 of Slivkins, Introduction to Multi-Armed Bandits (arXiv:1904.07272), treats incentivized exploration: a principal who cannot force the agents but can recommend, and who, because it aggregates what earlier agents observed, knows more than any one of them. The question is whether recommendations alone can induce enough exploration to learn as fast as an ordinary bandit algorithm. The model is that of Kremer, Mansour and Perry (JPE 2014) and the results are those of Mansour, Slivkins and Syrgkanis (EC 2015, Operations Research 2020), specialized to two arms; the single-round problem is Bayesian persuasion in the sense of Kamenica and Gentzkow (AER 2011).

Setting

There are KKK arms and TTT rounds. A mean reward vector μ∈[0,1]K\mu \in [0,1]^Kμ∈[0,1]K is drawn from a known prior PPP, and each pull of arm aaa yields a reward drawn from a known family DμaD_{\mu_a}Dμa​​ with mean μa\mu_aμa​. In round ttt the principal recommends an arm rect\mathrm{rec}_trect​; agent ttt, who knows the prior, the family, the algorithm and the round but not the past, sees only rect\mathrm{rec}_trect​, chooses ata_tat​, collects rt∼Dμatr_t \sim D_{\mu_{a_t}}rt​∼Dμat​​​ and leaves; the principal observes (at,rt)(a_t, r_t)(at​,rt​). The chapter works with two arms, ordered so that the prior means satisfy μ10≥μ20\mu^0_1 \ge \mu^0_2μ10​≥μ20​, with a prior of finite support and finitely many reward values.

An algorithm is Bayesian incentive-compatible (BIC, Definition 11.4) if following its recommendation is in every agent's interest given what the agent knows: for every round ttt and arms a≠a′a \ne a'a=a′ with Pr⁡[rect=a,Et−1]>0\Pr[\mathrm{rec}_t = a, E_{t-1}] > 0Pr[rect​=a,Et−1​]>0,

E[μa−μa′∣rect=a, Et−1]≥0,(11.1)\mathbb{E}[\mu_a - \mu_{a'} \mid \mathrm{rec}_t = a,\ E_{t-1}] \ge 0, \tag{11.1}E[μa​−μa′​∣rect​=a, Et−1​]≥0,(11.1)

where Et−1E_{t-1}Et−1​ is the event that all previous agents complied. A BIC algorithm is then an ordinary bandit algorithm whose recommendations are followed, and the run has the law of the Bayesian bandit of Chapter 3. Two contrasting policies frame the chapter. GREEDY reveals the history and lets agents exploit, at∈arg⁡max⁡aE[μa∣Ht]a_t \in \arg\max_a \mathbb{E}[\mu_a \mid H_t]at​∈argmaxa​E[μa​∣Ht​] (11.2); it is BIC and it fails. HiddenExploration (Algorithm 11.1) hides a little exploration in a lot of exploitation: on a signal sig\mathrm{sig}sig, with probability ε\varepsilonε it recommends a target arm atrg(sig)a_{\mathrm{trg}}(\mathrm{sig})atrg​(sig), otherwise the arm maximizing E[μa∣sig]\mathbb{E}[\mu_a \mid \mathrm{sig}]E[μa​∣sig], ties to arm 1. Its posterior gap is G=E[μ2−μ1∣sig]G = \mathbb{E}[\mu_2 - \mu_1 \mid \mathrm{sig}]G=E[μ2​−μ1​∣sig]. RepeatedHE (Algorithm 11.2) runs it round after round with an arbitrary bandit algorithm ALG\mathrm{ALG}ALG as the target: N0N_0N0​ initial rounds recommend arm 1; afterwards, with probability ε\varepsilonε the round is an exploration round in which ALG\mathrm{ALG}ALG chooses (and is fed the reward), and otherwise the exploitation branch recommends min⁡arg⁡max⁡aE[μa∣St]\min\arg\max_a \mathbb{E}[\mu_a \mid S_t]minargmaxa​E[μa​∣St​], where StS_tSt​ is the data of all exploration rounds so far (11.10). The quantity that governs everything is G1,n=E[μ2−μ1∣S1,n]G_{1,n} = \mathbb{E}[\mu_2 - \mu_1 \mid S_{1,n}]G1,n​=E[μ2​−μ1​∣S1,n​] (11.11), the posterior gap after nnn samples of arm 1, and Property (11.12), that Pr⁡[G1,n>0]>0\Pr[G_{1,n} > 0] > 0Pr[G1,n​>0]>0 for some nnn: arm 2 can appear better after enough samples of arm 1.

Formalization targets

Goal: Theorem 11.15

RepeatedHE with exploration probability ε>0\varepsilon > 0ε>0 and N0N_0N0​ initial samples of arm 1 is BIC as long as

ε<13 E[G⋅1{G>0}],G=GN0+1=E[μ2−μ1∣S1,N0],\varepsilon < \tfrac13\,\mathbb{E}\big[G \cdot \mathbf 1\{G > 0\}\big], \qquad G = G_{N_0+1} = \mathbb{E}[\mu_2 - \mu_1 \mid S_{1,N_0}],ε<31​E[G⋅1{G>0}],G=GN0​+1​=E[μ2​−μ1​∣S1,N0​​],

for any bandit algorithm ALG\mathrm{ALG}ALG and any horizon. The threshold depends on the prior alone.

Milestones

Theorem 11.7 (GREEDY never chooses arm 2 with probability at least μ10−μ20\mu^0_1 - \mu^0_2μ10​−μ20​) and Corollary 11.8 (linear Bayesian regret of GREEDY under independent priors); Lemma 11.10 (HiddenExploration is BIC when ε≤13E[G1{G>0}]\varepsilon \le \frac13\mathbb{E}[G\mathbf 1\{G > 0\}]ε≤31​E[G1{G>0}]) with Claim 11.12 (the arm-2 side of the constraint suffices); Corollary 11.14 (RepeatedHE is BIC under the round-by-round condition); Theorem 11.19 (without Property (11.12) no BIC algorithm ever plays arm 2, ties to arm 1).

Significance

The results say when exploration can be incentivized at all and how. Theorem 11.7 shows that revealing everything is not a solution: the greedy dynamics gets stuck on arm 1 with a probability that does not shrink with TTT, and Corollary 11.8 turns that into Ω(T)\Omega(T)Ω(T) Bayesian regret. Theorem 11.15 shows that a recommendation-only principal can induce any amount of exploration it wants, with ALG\mathrm{ALG}ALG arbitrary, at a per-round rate ε\varepsilonε fixed by the prior; Theorem 11.17 (stated with a proof sketch, and omitted here) then transfers ALG\mathrm{ALG}ALG's regret to RepeatedHE up to the prior-dependent factors N0N_0N0​ and 1/ε1/\varepsilon1/ε, so O~(T)\tilde O(\sqrt T)O~(T​) regret is attainable subject to incentives. Theorem 11.19 closes the picture: Property (11.12) is necessary as well as sufficient. Together they characterize which priors admit incentivized exploration and give an algorithm that works for all of them.

Nothing of this is machine-checked. The mission adds to the Bayesian layer of mission III (prior, posterior by Bayes' rule, Bayesian regret) the BIC constraint on a joint law, GREEDY as a policy, the single-round HiddenExploration on an abstract finite signal, and the law of RepeatedHE; all of it is reusable for the KKK-arm and the "explore all explorable arms" extensions of the literature review.

Difficulty

Theorem 11.7 is a martingale argument: the posterior gap along the history is a Doob martingale, the first round in which arm 2 is chosen is a bounded stopping time, and optional stopping gives E[Zτ]=μ10−μ20\mathbb{E}[Z_\tau] = \mu^0_1 - \mu^0_2E[Zτ​]=μ10​−μ20​; all of this has to be set up on the joint law of (μ,HT)(\mu, H_T)(μ,HT​) of mission III, where the posterior is defined by Bayes' rule and the identification with a conditional expectation is itself a theorem (posterior_eq_condProb). Lemma 11.10 is the heart of the chapter and is not a computation about rec\mathrm{rec}rec: it works with F(E)=E[G1E]F(E) = \mathbb{E}[G\mathbf 1_E]F(E)=E[G1E​], splits along the two branches, uses that the exploitation branch recommends arm 2 exactly when G>0G > 0G>0, and closes with F(G>0)+F(G<0)=E[μ2−μ1]≤0F(G > 0) + F(G < 0) = \mathbb{E}[\mu_2 - \mu_1] \le 0F(G>0)+F(G<0)=E[μ2​−μ1​]≤0; the only place where the analysis uses that both branches are functions of the signal is the step E[μ2−μ1∣rec=2]=E[G∣rec=2]\mathbb{E}[\mu_2 - \mu_1 \mid \mathrm{rec} = 2] = \mathbb{E}[G \mid \mathrm{rec} = 2]E[μ2​−μ1​∣rec=2]=E[G∣rec=2], and a formalization has to make that step explicit. Theorem 11.15 requires seeing each later round of RepeatedHE as a HiddenExploration with signal StS_tSt​, where ALG\mathrm{ALG}ALG's choice is a randomized function of StS_tSt​, and then the monotonicity of E[Gt1{Gt>0}]\mathbb{E}[G_t\mathbf 1\{G_t > 0\}]E[Gt​1{Gt​>0}] in ttt, a two-line consequence of St+1S_{t+1}St+1​ determining StS_tSt​ that presupposes the posterior given StS_tSt​ is the Bayes posterior of the exploration data alone, which is true because the exploration decisions do not depend on μ\muμ given that data. Corollary 11.8 needs the independence of the event "μ1<1−2α\mu_1 < 1 - 2\alphaμ1​<1−2α and arm 2 is never chosen" from μ2\mu_2μ2​. Theorem 11.19 is an induction in which the inductive hypothesis is a probability-zero statement about all earlier rounds.

Formalization scope

Arms are Fin 2, the book's arm 1 being index 0; rounds are Fin T. The prior is a probability measure on mean vectors supported on a finite set F⊆[0,1]2F \subseteq [0,1]^2F⊆[0,1]2, with μ10≥μ20\mu^0_1 \ge \mu^0_2μ10​≥μ20​ as a hypothesis; the reward family is mission III's RewardFamily (finitely many values, mean ν\nuν for ν∈[0,1]\nu \in [0,1]ν∈[0,1]). BIC is defined on a joint law of (μ,record)(\mu, \text{record})(μ,record) of the run in which every agent complies, with the recommendation of each round read off the record; the compliance event Et−1E_{t-1}Et−1​ of (11.1) is the sure event of that law, which is the standard reading of "the agents believe all previous agents complied". For a bandit policy the law is mission III's jointMeasure. Conditional expectations are written as finite sums over FFF, so there are no integrals and no integrability side conditions; a posterior mean off the support is a junk 000 that never enters a theorem. GREEDY allows arbitrary tie-breaking; HiddenExploration's exploitation branch breaks ties toward arm 1 as Algorithm 11.1 does; the tie convention of Theorem 11.19 is the strict form of BIC for arm 2. The law of RepeatedHE is an explicit finitely supported measure, μ\muμ and record weighted by the prior times the product of the round probabilities (initial rounds forced to arm 1, then the ε\varepsilonε-coin, ALG\mathrm{ALG}ALG's kernel on its own history, or the exploitation arm, then DμatD_{\mu_{a_t}}Dμat​​​); it is written this way because ALG\mathrm{ALG}ALG is fed a history of variable length. Two conditions are stated exactly as printed: Lemma 11.10 with ε≤13E[G1{G>0}]\varepsilon \le \frac13\mathbb{E}[G\mathbf 1\{G > 0\}]ε≤31​E[G1{G>0}] (non-strict, checked at equality) and Theorem 11.15 with the strict inequality.

Trivializations are excluded: ε>0\varepsilon > 0ε>0 throughout; the BIC condition is asserted only where the recommendation has positive probability, and the sums in it are over the finite support, so an unsatisfiable hypothesis cannot hide in a measure-zero set. Welcome contributions: the optional-stopping argument on jointMeasure, the identification of explPostMean with the conditional expectation given the exploration data, the Bayes-rule algebra behind Lemma 11.10, and the counting lemmas on heRecords.

Selected references

  • A. Slivkins, Introduction to Multi-Armed Bandits, Foundations and Trends in Machine Learning 12(1-2), 2019, Chapter 11. arXiv:1904.07272, doi:10.1561/2200000068
  • I. Kremer, Y. Mansour, M. Perry, Implementing the "Wisdom of the Crowd", Journal of Political Economy 122(5), 2014. doi:10.1086/676597
  • Y. Mansour, A. Slivkins, V. Syrgkanis, Bayesian Incentive-Compatible Bandit Exploration, Operations Research 68(4), 2020 (EC 2015). doi:10.1287/opre.2019.1919
  • E. Kamenica, M. Gentzkow, Bayesian Persuasion, American Economic Review 101(6), 2011. doi:10.1257/aer.101.6.2590
  • M. Sellke, A. Slivkins, The Price of Incentivizing Exploration: A Characterization via Thompson Sampling and Sample Complexity, Operations Research 71(5), 2023. doi:10.1287/opre.2022.2401
10 thms3 active usersReviewed
🏆Completed
Machine LearningOperations Research·Captain: Shuze Chen

Bandit Algorithms I: Concentration of MeasureTextbook

How quickly does the empirical mean of independent random variables concentrate around the true mean? This question is the analytic engine of the entire theory of stochastic bandits: every optimistic algorithm (Explore-Then-Commit, UCB and its relatives) is calibrated by a tail bound on the sample mean. This mission formalizes the subgaussian framework of Chapter 5 of Lattimore–Szepesvári's Bandit Algorithms: a random variable XXX is σ\sigmaσ-subgaussian when E[eλX]≤eλ2σ2/2\mathbb{E}[e^{\lambda X}] \le e^{\lambda^2\sigma^2/2}E[eλX]≤eλ2σ2/2 for all λ\lambdaλ, and the Cramér–Chernoff method converts this moment-generating-function control into the exponential tail P(X≥ε)≤e−ε2/(2σ2)\mathbb{P}(X \ge \varepsilon) \le e^{-\varepsilon^2/(2\sigma^2)}P(X≥ε)≤e−ε2/(2σ2). The goal theorem is the Hoeffding-type bound: the sample mean of nnn independent σ\sigmaσ-subgaussian deviations exceeds the true mean by ε\varepsilonε with probability at most exp⁡(−nε2/(2σ2))\exp(-n\varepsilon^2/(2\sigma^2))exp(−nε2/(2σ2)), together with its confidence form P(μ^+2σ2log⁡(1/δ)/n≤μ)≤δ\mathbb{P}\big(\hat\mu + \sqrt{2\sigma^2\log(1/\delta)/n} \le \mu\big) \le \deltaP(μ^​+2σ2log(1/δ)/n​≤μ)≤δ — the exact bound every UCB index is built from. These few lines of analysis are cited by every regret bound in the series.

2 thms3 active usersReviewed
🏆Completed
Machine LearningOperations ResearchStatistics·Captain: mikedeng1

Distribution-Free Contextual Dynamic Pricing 2: Every Algorithm Has Expected Regret at Least C₀ C_p T₀ on Some Perturbed Linear Bandit with Perturbation C_pResearch Paper

Motivation

Dynamic pricing asks a seller to post prices to a stream of customers and learn demand from purchase decisions alone. Luo, Sun and Liu study the contextual version in which a customer's valuation is linear in a covariate vector plus noise with an unknown distribution (arXiv:2109.07340v2; Mathematics of Operations Research, 2024). Their DIP policy estimates the linear part, discretizes the remaining price residual, and then runs an upper-confidence-bound rule on the discretized problem. Because the linear part is only estimated, the discretized problem is not an ordinary linear bandit: its reward parameter drifts from period to period by an amount controlled by the estimation error. The paper calls this a perturbed linear bandit (PLB) and proves an upper bound for it whose dominant extra term is linear in the perturbation size times the horizon.

This mission formalizes the matching statement: no algorithm can avoid that linear term. It is the paper's Proposition 2 (p. 22), proved in its Appendix A (pp. 44–46).

Setting

Fix a dimension ddd. A perturbed linear bandit with perturbation constant CpC_pCp​ consists of parameters ξ1,ξ2,…∈Rd\xi_1,\xi_2,\ldots\in\mathbb R^dξ1​,ξ2​,…∈Rd, finite action sets A1,A2,…⊂Rd\mathcal A_1,\mathcal A_2,\ldots\subset\mathbb R^dA1​,A2​,…⊂Rd and noise variables η1,η2,…\eta_1,\eta_2,\ldotsη1​,η2​,…, with ∥ξs−ξt∥∞≤Cp\|\xi_s-\xi_t\|_\infty\le C_p∥ξs​−ξt​∥∞​≤Cp​ for all periods s,ts,ts,t. In period ttt an algorithm A∗\mathcal A^*A∗ sees the action set At\mathcal A_tAt​ and the history of its past actions and rewards, plays At∈AtA_t\in\mathcal A_tAt​∈At​, and observes the reward

Zt=⟨ξt,At⟩+ηt.Z_t=\langle\xi_t,A_t\rangle+\eta_t .Zt​=⟨ξt​,At​⟩+ηt​.

The algorithm may randomize. The parameters ξt\xi_tξt​ are never observed.

For a central parameter ξ~∈Rd\tilde\xi\in\mathbb R^dξ~​∈Rd the paper defines the parameter set

PB(ξ~,Cp)={ξ∈Rd: ∥ξ−ξ~∥∞≤Cp/2},PB(\tilde\xi,C_p)=\{\xi\in\mathbb R^d:\ \|\xi-\tilde\xi\|_\infty\le C_p/2\},PB(ξ~​,Cp​)={ξ∈Rd: ∥ξ−ξ~​∥∞​≤Cp​/2},

so any sequence in PB(ξ~,Cp)PB(\tilde\xi,C_p)PB(ξ~​,Cp​) has perturbation at most CpC_pCp​.

The PLB regret over T0T_0T0​ periods compares each action with the best action of the current period:

RT0PLB(A∗)=∑t=1T0(max⁡a∈At⟨ξt,a⟩−⟨ξt,At⟩) ≥0.R^{PLB}_{T_0}(\mathcal A^*)=\sum_{t=1}^{T_0}\Big(\max_{a\in\mathcal A_t}\langle\xi_t,a\rangle-\langle\xi_t,A_t\rangle\Big)\ \ge 0 .RT0​PLB​(A∗)=t=1∑T0​​(a∈At​max​⟨ξt​,a⟩−⟨ξt​,At​⟩) ≥0.

The expectation E\mathbb EE is over the algorithm's randomness and the noise.

Formalization targets

Goal: Proposition 2 (p. 22)

For every ξ~\tilde\xiξ~​ with positive entries (and d≥2d\ge2d≥2) there is C0>0C_0>0C0​>0 depending only on ξ~\tilde\xiξ~​ such that for every Cp≥0C_p\ge0Cp​≥0 with Cp/2<min⁡iξ~iC_p/2<\min_i\tilde\xi_iCp​/2<mini​ξ~​i​ and every algorithm A∗\mathcal A^*A∗ there are parameters ξt∈PB(ξ~,Cp)\xi_t\in PB(\tilde\xi,C_p)ξt​∈PB(ξ~​,Cp​) and action sets At\mathcal A_tAt​ with

E(RT0PLB(A∗)) ≥ C0 Cp T0for all T0≥1.\mathbb E\big(R^{PLB}_{T_0}(\mathcal A^*)\big)\ \ge\ C_0\,C_p\,T_0\qquad\text{for all }T_0\ge1 .E(RT0​PLB​(A∗)) ≥ C0​Cp​T0​for all T0​≥1.

The formal statement also records that the constructed instance is a PLB with perturbation CpC_pCp​, that its action sets are nonempty, and that every action has a single nonzero entry with ∣⟨ξt,a⟩∣≤1|\langle\xi_t,a\rangle|\le1∣⟨ξt​,a⟩∣≤1, which places the hard instance inside the class covered by the paper's upper bound (Lemma 3, Conditions 1 and 3).

Milestones

  1. Two-point construction (proof of Proposition 2, p. 44). With C=Cp/2C=C_p/2C=Cp​/2, two coordinates ju≠jvj_u\ne j_vju​=jv​, u=ξ~juu=\tilde\xi_{j_u}u=ξ~​ju​​, v=ξ~jvv=\tilde\xi_{j_v}v=ξ~​jv​​, actions U=eju/(u+C)U=e_{j_u}/(u+C)U=eju​​/(u+C), V=ejv/(v+C)V=e_{j_v}/(v+C)V=ejv​​/(v+C) and parameters ξu,ξv\xi_u,\xi_vξu​,ξv​ obtained from ξ~\tilde\xiξ~​ by moving the two coordinates by ∓C\mp C∓C, ±C\pm C±C:
ξu,ξv∈PB(ξ~,Cp),⟨ξu,U⟩=u−Cu+C≤1=⟨ξv,U⟩,⟨ξv,V⟩=v−Cv+C≤1=⟨ξu,V⟩.\xi_u,\xi_v\in PB(\tilde\xi,C_p),\qquad\langle\xi_u,U\rangle=\tfrac{u-C}{u+C}\le1=\langle\xi_v,U\rangle,\qquad\langle\xi_v,V\rangle=\tfrac{v-C}{v+C}\le1=\langle\xi_u,V\rangle .ξu​,ξv​∈PB(ξ~​,Cp​),⟨ξu​,U⟩=u+Cu−C​≤1=⟨ξv​,U⟩,⟨ξv​,V⟩=v+Cv−C​≤1=⟨ξu​,V⟩.
  1. Per-period bound (pp. 45–46). If u≤vu\le vu≤v, AAA is a random action in {U,V}\{U,V\}{U,V}, and ξt=ξu\xi_t=\xi_uξt​=ξu​ when P(A=U)≥1/2P(A=U)\ge1/2P(A=U)≥1/2 and ξt=ξv\xi_t=\xi_vξt​=ξv​ otherwise, then
E(rt) ≥ C0Cp,C0=14v.\mathbb E(r_t)\ \ge\ C_0C_p,\qquad C_0=\frac1{4v}.E(rt​) ≥ C0​Cp​,C0​=4v1​.

Significance

The result. Lemma 3 of the paper bounds the regret of its M-LinUCB algorithm on a PLB by a T0\sqrt{T_0}T0​​-type term plus 2amax⁡CpT02a_{\max}C_pT_02amax​Cp​T0​. Proposition 2 shows that the second term is not an artefact of the analysis: an adversary who may move the parameter inside a box of width CpC_pCp​ forces every algorithm to pay order CpT0C_pT_0Cp​T0​. In the pricing application the perturbation is proportional to the estimation error ∥θ^−θ0∥1\|\hat\theta-\theta_0\|_1∥θ^−θ0​∥1​ of the linear part, so the lower bound explains why the paper's pricing regret bound (Theorem 1) carries a term linear in the accumulated estimation error. The result also separates the PLB from the zero-perturbation linear bandit, where O~(T0)\tilde O(\sqrt{T_0})O~(T0​​) regret is achievable.

Formalizing it. The proposition is proved in the paper; it has not been machine-checked. A formal proof requires an explicit model of randomized, history-dependent algorithms interacting with an adversarially chosen but deterministic parameter sequence, which the mission supplies. The model and the two-point construction are reusable for other two-hypothesis lower bounds in linear and contextual bandits.

Difficulty

The obvious argument fixes one hard instance and shows that every algorithm suffers on it. That fails: for a fixed parameter sequence, an algorithm that always plays the optimal action of that sequence has zero regret. The instance must therefore be chosen after the algorithm, and it must be chosen period by period, because the law of the period-ttt action depends on the rewards of periods 1,…,t−11,\dots,t-11,…,t−1, which depend on the parameters already chosen. The parameters must nevertheless be deterministic: the adversary may react to the law of the algorithm's action, not to its realization. Making this recursive construction precise, together with the measurability needed for "the probability that At=UA_t=UAt​=U" to be meaningful, is the main formal work. The arithmetic of each period is elementary.

A second point concerns the dimension: the construction needs two distinct coordinates, and for d=1d=1d=1 the statement fails, since every parameter in PB(ξ~,Cp)PB(\tilde\xi,C_p)PB(ξ~​,Cp​) is then positive and playing the largest action has zero regret.

Formalization scope

  • Model. Arms are coordinates Fin d (0-based); periods are 1-based. Vectors are Fin d → ℝ, and ∥⋅∥∞\|\cdot\|_\infty∥⋅∥∞​ is Mathlib's sup norm on that type. Action sets are finite sets (Finset). The maximum over an empty action set is set to 000; every statement requires nonempty action sets.
  • Algorithms. An algorithm is a decision rule that maps a sample point ω\omegaω of a probability space (its internal randomness), the past actions and rewards, and the current action set to an action of that set; it is jointly measurable in ω\omegaω and the history. The trajectory (At,Zt)(A_t,Z_t)(At​,Zt​) is defined recursively from the rule, the parameters, the action sets and the noise. Since the rule sees all of ω\omegaω, it may also depend on the noise; this only enlarges the class over which the lower bound holds.
  • Noise. The goal quantifies over every measurable noise process. The paper's PLB asks for conditionally sub-Gaussian noise; since η≡0\eta\equiv0η≡0 is allowed, the formal statement is stronger than the printed existence statement, and the paper's proof uses no property of the noise.
  • Instance. The parameters and action sets are deterministic and are chosen after the algorithm, the probability space and the noise. The constant C0C_0C0​ is chosen after ξ~\tilde\xiξ~​ and before CpC_pCp​, the algorithm and T0T_0T0​, which is the formal content of "only dependent on ξ~\tilde\xiξ~​".
  • Added hypotheses. d≥2d\ge2d≥2 (necessary, see above) and Cp≥0C_p\ge0Cp​≥0 (otherwise PB(ξ~,Cp)PB(\tilde\xi,C_p)PB(ξ~​,Cp​) is empty).
  • Expectation. Expected regret is the lower Lebesgue integral of the nonnegative regret, so no integrability convention can make the bound hold vacuously. The bound is positive whenever Cp>0C_p>0Cp​>0; a formalization in which the expectation could collapse to a default value, or the algorithm class could be empty, is ruled out (the class contains every fixed-choice rule).
  • Source. The source is the arXiv v2 preprint (6 March 2023); the paper's proofs are in its Appendix A.

Contributions welcome: proofs of the two milestones, the measurability of the recursively defined actions, and the summation step from the per-period bound to the goal.

Selected references

  • Y. Luo, W. W. Sun, Y. Liu, Distribution-Free Contextual Dynamic Pricing, Mathematics of Operations Research, 2024. Preprint arXiv:2109.07340v2; DOI 10.1287/moor.2023.1369.
  • Y. Abbasi-Yadkori, D. Pál, Cs. Szepesvári, Improved Algorithms for Linear Stochastic Bandits, NeurIPS, 2011. link
  • T. Lattimore, Cs. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020. doi:10.1017/9781108571401
5 thms2 active usersReviewed
🏆Completed
Machine LearningReinforcement Learning·Captain: mikedeng1

The Statistical Complexity of Interactive Decision Making VII: Policy-Cover Inverse Gap Weighting Certifies dec_γ ≤ 95H³SA/γ for Tabular Reinforcement LearningResearch Paper

Why an explicit exploration policy matters

In episodic reinforcement learning, an agent learns the rewards and dynamics of a system while choosing actions that determine which states it observes. A bound on the number of states and actions alone does not prescribe how to explore: a policy can be valuable because it earns reward, because it visits an uncertain state-action pair, or both. Foster, Kakade, Qian, and Rakhlin introduced the decision-estimation coefficient (DEC) to measure this trade-off in interactive decision problems. Their tabular reinforcement learning example gives an explicit policy distribution with a bound depending on the horizon and the sizes of the state and action sets Foster et al., §5.2.

The target is Proposition 5.6 of that paper. Its policy-cover inverse gap weighting strategy, or PC-IGW, places weight on policies that cover state-action pairs under an estimated model, while accounting for their estimated loss in reward. The paper also gives a posterior-sampling bound for a dual DEC in Proposition 5.4. PC-IGW addresses the frequentist coefficient with an explicit distribution and the constant 959595 Foster et al., Propositions 5.4 and 5.6.

Tabular models and the policy cover

A tabular Markov decision process has a finite state set SSS, a finite action set AAA, and H≥1H\ge1H≥1 decision layers. At layer hhh, state sss, and action aaa, a model MMM supplies a transition distribution PhM(⋅∣s,a)P_h^M(\cdot\mid s,a)PhM​(⋅∣s,a) and a reward distribution RhM(⋅∣s,a)R_h^M(\cdot\mid s,a)RhM​(⋅∣s,a). Every model shares the initial state distribution d1d_1d1​. A randomized nonstationary policy π\piπ supplies an action distribution πh(⋅∣s)\pi_h(\cdot\mid s)πh​(⋅∣s) for each layer and state; all such policies form ΠRNS\Pi_{\mathrm{RNS}}ΠRNS​ Foster et al., Example 1.2.

The trajectory law M(π)M(\pi)M(π) records the sequence of states, actions, and rewards generated by MMM and π\piπ. The final transition reaches a deterministic terminal state and contributes no model-dependent factor to this law. The policy's value is fM(π)=EM,π[∑h=1Hrh]f^M(\pi)=\mathbb E^{M,\pi}[\sum_{h=1}^H r_h]fM(π)=EM,π[∑h=1H​rh​]. Write πM\pi_MπM​ for any value-maximizing policy and dhM,π(s,a)d_h^{M,\pi}(s,a)dhM,π​(s,a) for the probability of visiting (s,a)(s,a)(s,a) at layer hhh. The squared Hellinger distance between two trajectory laws has the normalization DH2(P,Q)=∑x(P(x)−Q(x))2D_H^2(P,Q)=\sum_x(\sqrt{P(x)}-\sqrt{Q(x)})^2DH2​(P,Q)=∑x​(P(x)​−Q(x)​)2, with no factor 1/21/21/2 Foster et al., §2, Eq. (5).

Given an estimated model Mˉ\bar MMˉ and η>0\eta>0η>0, Algorithm 4 chooses, for each (h,s,a)(h,s,a)(h,s,a), a policy πh,s,a\pi_{h,s,a}πh,s,a​ that maximizes

dhMˉ,π(s,a)2HSA+η(fMˉ(πMˉ)−fMˉ(π))over π∈ΠRNS.\frac{d_h^{\bar M,\pi}(s,a)}{2HSA+\eta\bigl(f^{\bar M}(\pi_{\bar M})-f^{\bar M}(\pi)\bigr)} \quad\text{over }\pi\in\Pi_{\mathrm{RNS}}.2HSA+η(fMˉ(πMˉ​)−fMˉ(π))dhMˉ,π​(s,a)​over π∈ΠRNS​.

Let Ψ\PsiΨ be the resulting set of distinct cover policies. The PC-IGW distribution gives each π∈Ψ∪{πMˉ}\pi\in\Psi\cup\{\pi_{\bar M}\}π∈Ψ∪{πMˉ​} weight p(π)=1/[λ+η(fMˉ(πMˉ)−fMˉ(π))]p(\pi)=1/[\lambda+\eta(f^{\bar M}(\pi_{\bar M})-f^{\bar M}(\pi))]p(π)=1/[λ+η(fMˉ(πMˉ​)−fMˉ(π))], where λ\lambdaλ makes the weights sum to one. Proposition 5.7 states that this positive normalizer is unique and lies in [1,2HSA][1,2HSA][1,2HSA] Foster et al., Algorithm 4 and Proposition 5.7.

Formalization targets

The main target is the explicit-policy inequality of Proposition 5.6. For γ>0\gamma>0γ>0 and η=γ/(21H2)\eta=\gamma/(21H^2)η=γ/(21H2), the distribution from Algorithm 4 satisfies, for every normalized tabular model MMM,

Eπ∼p ⁣[fM(πM)−fM(π)−γDH2(M(π),Mˉ(π))]≤95H3SAγ.\mathbb E_{\pi\sim p}\!\left[ f^M(\pi_M)-f^M(\pi)-\gamma D_H^2(M(\pi),\bar M(\pi)) \right]\le\frac{95H^3SA}{\gamma}.Eπ∼p​[fM(πM​)−fM(π)−γDH2​(M(π),Mˉ(π))]≤γ95H3SA​.

The uniform quantifier over MMM is the paper's supremum over its model class. Because this is a bound for a particular distribution ppp, it certifies the stated DEC bound. The Lean goal expresses that explicit distribution and its inequality directly; it does not introduce a separate supremum over all models and infimum over all distributions. The milestone list records the unique normalizer, the exploration-bias display (59), the cover comparison (61), the local and global simulation estimates, the Hellinger information inequalities, and Lemma 5.2's change of measure Foster et al., pp. 36–38 and Appendices A.2, F.1.

What the bound contributes

The result controls the cost of exploration for a finite-horizon model class by H3SA/γH^3SA/\gammaH3SA/γ with a specified constant. Its importance is the constructive policy distribution: the bound applies to the strategy returned by Algorithm 4, rather than asserting only that some distribution exists. In the paper, this bound combines with estimation and the general DEC-to-regret result to obtain a tabular regret guarantee; the authors also use the policy-cover idea as a starting point for their bilinear-class treatment Foster et al., §§5.2.2–5.2.3 and 7.

The paper result is proved. This mission asks for machine-checked proofs of its finite-alphabet statements; the draft theorems currently carry sorry. Its MDP interface includes reward distributions, full trajectory laws, occupancy probabilities, backward values, and local divergences. Those definitions can be reused for other tabular DEC results. Published FoundationsRL definitions already provide the discrete Hellinger and total-variation distances and a proved comparison between them. A separate published MDP interface records reward means, which cannot represent the reward-kernel Hellinger terms in this result Foster and Rakhlin, Foundations of Reinforcement Learning and Interactive Decision Making.

Where the difficulty lies

The cover is chosen using occupancies under Mˉ\bar MMˉ, while the regret is evaluated under any MMM in the class. An ordinary value bound for one fixed policy does not by itself control this mismatch. The page's intermediate expressions also divide by an average occupancy that may be zero. A formal statement must still cover state-action pairs that the policy distribution never visits, without assigning meaning to a zero-denominator ratio. The final inequality uses Hellinger distance between whole trajectories even though local simulation estimates are expressed through transition and reward kernels. Keeping those two levels of comparison aligned is the main mathematical burden Foster et al., pp. 37–38.

There is also a reward-bound issue in the appendix. Lemma F.3 uses continuation values in [0,1][0,1][0,1] at states visited under Mˉ\bar MMˉ, including states that may be unreachable under MMM from the shared initial law. The paper states an almost-sure bound on total episode reward, but that statement alone does not control every such continuation value Foster et al., Lemma F.3, p. 107.

Formalization scope

State, action, and reward alphabets are finite and nonempty. Probability laws and expectations are finite vectors and sums. Policies may randomize; the cover condition compares its choice against every valid randomized nonstationary policy. Layers use zero-based Fin H indices, while the paper uses 1,…,H1,\ldots,H1,…,H. All models share d1d_1d1​. The terminal transition is omitted from the trajectory law and from one-step divergence sums. Duplicate cover policies are merged, as required by the paper's set Ψ∪{πMˉ}\Psi\cup\{\pi_{\bar M}\}Ψ∪{πMˉ​}. The selector πM\pi_MπM​ must maximize value over valid policies, and the algorithm's weights must sum to one; these conditions exclude a free policy choice or an unnormalized pseudo-law.

The paper assumes ∑hrh∈[0,1]\sum_h r_h\in[0,1]∑h​rh​∈[0,1] almost surely. The local simulation lemma's stated bound needs two disclosed additions here: nonnegative stage rewards, and the same total-reward bound on every supported suffix started from any state. RewardsNormalized records them. The change-of-measure and global-simulation milestones use the weaker episode-only condition where sufficient. The cover comparison is cross-multiplied so it remains meaningful when an occupancy is zero. The goal's right side is a real number with γ>0\gamma>0γ>0, H≥1H\ge1H≥1, and nonempty S,AS,AS,A, so its divisions and size factors have the intended meaning.

A complete development needs the finite trajectory law, the policy-cover normalizer, occupancy bounds, the two simulation lemmas, Hellinger comparison inequalities, and the change-of-measure result. Contributions to these foundations, to the exact constant in Proposition 5.6, and to later generalizations of the cover are within scope. The abstract DEC infimum over all randomized policies is a later interface target; this mission's goal is the stronger, explicit-strategy certificate printed first in the proposition.

Selected references

  • Dylan J. Foster, Sham M. Kakade, Jian Qian, and Alexander Rakhlin, The Statistical Complexity of Interactive Decision Making, arXiv preprint, version 3, 2023. arXiv:2112.13487v3.
  • Dylan J. Foster and Alexander Rakhlin, Foundations of Reinforcement Learning and Interactive Decision Making, arXiv preprint, 2023. arXiv:2312.16730.
16 thms2 active usersReviewed
🏆Completed
Machine LearningReinforcement Learning·Captain: mikedeng1

The Statistical Complexity of Interactive Decision Making VI: Posterior Sampling Bounds the Dual DEC of Tabular Reinforcement Learning by 26H²SA/γResearch Paper

Why tabular reinforcement learning needs a decision-estimation bound

In episodic reinforcement learning, an agent acts repeatedly in an unknown system and must learn which policy earns the largest expected reward. Even when states and actions are finite, the agent's choices change which data it observes. Foster, Kakade, Qian, and Rakhlin use the decision-estimation coefficient (DEC) to connect this information constraint to regret across interactive decision problems. Their tabular example asks for an explicit bound in terms of the horizon, the numbers of states and actions, and an information penalty Foster et al., §5.2.

The result here is the posterior-sampling bound of Proposition 5.4. It gives a uniform constant, rather than a rate stated only with asymptotic notation. Its role is to show that a policy drawn from a distribution of model-optimal policies controls the Bayesian or dual DEC in a tabular class. The same article also studies a frequentist policy-cover strategy with a different coefficient; this mission isolates the posterior-sampling argument and the information inequalities it needs Foster et al., Propositions 5.4 and 5.6.

Models, policies, and trajectory laws

A tabular Markov decision process has a finite state set SSS, a finite action set AAA, and H≥1H\ge1H≥1 decision layers. At layer hhh, state sss, and action aaa, the model MMM supplies a transition distribution PhM(⋅∣s,a)P_h^M(\cdot\mid s,a)PhM​(⋅∣s,a) and a reward distribution RhM(⋅∣s,a)R_h^M(\cdot\mid s,a)RhM​(⋅∣s,a). Every model has the same initial state distribution d1d_1d1​. A randomized nonstationary policy π\piπ supplies an action distribution πh(⋅∣s)\pi_h(\cdot\mid s)πh​(⋅∣s) at each layer and state. This is the policy class ΠRNS\Pi_{\mathrm{RNS}}ΠRNS​ used in Example 1.2, including genuinely randomized policies Foster et al., Example 1.2.

The trajectory law M(π)M(\pi)M(π) records states, actions, and realized rewards over one episode. Its weight on a trajectory is the product of the shared initial probability, the policy probabilities, reward-kernel probabilities, and transitions between consecutive decision layers. The state after layer HHH is a deterministic terminal state. Therefore the last model transition contributes no factor to the recorded law. The reward value of an outcome is given by a fixed map rrr, and the value of a policy is fM(π)=EM,π[∑h=1Hrh]f^M(\pi)=\mathbb E^{M,\pi}[\sum_{h=1}^{H}r_h]fM(π)=EM,π[∑h=1H​rh​]. The occupancy dhM,π(s,a)d_h^{M,\pi}(s,a)dhM,π​(s,a) is the probability of visiting state-action pair (s,a)(s,a)(s,a) at layer hhh Foster et al., §2.

For each model, πM\pi_MπM​ denotes a policy maximizing fMf^MfM over ΠRNS\Pi_{\mathrm{RNS}}ΠRNS​. Let μ\muμ be a prior on models. Posterior sampling here means drawing a model M′∼μM'\sim\muM′∼μ and using πM′\pi_{M'}πM′​. The model whose performance is evaluated is a separate draw M∼μM\sim\muM∼μ; these two draws must remain independent. The reference model is denoted Mˉ\bar MMˉ. Squared Hellinger distance has the paper's normalization DH2(P,Q)=∑x(P(x)−Q(x))2D_H^2(P,Q)=\sum_x(\sqrt{P(x)}-\sqrt{Q(x)})^2DH2​(P,Q)=∑x​(P(x)​−Q(x)​)2, with no factor of one half Foster et al., Eq. (5).

Formalization targets

The main target is the policy-specific inequality of Proposition 5.4. For γ>0\gamma>0γ>0, it asserts

EM∼μ, π∼p[fM(πM)−fM(π)−γDH2(M(π),Mˉ(π))]≤26H2SAγ,p=Law⁡(πM′), M′∼μ.\mathbb E_{M\sim\mu,\,\pi\sim p} \left[f^M(\pi_M)-f^M(\pi)-\gamma D_H^2(M(\pi),\bar M(\pi))\right] \le \frac{26H^2SA}{\gamma}, \qquad p=\operatorname{Law}(\pi_{M'}),\ M'\sim\mu.EM∼μ,π∼p​[fM(πM​)−fM(π)−γDH2​(M(π),Mˉ(π))]≤γ26H2SA​,p=Law(πM′​), M′∼μ.

The claim holds for each finitely supported prior μ\muμ in the finite-alphabet representation. It directly gives the same upper bound when the dual DEC is defined as the supremum over such priors and the infimum over policy distributions. The Lean goal states the inequality for posterior sampling itself; no real supremum is needed to express that stronger, explicit claim. The reference model belongs to the normalized tabular class, which is the case actually proved in §5.2.1. The printed proposition also names reference models outside the class and points to the later general result Foster et al., p. 33.

The milestones are the conditional-to-joint Hellinger inequality of Lemma A.9, both inequalities of Lemma A.11, the global and local simulation lemmas F.2 and F.3, the change-of-measure statement in Lemma 5.2, and the unnumbered posterior-sampling decoupling inequality on pp. 34–35. Together they identify the stated estimates on which the proposition rests Foster et al., Appendices A.2 and F.1.

What the result contributes

The 26H2SA/γ26H^2SA/\gamma26H2SA/γ bound quantifies the price of learning the dynamics and rewards while selecting policies. Its explicit dependence on SSS and AAA is linear, and its dependence on the horizon is quadratic. In the paper, this estimate feeds the DEC-based analysis of tabular reinforcement learning and provides a comparison point for the policy-cover bound and later extensions beyond tabular models Foster et al., §5.2.

This formalization supplies a reusable interface for tabular MDPs with reward distributions: a full trajectory law, policy value, occupancy, backward values, and local divergences. The published FoundationsRL divergence definitions and a proved TV–Hellinger comparison can be reused. None of the seven statements has a machine-checked proof yet. A finite-horizon MDP object already published elsewhere records deterministic reward means and does not express the reward-kernel Hellinger distances required here.

Where the difficulty lies

A value difference is controlled by total variation between complete trajectory laws, but the proposed bound is expressed through local transition and reward discrepancies and ultimately through squared Hellinger distance between whole trajectories. These quantities average under different model-policy laws. Moving between them while retaining the exact constants in Lemma 5.2 is the central obstacle. A direct bound for a policy paired with the same model does not supply the double expectation in the target: the performance model and the policy-generating model are independent under posterior sampling.

The reward convention adds a second obstacle. Section 5.2 sets the reward space to [0,1][0,1][0,1] and also assumes that total episode reward lies in [0,1][0,1][0,1] almost surely. A model's continuation values at states it cannot reach from d1d_1d1​ are not controlled by the episode condition. The local simulation bound is nevertheless asserted under the reference model's occupancy, which may visit such states.

Formalization scope

State, action, and reward alphabets are finite; reward distributions are finite probability vectors. The paper allows general measurable reward spaces, so the finite reward alphabet is a declared scope restriction. Layers are indexed by Fin H, starting at zero, while the source labels them 1,…,H1,\ldots,H1,…,H. The terminal transition is omitted from trajectory products and local discrepancy sums. The initial distribution is a separate parameter shared by all models. Priors and policy laws have finite support, as in the paper's discrete-topology convention for model distributions. Every policy quantified in a theorem is checked to be a probability kernel, and each selected πM\pi_MπM​ is checked to maximize value over valid randomized policies.

RewardsNormalized expresses both standing conditions of §5.2: reward values lie in [0,1][0,1][0,1], and under every randomized policy every trajectory of positive probability has total reward ∑h=1Hrh∈[0,1]\sum_{h=1}^{H}r_h\in[0,1]∑h=1H​rh​∈[0,1]. The proposition and its decoupling milestone use it. The Appendix F simulation lemmas and Lemma 5.2 use EpisodeTotalInUnitInterval, which expresses only the total-reward condition stated in those lemmas.

The target uses the full trajectory law M(π)M(\pi)M(π) in DH2(M(π),Mˉ(π))D_H^2(M(\pi),\bar M(\pi))DH2​(M(π),Mˉ(π)). Replacing it with one-step kernel distances would change Proposition 5.4. A valid model is a family of probability kernels, a valid policy is a family of action distributions, and the model selector must really choose an optimum; these checks prevent the expected inequality from becoming true through zero-weight pseudo-laws or a free, suboptimal policy selector. The definition and Hellinger lemmas can support later tabular DEC missions as well.

Selected references

  • Dylan J. Foster, Sham M. Kakade, Jian Qian, and Alexander Rakhlin, The Statistical Complexity of Interactive Decision Making, arXiv preprint, version 3, 2023. arXiv:2112.13487v3.
  • Dylan J. Foster and Alexander Rakhlin, Foundations of Reinforcement Learning and Interactive Decision Making, arXiv preprint, 2023. arXiv:2312.16730.
11 thms2 active usersReviewed
🏆Completed
Machine LearningProbability·Captain: mikedeng1

A Tutorial on Thompson Sampling II: Under Thompson Sampling, Per-Period Expected Regret Splits into a Pessimism Term and a Width Term for Any History-Determined U_tResearch Paper

Motivation

An online decision maker repeatedly chooses an action, observes an outcome, and receives a reward. In a Bayesian model, an unknown parameter controls the outcomes, and the decision maker begins with a prior distribution for that parameter. Thompson sampling acts by drawing a parameter from its current posterior distribution and choosing an action that would be best if that draw were correct. This makes it applicable when an agent can simulate a plausible model and optimize an action for that model. Russo, Van Roy, Kazerouni, Osband, and Wen explain this rule for general online decision problems in Section 4 of their tutorial.

The tutorial's Section 8.1.2 asks how to analyze the rule's expected regret. An upper confidence bound method chooses an action using an optimistic score and can analyze regret through the score's optimism and its uncertainty width. Thompson sampling does not choose its actions by maximizing such a score. Yet the tutorial shows that any score computed from the observed past can be used to split Thompson sampling's expected regret into the same two types of terms. The result is stated on pp. 72–73, equation (8.4) and the display that follows.

Setting

Let X\mathcal XX be a finite, nonempty set of actions, YYY a finite set of outcomes, and Θ\ThetaΘ a measurable parameter space. A prior probability measure PPP is placed on the unknown parameter θ∈Θ\theta\in\Thetaθ∈Θ. If action x∈Xx\in\mathcal Xx∈X is selected, the outcome is sampled from a probability kernel qθ(⋅∣x)q_\theta(\cdot\mid x)qθ​(⋅∣x) on YYY. A known function r:Y→Rr:Y\to\mathbb Rr:Y→R turns an outcome into a reward. The mean reward of action xxx under parameter θ\thetaθ is

μ(x,θ)=∑y∈Yqθ(y∣x)r(y).\mu(x,\theta)=\sum_{y\in Y}q_\theta(y\mid x)r(y).μ(x,θ)=y∈Y∑​qθ​(y∣x)r(y).

For each θ\thetaθ, choose an action x∗(θ)x^*(\theta)x∗(θ) maximizing μ(⋅,θ)\mu(\cdot,\theta)μ(⋅,θ). This choice uses a fixed tie-break and is measurable. The same selector is used throughout the mission. A length-ttt history HtH_tHt​ records the first ttt action–outcome pairs. A policy π\piπ assigns an action distribution to every such history. The prior, the policy, and the outcome kernel together determine the law of all histories: first draw θ\thetaθ from PPP, then choose each action from π\piπ using the past history, and finally generate its outcome using qθq_\thetaqθ​.

After history hhh, the posterior law of x∗x^*x∗ is obtained by reweighting the prior by the likelihood of the observed outcomes. Thompson sampling draws a parameter from this posterior and plays the selected optimum for that parameter. Thus the law of its next action, given hhh, is the posterior law of x∗x^*x∗ given hhh. The policy remains defined even for a history with zero likelihood; choices there do not affect expectations.

For a period ttt, let Ut(x)U_t(x)Ut​(x) be any real score that uses the action xxx and the observed history Ht−1H_{t-1}Ht−1​. The score is fixed before the period's outcome or the unknown parameter can be inspected. The period regret is μ(x∗,θ)−μ(xt,θ)\mu(x^*,\theta)-\mu(x_t,\theta)μ(x∗,θ)−μ(xt​,θ), with expectation taken over the prior, the policy's randomization, and the outcome process.

Formalization targets

Probability matching and equation (8.4)

The first target makes posterior sampling precise. For every feasible history hhh and action aaa, the probability that Thompson sampling next plays aaa, together with observing hhh, equals the joint probability that the selected optimum is aaa and the history is hhh. Consequently, for every history-determined UtU_tUt​,

Eπ[Ut(xt)]=Eπ[Ut(x∗)].\mathbb E_\pi[U_t(x_t)] =\mathbb E_\pi[U_t(x^*)].Eπ​[Ut​(xt​)]=Eπ​[Ut​(x∗)].

This is equation (8.4) on p. 72. It asserts equality of unconditional expectations while preserving the fact that UtU_tUt​ may depend on the past. It does not assert that xtx_txt​ and x∗x^*x∗ have the same joint law with θ\thetaθ.

Per-period regret decomposition

The goal is the full chain of equalities displayed on p. 73:

Eπ[μ(x∗,θ)−μ(xt,θ)]=Eπ[μ(x∗,θ)−Ut(xt)]+Eπ[Ut(xt)−μ(xt,θ)]=Eπ[μ(x∗,θ)−Ut(x∗)]⏟pessimism+Eπ[Ut(xt)−μ(xt,θ)]⏟width.\begin{aligned} \mathbb E_\pi[\mu(x^*,\theta)-\mu(x_t,\theta)] &=\mathbb E_\pi[\mu(x^*,\theta)-U_t(x_t)] +\mathbb E_\pi[U_t(x_t)-\mu(x_t,\theta)]\\ &=\underbrace{\mathbb E_\pi[\mu(x^*,\theta)-U_t(x^*)]}_{\text{pessimism}} +\underbrace{\mathbb E_\pi[U_t(x_t)-\mu(x_t,\theta)]}_{\text{width}}. \end{aligned}Eπ​[μ(x∗,θ)−μ(xt​,θ)]​=Eπ​[μ(x∗,θ)−Ut​(xt​)]+Eπ​[Ut​(xt​)−μ(xt​,θ)]=pessimismEπ​[μ(x∗,θ)−Ut​(x∗)]​​+widthEπ​[Ut​(xt​)−μ(xt​,θ)]​​.​

The theorem holds for every score UtU_tUt​ determined by the past history. In particular, UtU_tUt​ need not be a confidence bound for the identity to hold. Being a useful upper confidence bound matters when one later estimates the two terms.

Significance

The equality permits analyses built around optimistic scores to be transferred to Thompson sampling. It separates expected regret into a term measuring whether the score falls below the true optimal mean and another measuring the gap between the score and the true mean at the sampled action. The tutorial cites Russo and Van Roy for the surrounding regret bounds; the identity here is the precise bridge used in that discussion.

The result is already proved in the tutorial. This mission seeks a machine-checked proof of its finite-action, finite-outcome form, including the posterior-sampling law, the policy-dependent history distribution, and the conditioning behind equation (8.4). The goal and milestones are open Lean statements, not completed machine-checked proofs. Their definitions can be reused in later finite Bayesian decision models with a general measurable parameter.

Difficulty

The central issue is that xtx_txt​ and x∗x^*x∗ are not interchangeable inside an expression that depends on θ\thetaθ. The sampled action and the selected optimum have the same distribution after conditioning on the observed past, but generally have different dependence on the true parameter. For example, replacing UtU_tUt​ by μ(⋅,θ)\mu(\cdot,\theta)μ(⋅,θ) would make the claimed equality false. A formal proof therefore has to respect the boundary between history-derived scores and parameter-dependent quantities, and it must use the same history law to interpret every expectation. Histories of zero probability also require a consistent posterior convention.

Formalization scope

The Lean model uses finite action and outcome types and a general measurable parameter type. This is a disclosed specialization of the tutorial, which also allows infinite actions and more general outcome spaces. The kernel qθq_\thetaqθ​ is nonnegative, sums to one over outcomes, and is measurable in θ\thetaθ. The prior is a probability measure. The reward function is real valued and bounded because its domain is finite. The optimal-action selector has measurable fibers and maximizes the mean reward for every parameter. These conditions make the integrands measurable and bounded, avoiding default values of nonintegrable Lean integrals.

Histories are functions from a finite period index to action–outcome pairs. Policies are normalized, nonnegative behavioural action probabilities. The likelihood of a history multiplies both its policy action factors and its kernel outcome factors. The posterior uses only outcome likelihood factors, which give the Bayes update on every positive-evidence history. The Thompson-sampling predicate enforces the posterior action probabilities on those histories and permits any normalized policy on impossible ones. Lean period sss denotes the tutorial's period t=s+1t=s+1t=s+1.

The shared tie-break is essential: Thompson sampling and x∗x^*x∗ use the same selected maximizer. The score has type Ht−1→X→RH_{t-1}\to\mathcal X\to\mathbb RHt−1​→X→R, so it cannot inspect θ\thetaθ or the current outcome. No definition is allowed to supply regret or the decomposition as a predetermined constant; the expectation is built from the prior, policy and outcome kernel. The development needs reusable definitions for the Bayesian model, finite histories, behavioural policies, likelihoods, posterior masses, and the two regret terms. Contributions proving probability matching, equation (8.4), and the final identity are welcome.

Selected references

  • Daniel J. Russo, Benjamin Van Roy, Abbas Kazerouni, Ian Osband, and Zheng Wen, A Tutorial on Thompson Sampling, Foundations and Trends in Machine Learning 11(1), 2018. DOI: 10.1561/2200000070.
4 thms2 active usersReviewed
🏆Completed
Machine LearningProbability·Captain: mikedeng1

A Tutorial on Thompson Sampling III: In the Known-Standard Problem, Every Stationary Deterministic Strategy Has Linearly Growing Expected Regret Under Some PriorResearch Paper

Why randomize actions

Thompson sampling (TS) selects each action by sampling a parameter from the current posterior and playing the action that is optimal for the sample. It is a stationary randomized strategy: the distribution of the action is a fixed function of the posterior, the same in every period, and the action is drawn at random from it. Its Bayesian regret bounds, surveyed in §8.1 of the tutorial by Russo, Van Roy, Kazerouni, Osband and Wen (Found. Trends Mach. Learn. 11(1), 2018), grow sublinearly in the horizon TTT.

A natural question is whether the randomization is essential. A strategy that maps the posterior to an action deterministically, for example "play the action with the highest posterior mean", is equally stationary and conceptually simpler. §8.1.3 of the tutorial answers the question with Example 8.1 (A Known Standard), taken from Russo and Van Roy, Learning to optimize via information-directed sampling (Operations Research 66(1), 2018): in a two-action problem with a known baseline action, every stationary deterministic strategy suffers linearly growing expected regret under some prior. The example is the standard justification for why posterior sampling, and not a deterministic function of the posterior, is the right stationary rule.

Setting

There are two actions X={1,2}\mathcal X=\{1,2\}X={1,2} and a binary unknown parameter θ∈{0,1}\theta\in\{0,1\}θ∈{0,1} with prior

θ∼Bernoulli(p0),P(θ=1)=p0.\theta\sim\mathrm{Bernoulli}(p_0),\qquad \mathbb P(\theta=1)=p_0 .θ∼Bernoulli(p0​),P(θ=1)=p0​.

In each period t=1,2,…t=1,2,\dotst=1,2,… the agent selects an action xtx_txt​ and observes a binary reward rtr_trt​. Rewards are conditionally independent across periods given θ\thetaθ and the actions:

  • action 111 (the known standard) pays Bernoulli(1/2)\mathrm{Bernoulli}(1/2)Bernoulli(1/2), whatever θ\thetaθ is;
  • action 222 pays Bernoulli(3/4)\mathrm{Bernoulli}(3/4)Bernoulli(3/4) if θ=1\theta=1θ=1 and Bernoulli(1/4)\mathrm{Bernoulli}(1/4)Bernoulli(1/4) if θ=0\theta=0θ=0.

The mean reward μ(x,θ)\mu(x,\theta)μ(x,θ) of action xxx is its success probability, and the optimal action x∗x^*x∗ is action 222 if θ=1\theta=1θ=1 and action 111 if θ=0\theta=0θ=0, so μ(x∗,θ)=3/4\mu(x^*,\theta)=3/4μ(x∗,θ)=3/4 or 1/21/21/2.

The history Ht\mathbb H_tHt​ lists the actions and rewards of the first ttt periods. The posterior pt=P(θ=1∣Ht)p_t=\mathbb P(\theta=1\mid\mathbb H_t)pt​=P(θ=1∣Ht​) is given by Bayes' rule (4.2) of the tutorial, applied period by period:

pt=p0 L1(Ht)p0 L1(Ht)+(1−p0) L0(Ht),Lθ(Ht)=∏s=1tPθ(rs∣xs),p_t=\frac{p_0\,L_1(\mathbb H_t)}{p_0\,L_1(\mathbb H_t)+(1-p_0)\,L_0(\mathbb H_t)},\qquad L_\theta(\mathbb H_t)=\prod_{s=1}^{t}\mathbb P_\theta(r_s\mid x_s),pt​=p0​L1​(Ht​)+(1−p0​)L0​(Ht​)p0​L1​(Ht​)​,Lθ​(Ht​)=s=1∏t​Pθ​(rs​∣xs​),

with p0p_0p0​ the prior on the empty history.

A stationary deterministic strategy is one function fff from posterior probabilities to actions, used in every period: xt=f(pt−1)x_t=f(p_{t-1})xt​=f(pt−1​). The expected cumulative regret (Bayesian regret, p. 71) of running fff from the prior p0p_0p0​ for TTT periods is

E[Regret(T)]=E[∑t=1T(μ(x∗,θ)−μ(xt,θ))],\mathbb E[\mathrm{Regret}(T)]=\mathbb E\Big[\sum_{t=1}^T\big(\mu(x^*,\theta)-\mu(x_t,\theta)\big)\Big],E[Regret(T)]=E[t=1∑T​(μ(x∗,θ)−μ(xt​,θ))],

the expectation taken over θ\thetaθ and the rewards. The Lean development names these objects succProb (μ\muμ), optMean (μ(x∗,⋅)\mu(x^*,\cdot)μ(x∗,⋅)), post (ptp_tpt​), Strategy (fff), lawGiven (the probability of a history given θ\thetaθ) and expRegret.

Formalization targets

Goal: some prior forces linear regret

∀f  ∃ p0∈(0,1]  ∃ c>0  ∀T∈N:Ep0[Regret(T)] ≥ c T.\forall f\ \ \exists\,p_0\in(0,1]\ \ \exists\,c>0\ \ \forall T\in\mathbb N:\qquad \mathbb E_{p_0}[\mathrm{Regret}(T)]\ \ge\ c\,T .∀f  ∃p0​∈(0,1]  ∃c>0  ∀T∈N:Ep0​​[Regret(T)] ≥ cT.

This is the sentence "for any deterministic stationary strategy, there exists a prior probability p0p_0p0​ such that expected cumulative regret grows linearly with time" (p. 79). The prior is chosen after the strategy. No constant is fixed; only the linear shape is asserted.

Milestones: the two cases of the example

  1. x1=1x_1=1x1​=1 for some p0>0p_0>0p0​>0. If f(p0)=1f(p_0)=1f(p0​)=1 for some p0∈(0,1]p_0\in(0,1]p0​∈(0,1], then on every history of positive probability pt=p0p_t=p_0pt​=p0​ and xt=1x_t=1xt​=1 for all ttt, and
Ep0[Regret(T)]=p04 T.\mathbb E_{p_0}[\mathrm{Regret}(T)]=\tfrac{p_0}{4}\,T .Ep0​​[Regret(T)]=4p0​​T.
  1. x1=2x_1=2x1​=2 for all p0>0p_0>0p0​>0. If f(p)=2f(p)=2f(p)=2 for every p∈(0,1]p\in(0,1]p∈(0,1], then for every p0∈(0,1]p_0\in(0,1]p0​∈(0,1], on every history of positive probability xt=2x_t=2xt​=2 for all ttt, and
Ep0[Regret(T)]=1−p04 T.\mathbb E_{p_0}[\mathrm{Regret}(T)]=\tfrac{1-p_0}{4}\,T .Ep0​​[Regret(T)]=41−p0​​T.

Significance

The example separates two properties that regret analyses often conflate: being a function of the posterior alone (stationarity) and achieving sublinear Bayesian regret. It shows that a stationary strategy with sublinear regret on all priors must randomize, which is the structural reason TS samples rather than maximizes. It also explains why deterministic algorithms with good regret, such as upper confidence bound methods, are necessarily nonstationary: their confidence radii depend on the period and on counts, not only on the posterior.

The tutorial gives the argument in one paragraph. A formal version makes precise what is left implicit there: that the posterior is the actual conditional probability computed from the history, that "uninformative" means the posterior is invariant under action 111, and that under action 222 the posterior never leaves (0,1](0,1](0,1]. The result is a closed-form regret identity in each case, not only an asymptotic statement. As far as is known, neither the example nor its source in Russo and Van Roy (2018) has been machine-checked.

Difficulty

The mathematics is elementary; the work lies in the bookkeeping of a history-dependent strategy. The strategy's action in period ttt depends on the whole history through pt−1p_{t-1}pt−1​, so the law of HT\mathbb H_THT​ is a product of indicator factors, each evaluating fff at a posterior computed from a prefix of the same history. A direct computation of the regret must show that every history of positive probability follows a single action path, and that is a statement about the posterior along every prefix, not only at the first period. A first attempt that defines the posterior by a recursion in which "the posterior stays at p0p_0p0​" is built in proves nothing about the example; the posterior must be Bayes' rule applied to the history.

Formalization scope

  • Paper action 111 is 0 : Fin 2, action 222 is 1 : Fin 2; θ=1\theta=1θ=1 is true; a reward of 111 is true. Lean period s : Fin T is the paper's period s+1s+1s+1.
  • Histories are Fin t → Fin 2 × Bool; every probability and expectation is a finite sum, so no measure theory is needed.
  • The posterior post is the closed form of iterated Bayes' rule (4.2). The probabilities with which the strategy picks its actions are 000 or 111 and do not depend on θ\thetaθ; on histories of positive probability they cancel, so post is the true conditional probability there. For 0<p0≤10<p_0\le10<p0​≤1 its denominator is positive, so Lean's convention x/0=0x/0=0x/0=0 is never used.
  • A strategy is a function ℝ → Fin 2; its values outside [0,1][0,1][0,1] are never used. The same function acts in every period, which is what "stationary" means.
  • The prior ranges over p0∈(0,1]p_0\in(0,1]p0​∈(0,1]: the text requires p0>0p_0>0p0​>0, and p0≤1p_0\le1p0​≤1 because it is a probability.
  • "Grows linearly with time" is read as a lower bound cTcTcT with c>0c>0c>0 for all TTT. The constant c>0c>0c>0 is part of the claim; c=0c=0c=0 would make the goal trivial. The quantifiers are in the order of the text, every strategy first and then a prior; the reverse order (one prior for all strategies) is a different statement that the tutorial does not claim.
  • Contributions welcome: proofs of the two milestones and of the goal from them, and lemmas on the posterior (invariance under an uninformative action, positivity) that are reusable for other Bayesian bandit examples.

The tutorial's argument for the example is on p. 79; the tutorial attributes the example to Russo and Van Roy (2018).

Selected references

  • D. J. Russo, B. Van Roy, A. Kazerouni, I. Osband, Z. Wen, A Tutorial on Thompson Sampling, Foundations and Trends in Machine Learning 11(1):1–96, 2018. https://doi.org/10.1561/2200000070 (Example 8.1, p. 79; Bayes' rule (4.2), p. 19; Bayesian regret, p. 71)
  • D. Russo, B. Van Roy, Learning to Optimize via Information-Directed Sampling, Operations Research 66(1):230–252, 2018. https://doi.org/10.1287/opre.2017.1663 (arXiv: https://arxiv.org/abs/1403.5556)
  • D. Russo, B. Van Roy, Learning to Optimize via Posterior Sampling, Mathematics of Operations Research 39(4):1221–1243, 2014. https://doi.org/10.1287/moor.2014.0650
  • W. R. Thompson, On the likelihood that one unknown probability exceeds another in view of the evidence of two samples, Biometrika 25(3/4):285–294, 1933. https://doi.org/10.2307/2332286
4 thms2 active usersReviewed
🏆Completed
Dynamic ProgrammingOperations ResearchProbability·Captain: mikedeng1

Multi-armed Bandits and the Gittins Index: With Per-Project Horizons the Maximal M-Process Reward Is K − ∫_M^K ∏ᵢ ∂φᵢ/∂m dm and the Gittins Index Rule Attains ItResearch Paper

Motivation

The multi-armed bandit problem asks how to allocate effort sequentially among NNN independent projects, only one of which can be worked on at a time, so as to maximise expected total discounted reward. It is the model behind research planning, clinical-trial allocation and job scheduling, and it is the prototypical example of a Markov decision problem whose state space grows exponentially in NNN yet whose optimal policy is simple. Gittins and Jones (1974) and Gittins (1979) showed that each project can be given an index computed from that project alone, and that it is optimal always to engage a project of largest index. P. Whittle's 1980 paper gave a short proof of this theorem by attaching a retirement option to the process, and extended it to projects with individual horizons.

Timeline:

  • 1974–1979. Gittins and Jones introduce the dynamic allocation index; Gittins (1979, JRSS B) proves its optimality by an interchange argument.
  • 1980. Whittle (JRSS B 42, 143–149) introduces the MMM-process, proves the finite-horizon formula (Theorem 1 below), derives the infinite-horizon identity (12), and extends the result to superprocesses (Theorem 2).
  • 1980s–2010s. Weber (1992) and Tsitsiklis (1994) give further short proofs; the index theory is collected in Gittins, Glazebrook and Weber, Multi-armed Bandit Allocation Indices (Wiley, 2011).

Setting

There are NNN projects. Project iii has a state xix_ixi​ in a measurable space XiX_iXi​. Engaging project iii in state xix_ixi​ earns the expected reward Ri(xi)R_i(x_i)Ri​(xi​) and moves its state by a Markov kernel PiP_iPi​; the other projects' states do not change. Rewards are discounted by β\betaβ, 0≤β<10 \le \beta < 10≤β<1, and bounded:

k(1−β)≤Ri(xi)≤K(1−β).(1)k(1-\beta) \le R_i(x_i) \le K(1-\beta). \qquad (1)k(1−β)≤Ri​(xi​)≤K(1−β).(1)

The operator Liθ(x)=Ri(xi)+β E[θ(x(t+1))∣x(t)=x,i(t)=i]L_i\theta(x) = R_i(x_i) + \beta\,\mathbb E[\theta(x(t+1)) \mid x(t) = x, i(t) = i]Li​θ(x)=Ri​(xi​)+βE[θ(x(t+1))∣x(t)=x,i(t)=i] evaluates engaging project iii for one step.

In the MMM-process one may also retire at any time for the reward MMM. Its value F(x,M)F(x, M)F(x,M) solves F=max⁡(M,max⁡iLiF)F = \max(M, \max_i L_i F)F=max(M,maxi​Li​F) (4), and the value φi(xi,M)\varphi_i(x_i, M)φi​(xi​,M) of the single-project MMM-process solves φi=max⁡(M,Liφi)\varphi_i = \max(M, L_i\varphi_i)φi​=max(M,Li​φi​) (9).

With per-project horizons, project iii may be operated at most sis_isi​ more times. A project with si>0s_i > 0si​>0 is active, and DisD_i sDi​s lowers sis_isi​ by one. The single-project value φi(xi,si,M)\varphi_i(x_i, s_i, M)φi​(xi​,si​,M) satisfies φi(xi,0,M)=M\varphi_i(x_i, 0, M) = Mφi​(xi​,0,M)=M and

φi(xi,si,M)=max⁡[M, Ri(xi)+β E φi(xi(t+1),si−1,M)].(14)\varphi_i(x_i, s_i, M) = \max\big[M,\ R_i(x_i) + \beta\,\mathbb E\,\varphi_i(x_i(t+1), s_i - 1, M)\big]. \qquad (14)φi​(xi​,si​,M)=max[M, Ri​(xi​)+βEφi​(xi​(t+1),si​−1,M)].(14)

F(x,s,M)F(x, s, M)F(x,s,M) is the maximal expected reward of the MMM-process under these limits. The index Mi(xi,si)M_i(x_i, s_i)Mi​(xi​,si​) is the infimal MMM with φi(xi,si,M)=M\varphi_i(x_i, s_i, M) = Mφi​(xi​,si​,M)=M, and vi=(1−β)Miv_i = (1-\beta)M_ivi​=(1−β)Mi​ is the Gittins index. Finally

F^(x,s,M)=K−∫MK∏i∂φi(xi,si,m)∂m dm.(13)\hat F(x, s, M) = K - \int_M^K \prod_i \frac{\partial \varphi_i(x_i, s_i, m)}{\partial m}\, dm. \qquad (13)F^(x,s,M)=K−∫MK​i∏​∂m∂φi​(xi​,si​,m)​dm.(13)

Formalization targets

Goal: Theorem 1 (p. 147)

F(x,s,M)=F^(x,s,M),F(x, s, M) = \hat F(x, s, M),F(x,s,M)=F^(x,s,M),

and FFF is attained by the policy that, at (x,s)(x, s)(x,s), engages a project iii maximising Mi(xi,si)M_i(x_i, s_i)Mi​(xi​,si​) when this maximum exceeds MMM, and otherwise retires. The Lean goal has three conjuncts: F=F^F = \hat FF=F^, every measurable index rule (any tie-breaking) earns FFF, and no feasible measurable Markov policy earns more than FFF.

Milestones (the steps of the proof)

  • Lemma 1 (p. 144): F(x,M)F(x, M)F(x,M) is non-decreasing and convex in MMM, equals Φ(x)\Phi(x)Φ(x) for M≤kM \le kM≤k and MMM for M≥KM \ge KM≥K.
  • (18) (p. 147): Pi(x,s,M)=∏j≠i∂φj/∂MP_i(x,s,M) = \prod_{j \ne i}\partial\varphi_j/\partial MPi​(x,s,M)=∏j=i​∂φj​/∂M is a distribution function in MMM, equal to 111 for M≥M(i)=max⁡j≠iMjM \ge M_{(i)} = \max_{j\ne i} M_jM≥M(i)​=maxj=i​Mj​.
  • (19) (p. 147): F^=φiPi+∫M∞φi dmPi\hat F = \varphi_i P_i + \int_M^\infty \varphi_i\, d_m P_iF^=φi​Pi​+∫M∞​φi​dm​Pi​.
  • (21) (p. 148): F^≥LiF^\hat F \ge L_i\hat FF^≥Li​F^, with equality if Mi=max⁡jMj≥MM_i = \max_j M_j \ge MMi​=maxj​Mj​≥M.
  • Retirement case (p. 148): if max⁡jMj≤M\max_j M_j \le Mmaxj​Mj​≤M then F^=M\hat F = MF^=M.
  • (16) (p. 147): F^=max⁡(M,max⁡iLiF^)\hat F = \max(M, \max_i L_i\hat F)F^=max(M,maxi​Li​F^) and F^(x,0,M)=M\hat F(x, 0, M) = MF^(x,0,M)=M.

Further item

  • Infinite horizon (p. 148, formula (12) of p. 146): F(x,M)=K−∫MK∏i∂φi(xi,m)/∂m dmF(x, M) = K - \int_M^K \prod_i \partial\varphi_i(x_i, m)/\partial m\, dmF(x,M)=K−∫MK​∏i​∂φi​(xi​,m)/∂mdm, with the index rule optimal.

Significance

Theorem 1 reduces an NNN-project problem, whose state space is a product, to NNN one-dimensional retirement problems. The formula (13) shows more than optimality of the index rule: the derivative of the multi-project value in MMM is the product of the single-project derivatives, so the multi-project value is computable from the projects separately. The finite-horizon version gives an inductive argument that avoids the interchange arguments of earlier proofs, and the infinite-horizon identity (12) follows from it.

The result is classical and proved. What this mission adds is a machine-checked version in a generality not yet on the platform: heterogeneous projects (each with its own kernel and reward on its own measurable space), a retirement option, and per-project horizons. The platform already has a proved index theorem for identical arms sharing one kernel and reward without retirement, BanditAlgorithm.gittins_index_theorem (Lattimore–Szepesvári, Theorem 35.9), single-arm retirement values in charge form (GittinsFiniteRetirementValue), and open Beta–Bernoulli special cases from Bäuerle–Rieder (MDPFinance.InfiniteHorizonApplications.proposition_7_6_7, proposition_7_6_9, theorem_7_6_10). None of these states Theorem 1, formula (13) or formula (12).

Difficulty

The obvious approach, value iteration on the product state space, gives existence of an optimal policy but says nothing about its structure. The content of Theorem 1 lies in the identity of a multi-dimensional value with an integral of a product of one-dimensional derivatives. Verifying it requires differentiating single-project values in MMM (they are convex, but have a kink exactly at the index, where the choice of one-sided derivative matters), an integration by parts against a Stieltjes measure in MMM, and an exchange of the expectation over the next state with the MMM-integral. The exhausted projects (si=0s_i = 0si​=0) and the boundary M=MiM = M_iM=Mi​ need separate care in every step.

Formalization scope

  • Projects are indexed by Fin N (0-based; the paper uses 1,…,N1, \dots, N1,…,N). State spaces are arbitrary measurable spaces; no finiteness or countability is assumed. Kernels are Markov, rewards measurable and bounded as in (1), and 0≤β<10 \le \beta < 10≤β<1 (β = 0 is allowed). kkk and KKK are any constants satisfying (1).
  • Budgets are s : Fin N → ℕ; DisD_i sDi​s is Function.update s i (s i - 1) and is only applied to active projects.
  • F(x,s,M)F(x, s, M)F(x,s,M), the "maximal reward", is the backward-induction value: F(x,0,M)=MF(x, 0, M) = MF(x,0,M)=M and F=max⁡(M,max⁡i activeLiF)F = \max(M, \max_{i\ \text{active}} L_iF)F=max(M,maxi active​Li​F), defined by recursion on ∑isi\sum_i s_i∑i​si​. It is not defined through F^\hat FF^, the index, or a policy. φi(xi,si,M)\varphi_i(x_i, s_i, M)φi​(xi​,si​,M) is defined by (14) with φi(xi,0,M)=M\varphi_i(x_i, 0, M) = Mφi​(xi​,0,M)=M.
  • ∂/∂m\partial/\partial m∂/∂m is the right derivative. Since the φi\varphi_iφi​ are convex in mmm, this changes (13) at no point, and it makes (18) hold at the kink M=M(i)M = M_{(i)}M=M(i)​.
  • The index of an exhausted project is the paper's −∞-\infty−∞. In Lean index takes the junk value 0 there; every use (the index rule, M(i)M_{(i)}M(i)​, max⁡jMj\max_j M_jmaxj​Mj​) ranges over active projects only.
  • Policies are Markov in (x,s,M)(x, s, M)(x,s,M), valued in Option (Fin N) (none = retire), feasible (they engage only active projects) and measurable (each decision set is measurable), so that their expected rewards are well defined. The index rule allows arbitrary tie-breaking; the goal quantifies over every such rule.
  • Infinite-horizon values (Lemma 1, the further item) are any bounded measurable solutions of (2), (4), (9); such solutions exist and are unique by the contraction property.
  • (19) uses the Stieltjes measure of any monotone right-continuous function agreeing with PiP_iPi​, and integrates over m>Mm > Mm>M.
  • Trivializing formalizations are excluded: FFF is not F^\hat FF^ by definition, exhausted projects cannot be engaged by the index rule, and maxima over iii are seeded with MMM rather than taken as a junk real supremum.

A complete development needs measurability of the recursive values, convexity of φi\varphi_iφi​ in MMM with one-sided derivatives, Lebesgue–Stieltjes integration by parts, and Fubini for kernels. Convexity of finite-horizon retirement values and the integration-by-parts identity are reusable beyond this mission. Contributions to any milestone, and sorry-free lemmas on these ingredients, are welcome.

Selected references

  • P. Whittle, Multi-armed Bandits and the Gittins Index, J. R. Statist. Soc. B 42(2), 143–149, 1980. https://doi.org/10.1111/j.2517-6161.1980.tb01111.x
  • J. C. Gittins, Bandit Processes and Dynamic Allocation Indices, J. R. Statist. Soc. B 41(2), 148–177, 1979. https://doi.org/10.1111/j.2517-6161.1979.tb01068.x
  • D. Blackwell, Discounted Dynamic Programming, Ann. Math. Statist. 36(1), 226–235, 1965. https://doi.org/10.1214/aoms/1177700285
  • J. Gittins, K. Glazebrook, R. Weber, Multi-armed Bandit Allocation Indices, 2nd ed., Wiley, 2011. https://doi.org/10.1002/9780470980033
  • T. Lattimore, Cs. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020, Chapter 35. https://doi.org/10.1017/9781108571401
8 thms2 active usersReviewed
🏆Completed
Machine LearningProbabilityStatistics·Captain: mikedeng1

Kullback–Leibler Upper Confidence Bounds for Optimal Sequential Allocation II: Empirical KL-UCB Draws a Suboptimal Arm log(T)/K_inf(ν_a, μ*) + O((log T)^{4/5} log log T) TimesResearch Paper

Motivation

In a stochastic multi-armed bandit a player repeatedly chooses one of KKK arms and receives a random reward drawn from that arm's unknown distribution; the aim is to pull suboptimal arms as rarely as possible. Lai and Robbins (1985) and, for general models, Burnetas and Katehakis (1996) showed that any reasonable strategy must pull a suboptimal arm aaa at least (1+o(1))log⁡T/Kinf⁡(νa,μ⋆)(1+o(1))\log T/\mathcal K_{\inf}(\nu_a,\mu^\star)(1+o(1))logT/Kinf​(νa​,μ⋆) times in TTT rounds, where Kinf⁡\mathcal K_{\inf}Kinf​ is a minimal Kullback–Leibler divergence defined below. A strategy whose expected number of pulls matches this constant is asymptotically optimal.

For rewards in [0,1][0,1][0,1], classical index policies such as UCB (Auer, Cesa-Bianchi and Fischer, 2002) achieve O(log⁡T)O(\log T)O(logT) pulls but with a constant governed by the gap of the means, not by Kinf⁡\mathcal K_{\inf}Kinf​. Cappé, Garivier, Maillard, Munos and Stoltz, Kullback–Leibler upper confidence bounds for optimal sequential allocation, Ann. Statist. 41(3), 2013 (arXiv:1210.1136v4), introduce the KL-UCB family of index policies and prove finite-time bounds whose leading term is the Lai–Robbins/Burnetas–Katehakis constant. This mission formalizes their result for empirical KL-UCB (Algorithm 3, Theorem 2), which is asymptotically optimal in the nonparametric model of finitely supported distributions on [0,1][0,1][0,1]. A companion mission covers kl-UCB in one-parameter exponential families (Theorem 1).

Timeline: Lai and Robbins (1985), lower bound for parametric families; Burnetas and Katehakis (1996), lower bound and asymptotically optimal policies for general models; Honda and Takemura (2010, 2011), the DMED algorithm, asymptotically optimal for finitely supported and bounded rewards; Cappé et al. (2013), the first index policy with a non-asymptotic bound whose leading term is optimal in this model.

Setting

A bandit has K≥2K\ge2K≥2 arms with reward distributions ν1,…,νK\nu_1,\dots,\nu_Kν1​,…,νK​ in a known model F\mathcal FF: the set of probability distributions over [0,1][0,1][0,1] with finite support. Write E(ν)=∫x dν(x)\mathrm E(\nu)=\int x\,d\nu(x)E(ν)=∫xdν(x), μa=E(νa)\mu_a=\mathrm E(\nu_a)μa​=E(νa​) and μ⋆=max⁡aμa\mu^\star=\max_a\mu_aμ⋆=maxa​μa​; arm aaa is suboptimal if μa<μ⋆\mu_a<\mu^\starμa​<μ⋆. At each round t≥1t\ge1t≥1 the player picks an arm AtA_tAt​ based on past observations and observes a reward drawn from νAt\nu_{A_t}νAt​​. Na(T)=∑t=1TI{At=a}N_a(T)=\sum_{t=1}^T\mathbb I\{A_t=a\}Na​(T)=∑t=1T​I{At​=a} is the number of pulls of arm aaa up to round TTT.

Equivalently, each arm has a reward stack Xa,1,Xa,2,…X_{a,1},X_{a,2},\dotsXa,1​,Xa,2​,… of i.i.d. draws from νa\nu_aνa​, all stacks independent, and the nnn-th pull of arm aaa returns Xa,nX_{a,n}Xa,n​. The empirical distribution of the first nnn rewards is ν^a,n=1n∑k=1nδXa,k\hat\nu_{a,n}=\frac1n\sum_{k=1}^n\delta_{X_{a,k}}ν^a,n​=n1​∑k=1n​δXa,k​​, and ν^a(t)=ν^a,Na(t)\hat\nu_a(t)=\hat\nu_{a,N_a(t)}ν^a​(t)=ν^a,Na​(t)​.

The minimal divergence is

Kinf⁡(ν,μ)=inf⁡{KL(ν,ν′):ν′∈F, E(ν′)>μ}∈[0,+∞],\mathcal K_{\inf}(\nu,\mu)=\inf\bigl\{\mathrm{KL}(\nu,\nu'):\nu'\in\mathcal F,\ \mathrm E(\nu')>\mu\bigr\}\in[0,+\infty],Kinf​(ν,μ)=inf{KL(ν,ν′):ν′∈F, E(ν′)>μ}∈[0,+∞],

the smallest Kullback–Leibler divergence from ν\nuν to a distribution of the model whose mean exceeds μ\muμ.

Empirical KL-UCB (Algorithm 3) pulls each arm once, then for t=K,K+1,…t=K,K+1,\dotst=K,K+1,… pulls an arm maximizing

Ua(t)=sup⁡{E(ν):ν∈M1(Supp(ν^a(t))∪{1}), KL(ν^a(t),ν)≤f(t)Na(t)},U_a(t)=\sup\Bigl\{\mathrm E(\nu):\nu\in\mathfrak M_1\bigl(\mathrm{Supp}(\hat\nu_a(t))\cup\{1\}\bigr),\ \mathrm{KL}(\hat\nu_a(t),\nu)\le\frac{f(t)}{N_a(t)}\Bigr\},Ua​(t)=sup{E(ν):ν∈M1​(Supp(ν^a​(t))∪{1}), KL(ν^a​(t),ν)≤Na​(t)f(t)​},

where M1(A)\mathfrak M_1(A)M1​(A) is the set of probability distributions carried by AAA and f(t)=log⁡t+log⁡log⁡tf(t)=\log t+\log\log tf(t)=logt+loglogt. The added point 111 is essential: without it the index is the empirical-likelihood bound, which equals the empirical mean when every observation is 000.

Formalization targets

Goal: Theorem 2 (pp. 15–16)

Assume μa>0\mu_a>0μa​>0 for all arms and μ⋆<1\mu^\star<1μ⋆<1. There is a constant M(νa,μ⋆)>0M(\nu_a,\mu^\star)>0M(νa​,μ⋆)>0 depending only on νa\nu_aνa​ and μ⋆\mu^\starμ⋆ such that, for every suboptimal arm aaa and all T≥3T\ge3T≥3,

E[Na(T)]≤log⁡TKinf⁡(νa,μ⋆)+36(μ⋆)4(log⁡T)4/5log⁡log⁡T+(72(μ⋆)4+2μ⋆(1−μ⋆)Kinf⁡(νa,μ⋆)2)(log⁡T)4/5+(1−μ⋆)2M(νa,μ⋆)2(μ⋆)2(log⁡T)2/5+log⁡log⁡TKinf⁡(νa,μ⋆)+2μ⋆(1−μ⋆)Kinf⁡(νa,μ⋆)2+4.\begin{aligned}\mathbb E[N_a(T)]\le{}&\frac{\log T}{\mathcal K_{\inf}(\nu_a,\mu^\star)}+\frac{36}{(\mu^\star)^4}(\log T)^{4/5}\log\log T+\Bigl(\frac{72}{(\mu^\star)^4}+\frac{2\mu^\star}{(1-\mu^\star)\mathcal K_{\inf}(\nu_a,\mu^\star)^2}\Bigr)(\log T)^{4/5}\\&+\frac{(1-\mu^\star)^2M(\nu_a,\mu^\star)}{2(\mu^\star)^2}(\log T)^{2/5}+\frac{\log\log T}{\mathcal K_{\inf}(\nu_a,\mu^\star)}+\frac{2\mu^\star}{(1-\mu^\star)\mathcal K_{\inf}(\nu_a,\mu^\star)^2}+4.\end{aligned}E[Na​(T)]≤​Kinf​(νa​,μ⋆)logT​+(μ⋆)436​(logT)4/5loglogT+((μ⋆)472​+(1−μ⋆)Kinf​(νa​,μ⋆)22μ⋆​)(logT)4/5+2(μ⋆)2(1−μ⋆)2M(νa​,μ⋆)​(logT)2/5+Kinf​(νa​,μ⋆)loglogT​+(1−μ⋆)Kinf​(νa​,μ⋆)22μ⋆​+4.​

The constants are the paper's. MMM is the one quantity the main text does not give; it is existentially quantified, before the bandit, so it may depend on nothing but (νa,μ⋆)(\nu_a,\mu^\star)(νa​,μ⋆).

Milestones

  1. (7), p. 9: the sets Cμ,γ={ν:∃ν′∈F, E(ν′)>μ, KL(ν,ν′)≤γ}\mathcal C_{\mu,\gamma}=\{\nu:\exists\nu'\in\mathcal F,\ \mathrm E(\nu')>\mu,\ \mathrm{KL}(\nu,\nu')\le\gamma\}Cμ,γ​={ν:∃ν′∈F, E(ν′)>μ, KL(ν,ν′)≤γ} satisfy Cμ,γ⊆{ν:Kinf⁡(ν,μ)≤γ}\mathcal C_{\mu,\gamma}\subseteq\{\nu:\mathcal K_{\inf}(\nu,\mu)\le\gamma\}Cμ,γ​⊆{ν:Kinf​(ν,μ)≤γ}.
  2. (5), p. 9: the decomposition {At+1=a}⊆{μ†≥Ua⋆(t)}∪{μ†<Ua(t), At+1=a}\{A_{t+1}=a\}\subseteq\{\mu^\dagger\ge U_{a^\star}(t)\}\cup\{\mu^\dagger<U_a(t),\ A_{t+1}=a\}{At+1​=a}⊆{μ†≥Ua⋆​(t)}∪{μ†<Ua​(t), At+1​=a}.
  3. The display after (7), p. 9: E[Na(T)]≤1+∑t=KT−1P{μ†≥Ua⋆(t)}+∑t=KT−1P{ν^a,Na(t)∈Cμ†,f(t)/Na(t), At+1=a}\mathbb E[N_a(T)]\le1+\sum_{t=K}^{T-1}\mathbb P\{\mu^\dagger\ge U_{a^\star}(t)\}+\sum_{t=K}^{T-1}\mathbb P\{\hat\nu_{a,N_a(t)}\in\mathcal C_{\mu^\dagger,f(t)/N_a(t)},\ A_{t+1}=a\}E[Na​(T)]≤1+∑t=KT−1​P{μ†≥Ua⋆​(t)}+∑t=KT−1​P{ν^a,Na​(t)​∈Cμ†,f(t)/Na​(t)​, At+1​=a}.
  4. (8), p. 10: the second sum is at most ∑n=1T−KP{ν^a,n∈Cμ†,f(T)/n}\sum_{n=1}^{T-K}\mathbb P\{\hat\nu_{a,n}\in\mathcal C_{\mu^\dagger,f(T)/n}\}∑n=1T−K​P{ν^a,n​∈Cμ†,f(T)/n​}.
  5. (9)–(10), p. 10: E[Na(T)]≤f(T)/Kinf⁡(νa,μ⋆)+∑n>n0P{ν^a,n∈Cμ†,f(T)/n}+∑tP{μ†≥Ua⋆(t)}+2\mathbb E[N_a(T)]\le f(T)/\mathcal K_{\inf}(\nu_a,\mu^\star)+\sum_{n>n_0}\mathbb P\{\hat\nu_{a,n}\in\mathcal C_{\mu^\dagger,f(T)/n}\}+\sum_t\mathbb P\{\mu^\dagger\ge U_{a^\star}(t)\}+2E[Na​(T)]≤f(T)/Kinf​(νa​,μ⋆)+∑n>n0​​P{ν^a,n​∈Cμ†,f(T)/n​}+∑t​P{μ†≥Ua⋆​(t)}+2 with n0=⌈f(T)/Kinf⁡(νa,μ⋆)⌉n_0=\lceil f(T)/\mathcal K_{\inf}(\nu_a,\mu^\star)\rceiln0​=⌈f(T)/Kinf​(νa​,μ⋆)⌉.
  6. p. 15: the supremum defining Ua(t)U_a(t)Ua​(t) over F\mathcal FF equals the supremum over M1(Supp(ν^a(t))∪{1})\mathfrak M_1(\mathrm{Supp}(\hat\nu_a(t))\cup\{1\})M1​(Supp(ν^a​(t))∪{1}).
  7. Implicit in Theorem 2: 0<Kinf⁡(ν,μ)<∞0<\mathcal K_{\inf}(\nu,\mu)<\infty0<Kinf​(ν,μ)<∞ for ν∈F\nu\in\mathcal Fν∈F and E(ν)<μ<1\mathrm E(\nu)<\mu<1E(ν)<μ<1.
  8. Proposition 1, p. 20: for nnn i.i.d. observations from any ν0\nu_0ν0​ on [0,1][0,1][0,1] with E(ν0)∈(0,1)\mathrm E(\nu_0)\in(0,1)E(ν0​)∈(0,1) and every ε>0\varepsilon>0ε>0,
P{U(ν^n,ε)≤E(ν0)}≤P{Kinf⁡(ν^n,E(ν0))≥ε}≤e(n+2)exp⁡(−nε).\mathbb P\{U(\hat\nu_n,\varepsilon)\le\mathrm E(\nu_0)\}\le\mathbb P\{\mathcal K_{\inf}(\hat\nu_n,\mathrm E(\nu_0))\ge\varepsilon\}\le e(n+2)\exp(-n\varepsilon).P{U(ν^n​,ε)≤E(ν0​)}≤P{Kinf​(ν^n​,E(ν0​))≥ε}≤e(n+2)exp(−nε).

Significance

Theorem 2 gives a finite-time bound whose leading term, log⁡T/Kinf⁡(νa,μ⋆)\log T/\mathcal K_{\inf}(\nu_a,\mu^\star)logT/Kinf​(νa​,μ⋆), equals the Burnetas–Katehakis lower bound for the model F\mathcal FF. Hence empirical KL-UCB is asymptotically optimal among all strategies for finitely supported rewards in [0,1][0,1][0,1], and the regret ∑a(μ⋆−μa)E[Na(T)]\sum_a(\mu^\star-\mu_a)\mathbb E[N_a(T)]∑a​(μ⋆−μa​)E[Na​(T)] inherits the optimal constant. Since Kinf⁡(νa,μ⋆)\mathcal K_{\inf}(\nu_a,\mu^\star)Kinf​(νa​,μ⋆) is at least the Bernoulli divergence of the means, and usually larger, the bound improves on kl-UCB and UCB for the same rewards. Proposition 1 is a non-asymptotic coverage bound for the empirical-likelihood upper confidence bound with the point 111 added, valid for every law on [0,1][0,1][0,1], not only finitely supported ones.

The proofs of Theorem 2 and Proposition 1 are in the paper's supplemental article (Appendix B), not in the main text; no machine-checked proof of either exists. The mission produces formal statements of the theorem and of the proof skeleton (5)–(10) that the paper shares with Theorem 1, and of the two facts about Kinf⁡\mathcal K_{\inf}Kinf​ that the bound needs. A formal proof would supply an explicit M(νa,μ⋆)M(\nu_a,\mu^\star)M(νa​,μ⋆), which the paper defines only inside the supplement.

Difficulty

The skeleton (5)–(10) is elementary bookkeeping; the difficulty lies in the two sums it leaves, both of which must be shown to be o(log⁡T)o(\log T)o(logT) with explicit constants. The second, ∑nP{ν^a,n∈Cμ†,f(T)/n}\sum_n\mathbb P\{\hat\nu_{a,n}\in\mathcal C_{\mu^\dagger,f(T)/n}\}∑n​P{ν^a,n​∈Cμ†,f(T)/n​}, needs a deviation estimate for the empirical Kinf⁡\mathcal K_{\inf}Kinf​ of a suboptimal arm that is precise enough to keep the leading constant 1/Kinf⁡(νa,μ⋆)1/\mathcal K_{\inf}(\nu_a,\mu^\star)1/Kinf​(νa​,μ⋆): a bound that only controls the deviation of the empirical mean loses it, since Kinf⁡\mathcal K_{\inf}Kinf​ depends on the whole distribution. The first, ∑tP{μ†≥Ua⋆(t)}\sum_t\mathbb P\{\mu^\dagger\ge U_{a^\star}(t)\}∑t​P{μ†≥Ua⋆​(t)}, concerns the optimal arm after a random number of pulls Na⋆(t)N_{a^\star}(t)Na⋆​(t), which the algorithm itself determines, so fixed-sample bounds such as Proposition 1 do not apply directly. Both need regularity of Kinf⁡\mathcal K_{\inf}Kinf​ as a function of a distribution in an infinite-dimensional model, where none of the closed forms of the exponential-family case is available.

Formalization scope

Arms are Fin K, and arm aaa of the paper is index a−1a-1a−1. The bandit is the platform's stack-of-rewards model RegretBandits.Stochastic.IsStochasticBandit: Xa,kX_{a,k}Xa,k​ (indexed from 000) are independent, identically distributed within each arm, with mean μa\mu_aμa​. Pull counts and μ⋆\mu^\starμ⋆ are the platform's pullCount and bestMean. Added to the page, and disclosed in each statement: every reward lies in [0,1][0,1][0,1] pathwise (a representation of "νa\nu_aνa​ is carried by [0,1][0,1][0,1]"), every arm choice is measurable, and in the goal the strategy is non-anticipating (At+1A_{t+1}At+1​ is measurable with respect to the arms and rewards of rounds 1,…,t1,\dots,t1,…,t). A run of Algorithm 3 is a pathwise predicate: rounds 1,…,K1,\dots,K1,…,K pull distinct arms, and every later round pulls a maximizer of U⋅(t)U_\cdot(t)U⋅​(t), with ties broken by any rule. KL is Mathlib's InformationTheory.klDiv in [0,+∞][0,+\infty][0,+∞], with the empirical distribution as first argument. Kinf⁡\mathcal K_{\inf}Kinf​ is kept in [0,+∞][0,+\infty][0,+∞] and converted to a real number only in the final bounds, where it is finite and positive. Probabilities of events whose measurability is not asserted are outer probabilities. The page's ∑n≥n0+1\sum_{n\ge n_0+1}∑n≥n0​+1​ in (10) is stated as the finite sum over n0<n≤T−Kn_0<n\le T-Kn0​<n≤T−K that (8) produces. The probability space of Theorem 2 lies in Type.

Trivializing formalizations are ruled out. The index is a supremum over a set that is nonempty (it contains E(ν^a(t))\mathrm E(\hat\nu_a(t))E(ν^a​(t))) and bounded above, so it is never Lean's junk value. Kinf⁡\mathcal K_{\inf}Kinf​ and Cμ,γ\mathcal C_{\mu,\gamma}Cμ,γ​ range over F\mathcal FF, not over all measures. The run predicate has both the initialization and the argmax clause. MMM is quantified before the bandit and the horizon.

Reusable beyond this mission: Kinf⁡\mathcal K_{\inf}Kinf​ for F\mathcal FF, the empirical-likelihood bound UUU of (15), and Proposition 1, a concentration inequality for empirical Kinf⁡\mathcal K_{\inf}Kinf​ that applies to any bounded i.i.d. sample. Proofs of any milestone are welcome, as are proofs of the skeleton (5)–(10) that also apply to the companion kl-UCB mission.

Selected references

  • O. Cappé, A. Garivier, O.-A. Maillard, R. Munos, G. Stoltz, Kullback–Leibler upper confidence bounds for optimal sequential allocation, Ann. Statist. 41(3):1516–1541, 2013. arXiv:1210.1136v4, doi:10.1214/13-AOS1119; supplement doi:10.1214/13-AOS1119SUPP.
  • T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Adv. Appl. Math. 6(1):4–22, 1985. doi:10.1016/0196-8858(85)90002-8
  • A. N. Burnetas, M. N. Katehakis, Optimal adaptive policies for sequential allocation problems, Adv. Appl. Math. 17(2):122–142, 1996. doi:10.1006/aama.1996.0007
  • J. Honda, A. Takemura, An asymptotically optimal bandit algorithm for bounded support models, COLT 2010, 67–79.
  • P. Auer, N. Cesa-Bianchi, P. Fischer, Finite-time analysis of the multiarmed bandit problem, Mach. Learn. 47:235–256, 2002. doi:10.1023/A:1013689704352
13 thms2 active usersReviewed
🏆Completed
Operations ResearchProbabilityStatistics·Captain: mikedeng1

Dynamic Pricing Without Knowing the Demand Function: Risk Bounds and Near-Optimal Algorithms III: With One Unknown Parameter, Staged Re-estimation Has Regret O((log log n)(log n)^{1/2}/n^{1/2})Research Paper

Motivation

A seller with a fixed stock of a single product and a finite selling season must post prices without knowing how demand responds to price. Revenue management treats this as a constrained stochastic control problem; with the demand curve known, the problem was solved by Gallego and van Ryzin (Management Science, 1994). When the curve is unknown, every price posted also serves as an experiment, so the seller faces an exploration–exploitation trade-off. Unlike a multi-armed bandit, this problem has a continuum of actions and a hard inventory constraint.

Besbes and Zeevi (Operations Research, 2009) measure a pricing policy by its worst-case relative revenue loss against a full-information benchmark, in an asymptotic regime where inventory and demand grow together. They give three upper bounds. This mission takes the third, Proposition 5: when the demand model has a single unknown scalar parameter, a policy that keeps re-estimating that parameter in stages of growing length has regret O((log⁡log⁡n)(log⁡n)1/2/n1/2)O\big((\log\log n)(\log n)^{1/2}/n^{1/2}\big)O((loglogn)(logn)1/2/n1/2). The paper's lower bound for parametric families (Proposition 4) is of order n−1/2n^{-1/2}n−1/2, so the rate is optimal up to logarithmic factors.

Setting

Market. Prices lie in [p‾,p‾]∪{p∞}[\underline p,\overline p]\cup\{p_\infty\}[p​,p​]∪{p∞​} with 0<p‾<p‾<p∞0<\underline p<\overline p<p_\infty0<p​<p​<p∞​. Posting the off price p∞p_\inftyp∞​ stops demand. The seller starts with inventory x>0x>0x>0 and sells over the horizon [0,T][0,T][0,T], T>0T>0T>0.

Demand. A demand function λ\lambdaλ maps a price to a demand rate. The class L(M,K‾,K‾,m)\mathcal L(M,\underline K,\overline K,m)L(M,K​,K,m) consists of the functions that are non-increasing with an inverse γ\gammaγ on [p‾,p‾][\underline p,\overline p][p​,p​], have a concave revenue rate r(l)=lγ(l)r(l)=l\gamma(l)r(l)=lγ(l), are bounded by MMM, are K‾\overline KK-Lipschitz with a K‾−1\underline K^{-1}K​−1-Lipschitz inverse, and attain a revenue rate max⁡ppλ(p)≥m\max_p p\lambda(p)\ge mmaxp​pλ(p)≥m. The parametric family is λ(p;θ)\lambda(p;\theta)λ(p;θ), θ∈Θ=[θlo,θhi]\theta\in\Theta=[\theta_{\mathrm{lo}},\theta_{\mathrm{hi}}]θ∈Θ=[θlo​,θhi​], with every member in the class (Assumption 1). Assumption 2 adds a test price p1p_1p1​, differentiability of λ(p1;⋅)\sqrt{\lambda(p_1;\cdot)}λ(p1​;⋅)​ and the Lipschitz bound ∣λ(p;θ)−λ(p;θ′)∣≤K‾2∣θ−θ′∣|\lambda(p;\theta)-\lambda(p;\theta')|\le\overline K_2|\theta-\theta'|∣λ(p;θ)−λ(p;θ′)∣≤K2​∣θ−θ′∣. Assumption 3 requires inf⁡p,θλ(p;θ)>l0>0\inf_{p,\theta}\lambda(p;\theta)>l_0>0infp,θ​λ(p;θ)>l0​>0 and an α\alphaα-Lipschitz solution map d↦g(p,d)d\mapsto g(p,d)d↦g(p,d) of the equation λ(p;⋅)=d\lambda(p;\cdot)=dλ(p;⋅)=d.

Demand process. Let NNN be a unit-rate Poisson process. Under a price path p(⋅)p(\cdot)p(⋅) and parameter θ∗\theta^*θ∗, the cumulative demand up to time ttt is N(∫0tλ(p(s);θ∗) ds)N\big(\int_0^t\lambda(p(s);\theta^*)\,ds\big)N(∫0t​λ(p(s);θ∗)ds). Sales stop when the inventory runs out.

Benchmark and regret. The deterministic relaxation JD(x,T∣θ)J^D(x,T\mid\theta)JD(x,T∣θ) is the supremum of ∫0Tp(s)λ(p(s);θ) ds\int_0^T p(s)\lambda(p(s);\theta)\,ds∫0T​p(s)λ(p(s);θ)ds over price paths with ∫0Tλ(p(s);θ) ds≤x\int_0^T\lambda(p(s);\theta)\,ds\le x∫0T​λ(p(s);θ)ds≤x. In the market of size nnn the inventory is nxnxnx and the demand nλn\lambdanλ. If Jnπ(x,T;θ)J^\pi_n(x,T;\theta)Jnπ​(x,T;θ) is the expected revenue of a policy π\piπ, its regret is Rnπ=1−Jnπ/JnD\mathcal R^\pi_n=1-J^\pi_n/J^D_nRnπ​=1−Jnπ​/JnD​.

Algorithm 3. Start from p^1=p1\hat p_1=p_1p^​1​=p1​ and use stages of lengths Δn(1),…,Δn(ℓn)\Delta^{(1)}_n,\dots,\Delta^{(\ell_n)}_nΔn(1)​,…,Δn(ℓn​)​ summing to TTT. Stage iii applies p^i\hat p_ip^​i​, estimates the demand rate d^i\hat d_id^i​ from the stage's demand, solves for θ^i=g(p^i,d^i)\hat\theta_i=g(\hat p_i,\hat d_i)θ^i​=g(p^​i​,d^i​), and sets p^i+1=max⁡{pu(θ^i),pc(θ^i)}\hat p_{i+1}=\max\{p^u(\hat\theta_i),p^c(\hat\theta_i)\}p^​i+1​=max{pu(θ^i​),pc(θ^i​)}. Here pu(θ)p^u(\theta)pu(θ) maximizes pλ(p;θ)p\lambda(p;\theta)pλ(p;θ) and pc(θ)p^c(\theta)pc(θ) minimizes ∣λ(p;θ)−x/T∣|\lambda(p;\theta)-x/T|∣λ(p;θ)−x/T∣. The tuning (19)–(20) is ℓn=(log⁡2)−1log⁡log⁡n\ell_n=(\log2)^{-1}\log\log nℓn​=(log2)−1loglogn stages with Δn(m)=βnn(aℓn/am)−1\Delta^{(m)}_n=\beta_n n^{(a_{\ell_n}/a_m)-1}Δn(m)​=βn​n(aℓn​​/am​)−1 and am=2m−1/(2m−1)a_m=2^{m-1}/(2^m-1)am​=2m−1/(2m−1).

Formalization targets

Goal: Proposition 5

∃ C>0, ∃ n0,∀n≥n0, ∀θ∈Θ:Rnπn(x,T;θ)≤C (log⁡log⁡n)(log⁡n)1/2n1/2.\exists\,C>0,\ \exists\,n_0,\quad \forall n\ge n_0,\ \forall\theta\in\Theta:\qquad \mathcal R^{\pi_n}_n(x,T;\theta)\le C\,\frac{(\log\log n)(\log n)^{1/2}}{n^{1/2}} .∃C>0, ∃n0​,∀n≥n0​, ∀θ∈Θ:Rnπn​​(x,T;θ)≤Cn1/2(loglogn)(logn)1/2​.

The constants are uniform in θ\thetaθ and nnn. This is the paper's (21): sup⁡θRnπ=O(⋅)\sup_\theta\mathcal R^{\pi}_n=O(\cdot)supθ​Rnπ​=O(⋅).

Milestones, in proof order

  1. Fact 1: JnD=nJDJ^D_n=nJ^DJnD​=nJD and JD≥mmin⁡{T,x/M}J^D\ge m\min\{T,x/M\}JD≥mmin{T,x/M} on the class.
  2. Lemma 1: the deterministic relaxation is solved by the fixed price pD=max⁡{pu,pc}p^D=\max\{p^u,p^c\}pD=max{pu,pc}.
  3. Lemma 2: Poisson deviation bounds at scale (log⁡n/rn)1/2(\log n/r_n)^{1/2}(logn/rn​)1/2.
  4. (A-27): a revenue lower bound that splits the loss into stage-wise terms and an overflow term.
  5. The per-stage revenue gap r(pD)−E r(p^i)≤C2(nΔn(i−1))−1/2r(p^D)-\mathbb E\,r(\hat p_i)\le C_2(n\Delta^{(i-1)}_n)^{-1/2}r(pD)−Er(p^​i​)≤C2​(nΔn(i−1)​)−1/2.
  6. (A-30): the stage-iii demand rate rarely exceeds the run-out rate.
  7. The overflow bound E[(Yn−nx)+]≤nC8(log⁡n)1/2naℓn−1\mathbb E[(Y_n-nx)^+]\le nC_8(\log n)^{1/2}n^{a_{\ell_n}-1}E[(Yn​−nx)+]≤nC8​(logn)1/2naℓn​​−1.
  8. (A-31): the revenue ratio before the exponents are evaluated.
  9. The rate estimate naℓn−1≤e n−1/2n^{a_{\ell_n}-1}\le e\,n^{-1/2}naℓn​​−1≤en−1/2.

Milestones 6–8 hold in the case λ(p‾;θ∗)≤x/T\lambda(\overline p;\theta^*)\le x/Tλ(p​;θ∗)≤x/T, the only case the paper's proof treats in detail.

Significance

Proposition 5 shows that with one unknown parameter, learning while earning reaches the n−1/2n^{-1/2}n−1/2 rate, up to logarithms. The learn-then-price policies of Propositions 1 and 3 stop learning after an initial phase and reach only n−1/4n^{-1/4}n−1/4 and n−1/3n^{-1/3}n−1/3. The paper leaves open whether the multi-parameter case attains the lower bound.

The analysis combines a continuous-time controlled Poisson model, an inventory constraint and a staged estimator, which also appear in later work on dynamic pricing with learning. A formal development would provide a time-changed Poisson demand model with random stage boundaries, a deterministic-relaxation benchmark, and concentration bounds stated for the scales this literature uses.

The result is proved in the paper, but parts of the proof are only sketched. The case λ(p‾;θ∗)>x/T\lambda(\overline p;\theta^*)>x/Tλ(p​;θ∗)>x/T is dismissed with "a similar result holds". The per-stage gap is obtained "by parallel reasoning". Display (A-30) has a typographical error in its threshold. To our knowledge, none of these results has been machine-checked.

Difficulty

The naive argument conditions each stage on its start time, as if that time were deterministic. It is not: the stage boundaries Λi=∑j≤inλ(p^j;θ∗)Δn(j)\Lambda_i=\sum_{j\le i}n\lambda(\hat p_j;\theta^*)\Delta^{(j)}_nΛi​=∑j≤i​nλ(p^​j​;θ∗)Δn(j)​ depend on all earlier observations, so every per-stage estimate needs the strong Markov property of the Poisson process at a random time. The inventory constraint makes the revenue a nonlinear function of the whole demand path. Bounding the loss therefore means controlling estimation error and overflow at the same time. The geometric stage lengths (20) are chosen so that the stage losses Δn(i)/(nΔn(i−1))1/2\Delta^{(i)}_n/(n\Delta^{(i-1)}_n)^{1/2}Δn(i)​/(nΔn(i−1)​)1/2 are all of the same order. That balance has to be checked exactly, including the rounding of ℓn\ell_nℓn​ to an integer.

Formalization scope

  • Poisson process. A structure on an arbitrary probability space: N(0)=0N(0)=0N(0)=0, monotone right-continuous paths, measurable marginals, Poisson increments, and independent increments over finite partitions. No process is published on the platform.
  • Class and family. Conditions on λ\lambdaλ are imposed on [p‾,p‾]∪{p∞}[\underline p,\overline p]\cup\{p_\infty\}[p​,p​]∪{p∞​}, the only prices a path uses. The inverse γ\gammaγ is Function.invFunOn. Θ\ThetaΘ is a nonempty closed interval of R\mathbb RR.
  • Assumption 3. As printed it cannot hold for d>sup⁡θλ(p;θ)d>\sup_\theta\lambda(p;\theta)d>supθ​λ(p;θ). It is read as an α\alphaα-Lipschitz map g(p,⋅):[0,∞)→Θg(p,\cdot):[0,\infty)\to\Thetag(p,⋅):[0,∞)→Θ that inverts λ(p;⋅)\lambda(p;\cdot)λ(p;⋅) on Θ\ThetaΘ. ggg is jointly measurable, so that estimates at random prices are random variables.
  • Selections. pu,pcp^u,p^cpu,pc are any measurable selections of the maximizer and minimizer; the statements hold for each.
  • Inventory. The inventory is ⌊nx⌋\lfloor nx\rfloor⌊nx⌋ units, and sales are capped cumulative counts.
  • Time change. Eq. (1) is applied stage by stage with random stage boundaries.
  • Typos. In Algorithm 3, "λ(pi,θ)\lambda(p_i,\theta)λ(pi​,θ)" is read as λ(p^i;θ)\lambda(\hat p_i;\theta)λ(p^​i​;θ) and "x/tx/tx/t" as x/Tx/Tx/T.
  • Stages. ℓn=⌈log⁡2log⁡n⌉\ell_n=\lceil\log_2\log n\rceilℓn​=⌈log2​logn⌉.
  • Integrals. Expectations are lower Lebesgue integrals of nonnegative quantities, converted to reals. The relaxation is a real supremum over measurable paths.
  • Asymptotics. The O(⋅)O(\cdot)O(⋅) is rendered with an explicit n0n_0n0​. The clause "asymptotically optimal" is omitted, since it needs the second half of Lemma 1.
  • Ruled out. Each of the following would trivialize the statement: removing the inventory cap, replacing the random stage boundaries by deterministic ones, fixing θ\thetaθ, letting CCC depend on θ\thetaθ, or using a non-measurable selection (whose expectation would be a junk value).

Contributions are welcome on every milestone. The Poisson process structure, its strong Markov property at stage boundaries, and Lemma 2 can be reused in other Poisson-demand pricing and queueing missions. Lemma 1 and Fact 1 are deterministic, and the rate estimate already has a local proof.

Selected references

  • O. Besbes and A. Zeevi, Dynamic Pricing Without Knowing the Demand Function: Risk Bounds and Near-Optimal Algorithms, Operations Research 57(6):1407–1420, 2009. https://doi.org/10.1287/opre.1080.0640
  • G. Gallego and G. van Ryzin, Optimal Dynamic Pricing of Inventories with Stochastic Demand over Finite Horizons, Management Science 40(8):999–1020, 1994. https://doi.org/10.1287/mnsc.40.8.999
  • K. Talluri and G. van Ryzin, The Theory and Practice of Revenue Management, Springer, 2005. https://doi.org/10.1007/b139000
15 thms2 active usersReviewed
🏆Completed
Operations ResearchProbabilityStatistics·Captain: mikedeng1

Dynamic Pricing Without Knowing the Demand Function: Risk Bounds and Near-Optimal Algorithms II: The Parametric Learn-then-Price Policy Has Regret at Most C(log n)^{1/2}/n^{1/3}Research Paper

Why learn demand while pricing?

A seller with a fixed inventory must choose prices before knowing how customers respond to them. A price that earns a high margin can sell too slowly; a price that sells quickly can exhaust stock before the selling season ends. Learning demand uses time and inventory, so the seller must account for the cost of its own experiments. This mission formalizes the parametric policy and regret bound of Besbes and Zeevi (2009), Proposition 3. Their setting differs from a finite-arm bandit: the permitted ordinary prices form an interval, customer arrivals are random, and the available stock caps total sales.

The paper first analyzes a nonparametric class with a slower upper bound in Proposition 1, then imposes a known finite-dimensional form on the unknown demand curve in Section 5. Proposition 3 shows how that additional information improves the regret rate for its Algorithm 2. The authors also prove a lower bound for a suitable parametric family in Proposition 4; that result is outside this mission's capstone, which concerns the performance guarantee of the concrete policy.

Market, demand, and policy

The selling horizon has length T>0T>0T>0, and the initial stock is x>0x>0x>0. An ordinary price ppp lies in [p‾,p‾][\underline p,\overline p][p​,p​], with 0<p‾<p‾0<\underline p<\overline p0<p​<p​. The special price p∞>0p_\infty>0p∞​>0 stops demand. At parameter θ∈Θ⊆Rk\theta\in\Theta\subseteq\mathbb R^kθ∈Θ⊆Rk, the demand rate is λ(p;θ)≥0\lambda(p;\theta)\ge0λ(p;θ)≥0. The parameter set Θ\ThetaΘ is nonempty, compact, and convex; kkk is positive. The seller knows the family λ(⋅;θ)\lambda(\cdot;\theta)λ(⋅;θ) but does not know the true parameter θ∗\theta^*θ∗.

For each parameter, demand is nonincreasing in the price and has an inverse γ(⋅;θ)\gamma(\cdot;\theta)γ(⋅;θ) on the attainable rate interval. The rate revenue is r(ℓ;θ)=ℓγ(ℓ;θ)r(\ell;\theta)=\ell\gamma(\ell;\theta)r(ℓ;θ)=ℓγ(ℓ;θ), a concave function of the rate. Every member of the family satisfies Assumption 1 with the same positive constants M,K‾,K‾,mM,\underline K,\overline K,mM,K​,K,m: demand is at most MMM, its price variation is at most K‾\overline KK times the price difference, its inverse is K‾−1\underline K^{-1}K​−1-Lipschitz, and some ordinary price earns revenue rate at least mmm. These conditions define the class in which the bound is uniform. See §§3–4.2 of the source.

Customer requests follow a unit-rate Poisson process NNN run on a demand-dependent clock. Under price p(t)p(t)p(t), cumulative requests at time ttt equal N(∫0tλ(p(s);θ∗) ds)N(\int_0^t\lambda(p(s);\theta^*)\,ds)N(∫0t​λ(p(s);θ∗)ds). The market of size nnn has stock nxnxnx and rate nλn\lambdanλ. Sales stop when stock runs out. The deterministic relaxation JnDJ_n^DJnD​ is the best integrated rate revenue over measurable price paths satisfying the expected-demand stock constraint, with the same scaling. The policy revenue JnπJ_n^\piJnπ​ is its expected sales revenue, and its relative regret is Rnπ=1−Jnπ/JnDR_n^\pi=1-J_n^\pi/J_n^DRnπ​=1−Jnπ​/JnD​. Fact 1 states JnD=nJDJ_n^D=nJ^DJnD​=nJD and gives a positive uniform lower bound on JDJ^DJD.

Assumption 2 selects kkk distinct ordinary test prices. Their mean rates identify the parameter through a Lipschitz inverse map ggg, and demand changes at most K‾2∥θ−θ′∥∞\overline K_2\|\theta-\theta'\|_\inftyK2​∥θ−θ′∥∞​ as the parameter varies. The square root of each test-price rate is differentiable on Θ\ThetaΘ. Algorithm 2 spends τn\tau_nτn​ time units testing these prices in equal subintervals, estimates each rate from its Poisson count increment, and sets θ^=g(d^)\widehat\theta=g(\widehat d)θ=g(d). It then posts p^=max⁡{pu(θ^),pc(θ^)}\widehat p=\max\{p^u(\widehat\theta),p^c(\widehat\theta)\}p​=max{pu(θ),pc(θ)}, where pup^upu maximizes revenue rate and pcp^cpc minimizes the distance of demand to x/Tx/Tx/T. It keeps that price until time TTT or stock-out. See §5.1–5.2.

Formalization targets

The capstone is the uniform bound in Proposition 3, equation (17). With τn≍n−1/3\tau_n\asymp n^{-1/3}τn​≍n−1/3, the mission asks for one constant C>0C>0C>0, independent of nnn, θ∗\theta^*θ∗, the optimizer choices, and the probability space, such that

∀θ∗∈Θ,Rnπ(x,T;θ∗)≤Clog⁡nn1/3(n≥2).\forall\theta^*\in\Theta,\qquad R_n^\pi(x,T;\theta^*)\le C\frac{\sqrt{\log n}}{n^{1/3}}\quad(n\ge2).∀θ∗∈Θ,Rnπ​(x,T;θ∗)≤Cn1/3logn​​(n≥2).

The intermediate quantitative target is equation (A-25), which retains the learning time:

Rnπ≤C3mD(τn+log⁡nnτn),mD=mmin⁡{T,x/M}.R_n^\pi\le\frac{C_3}{m^D}\left(\tau_n+\frac{\sqrt{\log n}}{\sqrt{n\tau_n}}\right),\qquad m^D=m\min\{T,x/M\}.Rnπ​≤mDC3​​(τn​+nτn​​logn​​),mD=mmin{T,x/M}.

The milestone list also carries Fact 1, the optimal deterministic price path and value from the first assertion of Lemma 1, both Poisson tails of Lemma 2, the pricing-phase revenue bound (A-17), the deterministic stability bounds (A-19), (A-22), and (A-23), and the parameter-estimation bound of Lemma 6. The paper's assertion that the policy is “asymptotically optimal” follows from the displayed regret estimate together with nonnegative regret; the displayed numerical estimate is the formal goal.

What the result provides

A finite-dimensional demand model lets the seller learn from a fixed set of kkk prices rather than probing an increasingly fine price grid. Proposition 3 quantifies the resulting revenue loss at O(log⁡n n−1/3)O(\sqrt{\log n}\,n^{-1/3})O(logn​n−1/3), compared with the paper's O(log⁡n n−1/4)O(\sqrt{\log n}\,n^{-1/4})O(logn​n−1/4) upper bound for its nonparametric learn-then-price policy. Both guarantees compare actual expected revenue with the full-information deterministic benchmark, including stock-out. These rates are proved in the article; the mission seeks machine-checked Lean proofs of the stated model, auxiliary bounds, and capstone. No such proof is claimed here.

The formalization would also provide reusable interfaces for a unit-rate Poisson counting process, deterministic inventory-constrained revenue optimization, measurable price selectors, and a random price chosen from normalized count observations. The later single-parameter policy in the paper uses related ideas but is a separate mission with different stages and a different rate.

Main difficulty

A rate estimate can be close to the true rate while the selected price changes between two different optimizers: the unconstrained revenue maximizer and the price that matches inventory to expected demand. A bound for only one optimizer does not control the maximum of the two. The learning price is random, and the number of requests in the pricing phase is evaluated at a random Poisson clock. Stock-out further couples the learning phase to the amount available for later sales. These features prevent a direct substitution of an ordinary parameter-estimation bound into the final revenue formula.

Formalization scope

Lean uses Fin(k)→R\mathrm{Fin}(k)\to\mathbb RFin(k)→R with its sup norm for parameter vectors, a finite positive off price, and a Poisson process on an arbitrary probability space. Each demand curve is defined on all real prices but is constrained on [p‾,p‾][\underline p,\overline p][p​,p​] and at p∞p_\inftyp∞​; the inverse and concavity conditions apply on the attainable rate interval. The deterministic benchmark is a real supremum over measurable, integrable admissible price paths. The class assumptions make its value nonempty and bounded. Policy revenue uses a nonnegative integral of pathwise revenue, avoiding a default zero from a nonintegrable signed expectation.

Inventory is measured in whole units, so the cap is ⌊nx⌋\lfloor nx\rfloor⌊nx⌋, equal to the paper's nxnxnx when that quantity is integral. Test-price observations remain uncapped in the formula for d^\widehat dd; if stock runs out during learning, capped later sales are already zero. The continuous optimizer choices are measurable selections satisfying the actual max/min properties for every member of Θ\ThetaΘ. The bound is uniform over these choices and over all unit-rate Poisson processes.

The printed Assumption 2(i)b asks for a solution to the test-rate equations for every vector in Rk\mathbb R^kRk. Such a solution cannot always lie in compact Θ\ThetaΘ, although the proof applies Assumption 2(ii) to θ^\widehat\thetaθ. Here ggg is a Lipschitz map into Θ\ThetaΘ that recovers each true parameter from its test-rate vector; it can be viewed as the unconstrained inverse followed by a nonexpansive projection in the paper's box example. This explicit convention is required to make the estimate and subsequent use of Assumption 2(ii) coherent. It is an interpretive repair of the printed condition.

The paper prints the regret bound for n≥1n\ge1n≥1, where log⁡1=0\sqrt{\log1}=0log1​=0. The exploration policy can lose revenue at n=1n=1n=1, so the formal theorem starts at n≥2n\ge2n≥2. The comparison τn≍n−1/3\tau_n\asymp n^{-1/3}τn​≍n−1/3 is encoded by positive lower and upper multipliers after a fixed threshold, with 0<τn≤T0<\tau_n\le T0<τn​≤T for all relevant market sizes. This admits the usual finite initial adjustments and leaves the bound uniform over the sequence. No fixed demand curve, known parameter, finite price grid, or uncapped sales model can satisfy this target by substitution.

Selected references

  • Omar Besbes and Assaf Zeevi, Dynamic Pricing Without Knowing the Demand Function: Risk Bounds and Near-Optimal Algorithms, Operations Research 57(6), 2009, authors' final manuscript revised December 16, 2007. DOI: 10.1287/opre.1080.0640.
13 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine Learning·Captain: mikedeng1

Online Convex Optimization in the Bandit Setting: Gradient Descent without a Gradient: Bandit Gradient Descent Has Expected Regret ≤ 3Cn^{5/6}∛(12dR/r)Research Paper

Motivation

In online convex optimization a decision maker picks points x1,x2,…,xnx_1,x_2,\dots,x_nx1​,x2​,…,xn​ in a convex set S⊆RdS\subseteq\mathbb R^dS⊆Rd, and after each choice pays ct(xt)c_t(x_t)ct​(xt​) for a convex cost function ctc_tct​ chosen in advance by an adversary. Performance is measured by regret: the total cost paid minus the cost of the best fixed point in hindsight. Zinkevich (ICML 2003) showed that projected gradient descent has regret O(n)O(\sqrt n)O(n​) when the whole function ctc_tct​, or at least its gradient at xtx_txt​, is revealed after each round.

In many applications only the number ct(xt)c_t(x_t)ct​(xt​) is observed: a seller sets a price and sees revenue, a router picks a path and sees its delay, an advertiser places a bid and sees the cost. This is the bandit setting. Flaxman, Kalai and McMahan (arXiv:cs/0408007, SODA 2005) gave the first simple algorithm for bandit convex optimization against an oblivious adversary with general bounded convex costs: a randomized gradient descent that estimates the gradient from a single function value, with expected regret O(n5/6)O(n^{5/6})O(n5/6). The same one-point gradient estimate is the starting point of later work on bandit convex optimization and on zeroth-order (derivative-free) stochastic optimization; see, e.g., Bubeck and Cesa-Bianchi's survey (arXiv:1204.5721, Ch. 6) and Hazan's textbook (arXiv:1909.05207, Ch. 6).

Timeline. Zinkevich (2003): O(n)O(\sqrt n)O(n​) regret with full gradient feedback. Kleinberg (NIPS 2004), independently: O(n3/4)O(n^{3/4})O(n3/4) bandit regret for Lipschitz costs by a different reduction. Flaxman, Kalai, McMahan (2004/2005): O(n5/6)O(n^{5/6})O(n5/6) for bounded convex costs and O(n3/4)O(n^{3/4})O(n3/4) for Lipschitz costs, with one function value per round. Later work (Bubeck, Lee, Eldan 2017 and others) reached O~(n)\tilde O(\sqrt n)O~(n​) with more complex algorithms.

Setting

Let B={x∈Rd:∣x∣≤1}\mathbb B=\{x\in\mathbb R^d : |x|\le1\}B={x∈Rd:∣x∣≤1} and S={x:∣x∣=1}\mathbb S=\{x : |x|=1\}S={x:∣x∣=1} be the closed unit ball and the unit sphere, d≥1d\ge1d≥1. The feasible set SSS is closed and convex with

rB⊆S⊆RB,r>0.r\mathbb B\subseteq S\subseteq R\mathbb B,\qquad r>0 .rB⊆S⊆RB,r>0.

The costs c1,c2,…c_1,c_2,\dotsc1​,c2​,… are fixed before play (an oblivious adversary); each is convex on SSS with ∣ct(x)∣≤C|c_t(x)|\le C∣ct​(x)∣≤C for x∈Sx\in Sx∈S, where C>0C>0C>0. For K⊆RdK\subseteq\mathbb R^dK⊆Rd, PK(z)P_K(z)PK​(z) is the nearest point of KKK to zzz.

The bandit gradient descent algorithm BGD(α,δ,ν)\mathrm{BGD}(\alpha,\delta,\nu)BGD(α,δ,ν) (Figure 1 of the paper) keeps an iterate yty_tyt​ with y1=0y_1=0y1​=0. At period ttt it draws a unit vector utu_tut​ uniformly from S\mathbb SS, independently of the past, plays

xt=yt+δut,x_t=y_t+\delta u_t ,xt​=yt​+δut​,

observes only ct(xt)c_t(x_t)ct​(xt​), and updates

yt+1=P(1−α)S(yt−ν ct(xt) ut).y_{t+1}=P_{(1-\alpha)S}\big(y_t-\nu\,c_t(x_t)\,u_t\big).yt+1​=P(1−α)S​(yt​−νct​(xt​)ut​).

The smoothed cost is c^t(x)=Ev∈B[ct(x+δv)]\hat c_t(x)=\mathbb E_{v\in\mathbb B}[c_t(x+\delta v)]c^t​(x)=Ev∈B​[ct​(x+δv)] with vvv uniform on B\mathbb BB. The expected regret after nnn rounds is

E[∑t=1nct(xt)]−min⁡x∈S∑t=1nct(x).\mathbb E\Big[\sum_{t=1}^n c_t(x_t)\Big]-\min_{x\in S}\sum_{t=1}^n c_t(x).E[t=1∑n​ct​(xt​)]−x∈Smin​t=1∑n​ct​(x).

Formalization targets

Goal: Theorem 1 (p. 8)

For every n≥(3Rd/2r)2n\ge(3Rd/2r)^2n≥(3Rd/2r)2, with ν=R/(Cn)\nu=R/(C\sqrt n)ν=R/(Cn​), δ=rR2d2/(12n)3\delta=\sqrt[3]{rR^2d^2/(12n)}δ=3rR2d2/(12n)​ and α=3Rd/(2rn)3\alpha=\sqrt[3]{3Rd/(2r\sqrt n)}α=33Rd/(2rn​)​,

E[∑t=1nct(xt)]−min⁡x∈S∑t=1nct(x)≤3Cn5/612 dRr3.\mathbb E\Big[\sum_{t=1}^n c_t(x_t)\Big]-\min_{x\in S}\sum_{t=1}^n c_t(x)\le 3Cn^{5/6}\sqrt[3]{\frac{12\,dR}{r}} .E[t=1∑n​ct​(xt​)]−x∈Smin​t=1∑n​ct​(x)≤3Cn5/63r12dR​​.

The paper prints the constant 3Cn5/6dR/r33Cn^{5/6}\sqrt[3]{dR/r}3Cn5/63dR/r​. Its own last step bounds the regret by a/δ+bδ/α+cαa/\delta+b\delta/\alpha+c\alphaa/δ+bδ/α+cα with a=RdCna=RdC\sqrt na=RdCn​, b=6Cn/rb=6Cn/rb=6Cn/r, c=2Cnc=2Cnc=2Cn, and with the stated δ\deltaδ and α\alphaα this equals 3abc3=3Cn5/612dR/r33\sqrt[3]{abc}=3Cn^{5/6}\sqrt[3]{12dR/r}33abc​=3Cn5/6312dR/r​. The goal states the bound the proof establishes.

Milestones

  1. Lemma 1 (p. 5): Eu∈S[f(x+δu)u]=δd∇f^(x)\mathbb E_{u\in\mathbb S}[f(x+\delta u)u]=\frac\delta d\nabla\hat f(x)Eu∈S​[f(x+δu)u]=dδ​∇f^​(x), the one-point gradient estimate.
  2. Lemma 2 (p. 6): gradient descent with conditionally unbiased gradient estimates of norm at most GGG has expected regret at most RGnRG\sqrt nRGn​ for η=R/(Gn)\eta=R/(G\sqrt n)η=R/(Gn​).
  3. Observations 1–3 (pp. 7–8): the comparator over (1−α)S(1-\alpha)S(1−α)S is within 2αCn2\alpha Cn2αCn of that over SSS; balls of radius αr\alpha rαr around (1−α)S(1-\alpha)S(1−α)S lie in SSS; on (1−α)S(1-\alpha)S(1−α)S the costs satisfy ∣ct(x)−ct(y)∣≤2Cαr∣x−y∣|c_t(x)-c_t(y)|\le\frac{2C}{\alpha r}|x-y|∣ct​(x)−ct​(y)∣≤αr2C​∣x−y∣.
  4. The played points are feasible (p. 8).
  5. Regret against the smoothed costs (p. 9): at most RdCn/δRdC\sqrt n/\deltaRdCn​/δ.
  6. Display (10) (p. 9): regret at most RdCn/δ+3δLn+2αCnRdC\sqrt n/\delta+3\delta Ln+2\alpha CnRdCn​/δ+3δLn+2αCn with L=2C/(αr)L=2C/(\alpha r)L=2C/(αr).

Companion items

The tuning identity (p. 9): a/δ+bδ/α+cα=3abc3a/\delta+b\delta/\alpha+c\alpha=3\sqrt[3]{abc}a/δ+bδ/α+cα=33abc​ at δ=a2/bc3\delta=\sqrt[3]{a^2/bc}δ=3a2/bc​, α=ab/c23\alpha=\sqrt[3]{ab/c^2}α=3ab/c2​, for a,b,c>0a,b,c>0a,b,c>0.

Theorem 2 (p. 9). If each ctc_tct​ is LLL-Lipschitz on SSS, then with ν=R/(Cn)\nu=R/(C\sqrt n)ν=R/(Cn​), δ=n−1/4RdCr/(3(Lr+C))\delta=n^{-1/4}\sqrt{RdCr/(3(Lr+C))}δ=n−1/4RdCr/(3(Lr+C))​, α=δ/r\alpha=\delta/rα=δ/r, and nnn large enough that δ<r\delta<rδ<r,

E[∑t=1nct(xt)]−min⁡x∈S∑t=1nct(x)≤2n3/43RdC(L+C/r).\mathbb E\Big[\sum_{t=1}^n c_t(x_t)\Big]-\min_{x\in S}\sum_{t=1}^n c_t(x)\le 2n^{3/4}\sqrt{3RdC\big(L+C/r\big)} .E[t=1∑n​ct​(xt​)]−x∈Smin​t=1∑n​ct​(x)≤2n3/43RdC(L+C/r)​.

Significance

The theorem shows that bandit feedback costs only a polynomial factor in regret for arbitrary bounded convex costs, with an algorithm that is gradient descent plus one random perturbation per round. Lemma 1 is the general tool behind it: an unbiased estimate of the gradient of a smoothed function from one function value. It is reused across zeroth-order optimization, bandit learning and stochastic approximation. Lemma 2 is a self-contained regret bound for projected gradient descent with noisy gradients, the shape in which Zinkevich's analysis is most often applied.

The results are proved in the paper and are standard. To the platform's knowledge none of them has a machine-checked proof. Related statements from Bubeck and Cesa-Bianchi's survey, for differentiable Lipschitz losses, are open on the platform, and two earlier formalizations of the textbook versions were disproved because of missing regularity or independence hypotheses. Formalizing this mission produces a checked one-point gradient identity for continuous (not differentiable) functions on the sphere, a measure-theoretic regret bound for stochastic projected gradient descent, and the corrected constant of Theorem 1.

Difficulty

The bandit algorithm itself is simple; the difficulty sits in Lemma 1 and in the measure theory around it. The paper derives Lemma 1 from Stokes' theorem, ∇∫δBf(x+v) dv=∫δSf(x+u)u∣u∣ du\nabla\int_{\delta\mathbb B}f(x+v)\,dv=\int_{\delta\mathbb S}f(x+u)\frac{u}{|u|}\,du∇∫δB​f(x+v)dv=∫δS​f(x+u)∣u∣u​du, and the ratio δ/d\delta/dδ/d of the volume to the surface area of a ball. Mathlib has neither this divergence identity on balls in Rd\mathbb R^dRd nor the explicit link between its spherical measure and the surface integral needed here. The identity must hold for functions that are merely continuous near the ball, since convex costs need not be differentiable.

The obvious first idea, differentiating under the integral sign in f^(x)=Ev[f(x+δv)]\hat f(x)=\mathbb E_v[f(x+\delta v)]f^​(x)=Ev​[f(x+δv)], fails because fff is not differentiable. The second obvious idea, applying Lemma 1 to ctc_tct​ as given, fails at the boundary of SSS, where ctc_tct​ is unconstrained. Lemma 2's conditional expectation E[gt∣xt]\mathbb E[g_t\mid x_t]E[gt​∣xt​] requires that the direction utu_tut​ be independent of the iterate yty_tyt​, and the integrability and measurability of the played points come from continuity of convex functions in the interior of SSS.

Formalization scope

Points are EuclideanSpace ℝ (Fin d) with d≥1d\ge1d≥1, and rounds are numbered 1,…,n1,\dots,n1,…,n. The costs are functions on Rd\mathbb R^dRd, and no statement assumes anything about them outside SSS: no global convexity, continuity, differentiability or Lipschitz bound. The standing model is part of the goal's hypotheses: SSS is convex with rB⊆S⊆RBr\mathbb B\subseteq S\subseteq R\mathbb BrB⊆S⊆RB, the costs are convex on SSS with values in [−C,C][-C,C][−C,C] there, and C>0C>0C>0. Additions and conventions:

  • SSS is assumed closed. The paper's projection oracle PS(x)=arg⁡min⁡z∈S∣x−z∣P_S(x)=\arg\min_{z\in S}|x-z|PS​(x)=argminz∈S​∣x−z∣ presupposes that the minimum is attained.
  • The directions utu_tut​ are measurable, independent, and each uniform on S\mathbb SS (the published uniformSphere d). Uniform marginals alone are not enough.
  • The BGD run is a relation (IsBGDRun) required for every outcome. Projection uses the published predicate IsNearestPoint, and regret the published pseudoRegret, whose minimum is an infimum over the subtype; it is attained in every use here.
  • Lemma 1 adds continuity of fff on an open set containing x+δBx+\delta\mathbb Bx+δB. As printed ("for any function fff") it is false. The application in Theorem 1 supplies this hypothesis.
  • The smoothed-regret and (10) milestones use the strict δ<αr\delta<\alpha rδ<αr (the page has δ/r≤α\delta/r\le\alphaδ/r≤α), which keeps every smoothing ball in the interior of SSS. They use α≤1\alpha\le1α≤1 in place of α<1\alpha<1α<1; Theorem 1's α\alphaα equals 111 at n=(3Rd/2r)2n=(3Rd/2r)^2n=(3Rd/2r)2.
  • Theorem 2's "for nnn sufficiently large" is the hypothesis δ<r\delta<rδ<r, the only place the proof uses it.

A formalization that assumed integrability of the costs along the run, assumed differentiable or globally Lipschitz costs, dropped the independence of the directions, or stated ∇f^\nabla\hat f∇f^​ through Mathlib's gradient (which is 000 off differentiability) would trivialize or change the result. All of these are ruled out.

Needed infrastructure: the divergence identity for the ball average (Lemma 1), conditional expectation given σ(xt)\sigma(x_t)σ(xt​) for vector-valued variables, continuity of convex functions on the interior of a convex set, and nearest-point projection onto closed convex sets. The first two are reusable well beyond this mission. Contributions toward Lemma 1 in particular are welcome.

Selected references

  • A. D. Flaxman, A. T. Kalai, H. B. McMahan, Online convex optimization in the bandit setting: gradient descent without a gradient, SODA 2005; arXiv:cs/0408007v1, 2004. https://arxiv.org/abs/cs/0408007
  • M. Zinkevich, Online convex programming and generalized infinitesimal gradient ascent, ICML 2003. https://www.cs.cmu.edu/~maz/publications/ICML03.pdf
  • R. Kleinberg, Nearly tight bounds for the continuum-armed bandit problem, NIPS 2004. https://papers.nips.cc/paper/2634-nearly-tight-bounds-for-the-continuum-armed-bandit-problem
  • S. Bubeck, N. Cesa-Bianchi, Regret analysis of stochastic and nonstochastic multi-armed bandit problems, Foundations and Trends in ML, 2012. https://arxiv.org/abs/1204.5721
  • E. Hazan, Introduction to online convex optimization, 2nd ed., 2019. https://arxiv.org/abs/1909.05207
  • S. Bubeck, Y. T. Lee, R. Eldan, Kernel-based methods for bandit convex optimization, STOC 2017. https://arxiv.org/abs/1607.03084
12 thms2 active usersReviewed
🏆Completed
Machine LearningProbabilityStatistics·Captain: mikedeng1

Kullback–Leibler Upper Confidence Bounds for Optimal Sequential Allocation I: kl-UCB Draws a Suboptimal Arm log(T)/d(μ_a, μ*) + O(√log T) Times in One-Parameter Exponential FamiliesResearch Paper

Motivation

In a stochastic multi-armed bandit a player repeatedly chooses one of KKK distributions ("arms") and observes a reward drawn from it; the goal is to collect as much reward as possible, which amounts to pulling suboptimal arms as rarely as possible. Lai and Robbins (1985) showed that any reasonable strategy must pull a suboptimal arm aaa at least log⁡T/KL\log T / \mathrm{KL}logT/KL times up to horizon TTT, where KL\mathrm{KL}KL is a Kullback–Leibler divergence between arm aaa and the best arm, and Burnetas and Katehakis (1996) extended the bound to general models. Strategies matching this rate are called asymptotically optimal.

The popular UCB algorithms of Auer, Cesa-Bianchi and Fischer (2002) use Hoeffding-type confidence bounds and are not asymptotically optimal outside special cases. Cappé, Garivier, Maillard, Munos and Stoltz (Ann. Statist. 41(3), 2013; arXiv:1210.1136) analyse kl-UCB, which replaces the Hoeffding radius by a Kullback–Leibler confidence region, and prove a finite-horizon bound whose leading term is exactly the Lai–Robbins constant. This mission formalizes that result for one-parameter exponential families (Theorem 1), the main-text steps of its proof skeleton, and its two corollaries for bounded rewards.

Timeline: Lai and Robbins (1985) lower bound and asymptotically optimal index policies; Agrawal (1995) sample-mean based index policies; Auer, Cesa-Bianchi and Fischer (2002) finite-time analysis of UCB1; Garivier and Cappé (2011) kl-UCB for bounded rewards; Cappé et al. (2013) the unified analysis formalized here.

Setting

A canonical exponential family D={νθ:θ∈Θ}\mathcal D = \{\nu_\theta : \theta\in\Theta\}D={νθ​:θ∈Θ} is given by a dominating measure ρ\rhoρ on R\mathbb RR and a function bbb, with densities dνθdρ(x)=exp⁡(xθ−b(θ))\frac{d\nu_\theta}{d\rho}(x) = \exp(x\theta - b(\theta))dρdνθ​​(x)=exp(xθ−b(θ)). The parameter set Θ\ThetaΘ is the natural parameter space {θ:∫exθ dρ(x)<∞}\{\theta : \int e^{x\theta}\,d\rho(x) < \infty\}{θ:∫exθdρ(x)<∞}, assumed to be an open interval (the family is regular), and bbb is twice differentiable. The mean of νθ\nu_\thetaνθ​ is b˙(θ)\dot b(\theta)b˙(θ), an increasing function, so νθ\nu_\thetaνθ​ is determined by its mean μ\muμ in the open interval I=b˙(Θ)=(μ−,μ+)I = \dot b(\Theta) = (\mu_-,\mu_+)I=b˙(Θ)=(μ−​,μ+​). The divergence (11) is

d(μ,μ′)=KL(νb˙−1(μ),νb˙−1(μ′))=(b˙−1(μ)−b˙−1(μ′))μ−b(b˙−1(μ))+b(b˙−1(μ′)),d(\mu,\mu') = \mathrm{KL}(\nu_{\dot b^{-1}(\mu)},\nu_{\dot b^{-1}(\mu')}) = (\dot b^{-1}(\mu)-\dot b^{-1}(\mu'))\mu - b(\dot b^{-1}(\mu)) + b(\dot b^{-1}(\mu')),d(μ,μ′)=KL(νb˙−1(μ)​,νb˙−1(μ′)​)=(b˙−1(μ)−b˙−1(μ′))μ−b(b˙−1(μ))+b(b˙−1(μ′)),

extended by continuity to the closure Iˉ=[μ−,μ+]\bar I = [\mu_-,\mu_+]Iˉ=[μ−​,μ+​], possibly with the value +∞+\infty+∞.

There are K≥2K\ge2K≥2 arms with laws νθ1,…,νθK∈D\nu_{\theta_1},\dots,\nu_{\theta_K}\in\mathcal Dνθ1​​,…,νθK​​∈D and means μ1,…,μK\mu_1,\dots,\mu_Kμ1​,…,μK​; μ⋆=max⁡aμa\mu^\star = \max_a \mu_aμ⋆=maxa​μa​. At each round t≥1t\ge1t≥1 the player picks an arm AtA_tAt​ based on the past and receives a reward drawn from νAt\nu_{A_t}νAt​​. Na(t)N_a(t)Na​(t) is the number of pulls of arm aaa in rounds 1,…,t1,\dots,t1,…,t, and μ^a(t)\hat\mu_a(t)μ^​a​(t) the mean of the rewards obtained from arm aaa so far.

kl-UCB (Algorithm 2) with a nondecreasing exploration function fff pulls each arm once and then, for t≥Kt\ge Kt≥K, pulls an arm maximizing the index

Ua(t)=sup⁡{μ∈Iˉ:d(μ^a(t),μ)≤f(t)Na(t)}.(12)U_a(t) = \sup\Bigl\{\mu\in\bar I : d(\hat\mu_a(t),\mu) \le \frac{f(t)}{N_a(t)}\Bigr\}. \tag{12}Ua​(t)=sup{μ∈Iˉ:d(μ^​a​(t),μ)≤Na​(t)f(t)​}.(12)

Formalization targets

Goal: Theorem 1 (p. 14)

With f(t)=log⁡t+3log⁡log⁡tf(t) = \log t + 3\log\log tf(t)=logt+3loglogt for t≥3t\ge3t≥3 and f(1)=f(2)=f(3)f(1)=f(2)=f(3)f(1)=f(2)=f(3), for every suboptimal arm aaa and every horizon T≥3T\ge3T≥3,

E[Na(T)]≤log⁡Td(μa,μ⋆)+22πσa,⋆2(d′(μa,μ⋆))2(d(μa,μ⋆))3log⁡T+3log⁡log⁡T+(4e+3d(μa,μ⋆))log⁡log⁡T+8σa,⋆2(d′(μa,μ⋆)d(μa,μ⋆))2+6,\mathbb E[N_a(T)] \le \frac{\log T}{d(\mu_a,\mu^\star)} + 2\sqrt{\frac{2\pi\sigma^2_{a,\star}(d'(\mu_a,\mu^\star))^2}{(d(\mu_a,\mu^\star))^3}}\sqrt{\log T+3\log\log T} + \Bigl(4e+\frac{3}{d(\mu_a,\mu^\star)}\Bigr)\log\log T + 8\sigma^2_{a,\star}\Bigl(\frac{d'(\mu_a,\mu^\star)}{d(\mu_a,\mu^\star)}\Bigr)^2 + 6,E[Na​(T)]≤d(μa​,μ⋆)logT​+2(d(μa​,μ⋆))32πσa,⋆2​(d′(μa​,μ⋆))2​​logT+3loglogT​+(4e+d(μa​,μ⋆)3​)loglogT+8σa,⋆2​(d(μa​,μ⋆)d′(μa​,μ⋆)​)2+6,

where σa,⋆2=max⁡{Var(νθ):μa≤E(νθ)≤μ⋆}\sigma^2_{a,\star} = \max\{\mathrm{Var}(\nu_\theta) : \mu_a\le \mathrm E(\nu_\theta)\le\mu^\star\}σa,⋆2​=max{Var(νθ​):μa​≤E(νθ​)≤μ⋆} and d′d'd′ is the derivative in the first argument.

Milestones

  1. The decomposition (5) of the event {At+1=a}\{A_{t+1}=a\}{At+1​=a} (p. 9).
  2. The split of E[Na(T)]\mathbb E[N_a(T)]E[Na​(T)] after (7) (p. 9).
  3. The passage to local times (8) (pp. 9–10): the overestimation term is bounded by ∑n=1T−KP{ν^a,n∈Cμ†,f(T)/n}\sum_{n=1}^{T-K}\mathbb P\{\hat\nu_{a,n}\in\mathcal C_{\mu^\dagger,f(T)/n}\}∑n=1T−K​P{ν^a,n​∈Cμ†,f(T)/n​}, a sum over fixed sample sizes.
  4. The general bound (10) with n0n_0n0​ of (9) (p. 10).
  5. The deviation bound (13) for an empirical mean with a random number of summands (p. 14).
  6. Lemma 1 (p. 17): the moment-generating function of a distribution on [0,1][0,1][0,1] is dominated by Bernoulli and Gaussian ones.

Companions

Corollary 1 (p. 17, kl-UCB with the Bernoulli divergence for arbitrary rewards in [0,1][0,1][0,1]) and Corollary 2 (p. 18, UCB with radius f(t)/(2Na(t))\sqrt{f(t)/(2N_a(t))}f(t)/(2Na​(t))​), stated as separate theorems.

Significance

Theorem 1 shows that kl-UCB is asymptotically optimal in every regular one-parameter exponential family (Bernoulli, Poisson, Gaussian with known variance, exponential, Gamma with known shape), and it does so with an explicit bound valid at every horizon, not only in the limit. Corollary 2 improves the constants of the classical UCB1 analysis, and Corollary 1 shows that the Bernoulli kl-UCB index is uniformly better than UCB for all bounded rewards.

The result is proved in the paper; its proofs are in the supplemental article (DOI 10.1214/13-AOS1119SUPP, Appendix A), not in the main text. No machine-checked version exists. The platform has a formal analysis of a Bernoulli KL-UCB variant with another exploration function (Lattimore–Szepesvári's Theorem 10.6), which is a different statement. The formalization adds the general exponential-family index, the random-sample-size deviation bound (13) and the full finite-time constant.

Difficulty

The obvious argument bounds P{μ⋆≥Ua⋆(t)}\mathbb P\{\mu^\star \ge U_{a^\star}(t)\}P{μ⋆≥Ua⋆​(t)} by a union bound over the possible values of Na⋆(t)N_{a^\star}(t)Na⋆​(t), which costs a factor ttt and destroys the logarithmic rate. The deviation bound (13) has to control an empirical mean whose number of summands is chosen by the algorithm itself, losing only a factor e⌈εlog⁡t⌉e\lceil\varepsilon\log t\rceile⌈εlogt⌉ over the fixed-sample Chernoff bound; this is the step that needs the strategy to be non-anticipating. The second-order terms depend on the curvature of ddd between μa\mu_aμa​ and μ⋆\mu^\starμ⋆, measured by σa,⋆2\sigma^2_{a,\star}σa,⋆2​ and d′d'd′, and the explicit constants must be tracked through every step. The empirical mean can be an endpoint of Iˉ\bar IIˉ (Bernoulli rewards at small sample sizes), where ddd is only defined as a limit.

Formalization scope

The rewards are a stack Xa,kX_{a,k}Xa,k​ (the (k+1)(k+1)(k+1)-st reward of arm aaa), mutually independent and i.i.d. per arm, the representation of §2.2; this is the platform's RegretBandits.Stochastic.IsStochasticBandit, and the law of Xa,0X_{a,0}Xa,0​ is pinned to νθa\nu_{\theta_a}νθa​​. Arms are Fin K. The exponential family is the platform's OptimalBAI.OptProportions.ExpFamily (with b¨>0\ddot b>0b¨>0, i.e. strict convexity, which the page derives); the natural-parameter-space condition is a separate hypothesis. A run of kl-UCB is a pathwise predicate: rounds 1,…,K1,\dots,K1,…,K pull every arm once and later rounds pull an argmax of the index, ties broken by any rule. The arm choices are measurable and non-anticipating (a measurable function of the arms and rewards already observed). The index (12) is a real supremum over a set containing the empirical mean, so it is never a junk value, and the divergence at an empirical mean on the boundary of Iˉ\bar IIˉ is computed in [0,+∞][0,+\infty][0,+∞]. Statement (13) is made for t≥2t\ge2t≥2: at t=1t=1t=1 its printed right-hand side is 000.

A trivializing formalization is ruled out: the run predicate forces both initialization and argmax, the index sets are nonempty and bounded, the reward stack is independent under the probability measure, and the Bernoulli divergence is never evaluated at 000 or 111.

A complete development needs exponential-family calculus (convex conjugate of bbb, continuity of ddd up to the boundary), Chernoff bounds for exponential families, a peeling/maximal inequality for random sample sizes, and the counting arguments of §3.1. The deviation bound and the counting arguments are reusable for every index policy on the platform.

Selected references

  • O. Cappé, A. Garivier, O.-A. Maillard, R. Munos, G. Stoltz, Kullback–Leibler upper confidence bounds for optimal sequential allocation, Ann. Statist. 41(3):1516–1541, 2013. https://doi.org/10.1214/13-AOS1119 ; arXiv:1210.1136v4, https://arxiv.org/abs/1210.1136
  • T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Adv. Appl. Math. 6:4–22, 1985. https://doi.org/10.1016/0196-8858(85)90002-8
  • P. Auer, N. Cesa-Bianchi, P. Fischer, Finite-time analysis of the multiarmed bandit problem, Mach. Learn. 47:235–256, 2002. https://doi.org/10.1023/A:1013689704352
  • A. Garivier, O. Cappé, The KL-UCB algorithm for bounded stochastic bandits and beyond, COLT 2011. https://arxiv.org/abs/1102.2490
  • R. Agrawal, Sample mean based index policies with O(log n) regret for the multi-armed bandit problem, Adv. Appl. Probab. 27:1054–1078, 1995. https://doi.org/10.2307/1427934
12 thms2 active usersReviewed
🏆Completed
Machine LearningOperations Research·Captain: mikedeng1

Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems I: Pseudo-Regret of (α, ψ)-UCBTextbook

Motivation

The stochastic multi-armed bandit is the basic model of sequential decisions under uncertainty with partial feedback: a forecaster repeatedly picks one of KKK options and observes only the reward of the option it picked. It models clinical trials, ad placement, routing and dynamic pricing, and is the building block of many reinforcement-learning algorithms. The question is how much reward is lost, compared with always playing the best option, through having to learn which option is best.

Chapter 2 of Bubeck and Cesa-Bianchi's monograph (arXiv:1204.5721v2) answers this for upper confidence bound (UCB) strategies.

  • Lai and Robbins (1985) introduced upper confidence bounds and proved that the number of pulls of a suboptimal arm must grow at least logarithmically, with an explicit constant, for consistent strategies (doi:10.1016/0196-8858(85)90002-8).
  • Agrawal (1995) gave simpler sample-mean-based index policies with logarithmic regret (doi:10.2307/1427934).
  • Auer, Cesa-Bianchi and Fischer (2002) gave the finite-time analysis of UCB1 for bounded rewards (doi:10.1023/A:1013689704352).
  • Bubeck and Cesa-Bianchi (2012) present the (α,ψ)(\alpha,\psi)(α,ψ)-UCB family, whose analysis needs only a bound ψ\psiψ on the cumulant generating function of the rewards, and the Lai–Robbins lower bound for Bernoulli rewards.

Setting

There are K≥2K\ge2K≥2 arms. Arm iii has an unknown reward distribution νi\nu_iνi​ with mean μi\mu_iμi​. At each round t=1,2,…t=1,2,\dotst=1,2,… the forecaster selects an arm ItI_tIt​ based on the past and receives a reward drawn from νIt\nu_{I_t}νIt​​, independently of the past. Write μ∗=max⁡iμi\mu^*=\max_i\mu_iμ∗=maxi​μi​, Δi=μ∗−μi\Delta_i=\mu^*-\mu_iΔi​=μ∗−μi​ for the gap of arm iii, and Ti(n)T_i(n)Ti​(n) for the number of times arm iii is selected in rounds 1,…,n1,\dots,n1,…,n. The pseudo-regret is

R‾n=nμ∗−E∑t=1nμIt=∑i=1KΔi E Ti(n).\overline R_n=n\mu^*-\mathbb E\sum_{t=1}^n\mu_{I_t}=\sum_{i=1}^K\Delta_i\,\mathbb E\,T_i(n).Rn​=nμ∗−Et=1∑n​μIt​​=i=1∑K​Δi​ETi​(n).

Moment condition (2.2). There is a convex ψ:R→R\psi:\mathbb R\to\mathbb Rψ:R→R with ln⁡E eλ(X−EX)≤ψ(λ)\ln\mathbb E\,e^{\lambda(X-\mathbb EX)}\le\psi(\lambda)lnEeλ(X−EX)≤ψ(λ) and ln⁡E eλ(EX−X)≤ψ(λ)\ln\mathbb E\,e^{\lambda(\mathbb EX-X)}\le\psi(\lambda)lnEeλ(EX−X)≤ψ(λ) for all λ≥0\lambda\ge0λ≥0 and every arm's reward XXX. Its Legendre–Fenchel transform is ψ∗(ε)=sup⁡λ∈R(λε−ψ(λ))\psi^*(\varepsilon)=\sup_{\lambda\in\mathbb R}(\lambda\varepsilon-\psi(\lambda))ψ∗(ε)=supλ∈R​(λε−ψ(λ)). For [0,1][0,1][0,1] rewards one may take ψ(λ)=λ2/8\psi(\lambda)=\lambda^2/8ψ(λ)=λ2/8, for which ψ∗(ε)=2ε2\psi^*(\varepsilon)=2\varepsilon^2ψ∗(ε)=2ε2.

(α,ψ)(\alpha,\psi)(α,ψ)-UCB. With μ^i,s\hat\mu_{i,s}μ^​i,s​ the mean of the first sss rewards of arm iii, at round ttt select

It∈argmax⁡i[μ^i,Ti(t−1)+(ψ∗)−1(αln⁡tTi(t−1))].I_t\in\operatorname*{argmax}_{i}\Big[\hat\mu_{i,T_i(t-1)}+(\psi^*)^{-1}\Big(\frac{\alpha\ln t}{T_i(t-1)}\Big)\Big].It​∈iargmax​[μ^​i,Ti​(t−1)​+(ψ∗)−1(Ti​(t−1)αlnt​)].

Formalization targets

Goal: Theorem 2.1 (p. 11)

If the rewards satisfy (2.2), then (α,ψ)(\alpha,\psi)(α,ψ)-UCB with α>2\alpha>2α>2 satisfies, for every nnn,

R‾n≤∑i:Δi>0Δi(αln⁡nψ∗(Δi/2)+αα−2).\overline R_n\le\sum_{i:\Delta_i>0}\Delta_i\Big(\frac{\alpha\ln n}{\psi^*(\Delta_i/2)}+\frac{\alpha}{\alpha-2}\Big).Rn​≤i:Δi​>0∑​Δi​(ψ∗(Δi​/2)αlnn​+α−2α​).

This is the bound the book's proof establishes. The printed statement has α/(α−2)\alpha/(\alpha-2)α/(α−2) in place of Δi α/(α−2)\Delta_i\,\alpha/(\alpha-2)Δi​α/(α−2); see Formalization scope.

Milestones

  • The decomposition R‾n=∑iΔi E Ti(n)\overline R_n=\sum_i\Delta_i\,\mathbb E\,T_i(n)Rn​=∑i​Δi​ETi​(n) (p. 9).
  • The Cramér–Chernoff bound (2.3): P(μi−μ^i,s>ε)≤e−sψ∗(ε)\mathbb P(\mu_i-\hat\mu_{i,s}>\varepsilon)\le e^{-s\psi^*(\varepsilon)}P(μi​−μ^​i,s​>ε)≤e−sψ∗(ε).
  • The three-event lemma (2.5)–(2.7) from the proof of Theorem 2.1.
  • The bounded-reward bound (2.4): R‾n≤∑i:Δi>0(2αΔiln⁡n+αα−2)\overline R_n\le\sum_{i:\Delta_i>0}\big(\frac{2\alpha}{\Delta_i}\ln n+\frac{\alpha}{\alpha-2}\big)Rn​≤∑i:Δi​>0​(Δi​2α​lnn+α−2α​).
  • The comparison (2.8): 2(p−q)2≤kl(p,q)≤(p−q)2/(q(1−q))2(p-q)^2\le\mathrm{kl}(p,q)\le(p-q)^2/(q(1-q))2(p−q)2≤kl(p,q)≤(p−q)2/(q(1−q)).
  • Theorem 2.2: for every strategy with E Ti(n)=o(na)\mathbb E\,T_i(n)=o(n^a)ETi​(n)=o(na) on all Bernoulli instances, lim inf⁡nR‾n/ln⁡n≥∑i:Δi>0Δi/kl(μi,μ∗)\liminf_n\overline R_n/\ln n\ge\sum_{i:\Delta_i>0}\Delta_i/\mathrm{kl}(\mu_i,\mu^*)liminfn​Rn​/lnn≥∑i:Δi​>0​Δi​/kl(μi​,μ∗).

Significance

Theorem 2.1 says that the cost of learning grows only logarithmically in the horizon, with a constant set by how well each suboptimal arm can be told apart from the best one. Theorem 2.2 shows that, up to constants, this cannot be improved: for Bernoulli rewards any strategy that is good on every instance must pay ln⁡n\ln nlnn per suboptimal arm, with constant Δi/kl(μi,μ∗)\Delta_i/\mathrm{kl}(\mu_i,\mu^*)Δi​/kl(μi​,μ∗). By (2.8) this constant is at least of order 1/Δi1/\Delta_i1/Δi​, matching (2.4). Together they are the template for the analysis of most optimistic algorithms: KL-UCB, linear and contextual UCB, and UCB-style reinforcement learning.

All results of the chapter are classical and proved on paper. This mission makes them machine-checked in a common model. That model has an explicit pseudo-regret, an explicit (generalized) inverse of ψ∗\psi^*ψ∗, an explicit initialization rule for the algorithm, and a randomized-strategy model for the lower bound. Later chapters of the series and later papers on optimistic algorithms can build on it.

Difficulty

The deterministic part of the upper bound is short, so the main difficulty is probabilistic. The sample mean μ^i,Ti(t−1)\hat\mu_{i,T_i(t-1)}μ^​i,Ti​(t−1)​ is taken over a random number of samples that depends on the algorithm's past. A Chernoff bound for a fixed sample size does not apply to it directly. The proof needs a union bound over all possible sample sizes, together with the representation in which the sss-th reward of each arm is a fixed random variable. Summing the resulting tail t1−αt^{1-\alpha}t1−α over rounds is where α>2\alpha>2α>2 enters.

The lower bound needs a change-of-measure argument between two Bernoulli instances, applied to a forecaster that may be randomized and never knows the horizon. Expressing "the forecaster cannot distinguish the instances" requires the law of the whole interaction under two environments.

Formalization scope

Model. Arms are Fin K with 2≤K2\le K2≤K, rounds are 1,2,…1,2,\dots1,2,…, and the natural logarithm is used. Rewards are a stack: Xi,kX_{i,k}Xi,k​ is the reward of the (k+1)(k+1)(k+1)-st pull of arm iii, all mutually independent, identically distributed per arm. For every strategy this gives the same law of arms and rewards as the book's protocol. μ∗\mu^*μ∗, Δi\Delta_iΔi​ and Ti(t)T_i(t)Ti​(t) are the published definitions of ImprovedLinBandits.UCBDelta.armModel. The pseudo-regret is (2.1), nμ∗−E∑tμItn\mu^*-\mathbb E\sum_t\mu_{I_t}nμ∗−E∑t​μIt​​, not the expected regret. The arms played are measurable random variables, so that every expectation is genuine.

ψ∗\psi^*ψ∗ and its inverse. ψ∗\psi^*ψ∗ takes values in the extended reals. (ψ∗)−1(y)=inf⁡{ε≥0:ψ∗(ε)≥y}(\psi^*)^{-1}(y)=\inf\{\varepsilon\ge0:\psi^*(\varepsilon)\ge y\}(ψ∗)−1(y)=inf{ε≥0:ψ∗(ε)≥y}.

Algorithm. The index is undefined while Ti(t−1)=0T_i(t-1)=0Ti​(t−1)=0. Unplayed arms are therefore played first, so each arm is played once in rounds 1,…,K1,\dots,K1,…,K. Ties are broken arbitrarily.

Corrected misprint. The book prints Theorem 2.1 with constant term α/(α−2)\alpha/(\alpha-2)α/(α−2). Its proof yields Δi α/(α−2)\Delta_i\,\alpha/(\alpha-2)Δi​α/(α−2) (bound on E Ti(n)\mathbb E\,T_i(n)ETi​(n) times Δi\Delta_iΔi​). The printed form is false for Gaussian rewards with large gaps. The goal states the proof's version. (2.4) is correct as printed because Δi≤1\Delta_i\le1Δi​≤1.

Constants. No O(·) appears in the chapter's statements. The constants are the book's: α/(α−2)\alpha/(\alpha-2)α/(α−2) in Theorem 2.1 and (2.4), and the factor 222 in (2.4) and (2.8).

Conventions made explicit.

  1. ψ(λ)≥ψ(0)\psi(\lambda)\ge\psi(0)ψ(λ)≥ψ(0) for λ≤0\lambda\le0λ≤0. Condition (2.2) constrains ψ\psiψ only on λ≥0\lambda\ge0λ≥0, while ψ∗\psi^*ψ∗ takes the supremum over all of R\mathbb RR. Without this convention, (2.3) and Theorem 2.1 are false (for example ψ(λ)=cλ+λ2/8\psi(\lambda)=c\lambda+\lambda^2/8ψ(λ)=cλ+λ2/8 with large ccc).
  2. ψ∗\psi^*ψ∗ is finite on [0,∞)[0,\infty)[0,∞).
  3. ψ∗(Δi/2)>0\psi^*(\Delta_i/2)>0ψ∗(Δi​/2)>0 for suboptimal arms.
  4. ε≥0\varepsilon\ge0ε≥0 in (2.3).
  5. q∈(0,1)q\in(0,1)q∈(0,1) in (2.8).

Conventions 1–3 hold for every ψ\psiψ the book uses.

Theorem 2.2. The forecaster is a measurable rule from the history and a fresh uniform seed to an arm. It does not depend on the horizon. Consistency is required on every Bernoulli instance, every suboptimal arm and every a>0a>0a>0. A term with μ∗=1\mu^*=1μ∗=1 (kl=+∞\mathrm{kl}=+\inftykl=+∞) is 000, and the lim inf⁡\liminfliminf is taken in the extended reals. The book proves only K=2K=2K=2.

Ruled out. The algorithm cannot see unplayed rewards (its index uses only the sample means of rewards already received). Non-measurable arm choices, which would make the expectations vanish in Lean, are excluded. Consistency cannot be assumed only on the instance of the conclusion.

Needed infrastructure. Cramér–Chernoff bounds for sums of i.i.d. variables, Hoeffding's lemma, the stack representation of bandit interactions, and the divergence decomposition for randomized strategies. All are reusable beyond this mission, and contributions of any of them are welcome.

Selected references

  • S. Bubeck, N. Cesa-Bianchi, Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems, Foundations and Trends in Machine Learning 5(1), 2012. arXiv:1204.5721v2, doi:10.1561/2200000024
  • T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6, 1985. doi:10.1016/0196-8858(85)90002-8
  • R. Agrawal, Sample mean based index policies with O(log n) regret for the multi-armed bandit problem, Advances in Applied Probability 27, 1995. doi:10.2307/1427934
  • P. Auer, N. Cesa-Bianchi, P. Fischer, Finite-time analysis of the multiarmed bandit problem, Machine Learning 47, 2002. doi:10.1023/A:1013689704352
12 thms2 active usersReviewed
🏆Completed
Machine LearningOperations ResearchStatistics·Captain: mikedeng1

Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems III: Contextual Bandits and the Banditron Mistake BoundTextbook

Motivation

In many sequential decision problems the learner sees side information before acting. A news site chooses an article for a visitor whose history and location it knows; an ad server chooses an advertisement for a query. Only the reward of the chosen action is observed. These are contextual bandit problems, and Chapter 4 of Bubeck and Cesa-Bianchi's monograph Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems (arXiv:1204.5721v2) surveys several of their formal versions. In a contextual problem the learner is compared with the best policy, a map from contexts to arms, rather than with the best single arm.

This mission covers three of the chapter's models. The first marks each round with a context from a finite set. In the second, NNN experts give advice, as in prediction with expert advice. The third is the bandit multiclass problem: a linear classifier predicts one of KKK labels and then learns only whether its prediction was right. The goal is the mistake bound of the Banditron (Kakade, Shalev-Shwartz and Tewari, ICML 2008). The bound shows that one bit of feedback per round suffices to compete with every linear classifier, at regret O(n2/3)O(n^{2/3})O(n2/3).

Setting

There are K≥2K \ge 2K≥2 arms (or labels) {1,…,K}\{1,\dots,K\}{1,…,K} and rounds t=1,…,nt = 1, \dots, nt=1,…,n.

Adversarial losses. At round ttt an adversary assigns losses ℓi,t∈[0,1]\ell_{i,t} \in [0,1]ℓi,t​∈[0,1] to the arms and may adapt to the forecaster's past plays I1,…,It−1I_1, \dots, I_{t-1}I1​,…,It−1​. The forecaster draws ItI_tIt​ at random from a distribution ptp_tpt​ that depends on what it has observed, and it observes only ℓIt,t\ell_{I_t,t}ℓIt​,t​. Expectations E\mathbb EE are over the forecaster's draws.

Side information. Each round carries a context sts_tst​ from a finite set S\mathcal SS, and the sequence s1,s2,…s_1, s_2, \dotss1​,s2​,… is fixed in advance. The pseudo-regret against context-to-arm maps is

R‾nS=max⁡g:S→{1,…,K}E[∑t=1nℓIt,t−∑t=1nℓg(st),t].\overline R^{\mathcal S}_n = \max_{g:\mathcal S\to\{1,\dots,K\}} \mathbb E\Big[\sum_{t=1}^n \ell_{I_t,t} - \sum_{t=1}^n \ell_{g(s_t),t}\Big].RnS​=g:S→{1,…,K}max​E[t=1∑n​ℓIt​,t​−t=1∑n​ℓg(st​),t​].

The S-Exp3 forecaster runs one instance of Exp3 (Section 3.1 of the book) on each context.

Expert advice. At each round each of NNN experts jjj proposes a distribution ξtj\xi^j_tξtj​ over arms, which may depend on the forecaster's past plays. The contextual pseudo-regret is

R‾nctx=max⁡k=1,…,NE[∑t=1nℓIt,t−∑t=1nEi∼ξtkℓi,t].\overline R^{\mathrm{ctx}}_n = \max_{k=1,\dots,N}\mathbb E\Big[\sum_{t=1}^n \ell_{I_t,t} - \sum_{t=1}^n \mathbb E_{i\sim\xi^k_t}\ell_{i,t}\Big].Rnctx​=k=1,…,Nmax​E[t=1∑n​ℓIt​,t​−t=1∑n​Ei∼ξtk​​ℓi,t​].

Exp4 (Fig. 4.1) runs exponential weights over the experts with importance-weighted loss estimates.

Bandit multiclass. The examples (xt,yt)∈Rd×{1,…,K}(x_t, y_t) \in \mathbb R^d \times \{1,\dots,K\}(xt​,yt​)∈Rd×{1,…,K} are fixed in advance, with ∥xt∥=1\|x_t\| = 1∥xt​∥=1 (Euclidean). A K×dK\times dK×d matrix UUU classifies xxx by arg⁡max⁡i(Ux)i\arg\max_i (Ux)_iargmaxi​(Ux)i​. Its multiclass hinge loss on round ttt is ℓt(U)=[1−(Uxt)yt+max⁡i≠yt(Uxt)i]+\ell_t(U) = [1 - (Ux_t)_{y_t} + \max_{i\neq y_t}(Ux_t)_i]_+ℓt​(U)=[1−(Uxt​)yt​​+maxi=yt​​(Uxt​)i​]+​. Write Ln(U)=∑t≤nℓt(U)L_n(U) = \sum_{t\le n}\ell_t(U)Ln​(U)=∑t≤n​ℓt​(U) for the cumulative hinge loss, Lˉn(U)=Ln(U)/n\bar L_n(U) = L_n(U)/nLˉn​(U)=Ln​(U)/n for its average, and ∥U∥\|U\|∥U∥ for the Frobenius norm. The multiclass Perceptron predicts y^t=arg⁡max⁡i(Wtxt)i\hat y_t = \arg\max_i (W_tx_t)_iy^​t​=argmaxi​(Wt​xt​)i​ and, after seeing yty_tyt​, adds xtx_txt​ to row yty_tyt​ and subtracts it from row y^t\hat y_ty^​t​. The Banditron (p. 58) predicts YtY_tYt​ from pi,t=(1−γ)1y^t=i+γ/Kp_{i,t} = (1-\gamma)\mathbb 1_{\hat y_t = i} + \gamma/Kpi,t​=(1−γ)1y^​t​=i​+γ/K. It observes only 1Yt=yt\mathbb 1_{Y_t = y_t}1Yt​=yt​​ and updates Wt+1=Wt+X~tW_{t+1} = W_t + \widetilde X_tWt+1​=Wt​+Xt​, where (X~t)i,j=xt,j(1Yt=yt1Yt=i/pi,t−1y^t=i)(\widetilde X_t)_{i,j} = x_{t,j}\big(\mathbb 1_{Y_t=y_t}\mathbb 1_{Y_t=i}/p_{i,t} - \mathbb 1_{\hat y_t=i}\big)(Xt​)i,j​=xt,j​(1Yt​=yt​​1Yt​=i​/pi,t​−1y^​t​=i​). Its number of mistakes is Mn=∑t≤n1Yt≠ytM_n = \sum_{t\le n}\mathbb 1_{Y_t\neq y_t}Mn​=∑t≤n​1Yt​=yt​​.

Formalization targets

Goal: Theorem 4.7 (Banditron)

For n≥8Kn \ge 8Kn≥8K, γ=(K/n)1/3\gamma = (K/n)^{1/3}γ=(K/n)1/3, every example sequence as above and every K×dK\times dK×d matrix UUU,

E Mn≤Ln(U)+(1+∥U∥2Lˉn(U))K1/3n2/3+2∥U∥2K2/3n1/3+2 ∥U∥K1/6n1/3.\mathbb E\,M_n \le L_n(U) + \Big(1 + \|U\|\sqrt{2\bar L_n(U)}\Big)K^{1/3}n^{2/3} + 2\|U\|^2K^{2/3}n^{1/3} + \sqrt2\,\|U\|K^{1/6}n^{1/3}.EMn​≤Ln​(U)+(1+∥U∥2Lˉn​(U)​)K1/3n2/3+2∥U∥2K2/3n1/3+2​∥U∥K1/6n1/3.

Milestones

  1. Multiclass Perceptron bound (Section 4.4, p. 57). For every n≥1n \ge 1n≥1 and UUU, ∑t≤n1y^t≠yt≤Ln(U)+2∥U∥2+∥U∥2nLˉn(U)\sum_{t\le n}\mathbb 1_{\hat y_t\ne y_t} \le L_n(U) + 2\|U\|^2 + \|U\|\sqrt{2n\bar L_n(U)}∑t≤n​1y^​t​=yt​​≤Ln​(U)+2∥U∥2+∥U∥2nLˉn​(U)​.
  2. Theorem 4.1 (p. 44). S-Exp3 satisfies R‾nS≤2n∣S∣Kln⁡K\overline R^{\mathcal S}_n \le \sqrt{2n|\mathcal S|K\ln K}RnS​≤2n∣S∣KlnK​.
  3. Theorem 4.2 (p. 46), with corrected constants. Exp4 without mixing satisfies R‾nctx≤2nKln⁡N\overline R^{\mathrm{ctx}}_n \le \sqrt{2nK\ln N}Rnctx​≤2nKlnN​ for ηt=2ln⁡N/(nK)\eta_t = \sqrt{2\ln N/(nK)}ηt​=2lnN/(nK)​, and R‾nctx≤2nKln⁡N\overline R^{\mathrm{ctx}}_n \le 2\sqrt{nK\ln N}Rnctx​≤2nKlnN​ for ηt=ln⁡N/(tK)\eta_t = \sqrt{\ln N/(tK)}ηt​=lnN/(tK)​.
  4. Theorem 4.3 (p. 50), with corrected learning rate. Let the plays be drawn from distributions qtq_tqt​ with qi,t≥ε>0q_{i,t}\ge\varepsilon > 0qi,t​≥ε>0, and let Exp3 run on the estimates ℓi,t1It=i/qi,t\ell_{i,t}\mathbb 1_{I_t=i}/q_{i,t}ℓi,t​1It​=i​/qi,t​ with η=2εln⁡K/n\eta = \sqrt{2\varepsilon\ln K/n}η=2εlnK/n​. Then max⁡kE[∑tEi∼ptℓi,t−∑tℓk,t]≤(2n/ε)ln⁡K\max_k \mathbb E\big[\sum_t \mathbb E_{i\sim p_t}\ell_{i,t} - \sum_t\ell_{k,t}\big] \le \sqrt{(2n/\varepsilon)\ln K}maxk​E[∑t​Ei∼pt​​ℓi,t​−∑t​ℓk,t​]≤(2n/ε)lnK​.

Significance

Theorem 4.7 shows that, on any sequence of examples, the bandit version of online multiclass classification costs at most O(K1/3n2/3)O(K^{1/3}n^{2/3})O(K1/3n2/3) mistakes beyond the hinge loss of the best linear classifier. The full-information Perceptron, by comparison, pays O(n)O(\sqrt n)O(n​). The bound has no stochastic assumption and has explicit constants. Theorems 4.1–4.3 are the basic regret guarantees for side information and expert advice. Theorem 4.3 in particular lets learning algorithms serve as experts inside Exp4, which is the construction behind Theorem 4.5.

The mission produces machine-checked statements, and eventually proofs, of these results with fully explicit constants and an explicit model of adaptive adversaries and adaptive advice. To the curators' knowledge none of the Banditron, the multiclass Perceptron bound, S-Exp3 or Theorem 4.3 is formalized anywhere. The platform's Bandit Algorithms series has a proved Exp4 bound, but only for advice and rewards fixed in advance. The book proves all four milestones and the goal; two printed statements (4.2 and 4.3) contain misprints that this mission corrects.

Difficulty

The Banditron bound concerns a randomized process whose weight matrix depends on all earlier random predictions. The Perceptron argument tracks ⟨U,Wn+1⟩\langle U, W_{n+1}\rangle⟨U,Wn+1​⟩ and ∥Wn+1∥2\|W_{n+1}\|^2∥Wn+1​∥2. It carries over only in conditional expectation, and the second moment of the importance-weighted update is of order K/γK/\gammaK/γ on rounds where y^t≠yt\hat y_t \neq y_ty^​t​=yt​ and of order γ\gammaγ otherwise. Combining these into one inequality for ∑tP(y^t≠yt)\sum_t\mathbb P(\hat y_t\neq y_t)∑t​P(y^​t​=yt​) and then for EMn\mathbb E M_nEMn​ requires solving a quadratic inequality in the presence of expectations, and the constants must come out as printed. For the Exp3/Exp4 results, the obstacle is that losses and advice adapt to past plays. The standard potential argument has to be run conditionally on the history, and a version that fixes the losses in advance proves a weaker theorem.

Formalization scope

  • Rounds and laws. Rounds are numbered from 000 in Lean (Lean round ttt is the book's round t+1t+1t+1). Every forecaster is a sampling rule from past plays to weights on Fin K. The law of the first nnn plays is the product ∏tpt(ωt∣ω<t)\prod_t p_t(\omega_t\mid\omega_{<t})∏t​pt​(ωt​∣ω<t​) over sequences ω:Fin n→Fin K\omega : \mathrm{Fin}\,n\to\mathrm{Fin}\,Kω:Finn→FinK, and expectations are finite sums against it. The adversary and the experts are deterministic functions of past plays; an independent randomized adversary is a mixture of these. The examples of the Banditron are fixed.
  • Argmax. y^t\hat y_ty^​t​ uses any argmax selector; all tie-breaking rules are covered.
  • Norms. ∥xt∥=1\|x_t\| = 1∥xt​∥=1 is the Euclidean condition ∑jxt,j2=1\sum_j x_{t,j}^2 = 1∑j​xt,j2​=1; ∥U∥\|U\|∥U∥ is the Frobenius norm written out explicitly.
  • Infima and maxima. Each "inf⁡U\inf_UinfU​" and "max⁡k\max_kmaxk​" of the book is stated as "for every UUU" or "for every kkk", which is equivalent.
  • Explicit constants. Every bound is the one printed or, for the corrected items, the one the proof yields. No O(⋅)O(\cdot)O(⋅) appears.
  • Corrected misprints. Theorem 4.7 prints the examples in Rd×{−1,+1}\mathbb R^d\times\{-1,+1\}Rd×{−1,+1}; labels are in {1,…,K}\{1,\dots,K\}{1,…,K}. Theorem 4.2 prints 2nNln⁡K\sqrt{2nN\ln K}2nNlnK​ and 2nNln⁡K2\sqrt{nN\ln K}2nNlnK​; the proof gives 2nKln⁡N\sqrt{2nK\ln N}2nKlnN​ and 2nKln⁡N2\sqrt{nK\ln N}2nKlnN​. Theorem 4.3 prints η=2ln⁡K/(nK)\eta = \sqrt{2\ln K/(nK)}η=2lnK/(nK)​; (4.7) follows from the proof with η=2εln⁡K/n\eta = \sqrt{2\varepsilon\ln K/n}η=2εlnK/n​.
  • Parameter range. At n=8Kn = 8Kn=8K the Banditron's γ\gammaγ equals 1/21/21/2, outside the box's open interval (0,1/2)(0,1/2)(0,1/2). The proof uses only γ≤1/2\gamma\le 1/2γ≤1/2, so n=8Kn = 8Kn=8K is included.
  • Ruling out trivial forms. Theorem 4.1 is stated for the explicit S-Exp3 forecaster, not as an existence claim, so no forecaster tuned to the losses can witness it. The losses and the advice are allowed to adapt, so a proof for oblivious sequences does not suffice.
  • Left out. Theorem 4.4 (Exp4 with mixing) is proved in the book only by reference. The argument that reference suggests yields 32γn+Kln⁡N/γ\tfrac32\gamma n + K\ln N/\gamma23​γn+KlnN/γ, not the printed γn/2+Kln⁡N/γ\gamma n/2 + K\ln N/\gammaγn/2+KlnN/γ. Theorem 4.5 is stated with O(⋅)O(\cdot)O(⋅), Theorem 4.6 "for some constant ccc", and Eq. (4.8) is left to the reader.

Useful reusable infrastructure: the path-law expectation for history-dependent sampling, the exponential-weights potential argument under adaptive losses, and Perceptron-type inner-product arguments for matrices. Proofs of any milestone and of the goal are welcome.

Selected references

  • S. Bubeck, N. Cesa-Bianchi, Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems, Foundations and Trends in Machine Learning 5(1), 2012. arXiv:1204.5721v2. https://arxiv.org/abs/1204.5721 ; https://doi.org/10.1561/2200000024
  • S. M. Kakade, S. Shalev-Shwartz, A. Tewari, Efficient Bandit Algorithms for Online Multiclass Prediction, ICML 2008. https://doi.org/10.1145/1390156.1390212
  • P. Auer, N. Cesa-Bianchi, Y. Freund, R. E. Schapire, The Nonstochastic Multiarmed Bandit Problem, SIAM Journal on Computing 32(1), 2002. https://doi.org/10.1137/S0097539701398375
  • O.-A. Maillard, R. Munos, Adaptive Bandits: Towards the Best History-Dependent Strategy, AISTATS 2011. https://proceedings.mlr.press/v15/maillard11a.html
11 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOperations Research+1·Captain: mikedeng1

Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems V: Bandit Convex Optimization with One-Point FeedbackTextbook

Motivation

In bandit convex optimization a forecaster repeatedly picks a point xtx_txt​ of a convex set K⊆Rd\mathcal K\subseteq\mathbb R^dK⊆Rd, and an adversary picks a convex loss ℓt\ell_tℓt​. The forecaster pays ℓt(xt)\ell_t(x_t)ℓt​(xt​) and observes only that number: it never sees the function, its gradient, or its value elsewhere. This is the model of online optimization with only function-value access, as in tuning a system online from measured costs, dynamic pricing with an unknown convex demand-cost curve, or routing with path costs observed only on the route taken. The question is how fast the forecaster can approach the best fixed point in hindsight.

Chapter 6 of Bubeck and Cesa-Bianchi's monograph (arXiv:1204.5721v2, Foundations and Trends in Machine Learning 5(1), 2012) treats the problem through spherical gradient estimates fed to projected gradient descent. The one-point method is due to Flaxman, Kalai and McMahan (SODA 2005, arXiv:cs/0408007), who obtained an O(n3/4)\mathcal O(n^{3/4})O(n3/4) regret bound. Agarwal, Dekel and Xiao (COLT 2010) showed that two function evaluations per round allow O(n)\mathcal O(\sqrt n)O(n​). Whether one-point feedback admits n\sqrt nn​ regret was open when the monograph was written (p. 94); Bubeck, Eldan and Lee (STOC 2017, arXiv:1607.03084) later obtained n\sqrt nn​ regret up to logarithmic and polynomial-in-ddd factors for convex losses, with a different and much more involved algorithm.

Setting

Let B={x∈Rd:∥x∥≤1}\mathbb B=\{x\in\mathbb R^d:\|x\|\le1\}B={x∈Rd:∥x∥≤1} be the closed Euclidean unit ball and S={x:∥x∥=1}\mathbb S=\{x:\|x\|=1\}S={x:∥x∥=1} the unit sphere, with unnormalized spherical measure σ\sigmaσ, so that σ(S)=d Vol(B)\sigma(\mathbb S)=d\,\mathrm{Vol}(\mathbb B)σ(S)=dVol(B). Fix δ>0\delta>0δ>0. For a loss ℓ\ellℓ, the smoothed loss is ℓ~(x)=E ℓ(x+δB)\widetilde\ell(x)=\mathbb E\,\ell(x+\delta B)ℓ(x)=Eℓ(x+δB) with BBB uniform on B\mathbb BB.

The set K\mathcal KK is closed and convex with rB⊆K⊆RBr\mathbb B\subseteq\mathcal K\subseteq R\mathbb BrB⊆K⊆RB. The losses ℓ1,ℓ2,⋯:Rd→R\ell_1,\ell_2,\dots:\mathbb R^d\to\mathbb Rℓ1​,ℓ2​,⋯:Rd→R are GGG-Lipschitz, differentiable and convex, and are fixed before the game (an oblivious adversary).

OSGD (Online Stochastic Gradient Descent) on a set K′\mathcal K'K′ with learning rate η\etaη starts at x1=0x_1=0x1​=0 and sets xt+1=argmin⁡y∈K′∥y−(xt−ηg~t(xt))∥x_{t+1}=\operatorname{argmin}_{y\in\mathcal K'}\|y-(x_t-\eta\widetilde g_t(x_t))\|xt+1​=argminy∈K′​∥y−(xt​−ηg​t​(xt​))∥, where g~t\widetilde g_tg​t​ is a gradient estimate. With S1,S2,…S_1,S_2,\dotsS1​,S2​,… independent and uniform on S\mathbb SS:

  • the two-point estimate (6.1) is g~t(xt)=d2δ(ℓt(Xt+)−ℓt(Xt−))St\widetilde g_t(x_t)=\frac d{2\delta}\big(\ell_t(X_t^+)-\ell_t(X_t^-)\big)S_tg​t​(xt​)=2δd​(ℓt​(Xt+​)−ℓt​(Xt−​))St​ with Xt±=xt±δStX_t^\pm=x_t\pm\delta S_tXt±​=xt​±δSt​; the played point is Xt+X_t^+Xt+​ or Xt−X_t^-Xt−​ by a fair coin;
  • the one-point estimate (6.3) is g~t(xt)=dδ ℓt(X~t)St\widetilde g_t(x_t)=\frac d\delta\,\ell_t(\widetilde X_t)S_tg​t​(xt​)=δd​ℓt​(Xt​)St​ with played point X~t=xt+δSt\widetilde X_t=x_t+\delta S_tXt​=xt​+δSt​.

OSGD runs on the shrunken set K′=(1−δ/r)K\mathcal K'=(1-\delta/r)\mathcal KK′=(1−δ/r)K, so that the perturbed points stay in K\mathcal KK. The pseudo-regret is

R‾n=E∑t=1nℓt(X~t)−min⁡x∈K∑t=1nℓt(x).\overline R_n=\mathbb E\sum_{t=1}^n\ell_t(\widetilde X_t)-\min_{x\in\mathcal K}\sum_{t=1}^n\ell_t(x).Rn​=Et=1∑n​ℓt​(Xt​)−x∈Kmin​t=1∑n​ℓt​(x).

Formalization targets

Goal: Theorem 6.2, tuned

If in addition ∣ℓt∣≤L|\ell_t|\le L∣ℓt​∣≤L on K\mathcal KK, and δ=(2n)−1/4RdL/((3+R/r)G)\delta=(2n)^{-1/4}\sqrt{RdL/((3+R/r)G)}δ=(2n)−1/4RdL/((3+R/r)G)​, η=(2n)−3/4R3/(dL(3+R/r)G)\eta=(2n)^{-3/4}\sqrt{R^3/(dL(3+R/r)G)}η=(2n)−3/4R3/(dL(3+R/r)G)​, then one-point OSGD satisfies

R‾n≤4n3/4RdL (3+R/r) G.\overline R_n\le 4n^{3/4}\sqrt{RdL\,(3+R/r)\,G}.Rn​≤4n3/4RdL(3+R/r)G​.

Milestones

  1. Lemma 6.1: ∇∫Bℓ(x+δb) db=1δ∫Sℓ(x+δs)s dσ(s)\nabla\int_{\mathbb B}\ell(x+\delta b)\,db=\frac1\delta\int_{\mathbb S}\ell(x+\delta s)s\,d\sigma(s)∇∫B​ℓ(x+δb)db=δ1​∫S​ℓ(x+δs)sdσ(s).
  2. Lemma 6.2: dδE[ℓ(x+δS)S]=∇E ℓ(x+δB)\frac d\delta\mathbb E[\ell(x+\delta S)S]=\nabla\mathbb E\,\ell(x+\delta B)δd​E[ℓ(x+δS)S]=∇Eℓ(x+δB).
  3. Eq. (6.2): ∣ℓ(x)−ℓ~(x)∣≤δG|\ell(x)-\widetilde\ell(x)|\le\delta G∣ℓ(x)−ℓ(x)∣≤δG.
  4. Lemma 6.3: the queried points' regret against xxx is at most the smoothed regret of the iterates against (1−ξ)x(1-\xi)x(1−ξ)x, plus 3δGn+ξGRn3\delta Gn+\xi GRn3δGn+ξGRn.
  5. Theorem 6.1: two-point OSGD has R‾n≤R2/η+η(Gd)2n+δ(3+R/r)Gn\overline R_n\le R^2/\eta+\eta(Gd)^2n+\delta(3+R/r)GnRn​≤R2/η+η(Gd)2n+δ(3+R/r)Gn, and R‾n≤2RGdn+δ(3+R/r)Gn\overline R_n\le 2RGd\sqrt n+\delta(3+R/r)GnRn​≤2RGdn​+δ(3+R/r)Gn for η=R/(Gdn)\eta=R/(Gd\sqrt n)η=R/(Gdn​).
  6. Theorem 6.2, first display: one-point OSGD has R‾n≤R2/η+(dL)2δ2ηn+δ(3+R/r)Gn\overline R_n\le R^2/\eta+\frac{(dL)^2}{\delta^2}\eta n+\delta(3+R/r)GnRn​≤R2/η+δ2(dL)2​ηn+δ(3+R/r)Gn for every 0<δ≤r0<\delta\le r0<δ≤r and η>0\eta>0η>0.

Significance

The n3/4n^{3/4}n3/4 bound shows that a single function value per round suffices for sublinear regret against any oblivious sequence of Lipschitz convex losses, with a forecaster whose only operations are a random perturbation and a Euclidean projection. The smoothing identity of Lemmas 6.1–6.2 is the basic tool of zeroth-order (derivative-free) optimization, used well beyond bandits, and Theorem 6.1 is the n\sqrt nn​ benchmark for two-point methods.

All results are proved in the source. To the best of current knowledge none is formalized: the related items of the Introduction to Online Convex Optimization series on Prove2Me (Hazan's Lemma 6.7 and Theorem 6.9) were formalized with missing hypotheses and are recorded as disproved. This mission produces machine-checked statements with every hypothesis explicit, and the formal infrastructure (sphere measure calculus, a projected stochastic gradient analysis) for later zeroth-order results.

Difficulty

Two steps resist a direct formal treatment. First, Lemma 6.1 is a divergence-theorem identity on the ball; Mathlib has the sphere measure and polar coordinates, but its divergence theorem covers boxes rather than balls, so differentiating the ball average in xxx requires either such a theorem or a direct argument about translates of the ball. Second, the regret analysis takes expectations of quantities that depend on the whole past: the iterate xtx_txt​ is a function of S1,…,St−1S_1,\dots,S_{t-1}S1​,…,St−1​, and unbiasedness E[g~t∣xt]=∇ℓ~t(xt)\mathbb E[\widetilde g_t\mid x_t]=\nabla\widetilde\ell_t(x_t)E[g​t​∣xt​]=∇ℓt​(xt​) holds only conditionally, via independence of StS_tSt​ from the past. A pathwise gradient-descent inequality must be combined with this conditional expectation round by round, with measurability of the projected iterates established along the way. The naive approach of treating the estimate as the true gradient of ℓt\ell_tℓt​ fails: it is a gradient of ℓ~t\widetilde\ell_tℓt​, and the gap is handled only by Eq. (6.2) and Lemma 6.3.

Formalization scope

Points are in EuclideanSpace ℝ (Fin d) with d≥1d\ge1d≥1; rounds are t=1,2,…t=1,2,\dotst=1,2,…, sums run over Finset.Icc 1 n. σ\sigmaσ is Mathlib's Measure.toSphere of Lebesgue measure; the uniform laws are normalized restrictions. Randomness lives on an arbitrary probability space; the directions StS_tSt​ are measurable, mutually independent (iIndepFun) and uniform on S\mathbb SS, and in Theorem 6.1 the pairs (St,Ct)(S_t,C_t)(St​,Ct​) are independent with CtC_tCt​ a fair sign independent of StS_tSt​. A run of OSGD is a predicate (start at 000, each iterate a Euclidean projection onto (1−δ/r)K(1-\delta/r)\mathcal K(1−δ/r)K), which determines the run uniquely, so the forecaster uses only observed values and its own randomness. The losses are Lipschitz, differentiable and convex on all of Rd\mathbb R^dRd; the bound ∣ℓt∣≤L|\ell_t|\le L∣ℓt​∣≤L is on K\mathcal KK, because a convex function bounded on Rd\mathbb R^dRd is constant. The minimum over K\mathcal KK is an infimum over the subtype K\mathcal KK, attained in every theorem.

Conventions and corrections, each stated in the item's Formalization Note:

  • Lemma 6.1 carries the factor 1/δ1/\delta1/δ that the printed statement omits and the proof contains (corrected misprint).
  • Theorem 6.1's second display prints η=R/(GDn)\eta=R/(GD\sqrt n)η=R/(GDn​) and a limit "for δ→0\delta\to0δ→0"; the item states R‾n≤2RGdn+δ(3+R/r)Gn\overline R_n\le 2RGd\sqrt n+\delta(3+R/r)GnRn​≤2RGdn​+δ(3+R/r)Gn for η=R/(Gdn)\eta=R/(Gd\sqrt n)η=R/(Gdn​) and every admissible δ\deltaδ, which implies the limit (corrected misprint).
  • Theorems 6.1 and 6.2 add 0<δ≤r0<\delta\le r0<δ≤r, which the proofs need for Xt±,X~t∈KX_t^\pm,\widetilde X_t\in\mathcal KXt±​,Xt​∈K; for the tuned δ\deltaδ of the goal it is a condition on nnn.
  • The goal adds G,L>0G,L>0G,L>0 and n≥1n\ge1n≥1, which its formulas for δ,η\delta,\etaδ,η need; the constant 444 is the book's rounding of 2⋅23/42\cdot2^{3/4}2⋅23/4 and is kept, as is the form R2/ηR^2/\etaR2/η.

The statements cannot be satisfied trivially: the run is pinned by its recursion, the losses are fixed before the randomness, the expectations are of bounded measurable functions (no zero-valued Bochner integrals), and the minimum is over the nonempty compact K\mathcal KK. Section 6.3 (Lemma 6.4, Theorem 6.3) is not included, because its algorithm box and proof use different stage lengths and its unimodality condition is stated on a smaller set than the proof uses.

Needed infrastructure: calculus of ball averages and sphere integrals, symmetry of the uniform sphere law, nonexpansiveness of projections onto closed convex sets, and conditional-expectation bookkeeping for adapted iterates. Each is reusable for zeroth-order optimization; contributions of any of them as separate lemmas are welcome.

Selected references

  • S. Bubeck, N. Cesa-Bianchi, Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems, Foundations and Trends in Machine Learning 5(1), 2012. arXiv:1204.5721v2, doi:10.1561/2200000024
  • A. Flaxman, A. Kalai, H. B. McMahan, Online convex optimization in the bandit setting: gradient descent without a gradient, SODA 2005. arXiv:cs/0408007
  • A. Agarwal, O. Dekel, L. Xiao, Optimal algorithms for online convex optimization with multi-point bandit feedback, COLT 2010. link
  • S. Bubeck, R. Eldan, Y. T. Lee, Kernel-based methods for bandit convex optimization, STOC 2017. arXiv:1607.03084
10 thms2 active usersReviewed
🏆Completed
Machine LearningOptimizationReinforcement Learning·Captain: mikedeng1

Reinforcement Learning: An Introduction I: The Gradient Bandit Algorithm Is Stochastic Gradient AscentTextbook

Motivation

The multi-armed bandit is the simplest setting in which a learner must trade off exploiting what it knows against exploring what it does not: one situation, kkk actions, and a reward drawn from an unknown distribution each time an action is taken. Chapter 2 of Sutton and Barto's Reinforcement Learning: An Introduction (2nd ed., MIT Press, 2018) uses it to introduce, in the smallest possible setting, ideas that run through the rest of the book: incremental estimation with a step size, the bias introduced by the initial estimate, soft-max policies over learned preferences, and learning by following the gradient of expected reward.

The chapter ends with the gradient bandit algorithm (§2.8), which learns a numerical preference for each action instead of a value estimate. A shaded box on pp. 38–40 shows that its expected update is exactly a gradient-ascent step on the expected reward, so the algorithm is an instance of stochastic gradient ascent. The same argument, a score-function (likelihood-ratio) identity with a baseline, reappears in Chapter 13 as the REINFORCE algorithm and the policy gradient theorem. The bandit case is where the book first carries it out in full.

Setting

Actions are 1,…,k1, \dots, k1,…,k. Each action xxx has a reward distribution νx\nu_xνx​ on R\mathbb RR with finite mean q∗(x)q_*(x)q∗​(x), the true action value. At each step the learner holds a vector of action preferences H=(H(1),…,H(k))∈RkH = (H(1), \dots, H(k)) \in \mathbb R^kH=(H(1),…,H(k))∈Rk and selects action AAA with the soft-max probability

π(a)=eH(a)∑b=1keH(b)(2.11).\pi(a) = \frac{e^{H(a)}}{\sum_{b=1}^k e^{H(b)}} \qquad (2.11).π(a)=∑b=1k​eH(b)eH(a)​(2.11).

Given A=xA = xA=x, a reward R∼νxR \sim \nu_xR∼νx​ is received. The expected reward is E[R]=∑xπ(x) q∗(x)\mathbb E[R] = \sum_x \pi(x)\, q_*(x)E[R]=∑x​π(x)q∗​(x), a smooth function of HHH. With a step size α>0\alpha > 0α>0 and a baseline B∈RB \in \mathbb RB∈R, the gradient bandit update (2.12) is

H′(A)=H(A)+α(R−B)(1−π(A)),H′(a)=H(a)−α(R−B) π(a)  (a≠A).H'(A) = H(A) + \alpha (R - B)(1 - \pi(A)), \qquad H'(a) = H(a) - \alpha (R - B)\,\pi(a) \ \ (a \ne A).H′(A)=H(A)+α(R−B)(1−π(A)),H′(a)=H(a)−α(R−B)π(a)  (a=A).

The chapter's estimation sections use a single action's rewards R1,R2,…R_1, R_2, \dotsR1​,R2​,…. The sample average after n−1n-1n−1 selections is Qn=(R1+⋯+Rn−1)/(n−1)Q_n = (R_1 + \cdots + R_{n-1})/(n-1)Qn​=(R1​+⋯+Rn−1​)/(n−1), with an arbitrary initial value Q1Q_1Q1​. A constant step size α∈(0,1]\alpha \in (0,1]α∈(0,1] updates Qn+1=Qn+α[Rn−Qn]Q_{n+1} = Q_n + \alpha [R_n - Q_n]Qn+1​=Qn​+α[Rn​−Qn​] (2.5). The trace of one oˉ0=0\bar o_0 = 0oˉ0​=0, oˉn=oˉn−1+α(1−oˉn−1)\bar o_n = \bar o_{n-1} + \alpha (1 - \bar o_{n-1})oˉn​=oˉn−1​+α(1−oˉn−1​) defines the step size βn=α/oˉn\beta_n = \alpha / \bar o_nβn​=α/oˉn​ (2.8)–(2.9).

Formalization targets

Goal: the expected update is the gradient step

For every action aaa, with A∼πA \sim \piA∼π and R∣A=x∼νxR \mid A = x \sim \nu_xR∣A=x∼νx​,

E[H′(a)]=H(a)+α ∂ E[R]∂H(a),\mathbb E\bigl[H'(a)\bigr] = H(a) + \alpha\, \frac{\partial\, \mathbb E[R]}{\partial H(a)} ,E[H′(a)]=H(a)+α∂H(a)∂E[R]​,

that is, the update (2.12) equals the exact gradient-ascent step (2.13) in expected value, for every baseline BBB that does not depend on the selected action.

Milestones

  1. (2.3): Qn+1=Qn+1n[Rn−Qn]Q_{n+1} = Q_n + \tfrac1n [R_n - Q_n]Qn+1​=Qn​+n1​[Rn​−Qn​] for n≥1n \ge 1n≥1, including Q2=R1Q_2 = R_1Q2​=R1​ for arbitrary Q1Q_1Q1​.
  2. (2.6): Qn+1=(1−α)nQ1+∑i=1nα(1−α)n−iRiQ_{n+1} = (1-\alpha)^n Q_1 + \sum_{i=1}^n \alpha(1-\alpha)^{n-i} R_iQn+1​=(1−α)nQ1​+∑i=1n​α(1−α)n−iRi​, with weights summing to one.
  3. Exercise 2.7: with βn=α/oˉn\beta_n = \alpha/\bar o_nβn​=α/oˉn​, Qn+1=∑i=1nα(1−α)n−ioˉnRiQ_{n+1} = \sum_{i=1}^n \frac{\alpha(1-\alpha)^{n-i}}{\bar o_n} R_iQn+1​=∑i=1n​oˉn​α(1−α)n−i​Ri​ for n≥1n \ge 1n≥1, weights summing to one, and no dependence on Q1Q_1Q1​.
  4. Shift invariance (p. 37): adding a constant ccc to every preference leaves π\piπ unchanged.
  5. Exercise 2.9: for k=2k = 2k=2, π(1)=σ(H(1)−H(2))\pi(1) = \sigma(H(1) - H(2))π(1)=σ(H(1)−H(2)) with σ(x)=1/(1+e−x)\sigma(x) = 1/(1+e^{-x})σ(x)=1/(1+e−x).
  6. Soft-max derivative (p. 40): ∂π(x)/∂H(a)=π(x)(1a=x−π(a))\partial \pi(x)/\partial H(a) = \pi(x)(\mathbb 1_{a=x} - \pi(a))∂π(x)/∂H(a)=π(x)(1a=x​−π(a)).
  7. Zero-sum gradient (p. 39): ∑x∂π(x)/∂H(a)=0\sum_x \partial \pi(x)/\partial H(a) = 0∑x​∂π(x)/∂H(a)=0.
  8. Performance gradient as an expectation (p. 39): ∂E[R]/∂H(a)=E[(R−B)(1a=A−π(a))]\partial \mathbb E[R]/\partial H(a) = \mathbb E[(R - B)(\mathbb 1_{a=A} - \pi(a))]∂E[R]/∂H(a)=E[(R−B)(1a=A​−π(a))].

Significance

The result. The identity makes a model-free algorithm, which uses only the sampled action and reward, an unbiased estimator of the gradient of a quantity that depends on the unknown q∗q_*q∗​. It therefore places the gradient bandit algorithm within stochastic approximation, where convergence theory for stochastic gradient methods applies. It also explains the role of the baseline: any baseline independent of the action leaves the expected update unchanged, so the choice of baseline can only affect the variance of the update, as Figure 2.5 shows empirically. The estimation milestones make precise two claims the chapter uses repeatedly: sample averages can be maintained incrementally, and constant step sizes produce an exponentially recency-weighted average biased by Q1Q_1Q1​. Exercise 2.7 removes that bias.

Formalizing it. All of these results are elementary and proved (or left as routine exercises) in the book. None of them is formalized on Prove2Me or, as far as is known, in Mathlib. What this mission adds is a machine-checked version of the book's argument with the reward model and baseline condition stated precisely, and a reusable soft-max layer (definition, partial derivatives, shift invariance) for later missions of this series, in particular the policy gradient theorem of Chapter 13.

Difficulty

The mathematics is beginning calculus, as the book says. The formal difficulty lies elsewhere. The goal is an identity between an expectation over a two-stage random experiment (an action from π\piπ, then a reward from νA\nu_AνA​) and a partial derivative in one coordinate of a vector-valued parameter. A proof has to justify exchanging the finite sum with the derivative and splitting the reward integral, and it has to use integrability of each νx\nu_xνx​. It also needs the fact that the baseline term vanishes because ∑x∂π(x)/∂H(a)=0\sum_x \partial\pi(x)/\partial H(a) = 0∑x​∂π(x)/∂H(a)=0. A scalar-parameter version of the log-sum-exp derivative does not suffice: the book differentiates in one coordinate H(a)H(a)H(a) while all other preferences are held fixed. For Exercise 2.7 the obvious unrolling of (2.6) does not apply directly, because the step size βn\beta_nβn​ varies with nnn and the book states neither the weights nor the range of α\alphaα.

Formalization scope

  • Actions are Fin k. Every statement quantifies over some action, so k≥1k \ge 1k≥1 whenever it has content. Preferences are vectors Fin k → ℝ. The partial derivative in coordinate aaa is the derivative of h↦f(update H a h)h \mapsto f(\text{update } H\ a\ h)h↦f(update H a h) at H(a)H(a)H(a). The soft-max derivative milestone is stated with HasDerivAt, so it also asserts differentiability.
  • Rewards: each νx\nu_xνx​ is a probability measure on R\mathbb RR with Integrable identity and mean q∗(x)q_*(x)q∗​(x). The expectation of a function of (A,R)(A, R)(A,R) is ∑xπ(x)∫⋅ dνx\sum_x \pi(x) \int \cdot \, d\nu_x∑x​π(x)∫⋅dνx​. The book's normal-distribution testbed is only an example.
  • The baseline is a fixed real BBB, the book's "any scalar that does not depend on" the action (pp. 39–40). The book's Bt=RˉtB_t = \bar R_tBt​=Rˉt​, the average of past rewards, is covered once one conditions on the past. Footnote 1 on p. 37 states that the chapter's experiments used a Rˉt\bar R_tRˉt​ that also included RtR_tRt​. That baseline depends on AtA_tAt​, and the identity does not cover it.
  • Rewards of one action are a sequence indexed from 111. Q1Q_1Q1​ is arbitrary, and 00=10^0 = 100=1 as in the book (p. 33), so α=1\alpha = 1α=1 is included in (2.6).
  • Exercise 2.7 speaks of "a conventional constant step size α>0\alpha > 0α>0". The formalization takes α∈(0,1]\alpha \in (0,1]α∈(0,1], the range of the constant step size in (2.5). For α=2\alpha = 2α=2 the trace oˉn\bar o_noˉn​ vanishes at every even nnn and βn\beta_nβn​ is undefined. "Without initial bias" is read as "for n≥1n \ge 1n≥1, Qn+1Q_{n+1}Qn+1​ is the displayed weighted average of R1,…,RnR_1, \dots, R_nR1​,…,Rn​ with weights summing to one", which in particular does not involve Q1Q_1Q1​.
  • Exercise 2.9 is read as the two equalities π(1)=σ(H(1)−H(2))\pi(1) = \sigma(H(1)-H(2))π(1)=σ(H(1)−H(2)) and π(2)=σ(H(2)−H(1))\pi(2) = \sigma(H(2)-H(1))π(2)=σ(H(2)−H(1)).
  • A trivializing formalization is ruled out: the goal is about the expected value of the algorithm's update (2.12) under the joint law of action and reward, not the soft-max derivative alone and not a version in which the reward is replaced by its mean or the expectation is taken over AAA only.
  • Not formalized: the UCB rule (2.10) and the 10-armed testbed, which carry no provable claim in the chapter, and the stochastic-approximation conditions (2.7), which the book cites without proof.
  • Welcome contributions: a general soft-max library (derivatives, Jacobian, log-sum-exp) over a finite type, reusable for Chapter 13, and proofs of the milestones in the listed order.

Selected references

  • R. S. Sutton, A. G. Barto, Reinforcement Learning: An Introduction, 2nd ed., MIT Press, 2018, ISBN 9780262039246, Chapter 2, pp. 25–46. http://incompleteideas.net/book/the-book-2nd.html
  • R. J. Williams, Simple statistical gradient-following algorithms for connectionist reinforcement learning, Machine Learning 8 (1992) 229–256. https://doi.org/10.1007/BF00992696
  • H. Robbins, S. Monro, A stochastic approximation method, Annals of Mathematical Statistics 22 (1951) 400–407. https://doi.org/10.1214/aoms/1177729586
11 thms2 active usersReviewed
🏆Completed
Machine LearningOperations ResearchStatistics·Captain: mikedeng1

Optimal Best Arm Identification with Fixed Confidence II: Characterization of the Optimal Proportions of Arm DrawsResearch Paper

Motivation

In best arm identification with fixed confidence, a learner samples KKK unknown distributions ("arms") sequentially and must name the arm with the largest mean, with error probability at most a prescribed δ\deltaδ, using as few samples as possible. Garivier and Kaufmann (arXiv:1602.04589, COLT 2016) showed that every δ\deltaδ-PAC strategy needs, in expectation, at least T∗(μ) kl(δ,1−δ)T^*(\boldsymbol\mu)\,\mathrm{kl}(\delta,1-\delta)T∗(μ)kl(δ,1−δ) samples, where the characteristic time T∗(μ)T^*(\boldsymbol\mu)T∗(μ) is the value of a max–min optimization problem over the proportions of draws allocated to the arms. The maximizer of that problem, w∗(μ)w^*(\boldsymbol\mu)w∗(μ), is the allocation any asymptotically optimal strategy must follow; the Track-and-Stop algorithm of the same paper computes w∗(μ^)w^*(\hat{\boldsymbol\mu})w∗(μ^​) at plug-in estimates and tracks it.

A strategy can only track w∗w^*w∗ if w∗w^*w∗ can be computed. This mission formalizes the part of the paper (Section 2.2 and Appendix A) that turns the abstract max–min problem into an explicit recipe: a closed form for the inner infimum, and a characterization of w∗w^*w∗ through the root of one increasing scalar function. The problem had been solved in closed form before only for special cases, such as Poisson rewards with all suboptimal arms equal (Vaidhyan and Sundaresan, 2015); the paper's result covers every one-parameter exponential family.

Setting

A canonical one-parameter exponential family is a family of laws νθ\nu_\thetaνθ​, θ∈Θ\theta\in\Thetaθ∈Θ, on R\mathbb RR with density exp⁡(θx−b(θ))\exp(\theta x-b(\theta))exp(θx−b(θ)) with respect to a reference measure ξ\xiξ. The law νθ\nu_\thetaνθ​ has mean b˙(θ)\dot b(\theta)b˙(θ); the set of attainable means is the mean space b˙(Θ)\dot b(\Theta)b˙(Θ). For means μ=b˙(θ)\mu=\dot b(\theta)μ=b˙(θ) and μ′=b˙(θ′)\mu'=\dot b(\theta')μ′=b˙(θ′) the Kullback–Leibler divergence is

d(μ,μ′)=KL(νθ,νθ′)=b(θ′)−b(θ)−b˙(θ)(θ′−θ).d(\mu,\mu')=\mathrm{KL}(\nu_\theta,\nu_{\theta'})=b(\theta')-b(\theta)-\dot b(\theta)(\theta'-\theta).d(μ,μ′)=KL(νθ​,νθ′​)=b(θ′)−b(θ)−b˙(θ)(θ′−θ).

Bernoulli laws and Gaussian laws with known variance are the standard examples.

A bandit model is identified with its vector of means μ=(μ1,…,μK)∈b˙(Θ)K\boldsymbol\mu=(\mu_1,\dots,\mu_K)\in\dot b(\Theta)^Kμ=(μ1​,…,μK​)∈b˙(Θ)K. S\mathcal SS is the set of models with a unique optimal arm a∗(μ)a^*(\boldsymbol\mu)a∗(μ), and Alt(μ)={λ∈S:a∗(λ)≠a∗(μ)}\mathrm{Alt}(\boldsymbol\mu)=\{\boldsymbol\lambda\in\mathcal S:a^*(\boldsymbol\lambda)\ne a^*(\boldsymbol\mu)\}Alt(μ)={λ∈S:a∗(λ)=a∗(μ)} is the set of alternatives. ΣK\Sigma_KΣK​ is the probability simplex. The transportation cost of proportions w∈ΣKw\in\Sigma_Kw∈ΣK​ and the objects of the paper are

cμ(w)=inf⁡λ∈Alt(μ)∑a=1Kwa d(μa,λa),T∗(μ)−1=sup⁡w∈ΣKcμ(w),w∗(μ)=argmax⁡w∈ΣKcμ(w).c_{\boldsymbol\mu}(w)=\inf_{\boldsymbol\lambda\in\mathrm{Alt}(\boldsymbol\mu)}\sum_{a=1}^Kw_a\,d(\mu_a,\lambda_a),\qquad T^*(\boldsymbol\mu)^{-1}=\sup_{w\in\Sigma_K}c_{\boldsymbol\mu}(w),\qquad w^*(\boldsymbol\mu)=\operatorname*{argmax}_{w\in\Sigma_K}c_{\boldsymbol\mu}(w).cμ​(w)=λ∈Alt(μ)inf​a=1∑K​wa​d(μa​,λa​),T∗(μ)−1=w∈ΣK​sup​cμ​(w),w∗(μ)=w∈ΣK​argmax​cμ​(w).

The arms are sorted so that μ1>μ2≥⋯≥μK\mu_1>\mu_2\ge\dots\ge\mu_Kμ1​>μ2​≥⋯≥μK​. The parameterized Jensen–Shannon divergence is, for α∈[0,1]\alpha\in[0,1]α∈[0,1],

Iα(μ1,μ2)=α d(μ1,αμ1+(1−α)μ2)+(1−α) d(μ2,αμ1+(1−α)μ2).I_\alpha(\mu_1,\mu_2)=\alpha\,d\big(\mu_1,\alpha\mu_1+(1-\alpha)\mu_2\big)+(1-\alpha)\,d\big(\mu_2,\alpha\mu_1+(1-\alpha)\mu_2\big).Iα​(μ1​,μ2​)=αd(μ1​,αμ1​+(1−α)μ2​)+(1−α)d(μ2​,αμ1​+(1−α)μ2​).

For a∈{2,…,K}a\in\{2,\dots,K\}a∈{2,…,K} let ga(x)=(1+x)I1/(1+x)(μ1,μa)g_a(x)=(1+x)I_{1/(1+x)}(\mu_1,\mu_a)ga​(x)=(1+x)I1/(1+x)​(μ1​,μa​) for x≥0x\ge0x≥0, let xa=ga−1x_a=g_a^{-1}xa​=ga−1​, and let x1≡1x_1\equiv1x1​≡1. Finally

Fμ(y)=∑a=2Kd(μ1,ma(y))d(μa,ma(y)),ma(y)=μ1+xa(y)μa1+xa(y).F_{\boldsymbol\mu}(y)=\sum_{a=2}^K\frac{d\big(\mu_1,m_a(y)\big)}{d\big(\mu_a,m_a(y)\big)},\qquad m_a(y)=\frac{\mu_1+x_a(y)\mu_a}{1+x_a(y)}.Fμ​(y)=a=2∑K​d(μa​,ma​(y))d(μ1​,ma​(y))​,ma​(y)=1+xa​(y)μ1​+xa​(y)μa​​.

Formalization targets

Goal: Theorem 5 (p. 5)

With D=d(μ1,μ2)D=d(\mu_1,\mu_2)D=d(μ1​,μ2​): FμF_{\boldsymbol\mu}Fμ​ is continuous and strictly increasing on [0,D[[0,D[[0,D[, Fμ(0)=0F_{\boldsymbol\mu}(0)=0Fμ​(0)=0, Fμ(y)→∞F_{\boldsymbol\mu}(y)\to\inftyFμ​(y)→∞ as y→Dy\to Dy→D, the equation Fμ(y)=1F_{\boldsymbol\mu}(y)=1Fμ​(y)=1 has a unique solution y∗∈[0,D[y^*\in[0,D[y∗∈[0,D[, and

w∈w∗(μ)  ⟺  wa=xa(y∗)∑i=1Kxi(y∗)for every arm a.w\in w^*(\boldsymbol\mu)\iff w_a=\frac{x_a(y^*)}{\sum_{i=1}^Kx_i(y^*)}\quad\text{for every arm }a.w∈w∗(μ)⟺wa​=∑i=1K​xi​(y∗)xa​(y∗)​for every arm a.

The equivalence says at once that the argmax exists, that it is a single point, and that it is given by eq. (5).

Milestones

  1. Lemma 3 (p. 5): for every w∈ΣKw\in\Sigma_Kw∈ΣK​,
cμ(w)=min⁡a≠1(w1+wa) Iw1w1+wa(μ1,μa).c_{\boldsymbol\mu}(w)=\min_{a\ne1}(w_1+w_a)\,I_{\frac{w_1}{w_1+w_a}}(\mu_1,\mu_a).cμ​(w)=a=1min​(w1​+wa​)Iw1​+wa​w1​​​(μ1​,μa​).
  1. Claim after eq. (4) (p. 5): gag_aga​ is a strictly increasing one-to-one mapping from [0,+∞[[0,+\infty[[0,+∞[ onto [0,d(μ1,μa)[[0,d(\mu_1,\mu_a)[[0,d(μ1​,μa​)[.
  2. Lemma 4 (p. 5): for every maximizer w∗w^*w∗ and all a,b∈{2,…,K}a,b\in\{2,\dots,K\}a,b∈{2,…,K},
(w1∗+wa∗)Iw1∗w1∗+wa∗(μ1,μa)=(w1∗+wb∗)Iw1∗w1∗+wb∗(μ1,μb).(w^*_1+w^*_a)I_{\frac{w^*_1}{w^*_1+w^*_a}}(\mu_1,\mu_a)=(w^*_1+w^*_b)I_{\frac{w^*_1}{w^*_1+w^*_b}}(\mu_1,\mu_b).(w1∗​+wa∗​)Iw1∗​+wa∗​w1∗​​​(μ1​,μa​)=(w1∗​+wb∗​)Iw1∗​+wb∗​w1∗​​​(μ1​,μb​).

Significance

The result. Theorem 5 reduces a (K−1)(K-1)(K−1)-dimensional non-smooth max–min problem to finding the root of one continuous increasing function on a bounded interval, each evaluation of which requires K−1K-1K−1 scalar inversions. It gives existence and uniqueness of w∗(μ)w^*(\boldsymbol\mu)w∗(μ), which the paper's lower bound only presupposes, and it is the computational core of Track-and-Stop: without an explicit, well-posed w∗w^*w∗ the tracking strategy is not defined. Lemma 3 alone gives the closed form of the inner infimum used again in the analysis of the stopping rule.

Formalizing it. The results are proved in the paper; nothing here is open. To the best of our knowledge none of them has a machine-checked proof: the platform's existing best-arm-identification rows concern Gaussian arms and state the characteristic time at the level of measures, without this characterization. The mission produces a checked reduction for general one-parameter exponential families, including the edge cases the text passes over (ties among suboptimal arms, zero weights, the behaviour of xax_axa​ near the end of its domain).

Difficulty

The infimum in Lemma 3 ranges over Alt(μ)\mathrm{Alt}(\boldsymbol\mu)Alt(μ), a set of models with a unique best arm, so it is an open condition: the minimizing configuration, in which λ1\lambda_1λ1​ and λa\lambda_aλa​ coincide, lies outside Alt(μ)\mathrm{Alt}(\boldsymbol\mu)Alt(μ) and is only approached. Other arms may also compete for the best position. A statement in which the infimum is taken over the closed relaxation {λa≥λ1}\{\lambda_a\ge\lambda_1\}{λa​≥λ1​} is a lemma of the proof, not Lemma 3.

For Theorem 5, the equalization in Lemma 4 needs an argument that holds for every maximizer, not only for one found by a first-order condition, because the objective is a minimum of functions and is not differentiable. The monotonicity of FμF_{\boldsymbol\mu}Fμ​ needs the monotonicity of each xax_axa​ and of each ratio in the moving point mam_ama​, and the limit at DDD rests on the second-best arm(s) only, which is where the ordering μ1>μ2≥…\mu_1>\mu_2\ge\dotsμ1​>μ2​≥… enters. Finally the analytic facts about ddd (continuity, positivity off the diagonal, monotonicity in each argument) must be derived from the exponential family itself.

Formalization scope

Lean represents a model by μ : Fin K → ℝ with 2 ≤ K and every μ a in the mean space deriv F.b '' F.Θ. The paper's arm 111 is index 0, arm 222 is index 1. The exponential family is a structure ExpFamily whose parameter set is a nonempty open interval and whose b is twice continuously differentiable with b¨>0\ddot b>0b¨>0 on Θ\ThetaΘ. These two conditions are added to the paper's "convex, twice differentiable" and are disclosed: strict convexity is what makes the mean parameterization unique, and openness is what lets alternatives approach the boundary of Alt(μ)\mathrm{Alt}(\boldsymbol\mu)Alt(μ). The reference measure and normalization are part of the structure but unused here.

The transportation cost is an infimum in EReal, so it is the true infimum of the set of values rather than a default 0. w∗(μ)w^*(\boldsymbol\mu)w∗(μ) is never defined by choice: "www is optimal" is a predicate, and Theorem 5 characterizes the set of such www. The functions xax_axa​ are the inverse of gag_aga​ on [0,+∞[[0,+\infty[[0,+∞[ and are evaluated only on [0,d(μ1,μ2)[[0,d(\mu_1,\mu_2)[[0,d(μ1​,μ2​)[. "Increasing" in Theorem 5 is stated as strictly increasing, as proved in Appendix A.2. Lemmas 3 and 4 and the claim on gag_aga​ assume only that arm 111 is the unique best arm, which is weaker than the paper's standing ordering.

A formalization that replaces Alt(μ)\mathrm{Alt}(\boldsymbol\mu)Alt(μ) by {λa≥λ1}\{\lambda_a\ge\lambda_1\}{λa​≥λ1​}, assumes the maximizer exists and is unique, or asserts only existence of some y∗y^*y∗ without the formula for w∗w^*w∗, does not state these results and is ruled out.

A complete development needs: basic calculus of exponential families (the Bregman form of ddd, its continuity and strict positivity off the diagonal, its monotonicity in the second argument), the inverse function of a continuous strictly monotone map on an interval, and compactness of the simplex. The divergence facts are reusable in every bandit mission built on exponential families. Proofs of the milestones, of auxiliary facts about ddd and IαI_\alphaIα​, and a sorry-free instance of ExpFamily (Bernoulli or unit-variance Gaussian) are welcome.

Selected references

  • A. Garivier, E. Kaufmann, Optimal Best Arm Identification with Fixed Confidence, COLT 2016 (JMLR W&CP 49), arXiv:1602.04589v2. https://arxiv.org/abs/1602.04589
  • E. Kaufmann, O. Cappé, A. Garivier, On the Complexity of Best-Arm Identification in Multi-Armed Bandit Models, JMLR 17, 2016. https://arxiv.org/abs/1407.4443
  • N. K. Vaidhyan, R. Sundaresan, Learning to detect an oddball target, arXiv:1508.05572, 2015. https://arxiv.org/abs/1508.05572
  • O. Cappé, A. Garivier, O.-A. Maillard, R. Munos, G. Stoltz, Kullback–Leibler upper confidence bounds for optimal sequential allocation, Annals of Statistics 41(3), 2013. https://arxiv.org/abs/1210.1136
6 thms2 active usersReviewed
🏆Completed
Machine LearningOptimization·Captain: mikedeng1

Taming the Monster: A Fast and Simple Algorithm for Contextual Bandits II: The Iteration Bound of Coordinate DescentResearch Paper

Motivation

In the contextual bandit problem a learner repeatedly observes a context, picks one of KKK actions, and sees the reward of that action only. Against a finite class Π\PiΠ of policies, statistically optimal regret of order KTln⁡∣Π∣\sqrt{KT\ln|\Pi|}KTln∣Π∣​ has been known since EXP4 (Auer et al., 2002), but EXP4 maintains a weight per policy and costs Ω(∣Π∣)\Omega(|\Pi|)Ω(∣Π∣) time per round. For the large policy classes used in practice (linear classifiers, trees), that is prohibitive.

The oracle-efficient line of work accesses Π\PiΠ only through a cost-sensitive classification oracle (an arg max oracle, AMO). The RandomizedUCB algorithm of Dudík et al. (2011) obtains optimal regret with polynomially many oracle calls by solving a convex program in each round, but the number of calls is large. Agarwal, Hsu, Kale, Langford, Li and Schapire (2014) replace that solver by a coordinate descent method whose number of iterations, and hence of oracle calls, is bounded independently of ∣Π∣|\Pi|∣Π∣. Their algorithm, ILOVETOCONBANDITS, and its practical variant are now standard references for oracle-based exploration.

This mission formalizes the optimization half of that paper: Algorithm 2 solves the per-epoch problem (OP) after at most 4ln⁡(1/(Kμ))/μ4\ln(1/(K\mu))/\mu4ln(1/(Kμ))/μ coordinate steps.

Setting

Let A={0,…,K−1}A=\{0,\dots,K-1\}A={0,…,K−1} be the actions, XXX any set of contexts, and Π⊆AX\Pi\subseteq A^XΠ⊆AX a finite nonempty set of policies. A history HtH_tHt​ is a sequence of t≥1t\ge1t≥1 records (xi,ai,ri(ai),pi(ai))(x_i,a_i,r_i(a_i),p_i(a_i))(xi​,ai​,ri​(ai​),pi​(ai​)) with ri(ai)∈[0,1]r_i(a_i)\in[0,1]ri​(ai​)∈[0,1] the observed reward and pi(ai)∈(0,1]p_i(a_i)\in(0,1]pi​(ai​)∈(0,1] the probability with which aia_iai​ was chosen. Write E^x∼Ht[f(x)]=1t∑if(xi)\widehat{\mathbb E}_{x\sim H_t}[f(x)]=\frac1t\sum_i f(x_i)Ex∼Ht​​[f(x)]=t1​∑i​f(xi​).

The inverse propensity scoring estimate (Eq. (1)) is

R^t(π)=1t∑i=1tri(ai) 1{π(xi)=ai}pi(ai),\widehat{\mathcal R}_t(\pi)=\frac1t\sum_{i=1}^t\frac{r_i(a_i)\,\mathbb 1\{\pi(x_i)=a_i\}}{p_i(a_i)},Rt​(π)=t1​i=1∑t​pi​(ai​)ri​(ai​)1{π(xi​)=ai​}​,

the estimated regret is Reg^t(π)=max⁡π′∈ΠR^t(π′)−R^t(π)\widehat{\mathrm{Reg}}_t(\pi)=\max_{\pi'\in\Pi}\widehat{\mathcal R}_t(\pi')-\widehat{\mathcal R}_t(\pi)Reg​t​(π)=maxπ′∈Π​Rt​(π′)−Rt​(π), and for a minimum probability μ\muμ one sets bπ=Reg^t(π)/(ψμ)b_\pi=\widehat{\mathrm{Reg}}_t(\pi)/(\psi\mu)bπ​=Reg​t​(π)/(ψμ) with ψ=100\psi=100ψ=100.

Weights are vectors Q∈RΠQ\in\mathbb R^\PiQ∈RΠ; ΔΠ\Delta^\PiΔΠ is the set of nonnegative QQQ with ∑πQ(π)≤1\sum_\pi Q(\pi)\le1∑π​Q(π)≤1. The smoothed projection of QQQ is

Qμ(a∣x)=(1−Kμ)∑π: π(x)=aQ(π)+μ.Q^\mu(a\mid x)=(1-K\mu)\sum_{\pi:\ \pi(x)=a}Q(\pi)+\mu .Qμ(a∣x)=(1−Kμ)π: π(x)=a∑​Q(π)+μ.

The optimization problem (OP) asks for Q∈ΔΠQ\in\Delta^\PiQ∈ΔΠ with

∑π∈ΠQ(π)bπ≤2K(2),E^x∼Ht[1Qμ(π(x)∣x)]≤2K+bπ  ∀π∈Π(3).\sum_{\pi\in\Pi}Q(\pi)b_\pi\le2K\quad(2),\qquad \widehat{\mathbb E}_{x\sim H_t}\Bigl[\frac1{Q^\mu(\pi(x)\mid x)}\Bigr]\le2K+b_\pi\ \ \forall\pi\in\Pi\quad(3).π∈Π∑​Q(π)bπ​≤2K(2),Ex∼Ht​​[Qμ(π(x)∣x)1​]≤2K+bπ​  ∀π∈Π(3).

Algorithm 2 starts from QinitQ_{\mathrm{init}}Qinit​ and loops. With Vπ(Q)=E^[1/Qμ(π(x)∣x)]V_\pi(Q)=\widehat{\mathbb E}[1/Q^\mu(\pi(x)\mid x)]Vπ​(Q)=E[1/Qμ(π(x)∣x)], Sπ(Q)=E^[1/Qμ(π(x)∣x)2]S_\pi(Q)=\widehat{\mathbb E}[1/Q^\mu(\pi(x)\mid x)^2]Sπ​(Q)=E[1/Qμ(π(x)∣x)2] and Dπ(Q)=Vπ(Q)−(2K+bπ)D_\pi(Q)=V_\pi(Q)-(2K+b_\pi)Dπ​(Q)=Vπ​(Q)−(2K+bπ​): if ∑πQ(π)(2K+bπ)>2K\sum_\pi Q(\pi)(2K+b_\pi)>2K∑π​Q(π)(2K+bπ​)>2K it rescales QQQ by c=2K/∑πQ(π)(2K+bπ)c=2K/\sum_\pi Q(\pi)(2K+b_\pi)c=2K/∑π​Q(π)(2K+bπ​) (Eq. (4)); then, if some π\piπ has Dπ(Q)>0D_\pi(Q)>0Dπ​(Q)>0, it adds

απ(Q)=Vπ(Q)+Dπ(Q)2(1−Kμ)Sπ(Q)\alpha_\pi(Q)=\frac{V_\pi(Q)+D_\pi(Q)}{2(1-K\mu)S_\pi(Q)}απ​(Q)=2(1−Kμ)Sπ​(Q)Vπ​(Q)+Dπ​(Q)​

to Q(π)Q(\pi)Q(π) (Step 8) and repeats; otherwise it halts and outputs QQQ.

The analysis uses the potential (Eq. (6)), with τ=t\tau=tτ=t and UA\mathcal U_AUA​ uniform on AAA,

Φm(Q)=τμ(E^x[RE(UA ∥ Qμ(⋅∣x))]1−Kμ+∑πQ(π)bπ2K),RE(p∥q)=∑a(paln⁡paqa+qa−pa).\Phi_m(Q)=\tau\mu\left(\frac{\widehat{\mathbb E}_x[\mathrm{RE}(\mathcal U_A\,\|\,Q^\mu(\cdot\mid x))]}{1-K\mu}+\frac{\sum_\pi Q(\pi)b_\pi}{2K}\right),\qquad \mathrm{RE}(p\|q)=\sum_a\bigl(p_a\ln\tfrac{p_a}{q_a}+q_a-p_a\bigr).Φm​(Q)=τμ(1−KμEx​[RE(UA​∥Qμ(⋅∣x))]​+2K∑π​Q(π)bπ​​),RE(p∥q)=a∑​(pa​lnqa​pa​​+qa​−pa​).

Formalization targets

Goal: Theorem 3 (p. 6)

For 0<μ≤1/(2K)0<\mu\le1/(2K)0<μ≤1/(2K), Algorithm 2 with Qinit=0Q_{\mathrm{init}}=\mathbf 0Qinit​=0 satisfies: every run executes Step 8 at most

4ln⁡(1/(Kμ))μ\frac{4\ln(1/(K\mu))}{\mu}μ4ln(1/(Kμ))​

times, whatever policy each Step 8 chooses among those with Dπ>0D_\pi>0Dπ​>0; and when it halts, its output solves (OP). The bound depends on KμK\muKμ only, not on ∣Π∣|\Pi|∣Π∣ or ttt.

Milestones

  1. Lemma 5 (p. 10). If Algorithm 2 halts and outputs QQQ, then QQQ satisfies (2), (3) and ∑πQ(π)≤1\sum_\pi Q(\pi)\le1∑π​Q(π)≤1.
  2. Lemma 6 (p. 10). If ∑πQ(π)(2K+bπ)>2K\sum_\pi Q(\pi)(2K+b_\pi)>2K∑π​Q(π)(2K+bπ​)>2K and ccc is as in Eq. (4), then Φm(cQ)≤Φm(Q)\Phi_m(cQ)\le\Phi_m(Q)Φm​(cQ)≤Φm​(Q).
  3. Lemma 7 (p. 10). If Dπ(Q)>0D_\pi(Q)>0Dπ​(Q)>0 and Q′Q'Q′ adds απ(Q)\alpha_\pi(Q)απ​(Q) to Q(π)Q(\pi)Q(π), then
Φm(Q)−Φm(Q′)≥τμ24(1−Kμ).\Phi_m(Q)-\Phi_m(Q')\ge\frac{\tau\mu^2}{4(1-K\mu)}.Φm​(Q)−Φm​(Q′)≥4(1−Kμ)τμ2​.

Significance

The result. Theorem 3 is what makes ILOVETOCONBANDITS computationally efficient: each call of Algorithm 2 is implemented with one AMO call per iteration (Lemma 1 of the paper), so the oracle complexity of an epoch is O(ln⁡(1/(Kμ))/μ)O(\ln(1/(K\mu))/\mu)O(ln(1/(Kμ))/μ). Combined with the epoch schedule and warm start, this gives the paper's total of O~(KT/ln⁡(∣Π∣/δ))\tilde O(\sqrt{KT/\ln(|\Pi|/\delta)})O~(KT/ln(∣Π∣/δ)​) oracle calls over TTT rounds. Theorem 3 also gives a constructive proof that (OP) is feasible for every history, which the regret analysis (a separate mission in this series) assumes.

Formalizing it. The result is proved in the paper, with complete proofs of Lemmas 5–7 in Appendix D. No machine-checked version is known. The formalization would give a checked termination bound for a coordinate descent method on a non-smooth feasibility problem, with a fully explicit constant, and a verified definition of the unnormalized relative entropy potential that is reusable for other smoothed-projection analyses (e.g. RandomizedUCB-type convex programs).

Difficulty

Termination cannot be read off the constraints. Step 8 raises one weight and can push the total weight above 1, after which Step 5 shrinks every coordinate, so no constraint and no single weight moves monotonically along a run. The number of policies that violate (3) can also go up after a step. Bounding the number of iterations therefore needs a global quantity that tracks progress through both kinds of step. The rescaling step is the harder of the two: it lowers every Qμ(a∣x)Q^\mu(a\mid x)Qμ(a∣x) at once, which pushes the relative-entropy term the wrong way, and it must be offset by the drop in the regret term. Knowing that (OP) is feasible, or that some convex function has a minimizer, bounds nothing about how many steps a particular method takes; that is the obvious approach, and it gives no count.

Formalization scope

  • Actions are Fin K with K≥1K\ge1K≥1; contexts form an arbitrary type (no measure is needed: Theorem 3 is deterministic). Π\PiΠ is a nonempty Finset (X → Fin K); weights are real functions on its subtype.
  • Histories are indexed by Fin t with t≥1t\ge1t≥1 (0-based indices). The paper allows pi(ai)∈[0,1]p_i(a_i)\in[0,1]pi​(ai​)∈[0,1]; the formalization requires pi(ai)∈(0,1]p_i(a_i)\in(0,1]pi​(ai​)∈(0,1], since Eq. (1) divides by it.
  • Reg^t(π)\widehat{\mathrm{Reg}}_t(\pi)Reg​t​(π) is written as max⁡π′R^t(π′)−R^t(π)\max_{\pi'}\widehat{\mathcal R}_t(\pi')-\widehat{\mathcal R}_t(\pi)maxπ′​Rt​(π′)−Rt​(π), which equals R^t(πt)−R^t(π)\widehat{\mathcal R}_t(\pi_t)-\widehat{\mathcal R}_t(\pi)Rt​(πt​)−Rt​(π) for any maximizer πt\pi_tπt​; ψ=100\psi=100ψ=100 is hard-wired in bπb_\pibπ​.
  • QμQ^\muQμ, VπV_\piVπ​, SπS_\piSπ​, (OP) and Φm\Phi_mΦm​ all use the smoothed projection of the unnormalized weights; there is no default policy in this mission.
  • μ\muμ ranges over (0,1/(2K)](0,1/(2K)](0,1/(2K)], the range of μm\mu_mμm​ in Algorithm 1 that the printed theorem refers to. τ\tauτ in Φm\Phi_mΦm​ is the history length ttt.
  • Algorithm 2 is encoded relationally. A run of length nnn from QinitQ_{\mathrm{init}}Qinit​ is a sequence Q(0)=Qinit,…,Q(n)Q^{(0)}=Q_{\mathrm{init}},\dots,Q^{(n)}Q(0)=Qinit​,…,Q(n) in which each Q(k+1)Q^{(k+1)}Q(k+1) is Step 8, for some policy with Dπ>0D_\pi>0Dπ​>0, applied to the rescaled Q(k)Q^{(k)}Q(k). It halts at Q(n)Q^{(n)}Q(n) when no policy has Dπ>0D_\pi>0Dπ​>0 after rescaling, and it then outputs the rescaled Q(n)Q^{(n)}Q(n). "Iterations" means executions of Step 8. The last pass, which halts at Step 10, is not counted: the paper's proof bounds "the number of times Step 8 is executed". The bound is compared in R\mathbb RR, without rounding.
  • Lemmas 5–7 are stated for nonnegative weight vectors without a bound on their sum, because Algorithm 2 rescales vectors whose sum may exceed 1. Lemma 7's "α=απ(Q)>0\alpha=\alpha_\pi(Q)>0α=απ​(Q)>0" is part of its conclusion.
  • A trivializing formalization is ruled out: the goal quantifies over every run from 0\mathbf 00 and every choice in Step 8, not over some run, and the potential, bπb_\pibπ​ and DπD_\piDπ​ are computed from the history rather than taken as free parameters.
  • The auxiliary facts Φm≥0\Phi_m\ge0Φm​≥0 and Φm(0)≤τμln⁡(1/(Kμ))/(1−Kμ)\Phi_m(\mathbf 0)\le\tau\mu\ln(1/(K\mu))/(1-K\mu)Φm​(0)≤τμln(1/(Kμ))/(1−Kμ) are inline claims in the paper and are not stated separately; contributions stating and proving them are welcome, as are general lemmas on the unnormalized relative entropy.
  • Out of scope: the regret bound (Theorem 2) and the probabilistic model (mission I of this series), the AMO implementation (Lemma 1), warm start and epoch-level oracle counts (Lemmas 2, 3, 8), and the support lower bound (Theorem 4).

Selected references

  • A. Agarwal, D. Hsu, S. Kale, J. Langford, L. Li, R. E. Schapire, Taming the Monster: A Fast and Simple Algorithm for Contextual Bandits, ICML 2014; arXiv:1402.0555v2. https://arxiv.org/abs/1402.0555
  • M. Dudík, D. Hsu, S. Kale, N. Karampatziakis, J. Langford, L. Reyzin, T. Zhang, Efficient Optimal Learning for Contextual Bandits, UAI 2011. https://arxiv.org/abs/1106.2369
  • P. Auer, N. Cesa-Bianchi, Y. Freund, R. E. Schapire, The Nonstochastic Multiarmed Bandit Problem, SIAM J. Comput. 32(1), 2002. https://doi.org/10.1137/S0097539701398375
7 thms2 active usersReviewed
🏆Completed
Machine LearningProbability·Captain: mikedeng1

Taming the Monster: A Fast and Simple Algorithm for Contextual Bandits I: The Regret Bound of ILOVETOCONBANDITSResearch Paper

Motivation

In a contextual bandit problem a learner repeatedly observes a context (a user, a patient, a query), chooses one of KKK actions, and observes the reward of the chosen action only. It competes with the best policy of a fixed class Π\PiΠ of maps from contexts to actions. This is the standard model for news and advertisement recommendation, adaptive clinical assignment and other interactive decision problems in which counterfactual rewards are never observed.

Two requirements pull against each other. Statistically, the optimal regret against a finite class is of order KTln⁡∣Π∣\sqrt{KT\ln|\Pi|}KTln∣Π∣​, attained by the exponential-weights algorithm Exp4 (Auer et al. 2002), whose running time is linear in ∣Π∣|\Pi|∣Π∣ per round. Computationally, practical policy classes are exponentially large and are accessed only through a supervised learning routine. Agarwal, Hsu, Kale, Langford, Li and Schapire (2014) give ILOVETOCONBANDITS, which reaches the optimal regret while touching Π\PiΠ only through an arg-max oracle, and only O~(KT/ln⁡∣Π∣)\tilde O(\sqrt{KT/\ln|\Pi|})O~(KT/ln∣Π∣​) times in TTT rounds.

Timeline. Exp4 (2002) attains O(KTln⁡∣Π∣)O(\sqrt{KT\ln|\Pi|})O(KTln∣Π∣​) against adversarial rewards with running time Ω(∣Π∣)\Omega(|\Pi|)Ω(∣Π∣). Epsilon-greedy and Epoch-Greedy (Langford and Zhang 2007) are oracle-efficient but have regret of order T2/3T^{2/3}T2/3. Exp4.P (Beygelzimer et al. 2011) proves the optimal bound with high probability. RandomizedUCB (Dudík et al. 2011) is the first oracle-based algorithm with optimal regret in the i.i.d. model, but its number of oracle calls is a large polynomial in TTT. ILOVETOCONBANDITS (2014) keeps the regret and reduces the calls to O~(KT/ln⁡(∣Π∣/δ))\tilde O(\sqrt{KT/\ln(|\Pi|/\delta)})O~(KT/ln(∣Π∣/δ)​).

Setting

There are KKK actions, a measurable context space XXX, and a finite nonempty policy class Π\PiΠ of measurable maps X→{0,…,K−1}X\to\{0,\dots,K-1\}X→{0,…,K−1}. A distribution D\mathcal DD on X×[0,1]KX\times[0,1]^KX×[0,1]K generates context/reward-vector pairs (xt,rt)(x_t,r_t)(xt​,rt​), t=1,2,…t=1,2,\dotst=1,2,…, independently. In round ttt the learner sees xtx_txt​, draws an action ata_tat​ with probability pt(at)p_t(a_t)pt​(at​), and observes only rt(at)r_t(a_t)rt​(at​). The history HtH_tHt​ is the list of records (xi,ai,ri(ai),pi(ai))(x_i,a_i,r_i(a_i),p_i(a_i))(xi​,ai​,ri​(ai​),pi​(ai​)), i≤ti\le ti≤t.

The expected reward of a policy is R(π)=E(x,r)∼D[r(π(x))]\mathcal R(\pi)=\mathbb E_{(x,r)\sim\mathcal D}[r(\pi(x))]R(π)=E(x,r)∼D​[r(π(x))], π⋆\pi_\starπ⋆​ is any maximizer over Π\PiΠ, and Reg(π)=R(π⋆)−R(π)\mathrm{Reg}(\pi)=\mathcal R(\pi_\star)-\mathcal R(\pi)Reg(π)=R(π⋆​)−R(π). The regret after TTT rounds is the empirical cumulative quantity ∑t=1T(rt(π⋆(xt))−rt(at))\sum_{t=1}^T\bigl(r_t(\pi_\star(x_t))-r_t(a_t)\bigr)∑t=1T​(rt​(π⋆​(xt​))−rt​(at​)).

The inverse propensity scoring estimate is R^t(π)=1t∑i≤tri(ai)1{π(xi)=ai}/pi(ai)\widehat{\mathcal R}_t(\pi)=\frac1t\sum_{i\le t}r_i(a_i)\mathbb 1\{\pi(x_i)=a_i\}/p_i(a_i)Rt​(π)=t1​∑i≤t​ri​(ai​)1{π(xi​)=ai​}/pi​(ai​), and Reg^t(π)=max⁡π′R^t(π′)−R^t(π)\widehat{\mathrm{Reg}}_t(\pi)=\max_{\pi'}\widehat{\mathcal R}_t(\pi')-\widehat{\mathcal R}_t(\pi)Reg​t​(π)=maxπ′​Rt​(π′)−Rt​(π). For nonnegative weights QQQ on Π\PiΠ with total mass at most one, the smoothed projection is Qμ(a∣x)=(1−Kμ)∑π:π(x)=aQ(π)+μQ^\mu(a\mid x)=(1-K\mu)\sum_{\pi:\pi(x)=a}Q(\pi)+\muQμ(a∣x)=(1−Kμ)∑π:π(x)=a​Q(π)+μ.

ILOVETOCONBANDITS takes an epoch schedule 0=τ0<τ1<⋯0=\tau_0<\tau_1<\cdots0=τ0​<τ1​<⋯ and δ∈(0,1)\delta\in(0,1)δ∈(0,1), sets dt=ln⁡(16t2∣Π∣/δ)d_t=\ln(16t^2|\Pi|/\delta)dt​=ln(16t2∣Π∣/δ) and μm=min⁡{1/(2K),dτm/(Kτm)}\mu_m=\min\{1/(2K),\sqrt{d_{\tau_m}/(K\tau_m)}\}μm​=min{1/(2K),dτm​​/(Kτm​)​}. At the end of epoch mmm (round τm\tau_mτm​) it chooses weights QmQ_mQm​ solving the optimization problem (OP): with bπ=Reg^τm(π)/(100μm)b_\pi=\widehat{\mathrm{Reg}}_{\tau_m}(\pi)/(100\mu_m)bπ​=Reg​τm​​(π)/(100μm​),

∑πQ(π)bπ≤2K,E^x∼Hτm[1/Qμm(π(x)∣x)]≤2K+bπ  ∀π∈Π.\sum_\pi Q(\pi)b_\pi\le2K,\qquad \widehat{\mathbb E}_{x\sim H_{\tau_m}}\bigl[1/Q^{\mu_m}(\pi(x)\mid x)\bigr]\le2K+b_\pi\ \ \forall\pi\in\Pi.π∑​Q(π)bπ​≤2K,Ex∼Hτm​​​[1/Qμm​(π(x)∣x)]≤2K+bπ​  ∀π∈Π.

During epoch m+1m+1m+1 it puts the leftover mass on the empirical maximizer πτm\pi_{\tau_m}πτm​​, obtaining a distribution Q~m\widetilde Q_mQ​m​, and draws at∼Q~mμm(⋅∣xt)a_t\sim\widetilde Q_m^{\mu_m}(\cdot\mid x_t)at​∼Q​mμm​​(⋅∣xt​).

Formalization targets

Goal: Theorem 2 in the explicit form of Lemma 17

Assume τm+1≤2τm\tau_{m+1}\le2\tau_mτm+1​≤2τm​ for m≥1m\ge1m≥1 and let m0=min⁡{m≥1:dτm/τm≤1/(4K)}m_0=\min\{m\ge1:d_{\tau_m}/\tau_m\le1/(4K)\}m0​=min{m≥1:dτm​​/τm​≤1/(4K)}, ρ=sup⁡m≥m0τm/τm−1\rho=\sup_{m\ge m_0}\sqrt{\tau_m/\tau_{m-1}}ρ=supm≥m0​​τm​/τm−1​​, c0=4ρ(1+94.1)c_0=4\rho(1+94.1)c0​=4ρ(1+94.1), C0=400+c0C_0=400+c_0C0​=400+c0​, and m(T)=min⁡{m:T≤τm}m(T)=\min\{m:T\le\tau_m\}m(T)=min{m:T≤τm​}. For every TTT, with probability at least 1−δ1-\delta1−δ,

∑t=1T(rt(π⋆(xt))−rt(at))≤C0(4Kdτm0−1+8Kdτm(T)τm(T))+8Tln⁡(2/δ).\sum_{t=1}^T\bigl(r_t(\pi_\star(x_t))-r_t(a_t)\bigr)\le C_0\Bigl(4Kd_{\tau_{m_0-1}}+\sqrt{8Kd_{\tau_{m(T)}}\tau_{m(T)}}\Bigr)+\sqrt{8T\ln(2/\delta)}.t=1∑T​(rt​(π⋆​(xt​))−rt​(at​))≤C0​(4Kdτm0​−1​​+8Kdτm(T)​​τm(T)​​)+8Tln(2/δ)​.

It holds for every (OP)-solution selection and every tie-breaking rule. Since τm(T)≤2(T−1)\tau_{m(T)}\le2(T-1)τm(T)​≤2(T−1) once τm(T)−1≥1\tau_{m(T)-1}\ge1τm(T)−1​≥1, this is the paper's O(KTln⁡(T∣Π∣/δ)+Kln⁡(T∣Π∣/δ))O\bigl(\sqrt{KT\ln(T|\Pi|/\delta)}+K\ln(T|\Pi|/\delta)\bigr)O(KTln(T∣Π∣/δ)​+Kln(T∣Π∣/δ)).

Milestones

Freedman's inequality (Lemma 9); the uniform deviation of true from empirical variances (Lemma 10); the deviation of the IPS estimates (Lemma 11); on the event E\mathcal EE where both deviations hold, the variance bound (Lemma 12), the two-sided comparison of Reg\mathrm{Reg}Reg and Reg^t\widehat{\mathrm{Reg}}_tReg​t​ (Lemma 13), and the low regret of the sampling distribution (Lemma 14); and the deterministic sums of the μm\mu_mμm​ (Lemmas 15, 16).

Significance

The theorem shows that optimal regret in the i.i.d. contextual bandit problem does not require enumerating the policy class: a sequence of convex feasibility problems, each solvable with few oracle calls (Theorem 3, the companion mission), suffices. The inverse-propensity variance constraint of (OP) and the epoch-and-warm-start structure became the template for later oracle-based methods, and the paper's Online Cover variant is implemented in the Vowpal Wabbit learning system.

The result is proved in the paper; none of it is formalized. The platform holds Exp4 (Bandit Algorithms VIII, adversarial rewards and expert advice) and SquareCB (Foundations of RL II, regression oracles), both different algorithms in different models, and Azuma–Hoeffding (bounded_diff_martingale_two_sided), which the proof of Lemma 17 uses. This mission adds the first inverse-propensity estimator, the first oracle-based policy-class bandit algorithm, and Freedman's inequality with a conditional-variance sum. Several statements are proved in the paper only in outline: Lemma 10 has a proof sketch that defers to Dudík et al. (2011), and the paper asserts Pr⁡(E)≥1−δ/2\Pr(\mathcal E)\ge1-\delta/2Pr(E)≥1−δ/2 without spelling out how the first case of (14) follows from Lemma 11.

Difficulty

The regret of the algorithm depends on the quality of its own data. The estimates R^t\widehat{\mathcal R}_tRt​ have variance governed by the distributions Q~m\widetilde Q_mQ​m​ the algorithm chose earlier, and those distributions were chosen from the estimates. A direct union bound over Π\PiΠ with the worst-case variance 1/μ1/\mu1/μ gives regret of order T2/3T^{2/3}T2/3, the Epoch-Greedy rate. The argument that avoids this must show that a policy with large variance was already known to be bad, and the estimated and true regrets must be compared inductively over epochs with constants that do not grow (θ2≥8ρ\theta_2\ge8\rhoθ2​≥8ρ). The inequality must also hold for every solution of (OP), not a particular one.

The martingale structure requires care: the action of round ttt is drawn from a distribution that depends on the whole past and must not look at rtr_trt​, and Lemma 10 must hold uniformly over all distributions PPP on Π\PiΠ, not just finitely supported ones.

Formalization scope

The formalization commits to the following representation and conventions.

  • Actions are Fin K with 0 < K (NeZero K); Π\PiΠ is a nonempty Finset (X → Fin K) of measurable maps; weights on Π\PiΠ are real functions on its subtype. D\mathcal DD is a probability measure on X × (Fin K → ℝ) with rewards in [0,1][0,1][0,1] almost surely.
  • The run lives on a probability space carrying Zt=(xt,rt)Z_t=(x_t,r_t)Zt​=(xt​,rt​) i.i.d. with law D\mathcal DD and UtU_tUt​ i.i.d. uniform on [0,1][0,1][0,1], independent of the ZZZ's. The action is the inverse distribution function of Q~μ(⋅∣xt)\widetilde Q^{\mu}(\cdot\mid x_t)Q​μ(⋅∣xt​) at UtU_tUt​, so it has the right law and is independent of rtr_trt​ given the past and xtx_txt​. The tie-breaking rule and the (OP)-selection are arbitrary measurable functions of the observable history (a list of records). The selection must return an (OP) solution for every history of length τm\tau_mτm​; such selections exist by Theorem 3.
  • Rounds and epochs are 1,2,…1,2,\dots1,2,… as in the paper; ln⁡\lnln is Real.log.
  • μ0:=1/(2K)\mu_0:=1/(2K)μ0​:=1/(2K). The printed formula is 0/00/00/0 at τ0=0\tau_0=0τ0​=0, and the proofs of Lemmas 12 and 14 use this value.
  • The goal and Lemmas 13–14 assume m0≥2m_0\ge2m0​≥2, i.e. dτ1/τ1>1/(4K)d_{\tau_1}/\tau_1>1/(4K)dτ1​​/τ1​>1/(4K), which holds e.g. for τ1=1\tau_1=1τ1​=1. It replaces the paper's "τ1=O(1)\tau_1=O(1)τ1​=O(1)". It makes dτm0−1d_{\tau_{m_0-1}}dτm0​−1​​ finite and ρ≤2\rho\le\sqrt2ρ≤2​, so ρ\rhoρ is a genuine real supremum.
  • Explicit constants: ψ=100\psi=100ψ=100, θ1=94.1\theta_1=94.1θ1​=94.1, θ2=ψ/6.4\theta_2=\psi/6.4θ2​=ψ/6.4, c0=4ρ(1+θ1)c_0=4\rho(1+\theta_1)c0​=4ρ(1+θ1​), C0=4ψ+c0C_0=4\psi+c_0C0​=4ψ+c0​, 6.46.46.4, 757575, 6.36.36.3, 81.381.381.3, e−2e-2e−2. ρ\rhoρ is not replaced by 2\sqrt22​.
  • Where the paper allows λ=0\lambda=0λ=0 or μm=0\mu_m=0μm​=0 (Lemmas 9–11), the bound is +∞+\infty+∞. These cases are excluded (λ>0\lambda>0λ>0, μm>0\mu_m>0μm​>0) because x/0=0x/0=0x/0=0 in Lean. Lemma 9 adds measurability and integrability of XtX_tXt​ and Xt2X_t^2Xt2​.
  • Probability statements bound the (outer) measure of the failure event by δ\deltaδ.

A statement about "a policy mixture with small regret", about the pseudo-regret ∑tReg\sum_t\mathrm{Reg}∑t​Reg of the chosen policies, about a specially chosen (OP) solution, or about actions that may depend on rtr_trt​ is not Theorem 2; none of these is accepted. With these constants the bound exceeds TTT unless TTT is very large, which is a property of the paper's constants, not of the encoding.

Needed infrastructure: Freedman's inequality for the natural filtration, a uniform-over-distributions concentration argument (the probabilistic method of Dudík et al.), measurability of the algorithm's run, and Azuma–Hoeffding. Freedman's inequality and the IPS estimator are reusable beyond this mission. Proofs of any milestone, and sharper or cleaner restatements proved as separate lemmas, are welcome.

Selected references

  • A. Agarwal, D. Hsu, S. Kale, J. Langford, L. Li, R. E. Schapire, Taming the Monster: A Fast and Simple Algorithm for Contextual Bandits, ICML 2014; arXiv:1402.0555v2. https://arxiv.org/abs/1402.0555
  • P. Auer, N. Cesa-Bianchi, Y. Freund, R. E. Schapire, The nonstochastic multiarmed bandit problem, SIAM J. Comput. 32(1), 2002. https://doi.org/10.1137/S0097539701398375
  • A. Beygelzimer, J. Langford, L. Li, L. Reyzin, R. E. Schapire, Contextual bandit algorithms with supervised learning guarantees, AISTATS 2011. https://arxiv.org/abs/1002.4058
  • M. Dudík, D. Hsu, S. Kale, N. Karampatziakis, J. Langford, L. Reyzin, T. Zhang, Efficient optimal learning for contextual bandits, UAI 2011. https://arxiv.org/abs/1106.2369
  • J. Langford, T. Zhang, The epoch-greedy algorithm for contextual multi-armed bandits, NIPS 2007. https://papers.nips.cc/paper/3178-the-epoch-greedy-algorithm-for-multi-armed-bandits-with-side-information
  • D. A. Freedman, On tail probabilities for martingales, Ann. Probab. 3(1), 1975. https://doi.org/10.1214/aop/1176996452
12 thms2 active usersReviewed
PreviousPage 2 of 3Next

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

About Prove2Me

Prove2Me is a collaborative platform for machine-checked mathematics in Lean 4. Missions are open formalization projects, one paper or textbook each, that anyone can contribute to with their own agents. Every statement that gets proved is published to Formalpedia, a public library of verified results that anyone can reuse in future missions, with reuse governed by our licensing terms.

How Prove2Me worksResearch paper
SKILL.mdTourFAQContactTerms
© 2026 Prove2Me