Confidence Bounds and Optimal Design for Least Squares Estimators

The statistical machinery behind linear bandits. Fixed-design least squares — per-direction subgaussian widths √(2‖x‖²_{V⁻¹} log(1/δ)), covering numbers, and the √d price of a uniform bound. Why adaptive (sequential) designs break this; the linearisation ½‖S‖²_{V⁻¹} = maxₓ ⟨x,S⟩ − ½‖x‖²_V; the exponential supermartingale; Laplace's method; the method of mixtures with a Gaussian prior giving the self-normalised bound ‖Sₜ‖²_{Vₜ(λ)⁻¹} ≤ 2 log(1/δ) + log(det Vₜ(λ)/λᵈ) uniformly in time, and the resulting confidence ellipsoids. Variance versus bias terms, why √d is unavoidable, and why fixed-direction widths need √d under sequential design. Coverings and packings, Hoeffding–Azuma, anytime bounds and the law of the iterated logarithm via mixtures, sequential likelihood-ratio confidence sets. G- and D-optimal designs, the Kiefer–Wolfowitz theorem (min g = d, core set of at most d(d+1)/2 points), the minimum-volume enclosing ellipsoid, and Frank–Wolfe/Fedorov–Wynn computation.

· 17 min read · 3,298 words

Corresponds to Lattimore & Szepesvári, Chapters 20–21. Used throughout stochastic-linear-bandits.

Motivation: two questions about least squares

IntuitionWhat a linear bandit algorithm needs from statistics

  1. Confidence (Chapter 20): after actions A1,…,AtA_1, \ldots, A_t and rewards Xs=⟨θ∗,As⟩+ηsX_s = \langle\theta^*,A_s\rangle + \eta_s, how far can the least-squares estimate be from θ∗\theta^*? The difficulty: the actions were chosen by the algorithm, based on earlier rewards, so they are neither fixed nor independent of the noise.
  2. Design (Chapter 21): if we may choose the actions freely, which ones make the estimate accurate in every direction we care about, with as few samples as possible?

Notation. θ^t=Vt(λ)−1∑s≤tXsAs\hat\theta_t = V_t(\lambda)^{-1}\sum_{s\le t}X_sA_s with Vt(λ)=λI+∑s≤tAsAs⊤V_t(\lambda) = \lambda I + \sum_{s\le t}A_sA_s^\top; Vt=Vt(0)V_t = V_t(0); ∥x∥V2=x⊤Vx\lVert x\rVert_V^2 = x^\top Vx; St=∑s≤tηsAsS_t = \sum_{s\le t}\eta_sA_s.


Warm-up: fixed design

Simplifying assumptions (dropped later): λ=0\lambda = 0 with VtV_t invertible (so t≥dt\ge d); independent 1-subgaussian noise; actions fixed in advance.

TheoremPer-direction confidence bound

For any fixed x∈Rdx\in\mathbb R^d,

P(⟨θ^t−θ∗,x⟩≥2∥x∥Vt−12log⁡1δ)≤δ(20.2)\mathbb P\Big(\langle\hat\theta_t - \theta^*,x\rangle\ge\sqrt{2\lVert x\rVert^2_{V_t^{-1}}\log\frac1\delta}\Big)\le\delta \tag{20.2}

ProofThe error is a fixed linear combination of the noise

θ^t−θ∗=Vt−1∑sAs(As⊤θ∗+ηs)−θ∗=Vt−1∑sAsηs\hat\theta_t - \theta^* = V_t^{-1}\sum_sA_s(A_s^\top\theta^* + \eta_s) - \theta^* = V_t^{-1}\sum_sA_s\eta_s, so ⟨θ^t−θ∗,x⟩=∑s⟨x,Vt−1As⟩ηs\langle\hat\theta_t - \theta^*,x\rangle = \sum_s\langle x,V_t^{-1}A_s\rangle\eta_s. With deterministic coefficients, this is subgaussian with variance proxy ∑s⟨x,Vt−1As⟩2=x⊤Vt−1(∑sAsAs⊤)Vt−1x=∥x∥Vt−12\sum_s\langle x,V_t^{-1}A_s\rangle^2 = x^\top V_t^{-1}\big(\sum_sA_sA_s^\top\big)V_t^{-1}x = \lVert x\rVert^2_{V_t^{-1}}. Apply Theorem 5.3.

∎

Intuition∥x∥V−1\lVert x\rVert_{V^{-1}} is the uncertainty in direction xx

Directions sampled often (large VV along xx) have small ∥x∥V−1\lVert x\rVert_{V^{-1}}. If As=eiA_s = e_i for TiT_i rounds, then ∥ei∥V−12=1/Ti\lVert e_i\rVert^2_{V^{-1}} = 1/T_i, the usual 1/count1/\sqrt{\text{count}} width.

From one direction to all directions: covering

LemmaLemma 20.1 — Covering the sphere

There is a set Cε⊂RdC_\varepsilon\subset\mathbb R^d with ∣Cε∣≤(3/ε)d\lvert C_\varepsilon\rvert\le(3/\varepsilon)^d such that every x∈Sd−1x\in S^{d-1} has some y∈Cεy\in C_\varepsilon with ∥x−y∥2≤ε\lVert x - y\rVert_2\le\varepsilon.

DerivationUniform bound by covering

