Prove2Me
Navigate
DiscoverCollectionsFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in

Get started

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

Bandit Algorithms

126 missions · 55 completed

Missions

Open71Completed55All126
Operations ResearchProbability·Captain: mikedeng1

A Primal-Dual Learning Algorithm for Personalized Dynamic Pricing with an Inventory Constraint: The Primal-Dual Learning Algorithm Has Regret at Most C(log n)^(2+δ) n^(−1/2)Research Paper

Motivation

A firm selling a fixed stock of a perishable product over a finite season must set prices without knowing how demand responds to them. Online retail and travel platforms often see a customer's segment before quoting a price, so they can price-discriminate across consumer types. Learning a separate demand curve for each type, however, seems to cost more exploration as the number of types grows. Chen and Gallego (arXiv:1812.09234v3, 2021; Mathematics of Operations Research, 2022) give a learning algorithm whose regret does not depend on the number of types. Their key step is to learn a single scalar first: the shadow price of inventory.

Earlier work on the single-type problem frames the question. Besbes and Zeevi (Operations Research, 2009) defined the regret against the fluid benchmark and showed that no policy can achieve regret decaying faster than n−1/2n^{-1/2}n−1/2. Wang, Deng and Ye (Operations Research, 2014) gave a single-product algorithm whose regret matches this rate up to logarithmic factors. Chen and Gallego extend the rate to MMM types with one inventory constraint.

Setting

There are M≥1M\ge1M≥1 consumer types, a season [0,T][0,T][0,T] and an inventory ccc. When the firm charges price ppp to type mmm, type-mmm consumers arrive at rate dm(p)d_m(p)dm​(p). Prices range over [0,p∞][0,p_\infty][0,p∞​], where p∞p_\inftyp∞​ is a choke price with dm(p∞)=0d_m(p_\infty)=0dm​(p∞​)=0. Assumption 1 asks that each dmd_mdm​ be strictly decreasing, bounded and Lipschitz, and that the revenue rate rm(λ)=λdm−1(λ)r_m(\lambda)=\lambda d_m^{-1}(\lambda)rm​(λ)=λdm−1​(λ) be Lipschitz and twice differentiable, with 0<M3≤−rm′′≤M40<M_3\le-r_m''\le M_40<M3​≤−rm′′​≤M4​.

The fluid problem (2) maximizes ∑m∫0Tpm(t)dm(pm(t)) dt\sum_m\int_0^Tp_m(t)d_m(p_m(t))\,dt∑m​∫0T​pm​(t)dm​(pm​(t))dt subject to ∑m∫0Tdm(pm(t)) dt≤c\sum_m\int_0^Td_m(p_m(t))\,dt\le c∑m​∫0T​dm​(pm​(t))dt≤c. Its optimal value is JD(T,c)J^D(T,c)JD(T,c). Its dual is one-dimensional. With

Rm(z)=max⁡pdm(p)(p−z),Pm(z)=argmax⁡pdm(p)(p−z),g(z)=cz+T∑mRm(z),\mathcal R_m(z)=\max_{p}d_m(p)(p-z),\qquad \mathcal P_m(z)=\operatorname*{argmax}_pd_m(p)(p-z),\qquad g(z)=cz+T\sum_m\mathcal R_m(z),Rm​(z)=pmax​dm​(p)(p−z),Pm​(z)=pargmax​dm​(p)(p−z),g(z)=cz+Tm∑​Rm​(z),

the dual optimum is z∗=argmin⁡z≥0g(z)z^*=\operatorname{argmin}_{z\ge0}g(z)z∗=argminz≥0​g(z), and the optimal fluid prices are the constants pm∗=Pm(z∗)p^*_m=\mathcal P_m(z^*)pm∗​=Pm​(z∗).

In the nnn-th system demand and inventory scale together: type mmm arrives as a Poisson process with rate n dm(p)n\,d_m(p)ndm​(p), and the inventory is ncncnc. The firm does not know dmd_mdm​. It knows TTT, ccc, nnn, the constants of Assumption 1 and, by Assumption 2, intervals [p‾,p‾][\underline p,\overline p][p​,p​] and [0,z‾][0,\overline z][0,z] that contain every pm∗p^*_mpm∗​ and z∗z^*z∗. A policy π\piπ has expected revenue Jnπ(T,c)J^\pi_n(T,c)Jnπ​(T,c), and its regret is

Rnπ(T,c)=1−Jnπ(T,c)JnD(T,c).R^\pi_n(T,c)=1-\frac{J^\pi_n(T,c)}{J^D_n(T,c)}.Rnπ​(T,c)=1−JnD​(T,c)Jnπ​(T,c)​.

Algorithm 1, the primal-dual learning algorithm, runs phases k=1,…,Kk=1,\dots,Kk=1,…,K of shrinking lengths τ(k)\tau^{(k)}τ(k). In each learning phase it tests a price grid inside the current interval estimators of every pm∗p^*_mpm∗​. It then minimizes an empirical dual function on a grid of zzz and re-centres the interval estimators of z∗z^*z∗ and pm∗p^*_mpm∗​. In the last phase it either charges a marked-up price (sufficient capacity), or tests two prices and interpolates between them so that total sales end close to ncncnc (insufficient capacity). Its tuning depends on a constant ϵ>0\epsilon>0ϵ>0.

Formalization targets

Goal: Theorem 1 (p. 17)

For every δ>0\delta>0δ>0 one can choose ϵ>0\epsilon>0ϵ>0 and a constant CCC, independent of nnn, such that

Rnπ(T,c)≤C(log⁡n)2+δn−1/2for all n≥3.R^\pi_n(T,c)\le C(\log n)^{2+\delta}n^{-1/2}\qquad\text{for all }n\ge3.Rnπ​(T,c)≤C(logn)2+δn−1/2for all n≥3.

The two capacity regimes (Propositions 2–3)

z∗=0: Rnπ=O((log⁡n)2+16ϵn−1/2),z∗>0: Rnπ=O((log⁡n)2+18ϵn−1/2).z^*=0:\ R^\pi_n=O\big((\log n)^{2+16\epsilon}n^{-1/2}\big),\qquad z^*>0:\ R^\pi_n=O\big((\log n)^{2+18\epsilon}n^{-1/2}\big).z∗=0: Rnπ​=O((logn)2+16ϵn−1/2),z∗>0: Rnπ​=O((logn)2+18ϵn−1/2).

Milestones

  • Proposition 1: the fluid problem is solved by the constant prices p∗p^*p∗, with complementary slackness; and the monotonicity, convexity and derivative of Pm\mathcal P_mPm​ and Rm\mathcal R_mRm​.
  • Lemmas 9 and 10: a Poisson tail bound decaying faster than any power, and E∣X−λ∣≤λ\mathbb E|X-\lambda|\le\sqrt\lambdaE∣X−λ∣≤λ​.
  • Lemmas 1 and 2: K≤3log⁡n+3K\le3\log n+3K≤3logn+3, and the learning phases take at most half the season.
  • Lemmas 3–5: the interval estimators contain z∗z^*z∗, pm∗p^*_mpm∗​ and Pm\mathcal P_mPm​ on the dual interval, except with probability O((log⁡n)2/n)O((\log n)^2/n)O((logn)2/n).
  • Lemmas 6–7: the sales at the start of the last phase are close to the fluid level.
  • Lemma 8: when z∗>0z^*>0z∗>0, E∣S(T)−nc∣=O((log⁡n)ϵn1/2)\mathbb E|S(T)-nc|=O((\log n)^\epsilon n^{1/2})E∣S(T)−nc∣=O((logn)ϵn1/2).

Significance

Theorem 1 states that personalized pricing for MMM types under a shared inventory constraint is learnable at the single-type rate n−1/2n^{-1/2}n−1/2, up to logarithms, with a constant independent of nnn. Combined with the Besbes–Zeevi lower bound, the algorithm is rate-optimal. Learning the scalar shadow price z∗z^*z∗, instead of the joint optimum of an MMM-dimensional problem, is what avoids the curse of dimensionality.

The result is proved in the paper; no machine-checked version exists. A formalization would add to the platform a multi-type Poisson pricing model with stock-outs, the one-dimensional Lagrangian dual of the fluid revenue-management problem, an exactly specified multi-phase learning algorithm, and reusable Poisson deviation lemmas. Lemmas 1, 2, 9 and 10 and Proposition 1 can be proved independently of the probabilistic analysis.

Difficulty

The obvious argument estimates every pm∗p^*_mpm∗​ directly and charges the estimate. With interval estimators of width n−1/4n^{-1/4}n−1/4 this loses revenue of order n−1/4n^{-1/4}n−1/4 per unit time when the inventory constraint binds (z∗>0z^*>0z∗>0), because pm∗p^*_mpm∗​ is then not the unconstrained maximizer of p dm(p)p\,d_m(p)pdm​(p). The paper's (11) and (19) show the loss. The bound n−1/2n^{-1/2}n−1/2 requires steering the total sales to within O(n1/2)O(n^{1/2})O(n1/2) of the inventory, which is Lemma 8, and a second-order expansion of the revenue around p∗p^*p∗ that uses the first-order condition rm′(dm(pm∗))=z∗r_m'(d_m(p^*_m))=z^*rm′​(dm​(pm∗​))=z∗.

Formally, a further difficulty is the closed loop: the prices of a phase depend on the counts of earlier phases, and the counts are Poisson only conditionally, through the random time change Nm(n∫0tdm(Pm(s)) ds)N_m(n\int_0^td_m(P_m(s))\,ds)Nm​(n∫0t​dm​(Pm​(s))ds).

Formalization scope

All declarations live in the namespace PrimalDualPricing.Regret. Printed page numbers equal PDF page numbers in the cited preprint.

  • Model. Types are Fin M. Each dm:R→Rd_m:\mathbb R\to\mathbb Rdm​:R→R satisfies Assumption 1 on [0,p∞][0,p_\infty][0,p∞​] and is otherwise any extension with 0≤dm≤M10\le d_m\le M_10≤dm​≤M1​; the last-phase prices p‾m(K)−α\underline p^{(K)}_m-\alphap​m(K)​−α and p‾m(K)+α\overline p^{(K)}_m+\alphap​m(K)​+α can leave [0,p∞][0,p_\infty][0,p∞​]. dm(p∞)=0d_m(p_\infty)=0dm​(p∞​)=0 encodes the choke price. Pm\mathcal P_mPm​ and z∗z^*z∗ are fields carrying their optimality properties.
  • Assumption 2 is read with z∗∈[0,z‾)z^*\in[0,\overline z)z∗∈[0,z); the page prints (0,z‾)(0,\overline z)(0,z), which would drop the case z∗=0z^*=0z∗=0 that Proposition 2 and Theorem 1 cover. The price interval is placed in the domain, 0≤p‾<p‾≤p∞0\le\underline p<\overline p\le p_\infty0≤p​<p​≤p∞​.
  • Remark 2 assumes M3≤g′′≤M4M_3\le g''\le M_4M3​≤g′′≤M4​; this is read on [0,z‾][0,\overline z][0,z], because globally it is unsatisfiable. Each Pm\mathcal P_mPm​ is M2M_2M2​-Lipschitz.
  • The fluid value JDJ^DJD is the supremum of (2) over measurable price paths. The identity JD=T∑mpm∗dm(pm∗)J^D=T\sum_mp^*_md_m(p^*_m)JD=T∑m​pm∗​dm​(pm∗​) is Proposition 1.4, a milestone, and is not built into the definition.
  • The algorithm. The grid sizes N(k)N^{(k)}N(k), Nz(k)N_z^{(k)}Nz(k)​ are rounded up. Grid spacing is Δm/N\Delta_m/NΔm​/N for prices and Δz/(Nz+1)\Delta_z/(N_z+1)Δz​/(Nz​+1) for duals: the text and Table 1, respectively the text and line 15, agree on these, and Algorithm 1 line 7 and Table 1 print the other forms. Ties in (8)–(9) go to the smallest index. θ\thetaθ in (14) is projected onto [0,1][0,1][0,1], with θ=0\theta=0θ=0 when (14) is degenerate. An empty plp^lpl period in step 4 follows the pseudo-code. The decision rules never take dmd_mdm​, Pm\mathcal P_mPm​ or z∗z^*z∗ as arguments.
  • The system. MMM independent unit-rate Poisson processes (the published IsPoissonProcess) with right-continuous, jointly measurable paths. The inventory of the nnn-th system is ⌊nc⌋\lfloor nc\rfloor⌊nc⌋ units, and nothing is sold after the stock-out. Lemmas 6–8 use the uncapped sales of the paper's modified system (§4).
  • Asymptotics. "O(f(n))O(f(n))O(f(n))" is read as ∃C\exists C∃C, independent of nnn and the phase index, for all n≥3n\ge3n≥3. "Sufficiently small ϵ\epsilonϵ" is read as every ϵ∈(0,ϵ0)\epsilon\in(0,\epsilon_0)ϵ∈(0,ϵ0​). The algorithm needs n≥3n\ge3n≥3 for KKK to exist.
  • Corrected misprints. Proposition 1.2 prints Rm′=−∑mdm(Pm)\mathcal R_m'=-\sum_md_m(\mathcal P_m)Rm′​=−∑m​dm​(Pm​) with the index reused; the statement is −dm(Pm(z))-d_m(\mathcal P_m(z))−dm​(Pm​(z)). Proposition 1.3 and the global form of the bound in 1.1 fail at the kinks of Pm\mathcal P_mPm​. Part 1's derivative bound is stated as a Lipschitz bound, and part 3 is omitted. Lemmas 3–4 are stated in product form. Propositions 2–3 are stated through their regret conclusions. The printed comparison J~nπ≤Jnπ\tilde J^\pi_n\le J^\pi_nJ~nπ​≤Jnπ​ (pp. 13–14) can fail when a last-phase price exceeds p‾\overline pp​, so it is not used.

A formalization that lets the policy read the true demand, measures regret against anything other than JnDJ^D_nJnD​, ignores the stock-out in JnπJ^\pi_nJnπ​, or assumes the high-probability events of Lemma 5 would trivialize the goal; none of these is done here.

Contributions to the reusable parts are welcome: Poisson tails, the envelope theorem for Rm\mathcal R_mRm​, and the time-change construction of policy-driven Poisson sales.

Selected references

  • N. Chen, G. Gallego, A Primal-dual Learning Algorithm for Personalized Dynamic Pricing with an Inventory Constraint, arXiv:1812.09234v3, 2021; Mathematics of Operations Research, 2022. https://arxiv.org/abs/1812.09234
  • O. Besbes, A. Zeevi, Dynamic Pricing Without Knowing the Demand Function: Risk Bounds and Near-Optimal Algorithms, Operations Research 57(6), 2009. https://doi.org/10.1287/opre.1080.0640
  • Z. Wang, S. Deng, Y. Ye, Close the Gaps: A Learning-While-Doing Algorithm for Single-Product Revenue Management Problems, Operations Research 62(2), 2014. https://doi.org/10.1287/opre.2013.1249
  • G. Gallego, G. van Ryzin, A Multiproduct Dynamic Pricing Problem and Its Applications to Network Yield Management, Operations Research 45(1), 1997. https://doi.org/10.1287/opre.45.1.24
21 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
Machine LearningOperations ResearchStatistics·Captain: mikedeng1

Explore First, Exploit Next: The True Shape of Regret in Bandit Problems III: A Strategy Smarter Than Uniform Draws Every Arm at Least (T/K)(1 − √(2T K_inf)) TimesResearch Paper

Why early exploration needs a lower bound

A stochastic multi-armed bandit learner repeatedly selects one of finitely many arms and sees only the reward from the selected arm. The learner must gather information about uncertain rewards while trying to select high-mean arms. Expected regret summarizes the lost reward relative to always selecting an optimal arm, but regret alone does not explain how much sampling each particular arm must receive. Garivier, Ménard, and Stoltz study this allocation question directly in Explore First, Exploit Next: The True Shape of Regret in Bandit Problems. Their Section 3 gives lower bounds for small horizons, before the longer-run logarithmic behavior becomes dominant.

The paper's first small-horizon result concerns strategies that allocate at least a uniform share to every optimal arm, on every problem in a specified model. This requirement excludes a strategy that always selects one fixed label and succeeds only when that label happens to be best. Such a requirement still permits many adaptive policies; it constrains their performance on optimal arms rather than prescribing their decisions round by round. The result establishes a compulsory initial allocation to every arm, including one whose mean is below the best mean. It is Theorem 2 on page 10 of the pinned arXiv:1602.07182v3.

Bandit problems, models, and strategies

A bandit problem ν‾=(νa)a=1K\underline\nu=(\nu_a)_{a=1}^Kν​=(νa​)a=1K​ consists of KKK probability laws on R\mathbb RR. At each round, a strategy ψ\psiψ chooses an arm using the observed past and randomization, then receives an independent reward drawn from that arm's law. The mean reward of arm aaa is μa=Eνa[X]\mu_a=\mathbb E_{\nu_a}[X]μa​=Eνa​​[X]. Write μ∗=max⁡aμa\mu^*=\max_a\mu_aμ∗=maxa​μa​, and call an arm optimal when μa=μ∗\mu_a=\mu^*μa​=μ∗. The count Nψ,a(T)N_{\psi,a}(T)Nψ,a​(T) is the number of times arm aaa is chosen during the first TTT rounds. These objects and the information available to a strategy are specified in Section 1.1, pages 2–3.