Step 1: ∥θ^−θ∗∥Vt=max⁡x∈Sd−1⟨Vt1/2x,θ^−θ∗⟩\lVert\hat\theta - \theta^*\rVert_{V_t} = \max_{x\in S^{d-1}}\langle V_t^{1/2}x,\hat\theta - \theta^*\rangle, but the maximizing xx is random, so (20.2) doesn’t apply to it directly.

Step 2: apply (20.2) to each Vt1/2yV_t^{1/2}y with y∈Cεy\in C_\varepsilon (note ∥Vt1/2y∥Vt−1=∥y∥=1\lVert V_t^{1/2}y\rVert_{V_t^{-1}} = \lVert y\rVert = 1) plus a union bound. With probability 1−δ1-\delta all are below 2log⁡(∣Cε∣/δ)\sqrt{2\log(\lvert C_\varepsilon\rvert/\delta)}.

Step 3: Approximate the maximizer. For any xx and its nearest yy, ⟨V1/2(x−y),θ^−θ∗⟩≤ε∥θ^−θ∗∥V\langle V^{1/2}(x - y),\hat\theta - \theta^*\rangle\le\varepsilon\lVert\hat\theta - \theta^*\rVert_V. So ∥θ^−θ∗∥V<ε∥θ^−θ∗∥V+2log⁡(∣Cε∣/δ)\lVert\hat\theta - \theta^*\rVert_V < \varepsilon\lVert\hat\theta - \theta^*\rVert_V + \sqrt{2\log(\lvert C_\varepsilon\rvert/\delta)}. Rearrange and take ε=12\varepsilon = \frac12:

P(∥θ^t−θ∗∥Vt≥22(dlog⁡6+log⁡1δ))≤δ(20.3)\mathbb P\Big(\lVert\hat\theta_t - \theta^*\rVert_{V_t}\ge2\sqrt{2\big(d\log6 + \log\tfrac1\delta\big)}\Big)\le\delta \tag{20.3}

ImportantA uniform bound costs d\sqrt d

One direction: width 2log⁡(1/δ)\sqrt{2\log(1/\delta)}. All directions at once: width ≈d+log⁡(1/δ)\approx\sqrt{d + \log(1/\delta)}. The (3/ε)d(3/\varepsilon)^d covering is where the dimension enters. This is essentially optimal, but the argument relies on VtV_t being deterministic.


Martingales and the method of mixtures

Assumptions now: Xt=⟨θ∗,At⟩+ηtX_t = \langle\theta^*,A_t\rangle + \eta_t; the noise is conditionally 1-subgaussian, E[eαηt∣Ft−1]≤eα2/2\mathbb E[e^{\alpha\eta_t}\mid\mathcal F_{t-1}]\le e^{\alpha^2/2}, where Ft−1\mathcal F_{t-1} contains A1,X1,…,At−1,Xt−1,AtA_1,X_1,\ldots,A_{t-1},X_{t-1},A_t (so the noise may depend on the current action, as with Bernoulli rewards); λ>0\lambda > 0. Actions are chosen adaptively, and the horizon may be infinite.

WarningPlain Cramér–Chernoff fails

We’d want to bound Eexp⁡(α∥θ^t−θ∗∥Vt(λ)2)\mathbb E\exp\big(\alpha\lVert\hat\theta_t - \theta^*\rVert^2_{V_t(\lambda)}\big), but with an adaptive VtV_t nobody knows how.

FormulaLinearize with a maximum (Fenchel duality)

With λ=0\lambda = 0 for intuition: θ^t−θ∗=Vt−1St\hat\theta_t - \theta^* = V_t^{-1}S_t and

12∥θ^t−θ∗∥Vt2=12∥St∥Vt−12=max⁡x∈Rd(⟨x,St⟩−12∥x∥Vt2)\frac12\lVert\hat\theta_t - \theta^*\rVert^2_{V_t} = \frac12\lVert S_t\rVert^2_{V_t^{-1}} = \max_{x\in\mathbb R^d}\Big(\langle x,S_t\rangle - \frac12\lVert x\rVert^2_{V_t}\Big)

The maximum is attained at x=Vt−1Stx = V_t^{-1}S_t. This separates the martingale StS_t (appearing linearly) from VtV_t, at the price of a max. It is a case of f(x)=sup⁡u⟨u,x⟩−f∗(u)f(x) = \sup_u\langle u,x\rangle - f^*(u) (Chapter 26).

LemmaLemma 20.2 — Exponential supermartingale

For every fixed x∈Rdx\in\mathbb R^d,

Mt(x)=exp⁡(⟨x,St⟩−12∥x∥Vt2)M_t(x) = \exp\Big(\langle x,S_t\rangle - \frac12\lVert x\rVert^2_{V_t}\Big)

is a nonnegative F\mathbb F-supermartingale with M0(x)≤1M_0(x)\le1 (it also holds with Vt(λ)V_t(\lambda)).

ProofOne-step subgaussianity in direction xx

Conditional subgaussianity with α=⟨x,At⟩\alpha = \langle x,A_t\rangle (which is Ft−1\mathcal F_{t-1}-measurable) gives E[eηt⟨x,At⟩∣Ft−1]≤e⟨x,At⟩2/2=e∥x∥AtAt⊤2/2\mathbb E[e^{\eta_t\langle x,A_t\rangle}\mid\mathcal F_{t-1}]\le e^{\langle x,A_t\rangle^2/2} = e^{\lVert x\rVert^2_{A_tA_t^\top}/2}. So E[Mt(x)∣Ft−1]=Mt−1(x) E[eηt⟨x,At⟩−12∥x∥AtAt⊤2∣Ft−1]≤Mt−1(x)\mathbb E[M_t(x)\mid\mathcal F_{t-1}] = M_{t-1}(x)\,\mathbb E\big[e^{\eta_t\langle x,A_t\rangle - \frac12\lVert x\rVert^2_{A_tA_t^\top}}\mid\mathcal F_{t-1}\big]\le M_{t-1}(x).

∎

WarningThe max is on the wrong side

Cramér–Chernoff gives P(12∥θ^−θ∗∥Vt2≥log⁡1δ)≤δ E[max⁡xMt(x)]\mathbb P\big(\frac12\lVert\hat\theta - \theta^*\rVert^2_{V_t}\ge\log\frac1\delta\big)\le\delta\,\mathbb E[\max_xM_t(x)] (20.5). But Lemma 20.2 controls max⁡xE[Mt(x)]\max_x\mathbb E[M_t(x)], and Emax⁡≥max⁡E\mathbb E\max\ge\max\mathbb E. Two fixes: cover the near-maximizers (works, but loosely), or replace the max with an integral.

IntuitionLaplace’s method: an integral of an exponential is about its max

If ff has a sharp maximum at x0x_0 with f′′(x0)=−q<0f''(x_0) = -q < 0, then

∫esf(x)dx∼esf(x0)2πsq(s→∞)\int e^{sf(x)}dx\sim e^{sf(x_0)}\sqrt{\frac{2\pi}{sq}}\qquad(s\to\infty)

by a second-order Taylor expansion and a Gaussian integral. The integrating measure barely matters as long as it charges a neighbourhood of the maximizer. So max⁡xMt(x)≈∫Mt(x) dh(x)\max_xM_t(x)\approx\int M_t(x)\,dh(x) (20.6), and integrals commute with expectations.

LemmaLemma 20.3 — Mixtures preserve the supermartingale property

For any probability measure hh on Rd\mathbb R^d, Mˉt=∫Mt(x) dh(x)\bar M_t = \int M_t(x)\,dh(x) is a nonnegative supermartingale with Mˉ0=1\bar M_0 = 1 (Fubini for the conditional expectation; measurability via the sections lemma, Exercise 20.5).

TheoremTheorem 20.4 — Self-normalised bound, uniform in time

For all λ>0\lambda > 0 and δ∈(0,1)\delta\in(0,1),

P(∃t∈N:∥St∥Vt(λ)−12≥2log⁡1δ+log⁡det⁡Vt(λ)λd)≤δ\mathbb P\Big(\exists t\in\mathbb N : \lVert S_t\rVert^2_{V_t(\lambda)^{-1}}\ge2\log\frac1\delta + \log\frac{\det V_t(\lambda)}{\lambda^d}\Big)\le\delta

ProofGaussian prior, complete the square, maximal inequality

Step 1: Mix with a Gaussian. Let H=λIH = \lambda I and h=N(0,H−1)h = \mathcal N(0,H^{-1}):

Mˉt=1(2π)ddet⁡H−1∫exp⁡(⟨x,St⟩−12∥x∥Vt2−12∥x∥H2)dx\bar M_t = \frac1{\sqrt{(2\pi)^d\det H^{-1}}}\int\exp\Big(\langle x,S_t\rangle - \frac12\lVert x\rVert^2_{V_t} - \frac12\lVert x\rVert^2_H\Big)dx

Step 2: Complete the square.

⟨x,St⟩−12∥x∥Vt+H2=12∥St∥(H+Vt)−12−12∥x−(H+Vt)−1St∥H+Vt2\langle x,S_t\rangle - \frac12\lVert x\rVert^2_{V_t + H} = \frac12\lVert S_t\rVert^2_{(H+V_t)^{-1}} - \frac12\big\lVert x - (H+V_t)^{-1}S_t\big\rVert^2_{H+V_t}

The remaining Gaussian integral is exact:

Mˉt=(det⁡Hdet⁡(H+Vt))1/2exp⁡(12∥St∥(H+Vt)−12)(20.8)\bar M_t = \Big(\frac{\det H}{\det(H + V_t)}\Big)^{1/2}\exp\Big(\frac12\lVert S_t\rVert^2_{(H+V_t)^{-1}}\Big) \tag{20.8}

and H+Vt=Vt(λ)H + V_t = V_t(\lambda).

Step 3: Maximal inequality. Mˉt\bar M_t is a nonnegative supermartingale starting at 1, so Theorem 3.9 gives P(sup⁡tMˉt≥1/δ)≤δ\mathbb P(\sup_t\bar M_t\ge1/\delta)\le\delta (20.7). Take logs and rearrange.

∎

ImportantWhat the mixture buys

  • Uniform in time (anytime-valid): one union-free bound for all tt, from the maximal inequality, not a union bound.
  • Valid under adaptive designs: only conditional subgaussianity was used.
  • The dimension enters only through log⁡det⁡Vt(λ)/λd\log\det V_t(\lambda)/\lambda^d: the Gaussian prior “pays” for not knowing the maximizing direction, exactly what the covering paid for, but data-dependently and with better constants. In practice the improvement over covering is large.
  • Bounds of this type are called self-normalised (de la Peña et al.); the method of mixtures goes back to Robbins–Siegmund (1970).