A model D\mathcal DD is a collection of possible arm laws, each a probability measure with an expectation. The notation ν‾∈D\underline\nu\in\mathcal Dν​∈D means that every arm law νa\nu_aνa​ belongs to D\mathcal DD. A strategy is smarter than uniform on D\mathcal DD when, for every bandit problem in the model, every optimal arm a∗a^*a∗, and every integer T≥1T\ge1T≥1, it satisfies

Eν‾[Nψ,a∗(T)]≥TK.\mathbb E_{\underline\nu}[N_{\psi,a^*}(T)]\ge\frac{T}{K}.Eν​​[Nψ,a∗​(T)]≥KT​.

This is Definition 2 on page 10. It imposes a lower bound on each optimal arm, even when several arms tie. The requirement is evaluated across the whole model, so a strategy cannot be declared smarter based only on its behavior on the one problem used in the theorem.

For a law PPP, a real threshold xxx, and the model D\mathcal DD, the information number Kinf⁡(P,x,D)\mathcal K_{\inf}(P,x,\mathcal D)Kinf​(P,x,D) is the infimum of KL(P,Q)\mathrm{KL}(P,Q)KL(P,Q) over those Q∈DQ\in\mathcal DQ∈D whose mean exceeds xxx. The infimum is +∞+\infty+∞ if there is no such QQQ. This is the definition on page 4. It measures how close an arm law is, in Kullback–Leibler divergence, to another admissible law that would give it a mean above the present optimum.

Formalization targets

The main target is Theorem 2. For a model D\mathcal DD, a strategy ψ\psiψ smarter than uniform on it, a bandit problem ν‾∈D\underline\nu\in\mathcal Dν​∈D, every arm aaa, and every integer T≥1T\ge1T≥1, the finite-Kinf⁡\mathcal K_{\inf}Kinf​ case states

Eν‾[Nψ,a(T)]≥TK(1−2TKinf⁡(νa,μ∗,D)).\mathbb E_{\underline\nu}[N_{\psi,a}(T)]\ge \frac{T}{K}\left(1-\sqrt{2T\mathcal K_{\inf}(\nu_a,\mu^*,\mathcal D)}\right).Eν​​[Nψ,a​(T)]≥KT​(1−2TKinf​(νa​,μ∗,D)​).

The theorem includes optimal and suboptimal arms. Its printed companion says that when T≤1/(8Kinf⁡)T\le 1/(8\mathcal K_{\inf})T≤1/(8Kinf​), the arm receives at least T/(2K)T/(2K)T/(2K) expected selections. In the formal statement this threshold is 8TKinf⁡≤18T\mathcal K_{\inf}\le18TKinf​≤1, which also expresses the intended case Kinf⁡=0\mathcal K_{\inf}=0Kinf​=0. Both claims come from Theorem 2, page 10.

The milestone list follows the source's intermediate claims. Lemma 6, page 20 bounds Bernoulli relative entropy using the maximum of x(1−x)x(1-x)x(1−x) over an interval. The proof of Theorem 2 on page 11 states that Bernoulli relative entropy increases in its second argument on the interval starting at its first argument. The same page applies the paper's fundamental inequality (6) to the fraction of rounds spent on one arm, then states a lower bound against a single alternative arm law. These are the four curated milestones for the main theorem.

What the result supplies

The theorem turns a model-relative performance condition into a numerical constraint on early sampling. If an arm's law is close in divergence to a law that would make it best, then the bound keeps its expected pull count near the uniform allocation until the horizon is large enough for the term under the square root to matter. It identifies the divergence scale 1/Kinf⁡1/\mathcal K_{\inf}1/Kinf​ at which this particular guarantee becomes weak. The paper contrasts this absolute bound with later relative and collective bounds that capture different dependencies on the number of arms in Section 3, pages 10–14.

The mathematical result is proved in the paper. The remaining work in this mission is a machine-checked development of the statement and its source-backed intermediate inequalities in the canonical stochastic-bandit model. The published bandit structure, policy kernels, and information number provide reusable foundations. Shared definitions supply the Bernoulli divergence, model predicate, and expected pull counts; the local strategy class makes the hypothesis of this particular theorem explicit. The Bernoulli inequalities and the pull-count change-of-measure statement can be reused in other lower-bound results.

Where the difficulty lies

A performance condition about optimal arms does not directly constrain an arm that is suboptimal in the problem of interest. The small-horizon bound must remain valid even though the strategy may adapt to observed rewards, and the relevant alternative arm laws can have divergences approaching an infimum without attaining it. The paper's lower bound is therefore sensitive to the exact orientation of the two bandit problems, the expected count multiplying their divergence, and the boundary conventions for Bernoulli relative entropy. Replacing the infimum with a chosen alternative or changing a divergence to an ordinary real with an arbitrary boundary value would alter the statement.

Formalization scope

Lean uses arms Fin K, indexed from zero; the paper indexes them from one. The horizon is a natural number, and counts are integrated against the probability measure on TTT-round histories generated by a stochastic kernel policy. Such kernels represent the arm-and-reward histories induced by the paper's strategies with auxiliary randomization. The model is a set of probability measures on R\mathbb RR whose identity function is integrable. The published StochasticBandit, BanditPolicy, and ConsistentBanditPolicy definitions supply the basic problem, history measure, pull count, means, and Kinf⁡\mathcal K_{\inf}Kinf​.

Bernoulli divergence and Kinf⁡\mathcal K_{\inf}Kinf​ take values in [0,+∞][0,+\infty][0,+∞]. The real square-root formula assumes Kinf⁡<+∞\mathcal K_{\inf}<+\inftyKinf​<+∞: when it is infinite, the paper's right-hand side is −∞-\infty−∞, while converting infinity directly to Lean's real value would produce a false numerical bound. The starting bandit problem is explicitly in D\mathcal DD, a condition needed for the proof's use of Definition 2 on a one-arm modification, although the printed quantifier on page 10 omits it. The arm parameter guarantees K≥1K\ge1K≥1; every target assumes T≥1T\ge1T≥1, so its real divisions by KKK and TTT have the stated meaning.

A faithful development must retain Definition 2's quantification over all model problems and each optimal arm, rather than only over the bandit problem used in the conclusion. It must use the full Bernoulli relative entropy, including infinite endpoint values, and the infimum over precisely those model laws with mean greater than μ∗\mu^*μ∗. The local instance of (6) also assumes integrable arm rewards in both bandit problems, as Section 1.1 requires. Contributions that establish the four milestones, the companion threshold claim, or reusable count-integrability and divergence facts are in scope.

Selected references

  • Aurélien Garivier, Pierre Ménard, and Gilles Stoltz, Explore First, Exploit Next: The True Shape of Regret in Bandit Problems, Mathematics of Operations Research, 2019; arXiv:1602.07182v3.
  • Tor Lattimore and Csaba Szepesvári, Bandit Algorithms, Cambridge University Press, 2020; DOI:10.1017/9781108571401. The published bandit definitions used here follow its canonical framework.
11 thms2 active usersReviewed
Machine LearningOperations ResearchStatistics·Captain: mikedeng1

Explore First, Exploit Next: The True Shape of Regret in Bandit Problems IV: Under Pairwise Symmetry a Suboptimal Arm Is Drawn T/K Times or Almost as Often as an Optimal ArmResearch Paper

Motivation

Lower bounds for stochastic multi-armed bandits are usually stated for a large horizon: Lai and Robbins (1985) and Burnetas and Katehakis (1996) show that a consistent strategy pulls a suboptimal arm at least about ln⁡T/Kinf⁡\ln T / \mathcal{K}_{\inf}lnT/Kinf​ times. These bounds say nothing about the small-horizon regime, the first rounds in which a strategy has not yet had time to separate the arms. Garivier, Ménard and Stoltz (arXiv:1602.07182v3, Mathematics of Operations Research, 2019) describe that regime through one information-theoretic inequality, their (6), and derive several lower bounds from it. This mission formalizes their relative lower bound, Theorem 3 (p. 11): under a mild symmetry assumption on the strategy, a suboptimal arm is either pulled at the uniform rate T/KT/KT/K or, on average, almost as often as an optimal arm. It is the statement behind the title's claim that bandit strategies must first explore.

Setting

There are KKK arms. A bandit problem ν‾=(ν1,…,νK)\underline{\nu} = (\nu_1, \dots, \nu_K)ν​=(ν1​,…,νK​) assigns to each arm aaa a probability distribution νa\nu_aνa​ on R\mathbb{R}R with expectation μa\mu_aμa​. The optimal mean is μ⋆=max⁡aμa\mu^\star = \max_a \mu_aμ⋆=maxa​μa​, the gap of arm aaa is Δa=μ⋆−μa\Delta_a = \mu^\star - \mu_aΔa​=μ⋆−μa​; arm aaa is optimal if Δa=0\Delta_a = 0Δa​=0 and suboptimal if Δa>0\Delta_a > 0Δa​>0.

A model D\mathcal{D}D is a set of distributions on R\mathbb{R}R, each with an expectation; a bandit problem is in D\mathcal{D}D when every νa\nu_aνa​ belongs to D\mathcal{D}D.

At each round ttt a strategy ψ\psiψ picks an arm AtA_tAt​, possibly at random, as a function of the past arms and rewards, and receives a reward YtY_tYt​ drawn from νAt\nu_{A_t}νAt​​. The number of pulls of arm aaa in the first TTT rounds is

Nψ,a(T)=∑t=1T1{At=a},N_{\psi,a}(T) = \sum_{t=1}^{T} \mathbb{1}\{A_t = a\},Nψ,a​(T)=t=1∑T​1{At​=a},

and Eν‾\mathbb{E}_{\underline{\nu}}Eν​​ denotes expectation when ψ\psiψ plays against ν‾\underline{\nu}ν​. Write Nψ,a+(T)=max⁡{Nψ,a(T),1}N^+_{\psi,a}(T) = \max\{N_{\psi,a}(T), 1\}Nψ,a+​(T)=max{Nψ,a​(T),1}.

Kullback–Leibler divergence. KL(P,Q)∈[0,+∞]\mathrm{KL}(P, Q) \in [0, +\infty]KL(P,Q)∈[0,+∞] is the Kullback–Leibler divergence between distributions, and kl(p,q)\mathrm{kl}(p, q)kl(p,q) the divergence between Bernoulli laws of parameters ppp and qqq.

Pairwise symmetry (Definition 3, p. 11). A strategy ψ\psiψ is pairwise symmetric for optimal arms on D\mathcal{D}D if for every bandit problem ν‾\underline{\nu}ν​ in D\mathcal{D}D, every pair of optimal arms a⋆,a⋆a^\star, a_\stara⋆,a⋆​ with νa⋆=νa⋆\nu_{a^\star} = \nu_{a_\star}νa⋆​=νa⋆​​ and every T≥1T \ge 1T≥1, the pairs (Nψ,a⋆(T),Nψ,a⋆(T))(N_{\psi,a^\star}(T), N_{\psi,a_\star}(T))(Nψ,a⋆​(T),Nψ,a⋆​​(T)) and (Nψ,a⋆(T),Nψ,a⋆(T))(N_{\psi,a_\star}(T), N_{\psi,a^\star}(T))(Nψ,a⋆​​(T),Nψ,a⋆​(T)) have the same distribution. A strategy that uses only the observed payoffs, and not the labels of the arms, has this property.

Formalization targets

Goal: Theorem 3 (p. 11)

For every model D\mathcal{D}D, every strategy ψ\psiψ pairwise symmetric for optimal arms on D\mathcal{D}D, every bandit problem ν‾\underline{\nu}ν​ in D\mathcal{D}D, every suboptimal arm aaa, every optimal arm a⋆a^\stara⋆ with KL(νa,νa⋆)<∞\mathrm{KL}(\nu_a, \nu_{a^\star}) < \inftyKL(νa​,νa⋆​)<∞ and every T≥1T \ge 1T≥1,

Eν‾[Nψ,a(T)]≥TKorEν‾[Nψ,a+(T)Nψ,a⋆+(T)]≥1−22T KL(νa,νa⋆)K.\mathbb{E}_{\underline{\nu}}\big[N_{\psi,a}(T)\big] \ge \frac{T}{K} \quad\text{or}\quad \mathbb{E}_{\underline{\nu}}\left[\frac{N^+_{\psi,a}(T)}{N^+_{\psi,a^\star}(T)}\right] \ge 1 - 2\sqrt{\frac{2T\,\mathrm{KL}(\nu_a, \nu_{a^\star})}{K}} .Eν​​[Nψ,a​(T)]≥KT​orEν​​[Nψ,a⋆+​(T)Nψ,a+​(T)​]≥1−2K2TKL(νa​,νa⋆​)​​.

Companion: the "in particular" clause

For T≤K/(32 KL(νa,νa⋆))T \le K / (32\,\mathrm{KL}(\nu_a, \nu_{a^\star}))T≤K/(32KL(νa​,νa⋆​)) the second alternative becomes Eν‾[Nψ,a+(T)/Nψ,a⋆+(T)]≥1/2\mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}(T)/N^+_{\psi,a^\star}(T)] \ge 1/2Eν​​[Nψ,a+​(T)/Nψ,a⋆+​(T)]≥1/2.