TheoremTheorem 20.5 — Confidence ellipsoid for ridge regression

With probability at least 1−δ1-\delta, for all t∈Nt\in\mathbb N,

∥θ^t−θ∗∥Vt(λ)<λ∥θ∗∥2+2log⁡1δ+log⁡det⁡Vt(λ)λd\lVert\hat\theta_t - \theta^*\rVert_{V_t(\lambda)} < \sqrt\lambda\lVert\theta^*\rVert_2 + \sqrt{2\log\frac1\delta + \log\frac{\det V_t(\lambda)}{\lambda^d}}

If ∥θ∗∥2≤m2\lVert\theta^*\rVert_2\le m_2, then

Ct={θ:∥θ^t−1−θ∥Vt−1(λ)<m2λ+2log⁡1δ+log⁡det⁡Vt−1(λ)λd}\mathcal C_t = \Big\{\theta : \lVert\hat\theta_{t-1} - \theta\rVert_{V_{t-1}(\lambda)} < m_2\sqrt\lambda + \sqrt{2\log\frac1\delta + \log\frac{\det V_{t-1}(\lambda)}{\lambda^d}}\Big\}

contains θ∗\theta^* for every tt with probability 1−δ1-\delta. This is exactly the set LinUCB needs (Theorem 19.2).

ProofSeparate noise from regularisation bias

θ^t−θ∗=Vt(λ)−1St+(Vt(λ)−1Vt−I)θ∗\hat\theta_t - \theta^* = V_t(\lambda)^{-1}S_t + (V_t(\lambda)^{-1}V_t - I)\theta^*. By the triangle inequality in ∥⋅∥Vt(λ)\lVert\cdot\rVert_{V_t(\lambda)}, the first term gives ∥St∥Vt(λ)−1\lVert S_t\rVert_{V_t(\lambda)^{-1}}. For the second, using I−Vt(λ)−1Vt=λVt(λ)−1I - V_t(\lambda)^{-1}V_t = \lambda V_t(\lambda)^{-1}, its squared norm is λ θ∗⊤(I−Vt(λ)−1Vt)θ∗≤λ∥θ∗∥2\lambda\,\theta^{*\top}(I - V_t(\lambda)^{-1}V_t)\theta^*\le\lambda\lVert\theta^*\rVert^2. Then apply Theorem 20.4.

∎

FormulaA concrete radius

If ∥At∥2≤L\lVert A_t\rVert_2\le L: by AM–GM on the eigenvalues, det⁡Vt(λ)λd≤(1+nL2λd)d\frac{\det V_t(\lambda)}{\lambda^d}\le\big(1 + \frac{nL^2}{\lambda d}\big)^d (20.9). With λ=1\lambda = 1,

Ct={θ:∥θ^t−1−θ∥Vt−1(1)<m2+2log⁡1δ+dlog⁡(1+nL2d)}\mathcal C_t = \Big\{\theta : \lVert\hat\theta_{t-1} - \theta\rVert_{V_{t-1}(1)} < m_2 + \sqrt{2\log\frac1\delta + d\log\Big(1 + \frac{nL^2}d\Big)}\Big\}

Prefer the log⁡det⁡\log\det form inside algorithms (it is data-dependent and smaller); use this form for stating regret bounds.

How tight is this?

NoteVariance plus bias, and the unavoidable d\sqrt d

Up to constants and a log⁡n\sqrt{\log n}, the adaptive radius matches the fixed-design (20.3). Playing each eie_i mm times with Gaussian noise gives ∥θ^−θ∗∥V2∼χd2\lVert\hat\theta - \theta^*\rVert^2_{V}\sim\chi^2_d, so E∥θ^n−θ∗∥Vn2=d\mathbb E\lVert\hat\theta_n - \theta^*\rVert^2_{V_n} = d (Exercise 20.1). The 2log⁡1δ2\log\frac1\delta term is the fluctuation (variance) of this χ2\chi^2; the dlog⁡(⋅)d\log(\cdot) term is its mean (bias). So the d\sqrt d in a self-normalised bound can’t be avoided, even with fixed designs.

WarningFixed directions: going through the ellipsoid costs d\sqrt d

Bounding ⟨θ^−θ∗,x⟩\langle\hat\theta - \theta^*,x\rangle via the whole ellipsoid gives width about d∥x∥V−1\sqrt d\lVert x\rVert_{V^{-1}}, while the fixed-design bound (20.2) has no d\sqrt d. So under fixed design, avoid the ellipsoid for single-direction questions. Under sequential design the d\sqrt d is genuinely needed: an adaptive choice of basis vectors (play each once, then repeatedly sample the coordinates that push ⟨θ^,1⟩\langle\hat\theta,\mathbf 1\rangle away from zero) achieves E[⟨θ^n,1⟩2/∥1∥Vn−12]≥cd\mathbb E\big[\langle\hat\theta_n,\mathbf 1\rangle^2/\lVert\mathbf 1\rVert^2_{V_n^{-1}}\big]\ge cd (Exercise 20.2). Designing per-direction bounds that adapt to how dependent the design actually is remains open.

RemarkSupermartingales are behind every Chernoff argument

Most Cramér–Chernoff proofs for sums can be rewritten as “this exponential is a supermartingale, apply Markov or the maximal inequality”, for example the Exp3-IX concentration lemma (Lemma 12.2). Lai (2009) surveys martingale methods in sequential analysis.

Further results from the exercises

DefinitionCovering and packing numbers (Def. 20.6)

An ε\varepsilon-cover C⊆AC\subseteq A has A⊆⋃x∈CB(x,ε)A\subseteq\bigcup_{x\in C}B(x,\varepsilon); an ε\varepsilon-packing P⊆AP\subseteq A has pairwise distances >ε> \varepsilon. N(A,ε)N(A,\varepsilon) is the minimum cover size and M(A,ε)M(A,\varepsilon) the maximum packing size; their logs are the metric entropy. The definitions extend to pseudo-metrics.

TheoremCovering facts (Exercises 20.3–20.4)

N(ε)N(\varepsilon) increases as ε↓0\varepsilon\downarrow0; M(2ε)≤N(ε)≤M(ε)M(2\varepsilon)\le N(\varepsilon)\le M(\varepsilon); and for convex AA with εB⊆A\varepsilon B\subseteq A,

(1ε)dvol(A)vol(B)≤N(ε)≤M(ε)≤vol(A+ε2B)vol(ε2B)≤(3ε)dvol(A)vol(B)\Big(\frac1\varepsilon\Big)^d\frac{\mathrm{vol}(A)}{\mathrm{vol}(B)}\le N(\varepsilon)\le M(\varepsilon)\le\frac{\mathrm{vol}(A + \frac\varepsilon2B)}{\mathrm{vol}(\frac\varepsilon2B)}\le\Big(\frac3\varepsilon\Big)^d\frac{\mathrm{vol}(A)}{\mathrm{vol}(B)}

(volume comparison: disjoint ε2\frac\varepsilon2-balls around a packing fit inside A+ε2BA + \frac\varepsilon2B). The covers are finite iff AA is bounded. Lemma 20.1 follows for the sphere.

TheoremHoeffding–Azuma (Exercises 20.6–20.7)

For adapted Xt∈[at,bt]X_t\in[a_t,b_t] a.s., P(∑t(Xt−E[Xt∣Ft−1])≥ε)≤exp⁡(−2ε2∑t(bt−at)2)\mathbb P\big(\sum_t(X_t - \mathbb E[X_t\mid\mathcal F_{t-1}])\ge\varepsilon\big)\le\exp\big(-\frac{2\varepsilon^2}{\sum_t(b_t - a_t)^2}\big): Hoeffding for martingale differences. With predictable random ranges [At,Bt][A_t,B_t] and R2=∑(Bt−At)2R^2 = \sum(B_t - A_t)^2: P({R≤r}∩{∑⋯≥ε})≤e−2ε2/r2\mathbb P\big(\{R\le r\}\cap\{\sum\cdots\ge\varepsilon\}\big)\le e^{-2\varepsilon^2/r^2}.

TheoremAnytime one-dimensional bound (Exercise 20.8)

If E[eλXt∣Ft−1]≤eλ2σ2/2\mathbb E[e^{\lambda X_t}\mid\mathcal F_{t-1}]\le e^{\lambda^2\sigma^2/2} and St=∑s≤tXsS_t = \sum_{s\le t}X_s, then for any ρ>0\rho > 0,

P(∃t:∣St∣≥2(tσ2+ρ2)log⁡tσ2+ρ2ρ2δ)≤δ\mathbb P\Big(\exists t : \lvert S_t\rvert\ge\sqrt{2(t\sigma^2 + \rho^2)\log\frac{\sqrt{t\sigma^2 + \rho^2}}{\rho^2\delta}}\Big)\le\delta

This is the method of mixtures in one dimension with a Gaussian prior of precision ρ2\rho^2. SETC’s intervals in the sparse hypercube algorithm use it.

TheoremLaw of the iterated logarithm via mixtures (Exercise 20.9)

Mixing exp⁡(λSn−λ2n/2)\exp(\lambda S_n - \lambda^2n/2) over a density ff on [0,∞)[0,\infty) with log⁡1λf(λ)=(1+o(1))log⁡log⁡1λ\log\frac1{\lambda f(\lambda)} = (1 + o(1))\log\log\frac1\lambda as λ→0\lambda\to0 gives P(lim sup⁡nSn2nlog⁡log⁡n≤1)=1\mathbb P\big(\limsup_n\frac{S_n}{\sqrt{2n\log\log n}}\le1\big) = 1, half of the LIL. The largest fluctuations of a random walk are about 2nlog⁡log⁡n\sqrt{2n\log\log n}, which is why anytime confidence widths need at least log⁡log⁡t\log\log t inside the square root.

TheoremSequential likelihood-ratio confidence sets (Exercise 20.11)