Milestones (p. 12)

  1. The symmetry identity: for the alternative problem ν‾′\underline{\nu}'ν​′ obtained by replacing νa\nu_aνa​ by νa⋆\nu_{a^\star}νa⋆​, Eν‾′[Nψ,a+/(Nψ,a++Nψ,a⋆+)]=1/2\mathbb{E}_{\underline{\nu}'}[N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})] = 1/2Eν​′​[Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​)]=1/2.
  2. Inequality (12): Eν‾[Nψ,a(T)] KL(νa,νa′)≥kl(Eν‾[Nψ,a+/(Nψ,a++Nψ,a⋆+)],1/2)\mathbb{E}_{\underline{\nu}}[N_{\psi,a}(T)]\,\mathrm{KL}(\nu_a, \nu'_a) \ge \mathrm{kl}\big(\mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})], 1/2\big)Eν​​[Nψ,a​(T)]KL(νa​,νa′​)≥kl(Eν​​[Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​)],1/2).
  3. The Jensen display: Eν‾[Nψ,a+/(Nψ,a++Nψ,a⋆+)]≤r/(1+r)\mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})] \le r/(1+r)Eν​​[Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​)]≤r/(1+r) with r=Eν‾[Nψ,a+/Nψ,a⋆+]r = \mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}/N^+_{\psi,a^\star}]r=Eν​​[Nψ,a+​/Nψ,a⋆+​].
  4. The bound r/(1+r)≥1/2−T KL(νa,νa′)/(2K)r/(1+r) \ge 1/2 - \sqrt{T\,\mathrm{KL}(\nu_a,\nu'_a)/(2K)}r/(1+r)≥1/2−TKL(νa​,νa′​)/(2K)​ in the case Eν‾[Nψ,a(T)]≤T/K\mathbb{E}_{\underline{\nu}}[N_{\psi,a}(T)] \le T/KEν​​[Nψ,a​(T)]≤T/K, r≤1r \le 1r≤1.

Significance

The theorem shows that in the initial phase a symmetric strategy cannot discard a suboptimal arm quickly: unless the arm is already sampled at the uniform rate, it is sampled, in ratio, nearly as often as an optimal arm, as long as TTT is small compared with K/KL(νa,νa⋆)K / \mathrm{KL}(\nu_a, \nu_{a^\star})K/KL(νa​,νa⋆​). Together with Theorem 2 (absolute lower bound) and Theorem 4 (collective lower bound) it describes the regret curve before the logarithmic regime: roughly linear growth, then a transition. It requires no assumption on the model beyond the existence of expectations, and it involves the divergence between the two arms themselves rather than the model-dependent quantity Kinf⁡\mathcal{K}_{\inf}Kinf​.

The result is proved in the paper; to our knowledge it has no machine-checked proof. The mission produces a Lean statement of Theorem 3 on the canonical bandit model already used on the platform (StochasticBandit, BanditPolicy), a formal version of the symmetry assumption that sibling missions reuse (mission V, the collective lower bound, uses the same Definition 3), and the four intermediate steps of the proof as separate targets.

Difficulty

The proof compares ν‾\underline{\nu}ν​ with an alternative problem ν‾′\underline{\nu}'ν​′ that differs from it only in arm aaa, and transfers information between them through the fundamental inequality (6) of §2, a data-processing argument on the law of the whole history. The step specific to Theorem 3 is the choice of the test variable: the expectation of Nψ,a+/(Nψ,a++Nψ,a⋆+)N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​) is known exactly under ν‾′\underline{\nu}'ν​′ only because the strategy is pairwise symmetric, and the symmetry is a statement about joint laws of pull counts, not about their expectations. A version of the assumption that only equates expected pull counts does not determine the expectation of this ratio. The ratio expectation then has to be extracted from a bound on E[x/(1+x)]\mathbb{E}[x/(1+x)]E[x/(1+x)], which needs Jensen's inequality in the right direction and a case split on whether r≤1r \le 1r≤1.

Formalization scope

Arms are Fin K (0-based). The horizon TTT is a natural number. A strategy is a published BanditPolicy: a family of Markov kernels from the observed history of (arm, reward) pairs to the next arm. Every strategy of the paper, which may use auxiliary uniform random variables, induces such a kernel with the same law of arms and rewards. The law of the history is the published banditMeasure, and Nψ,a(T)N_{\psi,a}(T)Nψ,a​(T) is the published armPullCount. The model is a set of measures on R\mathbb{R}R, and every statement assumes the paper's standing hypothesis (§1.1, p. 3) that each member is a probability measure with an integrable identity, together with ν‾\underline{\nu}ν​ in D\mathcal{D}D.

Committed conventions:

  • KL\mathrm{KL}KL and kl\mathrm{kl}kl take values in [0,+∞][0, +\infty][0,+∞]; the Bernoulli divergence is the Kullback–Leibler divergence of two Bernoulli laws on R\mathbb{R}R, not a real formula with finite values at q∈{0,1}q \in \{0, 1\}q∈{0,1}.
  • Theorem 3 adds the hypothesis KL(νa,νa⋆)<∞\mathrm{KL}(\nu_a, \nu_{a^\star}) < \inftyKL(νa​,νa⋆​)<∞. When the divergence is infinite the paper's bound reads ≥−∞\ge -\infty≥−∞ and is empty; reading +∞+\infty+∞ as the real number 000 would make the statement false.
  • The "in particular" clause is stated as 32 T KL(νa,νa⋆)≤K32\,T\,\mathrm{KL}(\nu_a,\nu_{a^\star}) \le K32TKL(νa​,νa⋆​)≤K in [0,+∞][0, +\infty][0,+∞], which needs no finiteness assumption.
  • Definition 3 is an equality of joint laws (push-forwards of the history law), for every T≥1T \ge 1T≥1, only for optimal arms of problems in D\mathcal{D}D with equal distributions. Invariance under all permutations of the arms would strengthen the hypothesis and weaken the theorem.
  • Ratios use max⁡{N,1}\max\{N, 1\}max{N,1} as real numbers, so no division by zero occurs; all integrands are bounded by TTT.

A formalization in which the symmetry assumption equates only expected pull counts, in which the divergence is a real number with junk value 000 at +∞+\infty+∞, or in which the disjunction is replaced by its second alternative, is not this theorem.

The mission builds on the published definitions StochasticBandit and BanditPolicy. A complete development needs the fundamental inequality (6) for kernel policies (the divergence decomposition, published as BanditAlgorithm.bandit_divergence_decomposition, plus data processing for the Bernoulli test), Pinsker's inequality for Bernoulli laws (proved on the platform as BanditAlgorithm.bernoulli_relative_entropy_pinsker), Jensen's inequality for concave functions (Mathlib), and a lemma that the law of the history is unchanged in distribution when two identically distributed optimal arms are relabelled. The measurability of pull counts and the relation between the extended-real and real Bernoulli divergences are reusable beyond this mission. Proofs of any milestone, and of the general fundamental inequality, are welcome.

Selected references

  • A. Garivier, P. Ménard, G. Stoltz, Explore First, Exploit Next: The True Shape of Regret in Bandit Problems, Mathematics of Operations Research 44(2), 2019. arXiv:1602.07182v3, doi:10.1287/moor.2017.0928
  • T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6(1), 1985. doi:10.1016/0196-8858(85)90002-8
  • A. N. Burnetas, M. N. Katehakis, Optimal adaptive policies for sequential allocation problems, Advances in Applied Mathematics 17(2), 1996. doi:10.1006/aama.1996.0007
  • T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020. doi:10.1017/9781108571401
8 thms2 active usersReviewed
Machine LearningOperations ResearchStatistics·Captain: mikedeng1

Explore First, Exploit Next: The True Shape of Regret in Bandit Problems II: Uniformly Fast Convergent Strategies Draw Every Suboptimal Arm at Least ln T / K_inf Times AsymptoticallyResearch Paper

Why this lower bound matters

In a stochastic multi-armed bandit, a player repeatedly chooses one of finitely many arms and sees only the reward from the chosen arm. To earn a high reward, the player must favor arms that appear best. Yet an arm that currently appears inferior might be best under another plausible reward law. That uncertainty forces any strategy with uniformly strong long-run performance to keep testing apparently inferior arms. The asymptotic number of tests is a basic benchmark for comparing bandit algorithms: an algorithm that matches the bound uses no more exploration than the information in the model demands. Garivier, Ménard, and Stoltz restate this general benchmark as Theorem 1 and derive it within their analysis of the shape of regret Garivier–Ménard–Stoltz, §2.2.

The quantity controlling the benchmark comes from Burnetas and Katehakis. It measures the least information needed to replace one suboptimal arm by a plausible arm that is strictly better than the current best. The paper cites their quantity on page 4, and its Theorem 1 applies it to every suboptimal arm separately Garivier–Ménard–Stoltz, pp. 4, 9. This per-arm form matters when different arms have different reward laws and different amounts of information are needed to rule them out.

Bandit setting and notation

A bandit problem ν‾=(νa)a=1K\underline\nu=(\nu_a)_{a=1}^Kν​=(νa​)a=1K​ has KKK arms. Each νa\nu_aνa​ is a probability distribution on real rewards with a finite expectation μa=EνaX\mu_a=\mathbb E_{\nu_a}Xμa​=Eνa​​X. A strategy ψ\psiψ chooses an arm at each round using the preceding chosen arms and observed rewards, and it may randomize. Let Nψ,a(T)N_{\psi,a}(T)Nψ,a​(T) be the number of times arm aaa is chosen in the first TTT rounds. The optimal mean is μ⋆=max⁡aμa\mu^\star=\max_a\mu_aμ⋆=maxa​μa​, and the gap of arm aaa is Δa=μ⋆−μa\Delta_a=\mu^\star-\mu_aΔa​=μ⋆−μa​. An arm is suboptimal when Δa>0\Delta_a>0Δa​>0. These are the objects of §1.1 of the paper Garivier–Ménard–Stoltz, pp. 2–3.

A model D\mathcal DD is a collection of possible real-valued reward distributions, each with a finite expectation. A bandit problem belongs to D\mathcal DD when every one of its arm laws does. The Kullback–Leibler divergence KL(P,Q)\mathrm{KL}(P,Q)KL(P,Q) measures the information discrepancy from one reward law PPP to another QQQ; it may be infinite. For an arm law PPP and a real threshold xxx, define the minimal information cost

Kinf⁡(P,x,D)=inf⁡{KL(P,Q):Q∈D, EQX>x}.\mathcal K_{\inf}(P,x,\mathcal D) =\inf\{\mathrm{KL}(P,Q):Q\in\mathcal D,\ \mathbb E_QX>x\}.Kinf​(P,x,D)=inf{KL(P,Q):Q∈D, EQ​X>x}.

The infimum is +∞+\infty+∞ if there is no qualifying alternative. The strict inequality in the set matters: the replacement law must make the arm better than the original optimal mean, not merely tied with it Garivier–Ménard–Stoltz, p. 4.

Definition 1 calls ψ\psiψ uniformly fast convergent on D\mathcal DD when, on every bandit problem in the model, the expected draw count of every suboptimal arm grows more slowly than TαT^\alphaTα for each 0<α≤10<\alpha\le10<α≤1:

Eν‾Nψ,a(T)=o(Tα).\mathbb E_{\underline\nu}N_{\psi,a}(T)=o(T^\alpha).Eν​​Nψ,a​(T)=o(Tα).

Here TTT tends to infinity through integer horizons. This condition is uniform in which problem and suboptimal arm are considered; it does not mean one numerical convergence rate is shared by all problems Garivier–Ménard–Stoltz, p. 9, Definition 1.

Formalization target

The goal is the paper's Theorem 1: for every model D\mathcal DD, every uniformly fast convergent strategy ψ\psiψ, every bandit problem ν‾\underline\nuν​ in the model, and every suboptimal arm aaa,

lim inf⁡T→∞Eν‾Nψ,a(T)ln⁡T≥1Kinf⁡(νa,μ⋆,D).\liminf_{T\to\infty} \frac{\mathbb E_{\underline\nu}N_{\psi,a}(T)}{\ln T} \ge\frac1{\mathcal K_{\inf}(\nu_a,\mu^\star,\mathcal D)}.T→∞liminf​lnTEν​​Nψ,a​(T)​≥Kinf​(νa​,μ⋆,D)1​.

The bound states the minimum logarithmic exploration of each arm. It leaves the model and strategy arbitrary within the stated class. The milestones follow four statements from the proof on page 9: the Bernoulli entropy estimate in (11), the one-arm divergence chain in (10), the eventual bound on draws of the alternative's other arms, and the bound for one alternative law before taking the infimum. Their source text is recorded beside the local drafts Garivier–Ménard–Stoltz, p. 9.

What the result gives

If Kinf⁡\mathcal K_{\inf}Kinf​ is finite and positive, the theorem gives a concrete coefficient of ln⁡T\ln TlnT that any uniformly fast convergent strategy must spend on arm aaa. A small information cost means that the arm is hard to distinguish from a better alternative and must be sampled more often. If the cost is infinite because the model contains no better replacement, the reciprocal is zero and the bound makes no positive demand. If the cost is zero, the reciprocal is infinite: the normalized expected draw count must diverge. The extended-value convention is part of the target, not a choice of a finite stand-in Garivier–Ménard–Stoltz, pp. 4, 9.

The theorem is proved in the paper, while the Lean statements in this proposal are proof targets. A completed development would supply a machine-checked bridge from the canonical stochastic-bandit policy model through Bernoulli relative entropy and change of measure to the per-arm asymptotic claim. Its model and divergence definitions can also support lower bounds for other bandit classes. The existing Prove2Me StochasticBandit, BanditPolicy, and ConsistentBanditPolicy definitions provide the bandit law, policy, draw count, means, and Kinf⁡\mathcal K_{\inf}Kinf​ used here; Definition 1's exact strategy class is supplied locally.

Difficulty

The central issue is the quantification over every alternative law in an arbitrary model. A suboptimal arm may become uniquely optimal after changing only its law, while the strategy must remain uniformly fast on that new problem. The information cost is an infimum over such changes, including an empty set or costs arbitrarily close to zero. A statement that reasons about just one convenient alternative does not yield the coefficient in Theorem 1. The proof also passes from finite-horizon information inequalities to a liminf as TTT grows, so endpoint values of Bernoulli divergence and the extended-value arithmetic must agree with the source Garivier–Ménard–Stoltz, pp. 6, 9.

Formalization scope

Lean indexes the paper's arms 1,…,K1,\ldots,K1,…,K by Fin K, starting at zero, and uses natural-number horizons. A strategy is a history-dependent Markov kernel, whose generated probability measure describes the sequence of chosen arms and rewards. The pull count is bounded by TTT, so its expectation is a genuine Bochner integral. The model explicitly requires probability laws with integrable identity; this ensures that each mean in Kinf⁡\mathcal K_{\inf}Kinf​ is genuine. Bernoulli relative entropy and arm-law KL take values in extended nonnegative reals. The goal's liminf is also in extended nonnegative reals, avoiding a default real liminf for an unbounded sequence. The first two horizons, where ln⁡T≤0\ln T\le0lnT≤0, cannot affect an asymptotic liminf.

The theorem sentence on page 9 says “all bandit problems” without restricting the original problem to D\mathcal DD. Its proof applies Definition 1 to that problem, so the proposal states that its arms belong to D\mathcal DD. Without this condition a strategy can behave arbitrarily outside the model. The source's other standing conventions are kept: every reward law has an expectation, the improvement condition in Kinf⁡\mathcal K_{\inf}Kinf​ is strict, and uniform fast convergence ranges over all model problems, all suboptimal arms, and all 0<α≤10<\alpha\le10<α≤1. These choices rule out an empty strategy class or a special hard-coded bandit as a trivial route to the target.

Useful contributions include the finite-horizon change-of-measure inequality, endpoint-correct Bernoulli KL estimates, identities for sums of expected pull counts, and asymptotic lemmas for reciprocal infima. The milestone drafts identify the paper statements those contributions must support.

Selected references

  • Aurélien Garivier, Pierre Ménard, and Gilles Stoltz, Explore First, Exploit Next: The True Shape of Regret in Bandit Problems, Mathematics of Operations Research, 2019; cited here from arXiv:1602.07182v3, §§1–2.
  • Apostolos N. Burnetas and Michael N. Katehakis, Optimal Adaptive Policies for Sequential Allocation Problems, Advances in Applied Mathematics, 1996. DOI:10.1006/aama.1996.0007.
9 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
Machine LearningReinforcement Learning·Captain: mikedeng1

The Statistical Complexity of Interactive Decision Making V: The Multi-Armed Bandit Has Localized DEC at Least 2⁻⁶·A/γ, via Hard Families of ModelsResearch Paper

Motivation

The Decision-Estimation Coefficient (DEC) of Foster, Kakade, Qian and Rakhlin (arXiv:2112.13487) is a single complexity measure that bounds the minimax regret of interactive decision making (bandits, reinforcement learning, and their structured variants) from above and from below. Its lower bound (Theorem 3.2 of the paper) states that every algorithm suffers expected regret at least a constant times decγ(Mεγ∞(M‾),M‾)⋅T\mathsf{dec}_\gamma(\mathcal{M}^\infty_{\varepsilon_\gamma}(\overline{M}), \overline{M}) \cdot Tdecγ​(Mεγ​∞​(M),M)⋅T, so concrete lower bounds on regret follow from concrete lower bounds on the DEC of localized model classes.

The multi-armed bandit is the first test of this program. Its minimax regret Θ(AT)\Theta(\sqrt{AT})Θ(AT​) is classical (Auer et al., 2002; Audibert and Bubeck, 2010). Section 5.1 of the paper recovers it through the DEC: posterior sampling (Proposition 5.1) and inverse gap weighting (Proposition 5.2) bound the DEC by A/γA/\gammaA/γ from above, and Proposition 5.3 bounds it by 2−6A/γ2^{-6} A/\gamma2−6A/γ from below. The lower bound is proved through a hard family of models (Definition 5.1, Lemma 5.1), the device that the paper reuses for linear, ReLU and gap-dependent lower bounds in §§6–7.

Setting

A decision space Π\PiΠ is a finite set; here Π=[A]={1,…,A}\Pi = [A] = \{1, \dots, A\}Π=[A]={1,…,A}, the arms. A model MMM assigns to every arm π\piπ a distribution M(π)M(\pi)M(π) over rewards in R=[0,1]\mathcal{R} = [0,1]R=[0,1] (observations are trivial). Its mean reward is fM(π)=Er∼M(π)[r]f^M(\pi) = \mathbb{E}_{r \sim M(\pi)}[r]fM(π)=Er∼M(π)​[r], and πM∈arg⁡max⁡πfM(π)\pi_M \in \arg\max_\pi f^M(\pi)πM​∈argmaxπ​fM(π) is an optimal arm. The squared Hellinger distance between distributions P,QP, QP,Q on a finite set is DH2(P,Q)=∑y(P(y)−Q(y))2D^2_{\mathrm{H}}(P, Q) = \sum_y (\sqrt{P(y)} - \sqrt{Q(y)})^2DH2​(P,Q)=∑y​(P(y)​−Q(y)​)2 (no factor 12\tfrac1221​, the paper's (5)).

For a class M\mathcal{M}M of models, a reference model M‾\overline{M}M and γ>0\gamma > 0γ>0, the DEC (2) is

decγ(M,M‾)=inf⁡p∈Δ(Π)sup⁡M∈MEπ∼p[fM(πM)−fM(π)−γ⋅DH2(M(π),M‾(π))].\mathsf{dec}_\gamma(\mathcal{M}, \overline{M}) = \inf_{p \in \Delta(\Pi)} \sup_{M \in \mathcal{M}} \mathbb{E}_{\pi \sim p}\Bigl[ f^M(\pi_M) - f^M(\pi) - \gamma \cdot D^2_{\mathrm{H}}\bigl(M(\pi), \overline{M}(\pi)\bigr) \Bigr].decγ​(M,M)=p∈Δ(Π)inf​M∈Msup​Eπ∼p​[fM(πM​)−fM(π)−γ⋅DH2​(M(π),M(π))].

With gM(π)=fM(πM)−fM(π)g^M(\pi) = f^M(\pi_M) - f^M(\pi)gM(π)=fM(πM​)−fM(π), the L∞L_\inftyL∞​-localized class (12) is Mε∞(M‾)={M∈M:∣gM(π)−gM‾(π)∣≤ε ∀π}\mathcal{M}^\infty_\varepsilon(\overline{M}) = \{ M \in \mathcal{M} : |g^M(\pi) - g^{\overline{M}}(\pi)| \le \varepsilon\ \forall \pi \}Mε∞​(M)={M∈M:∣gM(π)−gM(π)∣≤ε ∀π}.

A reference model M‾∈M\overline{M} \in \mathcal MM∈M and models M1,…,MN∈MM_1, \dots, M_N \in \mathcal MM1​,…,MN​∈M, N≥2N \ge 2N≥2, form an (α,β,δ)(\alpha, \beta, \delta)(α,β,δ)-family (Definition 5.1) if there are ui,vi:Π→[0,1]u_i, v_i : \Pi \to [0,1]ui​,vi​:Π→[0,1] with ∑iui(π)≤N/2\sum_i u_i(\pi) \le N/2∑i​ui​(π)≤N/2, ∑ivi(π)≤1\sum_i v_i(\pi) \le 1∑i​vi​(π)≤1, fMi(πMi)−fMi(π)≥α(1−ui(π))f^{M_i}(\pi_{M_i}) - f^{M_i}(\pi) \ge \alpha(1 - u_i(\pi))fMi​(πMi​​)−fMi​(π)≥α(1−ui​(π)) and DH2(Mi(π),M‾(π))≤βvi(π)+δD^2_{\mathrm{H}}(M_i(\pi), \overline{M}(\pi)) \le \beta v_i(\pi) + \deltaDH2​(Mi​(π),M(π))≤βvi​(π)+δ. Ber(q)\mathrm{Ber}(q)Ber(q) is the Bernoulli distribution with mean qqq.

Formalization targets

Goal: Proposition 5.3

For A≥2A \ge 2A≥2 arms, M‾(π)=Ber(12)\overline{M}(\pi) = \mathrm{Ber}(\tfrac12)M(π)=Ber(21​) and every γ≥A/3\gamma \ge A/3γ≥A/3, with εγ=A12γ\varepsilon_\gamma = \frac{A}{12\gamma}εγ​=12γA​,

decγ(Mεγ∞(M‾),M‾)≥2−6⋅Aγ.\mathsf{dec}_\gamma\bigl(\mathcal{M}^\infty_{\varepsilon_\gamma}(\overline{M}), \overline{M}\bigr) \ge 2^{-6} \cdot \frac{A}{\gamma}.decγ​(Mεγ​∞​(M),M)≥2−6⋅γA​.

Milestones

  1. Lemma A.7: DH2(Ber(p),Ber(q))≤(p−q)2(1p+q+12−p−q)D^2_{\mathrm{H}}(\mathrm{Ber}(p), \mathrm{Ber}(q)) \le (p-q)^2\bigl(\frac{1}{p+q} + \frac{1}{2-p-q}\bigr)DH2​(Ber(p),Ber(q))≤(p−q)2(p+q1​+2−p−q1​), and in particular DH2(Ber(12+Δ),Ber(12))≤3Δ2D^2_{\mathrm{H}}(\mathrm{Ber}(\tfrac12 + \Delta), \mathrm{Ber}(\tfrac12)) \le 3\Delta^2DH2​(Ber(21​+Δ),Ber(21​))≤3Δ2 for Δ∈(0,12)\Delta \in (0, \tfrac12)Δ∈(0,21​).
  2. Lemma 5.1: an (α,β,δ)(\alpha, \beta, \delta)(α,β,δ)-family M′\mathcal{M}'M′ with β≥0\beta \ge 0β≥0 satisfies decγ(M′,M‾)≥α2−γ(βN+δ)\mathsf{dec}_\gamma(\mathcal{M}', \overline{M}) \ge \frac{\alpha}{2} - \gamma(\frac{\beta}{N} + \delta)decγ​(M′,M)≥2α​−γ(Nβ​+δ) for all γ>0\gamma > 0γ>0.
  3. The hard family of the proof: Mi(π)=Ber(12+Δ I{π=i})M_i(\pi) = \mathrm{Ber}(\tfrac12 + \Delta\,\mathbb{I}\{\pi = i\})Mi​(π)=Ber(21​+ΔI{π=i}) is a (Δ,3Δ2,0)(\Delta, 3\Delta^2, 0)(Δ,3Δ2,0)-family around Ber(12)\mathrm{Ber}(\tfrac12)Ber(21​).
  4. Localization: this family lies in MΔ∞(M‾)\mathcal{M}^\infty_\Delta(\overline{M})MΔ∞​(M).

Companions

The mission also states monotonicity of the DEC in the class (the step from the family to the localized class), Proposition 5.1 (posterior sampling bounds the dual square-loss DEC by A/γA/\gammaA/γ), and Proposition 6.7, the gap-dependent variant: for bandits with gap exactly Δ∈(0,18)\Delta \in (0, \tfrac18)Δ∈(0,81​), some M‾\overline{M}M in the class has decγ(MΔ∞(M‾),M‾)≥Δ4I{γ≤A48Δ}\mathsf{dec}_\gamma(\mathcal{M}^\infty_\Delta(\overline{M}), \overline{M}) \ge \frac{\Delta}{4}\mathbb{I}\{\gamma \le \frac{A}{48\Delta}\}decγ​(MΔ∞​(M),M)≥4Δ​I{γ≤48ΔA​}. Proposition 5.2 (inverse gap weighting) and the decoupling step of Proposition 5.1 are already proved on the platform and are included as references.

Significance

Proposition 5.3 is the DEC half of the Ω(AT)\Omega(\sqrt{AT})Ω(AT​) minimax lower bound for multi-armed bandits: combined with Theorem 3.2 and γ≍AT\gamma \asymp \sqrt{AT}γ≍AT​, it shows that the general DEC theory loses nothing in rate on the most basic interactive problem, and that the A/γA/\gammaA/γ upper bounds of Propositions 5.1 and 5.2 are tight up to constants. Lemma 5.1 is reusable: any lower bound on the DEC of a class can be obtained by exhibiting a hard family, and the paper uses it for linear bandits (Proposition 6.2), ReLU bandits (Proposition 6.6) and gap-dependent bounds (Propositions 6.7, 6.8).

The results are proved in the paper. As far as is known, none of them is machine-checked. A formal development provides a verified Bernoulli Hellinger bound, a reusable hard-family lemma for the DEC as published on the platform, and the first verified DEC lower bound for a concrete class.

Difficulty

The DEC is an inf-sup, so a lower bound must hold against every exploration distribution ppp. A natural first attempt picks a single alternative model and compares; this fails because ppp can concentrate on the arm where that alternative differs and then the Hellinger penalty cancels the regret. The family must be spread so that no single ppp is good for all of its members at once, and the constants must balance the regret term against the information term at the chosen Δ\DeltaΔ. In Lean, an additional difficulty is that the DEC is written with real sInf/sSup, which are 000 on empty or unbounded sets: every lower bound requires showing that the relevant sets are nonempty and bounded, and passing from the family to the localized class needs monotonicity of the DEC, which the paper uses without comment.

Formalization scope

  • Finite alphabets. The paper works over measurable spaces; this mission formalizes the finite-alphabet case, as the published FoundationsRL layer does. A model is m : S → Y → ℝ with every m π a probability vector; expectations are finite sums. The DEC is the published FoundationsRL.GeneralDM.decGf, the Hellinger distance hellingerSq, the mean reward fM.
  • Bernoulli subclass. The bandit class is the Bernoulli bandits (Y = Bool, reward 111 on true), a subclass of the paper's all-distributions-on-[0,1][0,1][0,1] class. The DEC is monotone in the class and the localized class shrinks with the class, so the Bernoulli lower bounds imply the paper's.
  • Optimal-arm selector. πM\pi_MπM​ is a parameter piStar with the hypothesis that it picks a maximizer of fMf^MfM; every statement holds for every such selector, so for whichever maximizer the paper meant (the reference Ber(12)\mathrm{Ber}(\tfrac12)Ber(21​) has every arm optimal).
  • Added or dropped hypotheses. Lemma 5.1 makes the page's nonempty decision space explicit; its optimal-arm selector also implies nonemptiness. It adds β≥0\beta \ge 0β≥0, necessary for the printed averaging argument: without it, even identical models can contradict (52). The family definition includes membership of the reference and family models in a probability-model class. The localization claim inherits A≥2A \ge 2A≥2 from Proposition 5.3. Proposition 6.7 states A≥2A \ge 2A≥2, implicit on the page since a gap needs two arms. Proposition 5.1 represents models by their mean vectors and priors as finitely supported (footnote 5).
  • Constants. All constants are the page's: 2−62^{-6}2−6, A/(12γ)A/(12\gamma)A/(12γ), γ≥A/3\gamma \ge A/3γ≥A/3, 3Δ23\Delta^23Δ2, 5Δ25\Delta^25Δ2, Δ/4\Delta/4Δ/4, A/(48Δ)A/(48\Delta)A/(48Δ). The page's proof of Proposition 6.7 names a (Δ,5Δ2,0)(\Delta, 5\Delta^2, 0)(Δ,5Δ2,0)-family but computes with 12Δ212\Delta^212Δ2; the stated proposition is the one its arithmetic supports.
  • No trivialization. A formalization in which the localized class could be empty, or the payoffs unbounded, would make decGf evaluate to the junk value 000; the goal is stated over a class containing M‾\overline{M}M with bounded rewards, so the inequality is the genuine one, and its right side 2−6A/γ2^{-6} A/\gamma2−6A/γ is strictly positive.

Contributions welcome: proofs of the milestones in attack order, sorry-free lemmas on nonemptiness and boundedness of the DEC's inner sets (reusable for every DEC lower bound in this series), and a proof of the gap-dependent companion.

Selected references

  • D. J. Foster, S. M. Kakade, J. Qian, A. Rakhlin, The Statistical Complexity of Interactive Decision Making, arXiv preprint, 2021 (v3, 2023). https://arxiv.org/abs/2112.13487
  • D. J. Foster, A. Rakhlin, Foundations of Reinforcement Learning and Interactive Decision Making, lecture notes, 2023. https://arxiv.org/abs/2312.16730
  • 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
  • J.-Y. Audibert, S. Bubeck, Regret Bounds and Minimax Policies under Partial Monitoring, Journal of Machine Learning Research 11, 2010. https://www.jmlr.org/papers/v11/audibert10a.html
  • D. J. Foster, A. Rakhlin, Beyond UCB: Optimal and Efficient Contextual Bandits with Regression Oracles, ICML, 2020. https://arxiv.org/abs/2002.04926
13 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
Information TheoryMachine LearningProbability·Captain: mikedeng1

A Tutorial on Thompson Sampling I: Any Algorithm Whose Information Ratio Is at Most Γ̄ Has Expected Regret at Most √(Γ̄·H(x*)·T)Research Paper

Motivation

Thompson sampling (TS) chooses each action with the posterior probability that it is optimal. Its practical success in online advertising, recommendation and experimentation raised the question of why it works and when it does not. The tutorial of Russo, Van Roy, Kazerouni, Osband and Wen (Found. Trends Mach. Learn. 11(1), 2018) answers part of that question in §8.1 with an information-theoretic regret bound due to Russo and Van Roy (JMLR 17, 2016). The bound applies to every algorithm, not only TS. It reduces the analysis of a particular algorithm to one quantity, the information ratio: the squared expected regret of a period divided by the information the period yields about the optimal action.

Timeline:

  • 2016: Russo and Van Roy, An information-theoretic analysis of Thompson sampling (JMLR 17), introduce the information ratio with the regret and the mutual information both conditioned on a realised history, and prove the regret bound and bounds on the ratio for TS (for example ∣X∣/2|\mathcal X|/2∣X∣/2 for ∣X∣|\mathcal X|∣X∣-armed bandits and d/2d/2d/2 for linear bandits).
  • 2018: Russo and Van Roy, Learning to optimize via information-directed sampling (Oper. Res. 66), design algorithms that minimise the information ratio directly.
  • 2018: Russo and Van Roy, Satisficing in time-sensitive bandit learning (arXiv:1803.02855), observe that the same bounds hold with the information ratio averaged over histories. This is the form (8.7) the tutorial uses (footnote 1, p. 75).
  • 2018: the tutorial gives the bound as (8.8), with a three-line proof on p. 76 whose last step is cited from the 2016 paper.

Setting

An agent acts in periods t=1,2,…t=1,2,\dotst=1,2,…. In each period it selects an action xtx_txt​ from a finite set X\mathcal XX and observes an outcome yty_tyt​ from a finite set Y\mathcal YY. An unknown parameter θ\thetaθ, taking values in a measurable space Θ\ThetaΘ, has a prior distribution PPP. Given θ\thetaθ and the action xxx, the outcome is drawn from a probability vector qθ(⋅∣x)q_\theta(\cdot\mid x)qθ​(⋅∣x) that depends measurably on θ\thetaθ. The agent receives the reward r(yt)r(y_t)r(yt​) for a known function rrr. The mean reward of action xxx is

μ(x,θ)=∑yqθ(y∣x) r(y),\mu(x,\theta)=\sum_{y}q_\theta(y\mid x)\,r(y),μ(x,θ)=y∑​qθ​(y∣x)r(y),

and the optimal action x∗=x∗(θ)x^*=x^*(\theta)x∗=x∗(θ) is a measurable selector of arg⁡max⁡xμ(x,θ)\arg\max_x\mu(x,\theta)argmaxx​μ(x,θ). Ties may be broken in any measurable way, and every result holds for every selector.

The history Ht\mathbb H_tHt​ is the list (x1,y1),…,(xt,yt)(x_1,y_1),\dots,(x_t,y_t)(x1​,y1​),…,(xt​,yt​). An algorithm π\piπ assigns to every history a probability vector over X\mathcal XX, from which the next action is drawn. The joint law of (θ,Ht)(\theta,\mathbb H_t)(θ,Ht​) is P(dθ)∏s≤tπ(xs∣Hs−1) qθ(ys∣xs)P(d\theta)\prod_{s\le t}\pi(x_s\mid\mathbb H_{s-1})\,q_\theta(y_s\mid x_s)P(dθ)∏s≤t​π(xs​∣Hs−1​)qθ​(ys​∣xs​), and every expectation below is taken under it. The expected cumulative regret, or Bayesian regret, 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 entropy of the prior distribution of x∗x^*x∗ is H(x∗)=−∑aP(x∗=a)log⁡P(x∗=a)H(x^*)=-\sum_a\mathbb P(x^*=a)\log\mathbb P(x^*=a)H(x∗)=−∑a​P(x∗=a)logP(x∗=a). The conditional mutual information I(x∗;(xt,yt)∣Ht−1)I(x^*;(x_t,y_t)\mid\mathbb H_{t-1})I(x∗;(xt​,yt​)∣Ht−1​) is the expected information that the period-ttt observation carries about x∗x^*x∗ given the past, averaged over the past. The information ratio of period ttt is (8.7):

Γt=(E[μ(x∗,θ)−μ(xt,θ)])2I(x∗;(xt,yt)∣Ht−1).\Gamma_t=\frac{\big(\mathbb E[\mu(x^*,\theta)-\mu(x_t,\theta)]\big)^2}{I(x^*;(x_t,y_t)\mid\mathbb H_{t-1})}.Γt​=I(x∗;(xt​,yt​)∣Ht−1​)(E[μ(x∗,θ)−μ(xt​,θ)])2​.

Formalization targets

Goal: (8.8)

For every model and every algorithm, every horizon TTT and every Γˉ≥0\bar\Gamma\ge0Γˉ≥0 with (E[μ(x∗,θ)−μ(xt,θ)])2≤Γˉ I(x∗;(xt,yt)∣Ht−1)\big(\mathbb E[\mu(x^*,\theta)-\mu(x_t,\theta)]\big)^2\le\bar\Gamma\,I(x^*;(x_t,y_t)\mid\mathbb H_{t-1})(E[μ(x∗,θ)−μ(xt​,θ)])2≤ΓˉI(x∗;(xt​,yt​)∣Ht−1​) for t=1,…,Tt=1,\dots,Tt=1,…,T,

E[Regret(T)]≤Γˉ H(x∗) T.\mathbb E[\mathrm{Regret}(T)]\le\sqrt{\bar\Gamma\,H(x^*)\,T}.E[Regret(T)]≤ΓˉH(x∗)T​.

When every mutual information is positive, the least such Γˉ\bar\GammaΓˉ is max⁡t≤TΓt\max_{t\le T}\Gamma_tmaxt≤T​Γt​, which is the tutorial's Γˉ\bar\GammaΓˉ. A companion theorem states that literal form under the positivity hypothesis.

Milestones, in the order of the p. 76 proof

  1. Regret is additive over periods:
E[Regret(T)]=∑t=1TE[μ(x∗,θ)−μ(xt,θ)].\mathbb E[\mathrm{Regret}(T)]=\sum_{t=1}^T\mathbb E[\mu(x^*,\theta)-\mu(x_t,\theta)].E[Regret(T)]=t=1∑T​E[μ(x∗,θ)−μ(xt​,θ)].
  1. The Jensen (Cauchy–Schwarz) step: if δt2≤ΓˉIt\delta_t^2\le\bar\Gamma I_tδt2​≤ΓˉIt​ for every ttt and ∑tIt≤H\sum_tI_t\le H∑t​It​≤H, then ∑tδt≤TΓˉH\sum_t\delta_t\le\sqrt{T\bar\Gamma H}∑t​δt​≤TΓˉH​. This milestone is the published, proved theorem BanditAlgorithm.information_ratio_cumulative_bound_signed. It states the step in the form the proof uses, with the entropy budget as a hypothesis.
  2. The entropy budget:
∑t=1TI(x∗;(xt,yt)∣Ht−1)≤H(x∗).\sum_{t=1}^T I\big(x^*;(x_t,y_t)\mid\mathbb H_{t-1}\big)\le H(x^*).t=1∑T​I(x∗;(xt​,yt​)∣Ht−1​)≤H(x∗).

Significance

The result. (8.8) separates the regret analysis of an algorithm into two independent problems. One is a per-period bound on the information ratio, which depends on the structure of the problem and the algorithm. The other is the entropy of the optimal action, which depends only on the prior. Combined with the bound Γˉ≤∣X∣/2\bar\Gamma\le|\mathcal X|/2Γˉ≤∣X∣/2 that Russo and Van Roy (2016) prove for TS on ∣X∣|\mathcal X|∣X∣-armed bandits, and with H(x∗)≤log⁡∣X∣H(x^*)\le\log|\mathcal X|H(x∗)≤log∣X∣, it gives ∣X∣ Tlog⁡∣X∣/2\sqrt{|\mathcal X|\,T\log|\mathcal X|/2}∣X∣Tlog∣X∣/2​. On the shortest-path problem of p. 76, Γˉ≤d/2\bar\Gamma\le d/2Γˉ≤d/2 makes the regret depend on the number ddd of edges and on H(x∗)H(x^*)H(x∗), not on the exponentially large number of paths. The same inequality is the design criterion of information-directed sampling, which chooses actions to minimise the ratio.

Formalizing it. The bound and its proof are known. The work here is to make the probabilistic model explicit, with a general parameter space, an outcome kernel and randomized algorithms; to formalize the per-period decomposition of the regret and the chain-rule argument behind the entropy budget, which the tutorial cites rather than proves; and to obtain (8.8) for every algorithm in one statement. The Proved platform results for Lattimore and Szepesvári's Bayesian adversarial bandit (Theorem 36.5 and its chain-rule lemmas) are about a different model: one algorithm (TS), the best arm in hindsight of a reward matrix, and the budget log⁡k\log klogk. To our knowledge, (8.8) itself has not been formalized.

Difficulty

The Cauchy–Schwarz step is elementary and already proved on the platform. Milestone 1 is bookkeeping: the cumulative regret integrates over histories of length TTT, while the period-ttt term integrates over histories of length ttt, and later periods must be integrated out.

The substance is the entropy budget. The obvious attempt bounds each period's information by H(x∗)H(x^*)H(x∗), which loses a factor TTT. The bound is uniform in TTT only because the per-period informations telescope: their sum is the information that the whole history carries about x∗x^*x∗, which is at most the entropy of x∗x^*x∗. Each step requires conditional distributions of x∗x^*x∗ given histories, where θ\thetaθ lives in a general measurable space and is integrated against the prior. Zero-probability histories must also be handled consistently with the convention 0log⁡0=00\log0=00log0=0.

Formalization scope

  • X\mathcal XX and Y\mathcal YY are finite types; Θ\ThetaΘ is an arbitrary measurable space with a probability measure PPP. The tutorial allows infinite action sets and general outcomes. The finite specialization makes H(x∗)H(x^*)H(x∗) and every mutual information a finite sum. The kernel qθ(⋅∣x)q_\theta(\cdot\mid x)qθ​(⋅∣x) of §4 replaces the noise representation yt=g(xt,θ,wt)y_t=g(x_t,\theta,w_t)yt​=g(xt​,θ,wt​) of p. 70; both describe the same laws.
  • Standing hypotheses: qθ(y∣x)≥0q_\theta(y\mid x)\ge0qθ​(y∣x)≥0, ∑yqθ(y∣x)=1\sum_yq_\theta(y\mid x)=1∑y​qθ​(y∣x)=1, measurability of θ↦qθ(y∣x)\theta\mapsto q_\theta(y\mid x)θ↦qθ​(y∣x), measurability of the level sets of x∗x^*x∗, and μ(x,θ)≤μ(x∗(θ),θ)\mu(x,\theta)\le\mu(x^*(\theta),\theta)μ(x,θ)≤μ(x∗(θ),θ). They make every integrand bounded and measurable, so no integrability hypothesis is added.
  • An algorithm is a behavioural policy, giving a probability vector over actions for each history. Under perfect recall this is equivalent to an algorithm with internal randomization. Periods are 0-based in Lean: Lean period sss is tutorial period t=s+1t=s+1t=s+1.
  • Logarithms are natural in both III and HHH. The conditional mutual information is defined directly as an average of Kullback–Leibler divergences, with joint probabilities. It is not defined as a difference of unconditional mutual informations: that would make the chain rule true by definition and empty milestone 3. Γt\Gamma_tΓt​ is never written as a quotient in the goal, so a period with zero information cannot turn the bound into one with Γˉ=0\bar\Gamma=0Γˉ=0.
  • Needed infrastructure: marginalisation over later periods of the history law, the chain rule of conditional mutual information for finite variables averaged against a general prior, and I(x∗;HT)≤H(x∗)I(x^*;\mathbb H_T)\le H(x^*)I(x∗;HT​)≤H(x∗). The Proved platform results InformationTheory.kernel_average_categorical_kl_le_marginal_entropy and its _general variant (average KL to the mixture is at most the mixture's entropy), and the adversarial-bandit lemmas BanditAlgorithm.bayesian_ts_conditional_information_gain_chain_rule and bayesian_history_mutualInformation_endpoint_bound, are related tools. A chain-rule library for this history model would be reusable beyond this mission, for example for information-directed sampling. Proofs of auxiliary lemmas are welcome.

The tutorial's argument is on pp. 75–76. The tutorial cites Russo and Van Roy (2016) for the original results, including the entropy budget.

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
  • D. Russo, B. Van Roy, An Information-Theoretic Analysis of Thompson Sampling, Journal of Machine Learning Research 17(68):1–30, 2016. https://jmlr.org/papers/v17/14-087.html (arXiv: https://arxiv.org/abs/1403.5341)
  • 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
  • D. Russo, B. Van Roy, Satisficing in Time-Sensitive Bandit Learning, arXiv preprint, 2018. https://arxiv.org/abs/1803.02855
  • T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020, Chapter 36. https://doi.org/10.1017/9781108571401
5 thms2 active usersReviewed
Machine LearningProbabilityStatistics·Captain: mikedeng1

Thompson Sampling: An Asymptotically Optimal Finite-Time Analysis: Bernoulli Thompson Sampling Has Regret (1+ε)Σ Δ_a(ln T + ln ln T)/K(μ_a, μ*) + CResearch Paper

Motivation

Thompson Sampling is the oldest bandit algorithm: Thompson proposed it in 1933 for allocating patients between two treatments (Thompson 1933). It keeps a posterior distribution over the unknown mean reward of every action, draws one sample from each posterior, and plays the action whose sample is largest. It is simple, widely used, and in experiments competitive with index policies designed for optimality. For decades it had no matching theoretical guarantee.

The benchmark for a stochastic bandit algorithm is the lower bound of Lai and Robbins (1985): every consistent policy draws a suboptimal arm aaa at least (1−o(1))ln⁡T/K(μa,μ∗)(1-o(1))\ln T/K(\mu_a,\mu^*)(1−o(1))lnT/K(μa​,μ∗) times in TTT rounds, where KKK is the Kullback–Leibler divergence. A policy that attains this rate is called asymptotically optimal.

Timeline.

  • 1933: Thompson introduces the algorithm, without analysis.
  • 1985: Lai and Robbins prove the lower bound and construct asymptotically optimal (but complex) policies.
  • 2011–2012: KL-UCB (Garivier and Cappé 2011) and Bayes-UCB (Kaufmann, Cappé and Garivier 2012) attain the bound for Bernoulli rewards with simpler index rules.
  • 2012: Agrawal and Goyal give the first logarithmic regret bound for Thompson Sampling, O((∑a1/Δa2)2ln⁡T)O\big((\sum_a 1/\Delta_a^2)^2\ln T\big)O((∑a​1/Δa2​)2lnT), which does not have the optimal constant.
  • 2012: Kaufmann, Korda and Munos (arXiv:1205.4217) prove that Thompson Sampling with uniform priors is asymptotically optimal for Bernoulli rewards, with a finite-time bound. This mission formalizes that result.

Setting

There are K≥2K\ge2K≥2 arms. Arm aaa yields reward 111 with probability μa∈(0,1)\mu_a\in(0,1)μa​∈(0,1) and 000 otherwise. Arm 111 is the unique optimal arm: μ∗=μ1>μa\mu^*=\mu_1>\mu_aμ∗=μ1​>μa​ for every a≠1a\neq1a=1. The gap of arm aaa is Δa=μ∗−μa\Delta_a=\mu^*-\mu_aΔa​=μ∗−μa​, and μ2\mu_2μ2​ denotes the largest mean among the suboptimal arms.

Thompson Sampling with uniform priors. Let Na,tN_{a,t}Na,t​ be the number of times arm aaa has been drawn before round ttt and Sa,tS_{a,t}Sa,t​ the number of rewards equal to 111 among those draws. With a uniform prior, the posterior of μa\mu_aμa​ is Beta(Sa,t+1, Na,t−Sa,t+1)\mathrm{Beta}(S_{a,t}+1,\,N_{a,t}-S_{a,t}+1)Beta(Sa,t​+1,Na,t​−Sa,t​+1). In round ttt the algorithm draws, independently, a sample θa,t\theta_{a,t}θa,t​ from each posterior, plays At=arg⁡max⁡aθa,tA_t=\arg\max_a\theta_{a,t}At​=argmaxa​θa,t​, and observes the reward of AtA_tAt​.

The regret after TTT rounds is

R(T)=Tμ∗−E[∑t=1TRt]=∑aΔa E[Na,T],\mathcal R(T)=T\mu^*-\mathbb E\Big[\sum_{t=1}^T R_t\Big]=\sum_a\Delta_a\,\mathbb E[N_{a,T}],R(T)=Tμ∗−E[t=1∑T​Rt​]=a∑​Δa​E[Na,T​],

where Na,TN_{a,T}Na,T​ now counts the draws of aaa in rounds 1,…,T1,\dots,T1,…,T. The Bernoulli divergence is K(p,q)=pln⁡pq+(1−p)ln⁡1−p1−qK(p,q)=p\ln\frac pq+(1-p)\ln\frac{1-p}{1-q}K(p,q)=plnqp​+(1−p)ln1−q1−p​.

The analysis compares the Thompson samples with two index rules. For an arm with SSS successes in NNN draws, at round ttt of a horizon TTT, the KL-UCB index is u=sup⁡{x∈[S/N,1):N K(S/N,x)≤ln⁡t+ln⁡ln⁡T}u=\sup\{x\in[S/N,1):N\,K(S/N,x)\le\ln t+\ln\ln T\}u=sup{x∈[S/N,1):NK(S/N,x)≤lnt+lnlnT}, and the Bayes-UCB index qqq is the quantile of order 1−1tln⁡T1-\frac1{t\ln T}1−tlnT1​ of the posterior Beta(S+1,N−S+1)\mathrm{Beta}(S+1,N-S+1)Beta(S+1,N−S+1).

Formalization targets

Goal: Theorem 1 (p. 3)

For every ϵ>0\epsilon>0ϵ>0 there is a constant CCC, depending on ϵ\epsilonϵ and the instance, such that for every horizon TTT,

R(T)≤(1+ϵ)∑a: μa≠μ∗Δa(ln⁡T+ln⁡ln⁡T)K(μa,μ∗)+C.\mathcal R(T)\le(1+\epsilon)\sum_{a:\,\mu_a\neq\mu^*}\frac{\Delta_a(\ln T+\ln\ln T)}{K(\mu_a,\mu^*)}+C.R(T)≤(1+ϵ)a:μa​=μ∗∑​K(μa​,μ∗)Δa​(lnT+lnlnT)​+C.

Only the existence of CCC is asserted; its value is not fixed, so improvements of the lower-order terms do not invalidate the goal.

Milestones

  1. Eq. (1): R(T)=Tμ∗−E[∑t=1TRt]=∑aΔa E[Na,T]\mathcal R(T)=T\mu^*-\mathbb E[\sum_{t=1}^T R_t]=\sum_a\Delta_a\,\mathbb E[N_{a,T}]R(T)=Tμ∗−E[∑t=1T​Rt​]=∑a​Δa​E[Na,T​].
  2. Theorem 2 (p. 4): for each suboptimal aaa and ϵ>0\epsilon>0ϵ>0, E[Na,T]≤(1+ϵ)ln⁡T+ln⁡ln⁡TK(μa,μ1)+C\mathbb E[N_{a,T}]\le(1+\epsilon)\frac{\ln T+\ln\ln T}{K(\mu_a,\mu_1)}+CE[Na,T​]≤(1+ϵ)K(μa​,μ1​)lnT+lnlnT​+C.
  3. Proposition 1 (p. 4): there is b∈(0,1)b\in(0,1)b∈(0,1) with ∑t≥1P(N1,t≤tb)<∞\sum_{t\ge1}\mathbb P(N_{1,t}\le t^b)<\infty∑t≥1​P(N1,t​≤tb)<∞.
  4. Display (4) (p. 5): E[Na,T]≤A+B+2\mathbb E[N_{a,T}]\le A+B+2E[Na,T​]≤A+B+2, an under-estimation term AAA for arm 111 plus a KL-UCB term BBB.
  5. Lemma 1 (p. 5): ∑tP(θ1,t≤μ1−6ln⁡t/N1,t)≤N0(b)+3+Cb\sum_t\mathbb P\big(\theta_{1,t}\le\mu_1-\sqrt{6\ln t/N_{1,t}}\big)\le N_0(b)+3+C_b∑t​P(θ1,t​≤μ1​−6lnt/N1,t​​)≤N0​(b)+3+Cb​.
  6. Lemma 2 (p. 6): B≤(1+ϵ)ln⁡T+ln⁡ln⁡TK(μa,μ1)+DB\le(1+\epsilon)\frac{\ln T+\ln\ln T}{K(\mu_a,\mu_1)}+DB≤(1+ϵ)K(μa​,μ1​)lnT+lnlnT​+D for large TTT.
  7. Lemma 3 (p. 9), in its analytic form (§3.4): a bound on the probability that the optimal arm's samples stay below μ2+δ\mu_2+\deltaμ2​+δ for fff rounds.
  8. Lemma 4 (p. 9): saturated suboptimal arms rarely produce samples above μa+δa\mu_a+\delta_aμa​+δa​.
  9. §2 (p. 3): the Beta–Binomial identity (referenced) and the comparison q<uq<uq<u of the two indices.

Significance

The result. Theorem 1 shows that the leading constant of Thompson Sampling's regret equals the Lai–Robbins constant, so no consistent policy does asymptotically better on Bernoulli bandits. It settled a question open since 1933 and is the starting point for later optimality proofs of Thompson Sampling for exponential families and Gaussian rewards. Proposition 1, a summable tail bound on the event that the optimal arm is under-sampled, is the reusable core: it is a deviation inequality for the number of draws of the optimal arm, which a bound on its expectation does not imply.

The formalization. The result is proved on paper; no machine-checked proof of an optimal-constant bound for Bernoulli Thompson Sampling exists. The platform already has the Gaussian frequentist analysis of Lattimore and Szepesvári (Theorem 36.3) and the non-optimal Agrawal–Goyal bound. This mission adds the Bernoulli, Beta-posterior case with the KL constant, the KL-UCB and Bayes-UCB indices with the ln⁡t+ln⁡ln⁡T\ln t+\ln\ln Tlnt+lnlnT exploration rate, and the optimal-arm tail bound. Corrected statements are part of the output: the explicit constant DDD printed at the end of Lemma 2's proof is too small, the constant Cλ,μ1,μ2C_{\lambda,\mu_1,\mu_2}Cλ,μ1​,μ2​​ printed for Lemma 3 has the wrong sign, and Lemma 3 as printed quantifies over arbitrary random intervals, which its proof does not support.

Difficulty

The standard proof for an optimistic index policy bounds the draws of a suboptimal arm by an under-estimation term ∑tP(l1,t<μ1)\sum_t\mathbb P(l_{1,t}<\mu_1)∑t​P(l1,t​<μ1​) and a concentration term. For Thompson Sampling the under-estimation term is not small: once the posterior of arm 111 is concentrated, P(θ1,t<μ1)\mathbb P(\theta_{1,t}<\mu_1)P(θ1,t​<μ1​) is close to 1/21/21/2 in every round. Comparing θ1,t\theta_{1,t}θ1,t​ with a lower confidence bound μ1−6ln⁡t/N1,t\mu_1-\sqrt{6\ln t/N_{1,t}}μ1​−6lnt/N1,t​​ instead only helps if N1,tN_{1,t}N1,t​ is large with overwhelming probability, and a bound on E[N1,t]\mathbb E[N_{1,t}]E[N1,t​] does not give that.

Proposition 1 supplies it, and it carries most of the difficulty. It must control long stretches of rounds in which arm 111 is never drawn while several suboptimal arms compete, and it needs the samples of arm 111 conditioned on its successes at a random time while the arm is not played, an optional-skipping step that the paper leaves implicit and that is the main measure-theoretic obstacle in Lean.

Formalization scope

Representation. The algorithm is the published Thompson Sampling run (AgrawalGoyalTS_TwoArmed_ThompsonSampling) on the published Bernoulli bandit (bernoulliRelativeEntropy). That run is Agrawal–Goyal's Algorithm 2, which passes each reward through a Bernoulli trial of success probability equal to the reward; for {0,1}\{0,1\}{0,1} rewards the trial returns the reward almost surely, so the posterior is exactly the paper's. Probabilities are values of tsLaw in [0,∞][0,\infty][0,∞], expectations of counts and the regret are lower integrals in [0,∞][0,\infty][0,∞], and every bound compares with the embedding of a real right-hand side.

Conventions.

  • Lean arm 000 is the paper's arm 111. Lean round ttt is the paper's round t+1t+1t+1; Na,tN_{a,t}Na,t​, Sa,tS_{a,t}Sa,t​ are the counts when θa,t\theta_{a,t}θa,t​ is drawn, and the paper's ln⁡t\ln tlnt is ln⁡(t+1)\ln(t+1)ln(t+1) at Lean round ttt.
  • 6ln⁡t/N1,t=∞\sqrt{6\ln t/N_{1,t}}=\infty6lnt/N1,t​​=∞ when N1,t=0N_{1,t}=0N1,t​=0 (p. 4), encoded by cases.
  • The KL-UCB supremum ranges over x<1x<1x<1, because the real-valued divergence at q=1q=1q=1 is a finite junk value; u=1u=1u=1 when N=0N=0N=0 or S=NS=NS=N, cases the page leaves undefined.
  • Every constant with a printed dependence list (CCC, DDD, NNN, bbb, CbC_bCb​) is chosen after the instance (and ϵ\epsilonϵ, the arm) and before TTT.

Added hypotheses and corrections.

  • 0<μa<10<\mu_a<10<μa​<1 for every arm, and K≥2K\ge2K≥2: not stated in the paper but used by its proof (it divides by μ1(1−μ1)\mu_1(1-\mu_1)μ1​(1−μ1​), and K(μa,1)=∞K(\mu_a,1)=\inftyK(μa​,1)=∞).
  • Arm 111 is the unique optimal arm (the paper's standing assumption of §2; its reduction of several optimal arms to one is cited, not proved).
  • Display (4) and the index comparison are stated for T≥3T\ge3T≥3 (the page's T≥eT\ge eT≥e) and rounds 1≤t≤T1\le t\le T1≤t≤T.
  • Lemma 2 asserts only the existence of DDD; Lemma 3 is formalized as the analytic bound of §3.4 with the sign-corrected constant; Lemma 1's CbC_bCb​ is read with the pre-draw count.

Trivializations ruled out. Thompson Sampling is not characterized by a predicate that an arbitrary policy could satisfy: it is the published run, which fixes the posterior draws, the argmax and the updates. The constant is quantified as ∀ϵ ∃C ∀T\forall\epsilon\,\exists C\,\forall T∀ϵ∃C∀T, never ∀T ∃C\forall T\,\exists C∀T∃C, and no real number is extracted from a possibly infinite expectation.

Infrastructure and contributions. A complete development needs Beta–Binomial duality, Chernoff–Hoeffding bounds for binomial tails, optional skipping for the reward stacks of the played arms, and the KL-UCB counting argument. The Beta–Binomial identity is an open milestone shared with the Agrawal–Goyal mission. Proofs of any milestone, in particular Proposition 1, Lemma 4 and the analytic Lemma 3, are welcome; Lemma 3's bound and the index comparison are pure real analysis and reusable beyond bandits.

Selected references

  • E. Kaufmann, N. Korda, R. Munos, Thompson Sampling: An Asymptotically Optimal Finite Time Analysis, ALT 2012 (LNCS 7568); arXiv:1205.4217v2. https://arxiv.org/abs/1205.4217
  • 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.2307/2332286
  • T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6, 1985. https://doi.org/10.1016/0196-8858(85)90002-8
  • S. Agrawal, N. Goyal, Analysis of Thompson Sampling for the multi-armed bandit problem, COLT 2012; arXiv:1111.1797. https://arxiv.org/abs/1111.1797
  • A. Garivier, O. Cappé, The KL-UCB algorithm for bounded stochastic bandits and beyond, COLT 2011; arXiv:1102.2490. https://arxiv.org/abs/1102.2490
  • E. Kaufmann, O. Cappé, A. Garivier, On Bayesian upper confidence bounds for bandit problems, AISTATS 2012. https://proceedings.mlr.press/v22/kaufmann12.html
17 thms2 active usersReviewed
Operations ResearchProbability·Captain: mikedeng1

Linearly Parameterized Bandits 2: Under Smooth Best Arm Response, the Phased Exploration and Greedy Exploitation Policy Has Regret at Most a₁(‖z‖ + 1/‖z‖)r√TResearch Paper

Motivation

In a linearly parameterized bandit, a decision maker repeatedly chooses an arm from a possibly infinite set of vectors and receives a noisy reward whose mean is linear in an unknown parameter. Classical bandit policies treat arms as independent and must try each arm, so their regret grows with the number of arms (Lai and Robbins 1985); in applications such as marketing, where similar products are expected to have similar sales, the linear structure lets one observation inform many arms, and the regret can be made independent of the number of arms. Rusmevichientong and Tsitsiklis (arXiv:0812.3465v2, Mathematics of Operations Research 2010) show that when the arms form the unit sphere of Rr\mathbb R^rRr, no policy has Bayes risk below order rTr\sqrt TrT​. This mission concerns their matching upper bound: a simple policy that alternates exploration and greedy exploitation in cycles of growing length attains regret of order rTr\sqrt TrT​ whenever the best arm depends smoothly on the parameter.

The linearly parameterized bandit was studied earlier by Abe and Long (1999), Auer (2002) and, independently of this paper, Dani, Hayes and Kakade (2008), whose policies rely on confidence regions for the parameter. The result here shows that for arm sets with a smooth best arm, such as balls and, by Polovinkin (1996), ellipsoids, a phase-based policy with a single least squares estimate per cycle suffices.

Setting

Let r≥2r \ge 2r≥2 and let Ur⊂Rr\mathcal U_r \subset \mathbb R^rUr​⊂Rr be a compact set of arms. Playing arm uuu in period ttt yields the reward

Xtu=u′Z+Wtu,X^u_t = u'Z + W^u_t,Xtu​=u′Z+Wtu​,

where Z∈RrZ \in \mathbb R^rZ∈Rr is an unknown parameter and the errors WtuW^u_tWtu​ are independent of each other and of ZZZ, identically distributed in ttt for each arm, with mean zero.

Assumption 1 fixes constants σ0,uˉ,λ0>0\sigma_0, \bar u, \lambda_0 > 0σ0​,uˉ,λ0​>0 that do not depend on rrr: (a) every error satisfies E[exWtu]≤ex2σ02/2\mathbb E[e^{xW^u_t}] \le e^{x^2\sigma_0^2/2}E[exWtu​]≤ex2σ02​/2 for all x∈Rx \in \mathbb Rx∈R; (b) every arm has ∥u∥≤uˉ\|u\| \le \bar u∥u∥≤uˉ, and Ur\mathcal U_rUr​ contains linearly independent arms b1,…,brb_1, \dots, b_rb1​,…,br​ with λmin⁡(∑kbkbk′)≥λ0\lambda_{\min}(\sum_k b_kb_k') \ge \lambda_0λmin​(∑k​bk​bk′​)≥λ0​.

The smooth best arm response condition SBAR(JJJ) requires that every z≠0z \ne 0z=0 have a unique best arm u∗(z)u^*(z)u∗(z), the maximizer of v′zv'zv′z over Ur\mathcal U_rUr​, and that ∥u∗(z)−u∗(y)∥≤J∥z−y∥\|u^*(z) - u^*(y)\| \le J\|z - y\|∥u∗(z)−u∗(y)∥≤J∥z−y∥ for unit vectors z,yz, yz,y.

The PEGE policy (Phased Exploration and Greedy Exploitation) runs in cycles c=1,2,…c = 1, 2, \dotsc=1,2,…. In cycle ccc it plays b1,…,brb_1, \dots, b_rb1​,…,br​ once each, computes the ordinary least squares estimate

Z^(c)=1c(∑k=1rbkbk′)−1∑s=1c∑k=1rbkXbk(s)\widehat Z(c) = \frac1c\Big(\sum_{k=1}^r b_kb_k'\Big)^{-1}\sum_{s=1}^c\sum_{k=1}^r b_k X^{b_k}(s)Z(c)=c1​(k=1∑r​bk​bk′​)−1s=1∑c​k=1∑r​bk​Xbk​(s)

from all exploration rewards so far, and then plays a greedy arm G(c)∈arg⁡max⁡v∈Urv′Z^(c)G(c) \in \arg\max_{v \in \mathcal U_r} v'\widehat Z(c)G(c)∈argmaxv∈Ur​​v′Z(c) for ccc periods, ties broken arbitrarily.

The regret given Z=zZ = zZ=z over TTT periods is Regret(z,T,PEGE)=∑t=1TE[max⁡v∈Urv′z−Ut′z∣Z=z]\mathrm{Regret}(z, T, \mathrm{PEGE}) = \sum_{t=1}^T \mathbb E[\max_{v \in \mathcal U_r} v'z - U_t'z \mid Z = z]Regret(z,T,PEGE)=∑t=1T​E[maxv∈Ur​​v′z−Ut′​z∣Z=z], where UtU_tUt​ is the arm played in period ttt, and the risk is Risk(T,PEGE)=E[Regret(Z,T,PEGE)]\mathrm{Risk}(T, \mathrm{PEGE}) = \mathbb E[\mathrm{Regret}(Z, T, \mathrm{PEGE})]Risk(T,PEGE)=E[Regret(Z,T,PEGE)] for a prior on ZZZ.

Formalization targets

Goal: Theorem 3.1 (p. 15)

Under Assumption 1 and SBAR(JJJ), there is a1>0a_1 > 0a1​>0 depending only on σ0,uˉ,λ0,J\sigma_0, \bar u, \lambda_0, Jσ0​,uˉ,λ0​,J such that for every r≥2r \ge 2r≥2, every z≠0z \ne 0z=0 and T≥rT \ge rT≥r,

Regret(z,T,PEGE)≤a1(∥z∥+1∥z∥)rT,\mathrm{Regret}(z, T, \mathrm{PEGE}) \le a_1\Big(\|z\| + \frac1{\|z\|}\Big) r\sqrt T,Regret(z,T,PEGE)≤a1​(∥z∥+∥z∥1​)rT​,

and, if E∥Z∥≤M\mathbb E\|Z\| \le ME∥Z∥≤M and E[1/∥Z∥]≤M\mathbb E[1/\|Z\|] \le ME[1/∥Z∥]≤M for every rrr, there is a2a_2a2​ depending only on σ0,uˉ,λ0,J,M\sigma_0, \bar u, \lambda_0, J, Mσ0​,uˉ,λ0​,J,M with

Risk(T,PEGE)≤a2 rT.\mathrm{Risk}(T, \mathrm{PEGE}) \le a_2\, r\sqrt T.Risk(T,PEGE)≤a2​rT​.

The constants are unspecified; only the order rTr\sqrt TrT​ and its uniformity in rrr are asserted.

Milestones on the way

  • Lemma 3.4 (p. 16): E[∥Z^(c)−z∥2∣Z=z]≤h1r/c\mathbb E[\|\widehat Z(c) - z\|^2 \mid Z = z] \le h_1 r/cE[∥Z(c)−z∥2∣Z=z]≤h1​r/c.
  • Lemma 3.5 (p. 17): ∥w/∥w∥−z/∥z∥∥≤2∥w−z∥/max⁡{∥z∥,∥w∥}\|w/\|w\| - z/\|z\|\| \le 2\|w - z\|/\max\{\|z\|, \|w\|\}∥w/∥w∥−z/∥z∥∥≤2∥w−z∥/max{∥z∥,∥w∥}.
  • Lemma 3.6 (p. 17): E[max⁡uz′(u−G(c))∣Z=z]≤rh2/(c∥z∥)\mathbb E[\max_{u} z'(u - G(c)) \mid Z = z] \le r h_2/(c\|z\|)E[maxu​z′(u−G(c))∣Z=z]≤rh2​/(c∥z∥).
  • Sec. 3.1 (p. 18): Regret(z,rK+∑c=1Kc,PEGE)≤h3r∥z∥K+h4∑c=1Kr/∥z∥\mathrm{Regret}(z, rK + \sum_{c=1}^K c, \mathrm{PEGE}) \le h_3 r\|z\|K + h_4\sum_{c=1}^K r/\|z\|Regret(z,rK+∑c=1K​c,PEGE)≤h3​r∥z∥K+h4​∑c=1K​r/∥z∥.

Further results

  • Lemma 3.2 (p. 16): a density condition near the origin gives E[1/∥Z∥]≤M\mathbb E[1/\|Z\|] \le ME[1/∥Z∥]≤M; for Z∼N(0,Ir/r)Z \sim N(0, I_r/r)Z∼N(0,Ir​/r), E∥Z∥≤1\mathbb E\|Z\| \le 1E∥Z∥≤1 and E[1/∥Z∥]≤π\mathbb E[1/\|Z\|] \le \sqrt\piE[1/∥Z∥]≤π​.
  • The unit ball and the unit sphere satisfy SBAR(1) (pp. 15–16).
  • Corollary 3.3 (p. 16): on the unit sphere with standard normal errors, Regret≤a3(∥z∥+1/∥z∥)rT\mathrm{Regret} \le a_3(\|z\| + 1/\|z\|)r\sqrt TRegret≤a3​(∥z∥+1/∥z∥)rT​ and, under the prior N(0,Ir/r)N(0, I_r/r)N(0,Ir​/r), Risk≤a3rT\mathrm{Risk} \le a_3 r\sqrt TRisk≤a3​rT​.

Significance

Theorem 3.1 with the lower bound of Theorem 2.1 of the same paper identifies rTr\sqrt TrT​ as the optimal order of the regret and the Bayes risk on the unit sphere, and Corollary 3.3 makes this precise. The bound is achieved by a policy with no confidence sets and no per-period optimization beyond one linear maximization per cycle. The factor ∥z∥+1/∥z∥\|z\| + 1/\|z\|∥z∥+1/∥z∥ shows how the difficulty depends on the size of the parameter; combined with the trivial bound 2uˉ∥z∥T2\bar u\|z\|T2uˉ∥z∥T, the regret does not deteriorate as ∥z∥→0\|z\| \to 0∥z∥→0.

The results are proved in the paper; none of them is machine-checked. A formalization yields a verified dimension-explicit regret bound for an explore–exploit policy with a continuum of arms, the least squares error bound under sub-Gaussian noise (Lemma 3.4), and Lipschitz properties of normalization and of the best-arm map on balls, all of which are reusable.

Difficulty

The expected instantaneous regret of a greedy arm computed from a noisy estimate is not obviously small: for a general compact arm set the best arm may jump under a small perturbation of the parameter, and the regret is then of order ∥z∥\|z\|∥z∥ however good the estimate is. The SBAR condition prevents such jumps, but only for unit vectors, so the estimation error must be transferred to the direction of the estimate, which costs a factor 1/∥z∥1/\|z\|1/∥z∥. The estimate may also vanish with positive probability when the errors are discrete, where the greedy arm is arbitrary. Finally, the constants must be uniform in the dimension rrr: a bound with an rrr-dependent constant is immediate from Regret≤2uˉ∥z∥T\mathrm{Regret} \le 2\bar u\|z\|TRegret≤2uˉ∥z∥T, so the content of the theorem lies in keeping every constant independent of rrr and of the arm set.

Formalization scope

Vectors live in EuclideanSpace ℝ (Fin r), so the norm is Euclidean and the inner product is u′zu'zu′z. The model's standing assumptions are encoded as follows: Ur\mathcal U_rUr​ is compact; the errors of each arm have mean zero and a law νu\nu_uνu​; Assumption 1(a) requires the moment generating function to exist and satisfy the sub-Gaussian bound; Assumption 1(b) writes λmin⁡(∑kbkbk′)≥λ0\lambda_{\min}(\sum_k b_kb_k') \ge \lambda_0λmin​(∑k​bk​bk′​)≥λ0​ as λ0∥x∥2≤∑k(bk′x)2\lambda_0\|x\|^2 \le \sum_k (b_k'x)^2λ0​∥x∥2≤∑k​(bk′​x)2. Every constant is quantified before the dimension rrr, the arm set, the error laws, the exploration arms, the tie-breaking rule, the prior, zzz and TTT; a version with constants depending on rrr or on the instance is trivial and is excluded by this quantifier order.

PEGE uses exactly the exploration arms bkb_kbk​ of Assumption 1(b), which are an argument of the policy. The greedy rule is any measurable selection of a maximizer; measurability, implicit on the page, is a hypothesis. Since PEGE uses only exploration rewards, the expectation is over an independent table of exploration errors Wbk(s)W^{b_k}(s)Wbk​(s) with laws νbk\nu_{b_k}νbk​​, with the parameter fixed to zzz ("given Z=zZ = zZ=z"). Regret and risk are lower Lebesgue integrals of nonnegative gaps, valued in [0,∞][0, \infty][0,∞], so an upper bound is never satisfied by a default value of an integral. The moment conditions on the prior are lower Lebesgue integrals, so E[1/∥Z∥]=∞\mathbb E[1/\|Z\|] = \inftyE[1/∥Z∥]=∞ if the prior charges the origin.

Two printed statements are adjusted. Lemma 3.6 is printed "for any z∈Rrz \in \mathbb R^rz∈Rr", but its bound rh2/(c∥z∥)r h_2/(c\|z\|)rh2​/(c∥z∥) is undefined at z=0z = 0z=0; it is stated for z≠0z \ne 0z=0. The p. 18 cycle-sum display is likewise stated for z≠0z \ne 0z=0. In Corollary 3.3, the exploration arms are those of Assumption 1(b) with uˉ=λ0=1\bar u = \lambda_0 = 1uˉ=λ0​=1, as in its proof. Measurability of the density in Lemma 3.2(a) is added.

The source is arXiv:0812.3465v2; its printed page numbers equal the PDF page numbers. Contributions welcome: proofs of the milestones, a sub-Gaussian least squares library, and the Lipschitz analysis of best-arm maps.

Selected references

  • P. Rusmevichientong, J. N. Tsitsiklis, Linearly Parameterized Bandits, arXiv:0812.3465v2, 2010; Mathematics of Operations Research 35(2), 2010. https://arxiv.org/abs/0812.3465v2
  • V. Dani, T. P. Hayes, S. M. Kakade, Stochastic Linear Optimization under Bandit Feedback, COLT 2008. https://www.learningtheory.org/colt2008/papers/80-Dani.pdf
  • P. Auer, Using Confidence Bounds for Exploitation-Exploration Trade-offs, Journal of Machine Learning Research 3, 2002. https://www.jmlr.org/papers/v3/auer02a.html
  • E. S. Polovinkin, Strongly convex analysis, Sbornik: Mathematics 187(2), 1996. https://doi.org/10.1070/SM1996v187n02ABEH000111
13 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
Operations ResearchProbabilityStatistics·Captain: mikedeng1

Some Aspects of the Sequential Design of Experiments I: Greedy Sampling with Density-Zero Forced Draws Earns max(α, β) per Draw from Any Two Populations with Finite MeansResearch Paper

Motivation

An experimenter can draw observations from two populations AAA and BBB, one at a time, and wants the sum of the observations to be as large as possible. The populations might be two treatments, two seed varieties or two manufacturing processes; their means are unknown. Each draw both earns a reward and tells the experimenter something about the population it came from. This is the two-armed bandit problem. Herbert Robbins posed it in its modern form in his 1952 address to the American Mathematical Society, Some aspects of the sequential design of experiments, the paper usually credited with founding the field.

Section 2 of that paper asks a basic question (display (12), p. 532): is there any sampling rule whose average reward per draw approaches the larger of the two means, whatever the populations are? Robbins answers yes, for any two populations with finite means, using one explicit rule: forced sampling along a sparse schedule, greedy choice otherwise. This mission formalizes that answer.

Timeline:

  • 1933. Thompson proposes a randomized rule for choosing between two treatments (Biometrika 25).
  • 1952. Robbins states the problem for general populations and gives the rule Rˉ\bar RRˉ with lim⁡nLn(A,B,Rˉ)=0\lim_n L_n(A,B,\bar R) = 0limn​Ln​(A,B,Rˉ)=0 for any two populations with finite means. The argument is only indicated ("it can be shown").
  • 1985. Lai and Robbins replace o(n)o(n)o(n) loss by the optimal logarithmic regret for parametric families (Adv. Appl. Math. 6).
  • 2020. Lattimore and Szepesvári give the textbook account of explore-then-commit and ε\varepsilonε-greedy rules, which descend from forced sampling (Bandit Algorithms).

Setting

Population AAA has distribution function FFF and population BBB has GGG, with expectations

α=∫−∞∞x dF(x),β=∫−∞∞x dG(x),\alpha = \int_{-\infty}^{\infty} x\,dF(x), \qquad \beta = \int_{-\infty}^{\infty} x\,dG(x),α=∫−∞∞​xdF(x),β=∫−∞∞​xdG(x),

assumed to exist, i.e. to be finite (display (1), p. 529). Nothing else is assumed about FFF and GGG.

A sampling rule RRR draws x1,x2,…x_1, x_2, \dotsx1​,x2​,… one at a time; the population of the iii-th draw may depend on x1,…,xi−1x_1, \dots, x_{i-1}x1​,…,xi−1​, and given that choice xix_ixi​ is a fresh draw from the chosen population. Write Sn=x1+⋯+xnS_n = x_1 + \cdots + x_nSn​=x1​+⋯+xn​. The loss of RRR after nnn draws (display (9), p. 531) is

Ln(A,B,R)=max⁡(α,β)−E ⁣(Snn),L_n(A, B, R) = \max(\alpha, \beta) - E\!\left(\frac{S_n}{n}\right),Ln​(A,B,R)=max(α,β)−E(nSn​​),

the average shortfall per draw compared with someone who knows the better population and samples only from it.

The forcing schedule (display (13), p. 532) is a pair of disjoint, strictly increasing sequences of positive integers

1=a1<a2<⋯ ,2=b1<b2<⋯1 = a_1 < a_2 < \cdots, \qquad 2 = b_1 < b_2 < \cdots1=a1​<a2​<⋯,2=b1​<b2​<⋯

of density zero: the fraction of the integers 1,…,n1, \dots, n1,…,n that are an aaa or a bbb tends to 000.

The rule Rˉ\bar RRˉ: at draw iii, sample AAA if iii is an aaa, sample BBB if iii is a bbb, and otherwise sample AAA if the mean of the previous observations from AAA is strictly larger than the mean of the previous observations from BBB, and BBB if it is not (a tie goes to BBB).

Formalization targets

Goal: Section 2, Eq. (16)

For every pair of populations with finite means and every forcing schedule,

lim⁡n→∞Ln(A,B,Rˉ)=0.\lim_{n\to\infty} L_n(A, B, \bar R) = 0.n→∞lim​Ln​(A,B,Rˉ)=0.

The goal is about the specific rule Rˉ\bar RRˉ and every schedule, not about the mere existence of a rule. It contains no rate and no constant, so it stays valid when the rule is analysed more finely.

Milestone: Eq. (14), almost-sure average

lim⁡n→∞Snn=max⁡(α,β)with probability 1.\lim_{n\to\infty} \frac{S_n}{n} = \max(\alpha, \beta) \quad\text{with probability } 1.n→∞lim​nSn​​=max(α,β)with probability 1.

Milestone: Eq. (15), expected average

lim⁡n→∞E ⁣(Snn)=max⁡(α,β).\lim_{n\to\infty} E\!\left(\frac{S_n}{n}\right) = \max(\alpha, \beta).n→∞lim​E(nSn​​)=max(α,β).

The goal follows from (15) by the definition (9); the paper's chain is (14) ⇒ (15) ⇒ (16).

Significance

The result. The theorem answers question (12) and shows that "learning while earning" costs nothing in the long run, per draw, under no assumption beyond finite means: no bounded support, no variance, no parametric family. It is the first consistency result for an adaptive allocation rule, and it fixes the benchmark later refined by logarithmic-regret theory. In modern terms, a forced-exploration greedy policy has regret o(n)o(n)o(n) on every two-armed bandit with integrable rewards.

Formalizing it. The paper gives no proof; it says the results "can be shown". A machine-checked proof records the argument in full. The platform already has finite-horizon regret bounds for ε\varepsilonε-greedy with bounded rewards (FoundationsRL.Bandits.epsilon_greedy_regret_bound), consistency in the stronger Lai–Robbins sense as a definition (ConsistentBanditPolicy), and a forced-sample tail bound for the Lasso bandit (BastaniBayati.LassoBandit.forced_sample_estimator_tail). None of these covers a deterministic greedy rule, almost-sure convergence, or unbounded rewards with only a first moment, and none of them proves (14)–(16).

Difficulty

The obvious argument is: each population is sampled infinitely often because of the forced draws, so each running mean converges by the strong law, so greedy eventually picks the better population, and the forced draws are negligible. Two steps are not routine.

First, the strong law applies to i.i.d. sequences, while the observations from AAA arrive at random, history-dependent times. The values seen from AAA have to be identified, in law, with an i.i.d. sequence that is independent of the times at which they are used.

Second, passing from (14) to (15) needs uniform integrability of Sn/nS_n/nSn​/n, and only first moments are available. No second moment is assumed, so variance bounds are not available.

When α=β\alpha = \betaα=β greedy need not settle on either population; the statement must hold anyway, and an argument that waits for greedy to "lock in" does not cover this case.

Formalization scope

The populations are a published two-armed BanditAlgorithm.StochasticBandit 2. AAA is arm 0 and BBB is arm 1; α\alphaα and β\betaβ are banditArmMean. A rule is a published BanditPolicy 2, and EEE is the integral against the published infinite-horizon law banditTrajMeasure, whose coordinate ttt is the paper's draw t+1t+1t+1. Explicit readings of the paper's phrases:

  • "the expectations exist" means both reward laws are Integrable. This is the only assumption on the populations: the class DDD of the paper is left arbitrary, and α=β\alpha = \betaα=β is allowed.
  • "density 0" means #({1,…,n} ∩ (range a ∪ range b)) / n → 0 in R\mathbb RR. The sequences are infinite and 0-indexed (a 0 = 1, b 0 = 2).
  • "according as … exceeds or does not exceed": AAA exactly when meanA_AA​ > meanB_BB​, so ties go to BBB.
  • "with probability 1" is ∀ᵐ under the trajectory law, and "lim" is Tendsto … atTop along n∈Nn \in \mathbb Nn∈N.
  • Sn/nS_n/nSn​/n uses real division; its value at n=0n = 0n=0 plays no role.

Trivializing formalizations are excluded. Without integrability the means and expectations are junk zeros; this is ruled out by the integrability hypothesis. Replacing Rˉ\bar RRˉ by "there exists a rule" would weaken the goal to (12). Finite forcing sets would make the statement false.

The needed infrastructure is a representation of the adaptive sampling process by per-population i.i.d. reward sequences, the strong law along random subsequences, and a uniform-integrability argument for averages under only a first moment. All three are reusable for other forced-exploration and greedy policies. Contributions of these general lemmas as separate theorems are welcome.

Selected references

  • H. Robbins, Some aspects of the sequential design of experiments, Bull. Amer. Math. Soc. 58(5) (1952), 527–535. https://doi.org/10.1090/S0002-9904-1952-09620-8
  • W. R. Thompson, On the likelihood that one unknown probability exceeds another in view of the evidence of two samples, Biometrika 25 (1933), 285–294. https://doi.org/10.2307/2332286
  • T. L. Lai and H. Robbins, Asymptotically efficient adaptive allocation rules, Adv. Appl. Math. 6 (1985), 4–22. https://doi.org/10.1016/0196-8858(85)90002-8
  • T. Lattimore and Cs. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020. 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
Machine LearningOperations ResearchProbability·Captain: mikedeng1

The Best of Both Worlds: Stochastic and Adversarial Bandits: SAO Has Pseudo-Regret O(K log K log²β/Δ) on Stochastic Rewards and Regret Õ(√(nK)) Against Adaptive AdversariesResearch Paper

Motivation

In a multi-armed bandit problem a learner chooses one of KKK actions in each of nnn rounds and observes only the reward of the chosen action. Two models of the rewards have separate theories. In the stochastic model, each arm pays independent draws from a fixed distribution; algorithms such as UCB1 (Auer, Cesa-Bianchi & Fischer 2002) have regret of order ∑ilog⁡(n)/Δi\sum_i \log(n)/\Delta_i∑i​log(n)/Δi​, logarithmic in nnn. In the adversarial model, an adversary chooses the rewards; Exp3 and its variants (Auer, Cesa-Bianchi, Freund & Schapire 2002) have regret of order nK\sqrt{nK}nK​, which is optimal there. An algorithm tuned for one model fails in the other: stochastic algorithms can suffer linear regret against an adversary, and adversarial algorithms pay n\sqrt nn​ even when the rewards are i.i.d.

Bubeck and Slivkins (arXiv:1202.4473, COLT 2012) asked whether one algorithm can be near-optimal in both models without knowing which one it faces. They answered yes with the algorithm SAO. This result started the "best of both worlds" line of work on bandits. Later contributions include EXP3++ (Seldin & Slivkins 2014) and Tsallis-INF (Zimmert & Seldin 2021).

Setting

There are K≥2K\ge2K≥2 arms and n≥Kn\ge Kn≥K rounds. On round ttt the algorithm draws an arm ItI_tIt​ from a probability vector pt=(p1,t,…,pK,t)p_t=(p_{1,t},\dots,p_{K,t})pt​=(p1,t​,…,pK,t​) computed from the history it has observed. At the same time a reward vector gt∈[0,1]Kg_t\in[0,1]^Kgt​∈[0,1]K is fixed, and the algorithm observes only gIt,tg_{I_t,t}gIt​,t​.

  • Adversarial model. The vector gtg_tgt​ is chosen by an adaptive adversary: a function of the arms I1,…,It−1I_1,\dots,I_{t-1}I1​,…,It−1​ played earlier, but not of ItI_tIt​. The regret is Rn=max⁡i∑t=1ngi,t−∑t=1ngIt,tR_n=\max_i\sum_{t=1}^n g_{i,t}-\sum_{t=1}^n g_{I_t,t}Rn​=maxi​∑t=1n​gi,t​−∑t=1n​gIt​,t​.
  • Stochastic model. There are distributions ν1,…,νK\nu_1,\dots,\nu_Kν1​,…,νK​ on [0,1][0,1][0,1] with means μi\mu_iμi​, and all gi,t∼νig_{i,t}\sim\nu_igi,t​∼νi​ are independent. The pseudo-regret is R‾n=∑t=1n(max⁡iμi−μIt)\overline R_n=\sum_{t=1}^n(\max_i\mu_i-\mu_{I_t})Rn​=∑t=1n​(maxi​μi​−μIt​​). The gap of arm iii is Δi=max⁡jμj−μi\Delta_i=\max_j\mu_j-\mu_iΔi​=maxj​μj​−μi​, and the minimal gap is Δ=min⁡i:Δi>0Δi\Delta=\min_{i:\Delta_i>0}\Delta_iΔ=mini:Δi​>0​Δi​.

The analysis uses importance-weighted estimates H~i,t=1t∑s≤tgi,s1{Is=i}/pi,s\widetilde H_{i,t}=\frac1t\sum_{s\le t}g_{i,s}\mathbb 1_{\{I_s=i\}}/p_{i,s}Hi,t​=t1​∑s≤t​gi,s​1{Is​=i}​/pi,s​, the sample means H^i,t\widehat H_{i,t}Hi,t​, the averages Hi,t=1t∑s≤tgi,sH_{i,t}=\frac1t\sum_{s\le t}g_{i,s}Hi,t​=t1​∑s≤t​gi,s​, and the play counts Ti(t)T_i(t)Ti​(t).

SAO (Algorithm 1 of the paper) takes a parameter β>1\beta>1β>1. It keeps a set of active arms, initially all arms, and samples them uniformly at first. On each round it applies a test, (12), that deactivates an arm whose estimate H~i,t\widetilde H_{i,t}Hi,t​ falls far below the best active one. The probability of a deactivated arm then decays as qiτi/tq_i\tau_i/tqi​τi​/t, where τi\tau_iτi​ is the deactivation time and qiq_iqi​ the arm's probability at that moment. Three further tests, (13)–(15), check that the observations stay consistent with stochastic rewards. If any of them fails on round τ0\tau_0τ0​, SAO switches permanently to the adversarial algorithm Exp3.P (Bubeck & Cesa-Bianchi 2012, Fig. 3.1) for the remaining rounds.

Formalization targets

Goal: Theorem 4.1, high-probability form

For every δ∈(0,1)\delta\in(0,1)δ∈(0,1) let β=10Kn3δ−1\beta=10Kn^3\delta^{-1}β=10Kn3δ−1. With probability at least 1−δ1-\delta1−δ, SAO with parameter β\betaβ satisfies, in the stochastic model (whenever some arm has Δi>0\Delta_i>0Δi​>0),

R‾n≤260K(1+log⁡K)log⁡2(β)Δ,\overline R_n\le\frac{260K(1+\log K)\log^2(\beta)}{\Delta},Rn​≤Δ260K(1+logK)log2(β)​,

and, against every adaptive adversary with rewards in [0,1][0,1][0,1],

Rn≤60(1+log⁡K)(1+log⁡n)nKlog⁡(β)+5K2log⁡2(β)+200K2log⁡2(β).R_n\le60(1+\log K)(1+\log n)\sqrt{nK\log(\beta)+5K^2\log^2(\beta)}+200K^2\log^2(\beta).Rn​≤60(1+logK)(1+logn)nKlog(β)+5K2log2(β)​+200K2log2(β).

Milestones

The milestones follow the paper's proof in order:

  • Freedman's inequality (Theorem 4.3) in the paper's two-sided form, and its variance-adaptive form, Lemma 4.4.
  • The concentration lemmas for SAO's estimates (Lemmas 4.5, 4.6, 4.7) and the Exp3.P phase (Lemma 4.8).
  • The two good events of §4.1, (21)–(25).
  • The deterministic consequences on those events: Exp3.P is never started in the stochastic model; suboptimal arms are deactivated by time 260Klog⁡(β)/Δi2260K\log(\beta)/\Delta_i^2260Klog(β)/Δi2​; ∑iqi≤1+log⁡K\sum_iq_i\le1+\log K∑i​qi​≤1+logK, (27); and the adversarial regret bound of §4.3.
  • The two halves of Theorem 4.1.

Significance

The theorem shows that the stochastic and adversarial regret rates are not in conflict. A single algorithm, with no information about the model, gets O(Klog⁡Klog⁡2(n/δ)/Δ)O(K\log K\log^2(n/\delta)/\Delta)O(KlogKlog2(n/δ)/Δ) pseudo-regret on stochastic rewards and O~(nK)\tilde O(\sqrt{nK})O~(nK​) regret against adaptive adversaries. Each rate is within polylogarithmic factors of optimal for its model. Later algorithms improved the logarithmic factors and removed the explicit switching, but they are compared against this result.

The theorem is proved in the paper. It is not known to have a machine-checked proof. Formalizing it requires a precise model of an adaptive adversary interacting with a randomized algorithm, martingale concentration with random variance (Lemma 4.4), and an exact statement of SAO including its boundary cases. The pieces are reusable: the interaction model, the estimators, Exp3.P and its high-probability guarantee all apply to other adversarial bandit results.

Difficulty

Neither standard analysis carries over. In the stochastic model, SAO's sampling probabilities are random and depend on the past, and a deactivated arm's probability keeps changing. Hoeffding-type bounds for a fixed sampling scheme therefore do not apply to H~i,t\widetilde H_{i,t}Hi,t​. The variance of the importance-weighted estimate grows like ∑s1/pi,s\sum_s1/p_{i,s}∑s​1/pi,s​, which is controlled only through the algorithm's own schedule (16). This is why Lemma 4.5 has the two-part radius with max⁡(t−τi,0)/(qiτit)\max(t-\tau_i,0)/(q_i\tau_it)max(t−τi​,0)/(qi​τi​t). In the adversarial model, the deterministic argument has to show that whenever the consistency tests pass, the regret accumulated before the switch is already small, for an adversary that adapts to the arms played. A union bound over all quantities, all arms and all times (§4.1) is needed before any deterministic reasoning, so every constant in the event matters.

Formalization scope

All declarations live in the namespace BestBothWorlds.SAO.

  • Arms and paths. Arms are Fin K and rounds are 1,…,n1,\dots,n1,…,n. An arm path is Fin n → Fin K.
  • Algorithms and adversaries. An algorithm is a deterministic map from the observed history to a probability vector. A deterministic adaptive adversary is a map from the list of earlier arms to a reward vector; randomized adversaries are mixtures of these.
  • Probabilities. For a fixed adversary, the probability of an event is ∑I∈E∏tpIt,t\sum_{I\in E}\prod_tp_{I_t,t}∑I∈E​∏t​pIt​,t​. In the stochastic model this is integrated against the product law of the reward table.
  • Logarithms and constants. Real.log is the natural logarithm. All constants of Theorem 4.1 are explicit, with β=10Kn3δ−1\beta=10Kn^3\delta^{-1}β=10Kn3δ−1.
  • SAO. It is defined exactly as Algorithm 1. Arms are tested in order within a round, and the active set changes during the loop. Test (13) is false when Ti(t)=0T_i(t)=0Ti​(t)=0, and test (14) is false when τi=1\tau_i=1τi​=1.
  • Exp3.P. After the switch, Exp3.P runs from scratch for n−τ0n-\tau_0n−τ0​ rounds. Its parameters are those of Bubeck–Cesa-Bianchi Theorem 3.2 with confidence K/βK/\betaK/β, and γ\gammaγ and βP\beta_{\mathrm P}βP​ are clipped at 111.
  • §4 notation. τ0\tau_0τ0​, τi←min⁡(τi,τ0)\tau_i\leftarrow\min(\tau_i,\tau_0)τi​←min(τi​,τ0​) and qi=pi,min⁡(τi,τ0)q_i=p_{i,\min(\tau_i,\tau_0)}qi​=pi,min(τi​,τ0​)​ are computed from the run, never assumed.

A trivializing formalization is ruled out. The goal's hypotheses concern only the instance (KKK, nnn, δ\deltaδ, the distributions or the adversary). The algorithm's quantities (τ0\tau_0τ0​, τi\tau_iτi​, qiq_iqi​, the sampling probabilities) are computed by the definition of SAO and are never free variables or hypotheses. The adversarial half covers adaptive adversaries, not only oblivious reward tables.

The expectation form of Theorem 4.1 (O(⋅)O(\cdot)O(⋅) bounds with β=n4\beta=n^4β=n4), Theorem 1.1 and the two-armed warm-up of §3 are out of scope. Proofs of any milestone are welcome, as are alternative proofs of the concentration lemmas from Mathlib's martingale library.

Selected references

  • S. Bubeck and A. Slivkins, The best of both worlds: stochastic and adversarial bandits, COLT 2012; arXiv:1202.4473v1. https://arxiv.org/abs/1202.4473
  • S. Bubeck and N. Cesa-Bianchi, Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems, Foundations and Trends in Machine Learning 5(1), 2012. https://doi.org/10.1561/2200000024
  • D. A. Freedman, On tail probabilities for martingales, Annals of Probability 3(1), 1975. https://doi.org/10.1214/aop/1176996452
  • 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
  • P. Auer, N. Cesa-Bianchi, Y. Freund and R. E. Schapire, The nonstochastic multiarmed bandit problem, SIAM Journal on Computing 32(1), 2002. https://doi.org/10.1137/S0097539701398375
  • Y. Seldin and A. Slivkins, One practical algorithm for both stochastic and adversarial bandits, ICML 2014. https://proceedings.mlr.press/v32/seldinb14.html
  • J. Zimmert and Y. Seldin, Tsallis-INF: an optimal algorithm for stochastic and adversarial bandits, JMLR 22, 2021. https://jmlr.org/papers/v22/19-753.html
20 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
Machine LearningOperations ResearchProbability·Captain: mikedeng1

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

Motivation

Thompson Sampling (TS) is the oldest heuristic for the stochastic multi-armed bandit problem: it was proposed by Thompson in 1933 (Biometrika 25) and is used in practice for online advertising and recommendation, where it often performs as well as or better than upper-confidence-bound methods (Chapelle and Li, NIPS 2011; Scott 2010). Until 2012 its theoretical guarantees for the frequentist regret were weak: earlier analyses gave only o(T)o(T)o(T) regret in time TTT (Granmo 2010; May, Korda, Lee and Leslie 2011).

Agrawal and Goyal, Analysis of Thompson Sampling for the Multi-armed Bandit Problem (arXiv:1111.1797v3, COLT 2012), gave the first logarithmic finite-time bound on the expected regret of TS. This mission formalizes their two-armed result, Theorem 1. A companion mission covers the NNN-armed bound, Theorem 2.

Timeline. Lai and Robbins (1985) proved that every consistent algorithm has regret at least [∑iΔi/D(μi∥μ∗)+o(1)]ln⁡T\big[\sum_i \Delta_i/D(\mu_i\|\mu^*)+o(1)\big]\ln T[∑i​Δi​/D(μi​∥μ∗)+o(1)]lnT. Auer, Cesa-Bianchi and Fischer (2002) gave UCB1 with an O(∑iln⁡T/Δi)O(\sum_i \ln T/\Delta_i)O(∑i​lnT/Δi​) finite-time bound. Agrawal and Goyal (2012) proved O(ln⁡T/Δ+1/Δ3)O(\ln T/\Delta+1/\Delta^3)O(lnT/Δ+1/Δ3) for two-armed TS. Kaufmann, Korda and Munos (ALT 2012) and Agrawal and Goyal (AISTATS 2013) later proved asymptotically optimal bounds for Bernoulli TS.

Setting

There are two arms. Arm i∈{1,2}i\in\{1,2\}i∈{1,2} has a fixed, unknown reward distribution DiD_iDi​ supported in [0,1][0,1][0,1], with mean μi\mu_iμi​. Plays of an arm give i.i.d. rewards, independent of the other arm. Arm 1 is the unique optimal arm, μ1>μ2\mu_1>\mu_2μ1​>μ2​, and Δ=μ1−μ2\Delta=\mu_1-\mu_2Δ=μ1​−μ2​ is the gap.

Thompson Sampling for general stochastic bandits (Algorithm 2 of the paper) keeps, for each arm iii, a success count SiS_iSi​ and a failure count FiF_iFi​, both starting at 000. In each round t=1,2,…t=1,2,\dotst=1,2,… it

  1. samples, independently for each arm, θ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);
  2. plays i(t)=arg⁡max⁡iθi(t)i(t)=\arg\max_i\theta_i(t)i(t)=argmaxi​θi​(t) and observes a reward r~t∼Di(t)\tilde r_t\sim D_{i(t)}r~t​∼Di(t)​;
  3. performs a Bernoulli trial with success probability r~t\tilde r_tr~t​, with outcome rt∈{0,1}r_t\in\{0,1\}rt​∈{0,1};
  4. increments Si(t)S_{i(t)}Si(t)​ if rt=1r_t=1rt​=1 and Fi(t)F_{i(t)}Fi(t)​ otherwise.

ki(t)k_i(t)ki​(t) is the number of plays of arm iii before round ttt. The expected regret in time TTT is

E[R(T)]=E[∑t=1T(μ1−μi(t))],\mathbb E[\mathcal R(T)]=\mathbb E\Big[\sum_{t=1}^T(\mu_1-\mu_{i(t)})\Big],E[R(T)]=E[t=1∑T​(μ1​−μi(t)​)],

the expectation being over the rewards and the algorithm's randomness.

The analysis uses the Beta cdf Fα,βbetaF^{beta}_{\alpha,\beta}Fα,βbeta​, the binomial cdf Fn,pBF^B_{n,p}Fn,pB​, and the random variable X(j,s,y)X(j,s,y)X(j,s,y): the number of independent Beta(s+1,j−s+1)\mathrm{Beta}(s+1,j-s+1)Beta(s+1,j−s+1) draws made before one exceeds yyy.

Formalization targets

Goal: Theorem 1 (p. 3)

There is an absolute constant C>0C>0C>0 such that for every two-armed instance with rewards in [0,1][0,1][0,1] and μ1>μ2\mu_1>\mu_2μ1​>μ2​, and every T≥2T\ge 2T≥2,

E[R(T)]≤C(ln⁡TΔ+1Δ3).\mathbb E[\mathcal R(T)]\le C\Big(\frac{\ln T}{\Delta}+\frac1{\Delta^3}\Big).E[R(T)]≤C(ΔlnT​+Δ31​).

The constant is not fixed numerically: the paper states the theorem in O(⋅)O(\cdot)O(⋅) form (footnote 1), and the explicit display it reports on p. 8, 40ln⁡T/Δ+48/Δ3+18Δ40\ln T/\Delta+48/\Delta^3+18\Delta40lnT/Δ+48/Δ3+18Δ, is not the formal claim.

Milestones

  • Fact 1 (p. 12): Fα,βbeta(y)=1−Fα+β−1,yB(α−1)F^{beta}_{\alpha,\beta}(y)=1-F^B_{\alpha+\beta-1,y}(\alpha-1)Fα,βbeta​(y)=1−Fα+β−1,yB​(α−1) for positive integers α,β\alpha,\betaα,β.
  • Lemma 1 (p. 6): 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.
  • Lemma 6 (p. 13): Hoeffding-type bounds (10)–(11) on binomial cdfs.
  • Fact 2 (p. 13): every median of Binomial(n,p)\mathrm{Binomial}(n,p)Binomial(n,p) is ⌊np⌋\lfloor np\rfloor⌊np⌋ or ⌈np⌉\lceil np\rceil⌈np⌉.
  • Lemma 2 (p. 7): Pr⁡(E2(t))≥1−2/T2\Pr(E_2(t))\ge 1-2/T^2Pr(E2​(t))≥1−2/T2, where E2(t)={θ2(t)≤μ2+Δ/2 or k2(t)<24ln⁡T/Δ2}E_2(t)=\{\theta_2(t)\le\mu_2+\Delta/2\ \text{or}\ k_2(t)<24\ln T/\Delta^2\}E2​(t)={θ2​(t)≤μ2​+Δ/2 or k2​(t)<24lnT/Δ2}.
  • Lemma 3 (p. 7): a three-case bound on E[E[min⁡{X(j,s(j),y),T}∣s(j)]]\mathbb E\big[\mathbb E[\min\{X(j,s(j),y),T\}\mid s(j)]\big]E[E[min{X(j,s(j),y),T}∣s(j)]] for s(j)∼Binomial(j,μ1)s(j)\sim\mathrm{Binomial}(j,\mu_1)s(j)∼Binomial(j,μ1​).
  • Eq. (1) (p. 7): E[k2(T)]≤C(ln⁡T/Δ2+1/Δ4)\mathbb E[k_2(T)]\le C(\ln T/\Delta^2+1/\Delta^4)E[k2​(T)]≤C(lnT/Δ2+1/Δ4).

Significance

The result. Theorem 1 shows that TS, a randomized Bayesian heuristic with no explicit confidence bonus, has regret logarithmic in TTT on every two-armed instance, matching the order in TTT of the Lai–Robbins lower bound. The proof introduced a way to control the optimal arm's waiting time between plays through the Beta–Binomial duality, and later analyses of TS reuse that device.

Formalizing it. The result is proved on paper and has no machine-checked proof that we know of. The platform's existing TS results concern Gaussian TS (Lattimore and Szepesvári, Ch. 36) and Bayesian regret, which are different algorithms or regret notions. A formalization adds a reusable Lean model of Algorithm 2 on [0,1][0,1][0,1]-valued rewards, Beta–Binomial facts (Fact 1, Lemma 1), a binomial-median theorem, and binomial Hoeffding bounds. It also produces a proof with a constant that has been checked, since the printed constants contain an arithmetic slip.

Difficulty

The standard UCB argument does not transfer to TS. For UCB, the optimal arm's index exceeds its mean with high probability however often the arm has been played, because the exploration bonus is deterministic; the analysis then only has to count plays of the suboptimal arm until its own index concentrates, after Θ(ln⁡T/Δ2)\Theta(\ln T/\Delta^2)Θ(lnT/Δ2) plays. Under TS the optimal arm's sample θ1(t)\theta_1(t)θ1​(t) is random and, if the arm has been played rarely or its early rewards were poor, it falls below μ2\mu_2μ2​ with constant probability. The optimal arm may then wait a long, random time between plays, and the length of that wait depends on the arm's posterior, which in turn depends on how long it has waited. Counting plays of the suboptimal arm with a union bound over rounds, under the assumption that the optimal arm is already concentrated, therefore does not work; controlling these waiting times is the central difficulty and is where the 1/Δ31/\Delta^31/Δ3 dependence enters.

Formalization scope

  • Model. The instance is the platform's StochasticBandit 2 (a probability measure on R\mathbb RR per arm, mean banditArmMean), with the hypothesis that each reward law gives mass 111 to [0,1][0,1][0,1]. Lean arm 0 is the paper's arm 1 and Lean arm 1 the paper's arm 2. Lean rounds are indexed from 000.
  • Algorithm. Algorithm 2 is realized on one probability space with three independent i.i.d. tables: Beta draws W(i,t,a,b)∼Beta(a+1,b+1)W(i,t,a,b)\sim\mathrm{Beta}(a+1,b+1)W(i,t,a,b)∼Beta(a+1,b+1), rewards X(i,t)∼DiX(i,t)\sim D_iX(i,t)∼Di​, and uniforms V(i,t)V(i,t)V(i,t). Round ttt uses θi(t)=W(i,t,Si(t),Fi(t))\theta_i(t)=W(i,t,S_i(t),F_i(t))θi​(t)=W(i,t,Si​(t),Fi​(t)), r~t=X(i(t),t)\tilde r_t=X(i(t),t)r~t​=X(i(t),t) and rt=1{V(i(t),t)<r~t}r_t=\mathbf 1\{V(i(t),t)<\tilde r_t\}rt​=1{V(i(t),t)<r~t​}. Ties in the arg max go to the smaller index (a null event).
  • Values. Regret and expectations are lower Lebesgue integrals in [0,∞][0,\infty][0,∞]. X(j,s,y)X(j,s,y)X(j,s,y) is N∪{∞}\mathbb N\cup\{\infty\}N∪{∞}-valued, so Lemma 1 at y=1y=1y=1 reads ∞=∞\infty=\infty∞=∞, as in the paper.
  • O(·). The paper's O(⋅)O(\cdot)O(⋅) (footnote 1: f≤cgf\le cgf≤cg for n≥n0n\ge n_0n≥n0​) is stated with one universal constant C>0C>0C>0, quantified before the instance, the means and the horizon, for all T≥2T\ge 2T≥2. Eq. (1) is stated the same way, without its printed numerals.
  • Not trivial. The goal is about Algorithm 2 itself, with fresh Beta samples, fresh rewards and the Bernoulli coin. A statement about "any policy satisfying Lemma 2's event bound", or one whose constant depends on Δ\DeltaΔ, the reward laws or TTT, would not be Theorem 1.
  • Edge cases. μ1<1\mu_1<1μ1​<1 is assumed only in Lemma 3, where the paper's RRR and DDD require it. It is not a hypothesis of the goal.
  • Infrastructure. A complete proof needs: inverse-transform or order-statistics facts for Beta laws (Fact 1); geometric expectations; Hoeffding's inequality for sums of Bernoulli variables (Mathlib has Hoeffding/Azuma); the binomial median theorem (Jogdeo–Samuels; Kaas–Buhrman); and the coupling from the reward tables to the per-arm i.i.d. output stacks the paper reasons with. Fact 1, Lemma 6 and Fact 2 are reusable beyond this mission. Contributions to any milestone are welcome, and so is a direct proof of the regret bound with an explicit constant.

Selected references

  • S. Agrawal and N. Goyal, Analysis of Thompson Sampling for the Multi-armed Bandit Problem, COLT 2012; arXiv:1111.1797v3. 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) 285–294. https://doi.org/10.2307/2332286
  • T. L. Lai and H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6 (1985) 4–22. 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) 235–256. https://doi.org/10.1023/A:1013689704352
  • K. Jogdeo and S. M. Samuels, Monotone convergence of binomial probabilities and a generalization of Ramanujan's equation, Annals of Mathematical Statistics 39 (1968) 1191–1195. https://doi.org/10.1214/aoms/1177698243
  • R. Kaas and J. M. Buhrman, Mean, median and mode in binomial distributions, Statistica Neerlandica 34 (1980) 13–18. https://doi.org/10.1111/j.1467-9574.1980.tb00681.x
  • 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
11 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
PreviousPage 3 of 6Next

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