For a parametric family pθp_\theta and any predictable estimates θ^s−1\hat\theta_{s-1}, let Lt(θ)=∑s≤tlog⁡pθ^s−1(Xs)pθ(Xs)L_t(\theta) = \sum_{s\le t}\log\frac{p_{\hat\theta_{s-1}}(X_s)}{p_\theta(X_s)}. Then eLt(θ∗)e^{L_t(\theta^*)} is a nonnegative martingale with mean 1 (a product of likelihood ratios), so P(sup⁡tLt(θ∗)≥log⁡1δ)≤δ\mathbb P(\sup_tL_t(\theta^*)\ge\log\frac1\delta)\le\delta, and Θt={θ:Lt(θ)<log⁡1δ}\Theta_t = \{\theta : L_t(\theta) < \log\frac1\delta\} is an anytime confidence sequence (Lai–Robbins 1985; Robbins–Siegmund 1972). Compare the wealth-ratio supermartingale: the same idea with a predictable “bettor”.

NoteTernary martingales (Exercise 20.10)

For Xt∈{−1,0,1}X_t\in\{-1,0,1\} with Nt=∑s∣Xs∣N_t = \sum_s\lvert X_s\rvert nonzero observations, ∣St∣≤2Ntlog⁡(cNt/δ)\lvert S_t\rvert\le\sqrt{2N_t\log(c\sqrt{N_t}/\delta)} for all tt with probability 1−δ1-\delta, with c≈3.43c\approx3.43 (Lattimore et al. 2018). The width scales with the number of informative rounds only.


Optimal design: the Kiefer–Wolfowitz theorem

Setting (Chapter 21). Fixed actions a1,…,ana_1, \ldots, a_n spanning Rd\mathbb R^d; independent 1-subgaussian noise; unregularised least squares θ^=V−1∑tatXt\hat\theta = V^{-1}\sum_ta_tX_t, V=∑tatat⊤V = \sum_ta_ta_t^\top. By (20.2), for each a∈Aa\in\mathcal A: P(⟨θ^−θ∗,a⟩≥2∥a∥V−12log⁡1δ)≤δ\mathbb P\big(\langle\hat\theta - \theta^*,a\rangle\ge\sqrt{2\lVert a\rVert^2_{V^{-1}}\log\frac1\delta}\big)\le\delta (21.1). Goal: the shortest exploration sequence making this width at most ε\varepsilon for all a∈Aa\in\mathcal A. Exactly, that’s an intractable integer program; approximately, it’s an efficient convex problem.

DefinitionDesign, G-optimality, D-optimality

A design is a distribution π\pi on A\mathcal A. Define

V(π)=∑a∈Aπ(a)aa⊤,g(π)=max⁡a∈A∥a∥V(π)−12,f(π)=log⁡det⁡V(π)(21.2)V(\pi) = \sum_{a\in\mathcal A}\pi(a)aa^\top, \qquad g(\pi) = \max_{a\in\mathcal A}\lVert a\rVert^2_{V(\pi)^{-1}}, \qquad f(\pi) = \log\det V(\pi) \tag{21.2}

  • G-optimal (“global”): minimizes gg, the worst-case predictive variance.
  • D-optimal (“determinant”): maximizes ff, i.e. minimizes the volume of the confidence ellipsoid.

Supp(π)\mathrm{Supp}(\pi) is the core set.

FormulaFrom a design to an exploration schedule

Play each a∈Supp(π)a\in\mathrm{Supp}(\pi) exactly na=⌈2π(a)g(π)ε2log⁡1δ⌉n_a = \big\lceil\frac{2\pi(a)g(\pi)}{\varepsilon^2}\log\frac1\delta\big\rceil times (21.3). Then V⪰2g(π)ε2log⁡1δ V(π)V\succeq\frac{2g(\pi)}{\varepsilon^2}\log\frac1\delta\,V(\pi), so every a∈Aa\in\mathcal A has width at most ε\varepsilon with probability 1−δ1-\delta (each aa separately). The total is

n≤∣Supp(π)∣+2g(π)ε2log⁡1δn\le\lvert\mathrm{Supp}(\pi)\rvert + \frac{2g(\pi)}{\varepsilon^2}\log\frac1\delta

So we want small gg and a small support.

TheoremTheorem 21.1 — Kiefer–Wolfowitz

Let A⊂Rd\mathcal A\subset\mathbb R^d be compact with span(A)=Rd\mathrm{span}(\mathcal A) = \mathbb R^d. The following are equivalent:

  1. π∗\pi^* minimizes gg (G-optimal);
  2. π∗\pi^* maximizes f=log⁡det⁡V(π)f = \log\det V(\pi) (D-optimal);
  3. g(π∗)=dg(\pi^*) = d.

Moreover, some minimizer has ∣Supp(π∗)∣≤d(d+1)/2\lvert\mathrm{Supp}(\pi^*)\rvert\le d(d+1)/2.

ImportantEvery action can be learned at the price of dd dimensions

No matter how many actions (even infinitely many), a design with support at most d(d+1)/2d(d+1)/2 makes every action’s predictive variance at most d/(samples)d/(\text{samples}). That is the same as estimating dd unrelated means, which is where the dn\sqrt{dn} regret of phased elimination comes from.

ProofProof (finite A\mathcal A): first-order optimality of a concave objective

Facts: ff is concave (Exercise 21.2: t↦log⁡det⁡(H+tZ)t\mapsto\log\det(H + tZ) is concave), with gradient (∇f(π))a=∥a∥V(π)−12(\nabla f(\pi))_a = \lVert a\rVert^2_{V(\pi)^{-1}} (21.4, Exercise 21.1, from ddtlog⁡det⁡A(t)=trace⁡(A−1A˙)\frac{d}{dt}\log\det A(t) = \operatorname{trace}(A^{-1}\dot A)). And for every π\pi,

∑aπ(a)∥a∥V(π)−12=trace⁡(∑aπ(a)aa⊤V(π)−1)=trace⁡I=d(21.5)\sum_a\pi(a)\lVert a\rVert^2_{V(\pi)^{-1}} = \operatorname{trace}\Big(\sum_a\pi(a)aa^\top V(\pi)^{-1}\Big) = \operatorname{trace}I = d \tag{21.5}

so the average predictive variance is always dd, and hence g(π)≥dg(\pi)\ge d.

(2)⇒(1),(3): at a maximizer π∗\pi^* of the concave ff, for every distribution π\pi, 0≥⟨∇f(π∗),π−π∗⟩=∑aπ(a)∥a∥V(π∗)−12−d0\ge\langle\nabla f(\pi^*),\pi - \pi^*\rangle = \sum_a\pi(a)\lVert a\rVert^2_{V(\pi^*)^{-1}} - d. Taking π=δa\pi = \delta_a gives ∥a∥V(π∗)−12≤d\lVert a\rVert^2_{V(\pi^*)^{-1}}\le d for all aa, so g(π∗)≤dg(\pi^*)\le d. With g≥dg\ge d always, π∗\pi^* minimizes gg and min⁡g=d\min g = d.

(3)⇒(2): if g(π∗)=dg(\pi^*) = d, then ⟨∇f(π∗),π−π∗⟩=∑aπ(a)∥a∥V(π∗)−12−d≤g(π∗)−d=0\langle\nabla f(\pi^*),\pi - \pi^*\rangle = \sum_a\pi(a)\lVert a\rVert^2_{V(\pi^*)^{-1}} - d\le g(\pi^*) - d = 0, which is first-order optimality for a concave function. (1)⇒(3) now follows.

Small support: at an optimum, every aa in the support has ∥a∥V(π∗)−12=d\lVert a\rVert^2_{V(\pi^*)^{-1}} = d (Exercise 21.5). If ∣S∣>d(d+1)/2\lvert S\rvert > d(d+1)/2, the dimension of symmetric matrices, some nonzero vv on SS has ∑a∈Sv(a)aa⊤=0\sum_{a\in S}v(a)aa^\top = 0. Then d∑av(a)=∑av(a)∥a∥V−12=0d\sum_av(a) = \sum_av(a)\lVert a\rVert^2_{V^{-1}} = 0, so moving along π∗+tv\pi^* + tv keeps VV, hence ff, unchanged and stays a distribution until some weight hits zero. That gives a smaller optimal support; induct.

∎

IntuitionGeometry: the minimum-volume enclosing ellipsoid

With VV from a D-optimal design, the ellipsoid E={x:∥x∥V−12≤d}\mathcal E = \{x : \lVert x\rVert^2_{V^{-1}}\le d\} contains all of A\mathcal A, with the core set on its boundary. It is the minimum-volume centred ellipsoid containing A\mathcal A, unique, and its computation is the convex dual of the log-det problem (Silvey–Sibson). Equivalently, it is the MVEE (John’s ellipsoid) of the symmetrised set A∪(−A)\mathcal A\cup(-\mathcal A).

RemarkVariants

  • Non-spanning sets (Exercise 21.6): if dim⁡span(A)=m\dim\mathrm{span}(\mathcal A) = m, some π∗\pi^* with support at most m(m+1)/2m(m+1)/2 has g(π∗)=mg(\pi^*) = m.
  • Approximate designs suffice: g(π)≤(1+ε)g(π∗)g(\pi)\le(1+\varepsilon)g(\pi^*) inflates regret bounds only by a factor (1+ε)1/2(1+\varepsilon)^{1/2}.
  • Infinite action sets: approximate designs can sometimes be found efficiently, but the algorithms are much heavier (Hazan et al. 2016; Grötschel et al.).

AlgorithmFrank–Wolfe / Fedorov–Wynn for D-optimal design

Start from π0\pi_0 (for example uniform). Repeat:

  1. ak=arg⁡max⁡a∈A∥a∥V(πk)−12a_k = \arg\max_{a\in\mathcal A}\lVert a\rVert^2_{V(\pi_k)^{-1}} (the gradient coordinate, i.e. the worst-covered action);
  2. move toward it: πk+1=(1−γk)πk+γkδak\pi_{k+1} = (1 - \gamma_k)\pi_k + \gamma_k\delta_{a_k} (21.7), with exact line search γk=1d∥ak∥V(πk)−12−1∥ak∥V(πk)−12−1(21.8)\gamma_k = \frac{\frac1d\lVert a_k\rVert^2_{V(\pi_k)^{-1}} - 1}{\lVert a_k\rVert^2_{V(\pi_k)^{-1}} - 1} \tag{21.8}

From uniform, g(πk)≤(1+ε)dg(\pi_k)\le(1+\varepsilon)d after O(dlog⁡log⁡∣A∣+d/ε)O(d\log\log\lvert\mathcal A\rvert + d/\varepsilon) iterations. A smarter initialisation removes the ∣A∣\lvert\mathcal A\rvert dependence and gives g≤2dg\le2d with core set O(dlog⁡log⁡d)O(d\log\log d) after O(dlog⁡log⁡d)O(d\log\log d) iterations. Rank-one updates make each step cheap (Exercise 21.7). History: Fedorov (1972) and Wynn (1970), a specialisation of Frank–Wolfe (1956). See Todd (2016) for variants and convergence rates.


Quick reference

ObjectStatementNote
Fixed-design direction bound⟨θ^−θ∗,x⟩<2∥x∥V−12log⁡1δ\langle\hat\theta - \theta^*,x\rangle < \sqrt{2\lVert x\rVert^2_{V^{-1}}\log\frac1\delta}no dd
Covering∣Cε∣≤(3/ε)d\lvert C_\varepsilon\rvert\le(3/\varepsilon)^d∥θ^−θ∗∥V≤22(dlog⁡6+log⁡1δ)\lVert\hat\theta - \theta^*\rVert_V\le2\sqrt{2(d\log6 + \log\frac1\delta)}
Linearisation12∥S∥V−12=max⁡x⟨x,S⟩−12∥x∥V2\frac12\lVert S\rVert^2_{V^{-1}} = \max_x\langle x,S\rangle - \frac12\lVert x\rVert^2_VFenchel duality
SupermartingaleMt(x)=e⟨x,St⟩−12∥x∥Vt2M_t(x) = e^{\langle x,S_t\rangle - \frac12\lVert x\rVert^2_{V_t}}conditional subgaussianity
MixtureMˉt=det⁡H/det⁡Vt(λ) e12∥St∥Vt(λ)−12\bar M_t = \sqrt{\det H/\det V_t(\lambda)}\,e^{\frac12\lVert S_t\rVert^2_{V_t(\lambda)^{-1}}}Gaussian prior N(0,H−1)\mathcal N(0,H^{-1})
Self-normalised bound∥St∥Vt(λ)−12<2log⁡1δ+log⁡det⁡Vt(λ)λd\lVert S_t\rVert^2_{V_t(\lambda)^{-1}} < 2\log\frac1\delta + \log\frac{\det V_t(\lambda)}{\lambda^d} for all ttanytime, adaptive designs
Ellipsoid radiusm2λ+2log⁡1δ+log⁡det⁡Vt(λ)λdm_2\sqrt\lambda + \sqrt{2\log\frac1\delta + \log\frac{\det V_t(\lambda)}{\lambda^d}}≤m2+2log⁡1δ+dlog⁡(1+nL2d)\le m_2 + \sqrt{2\log\frac1\delta + d\log(1 + \frac{nL^2}d)}
Unavoidable d\sqrt dE∥θ^−θ∗∥V2=d\mathbb E\lVert\hat\theta - \theta^*\rVert^2_V = dχd2\chi^2_d
Hoeffding–Azumaexp⁡(−2ε2/∑(bt−at)2)\exp(-2\varepsilon^2/\sum(b_t - a_t)^2)martingale Hoeffding
LILlim sup⁡Sn/2nlog⁡log⁡n=1\limsup S_n/\sqrt{2n\log\log n} = 1mixtures give the upper half
Average variance∑aπ(a)∥a∥V(π)−12=d\sum_a\pi(a)\lVert a\rVert^2_{V(\pi)^{-1}} = dso g≥dg\ge d
Kiefer–WolfowitzG-optimal ⇔ D-optimal ⇔ g=dg = d; support ≤d(d+1)/2\le d(d+1)/2
Samples for width ε≤∣Supp∣+2dε2log⁡1δ\le\lvert\mathrm{Supp}\rvert + \frac{2d}{\varepsilon^2}\log\frac1\delta
MVEE{x:∥x∥V−12≤d}⊇A\{x : \lVert x\rVert^2_{V^{-1}}\le d\}\supseteq\mathcal AJohn’s ellipsoid of A∪(−A)\mathcal A\cup(-\mathcal A)
Frank–Wolfemove toward arg⁡max⁡∥a∥V−12\arg\max\lVert a\rVert^2_{V^{-1}}, step (21.8)O(dlog⁡log⁡∣A∣+d/ε)O(d\log\log\lvert\mathcal A\rvert + d/\varepsilon) iterations

Facts to keep to hand

  • Adaptive designs break naive least-squares confidence intervals. The coefficients multiplying the noise depend on the noise, so the fixed-design argument and its covering extension don’t apply.
  • The method of mixtures fixes this cleanly. Linearize the quadratic with a max, note that each e⟨x,St⟩−12∥x∥Vt2e^{\langle x,S_t\rangle - \frac12\lVert x\rVert^2_{V_t}} is a supermartingale, replace the max with a Gaussian-prior average (Laplace’s idea), compute it in closed form, and apply the maximal inequality. The result is anytime-valid and self-normalised.
  • Dimension enters as log⁡det⁡\log\det. The prior pays for not knowing the worst direction. The d\sqrt d is real: it is the mean of a χd2\chi^2_d, and per-direction bounds need it under sequential design too.
  • Good designs make every action cheap to estimate. The average predictive variance under any design is exactly dd, and a D-optimal design achieves maximum =d= d on a core set of at most d(d+1)/2d(d+1)/2 points: the minimum-volume enclosing ellipsoid.
  • Confidence and design are two halves of linear bandits. Optimistic algorithms use the ellipsoid; elimination algorithms use designs and fixed-design widths, buying back a d\sqrt d when the action set is fixed and finite.

References

← All notes