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.
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
- Confidence (Chapter 20): after actions and rewards , how far can the least-squares estimate be from ? The difficulty: the actions were chosen by the algorithm, based on earlier rewards, so they are neither fixed nor independent of the noise.
- 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. with ; ; ; .
Warm-up: fixed design
Simplifying assumptions (dropped later): with invertible (so ); independent 1-subgaussian noise; actions fixed in advance.
TheoremPer-direction confidence bound
For any fixed ,
ProofThe error is a fixed linear combination of the noise
, so . With deterministic coefficients, this is subgaussian with variance proxy . Apply Theorem 5.3.
∎
Intuition is the uncertainty in direction
Directions sampled often (large along ) have small . If for rounds, then , the usual width.
From one direction to all directions: covering
LemmaLemma 20.1 — Covering the sphere
There is a set with such that every has some with .
DerivationUniform bound by covering
Step 1: , but the maximizing is random, so (20.2) doesn’t apply to it directly.
Step 2: apply (20.2) to each with (note ) plus a union bound. With probability all are below .
Step 3: Approximate the maximizer. For any and its nearest , . So . Rearrange and take :
ImportantA uniform bound costs
One direction: width . All directions at once: width . The covering is where the dimension enters. This is essentially optimal, but the argument relies on being deterministic.
Martingales and the method of mixtures
Assumptions now: ; the noise is conditionally 1-subgaussian, , where contains (so the noise may depend on the current action, as with Bernoulli rewards); . Actions are chosen adaptively, and the horizon may be infinite.
WarningPlain Cramér–Chernoff fails
We’d want to bound , but with an adaptive nobody knows how.
FormulaLinearize with a maximum (Fenchel duality)
With for intuition: and
The maximum is attained at . This separates the martingale (appearing linearly) from , at the price of a max. It is a case of (Chapter 26).
LemmaLemma 20.2 — Exponential supermartingale
For every fixed ,
is a nonnegative -supermartingale with (it also holds with ).
ProofOne-step subgaussianity in direction
Conditional subgaussianity with (which is -measurable) gives . So .
∎
WarningThe max is on the wrong side
Cramér–Chernoff gives (20.5). But Lemma 20.2 controls , and . 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 has a sharp maximum at with , then
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 (20.6), and integrals commute with expectations.
LemmaLemma 20.3 — Mixtures preserve the supermartingale property
For any probability measure on , is a nonnegative supermartingale with (Fubini for the conditional expectation; measurability via the sections lemma, Exercise 20.5).
TheoremTheorem 20.4 — Self-normalised bound, uniform in time
For all and ,
ProofGaussian prior, complete the square, maximal inequality
Step 1: Mix with a Gaussian. Let and :
Step 2: Complete the square.
The remaining Gaussian integral is exact:
and .
Step 3: Maximal inequality. is a nonnegative supermartingale starting at 1, so Theorem 3.9 gives (20.7). Take logs and rearrange.
∎
ImportantWhat the mixture buys
- Uniform in time (anytime-valid): one union-free bound for all , from the maximal inequality, not a union bound.
- Valid under adaptive designs: only conditional subgaussianity was used.
- The dimension enters only through : 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 , for all ,
If , then
contains for every with probability . This is exactly the set LinUCB needs (Theorem 19.2).
ProofSeparate noise from regularisation bias
. By the triangle inequality in , the first term gives . For the second, using , its squared norm is . Then apply Theorem 20.4.
∎
FormulaA concrete radius
If : by AM–GM on the eigenvalues, (20.9). With ,
Prefer the 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
Up to constants and a , the adaptive radius matches the fixed-design (20.3). Playing each times with Gaussian noise gives , so (Exercise 20.1). The term is the fluctuation (variance) of this ; the term is its mean (bias). So the in a self-normalised bound can’t be avoided, even with fixed designs.
WarningFixed directions: going through the ellipsoid costs
Bounding via the whole ellipsoid gives width about , while the fixed-design bound (20.2) has no . So under fixed design, avoid the ellipsoid for single-direction questions. Under sequential design the is genuinely needed: an adaptive choice of basis vectors (play each once, then repeatedly sample the coordinates that push away from zero) achieves (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 -cover has ; an -packing has pairwise distances . is the minimum cover size and the maximum packing size; their logs are the metric entropy. The definitions extend to pseudo-metrics.
TheoremCovering facts (Exercises 20.3–20.4)
increases as ; ; and for convex with ,
(volume comparison: disjoint -balls around a packing fit inside ). The covers are finite iff is bounded. Lemma 20.1 follows for the sphere.
TheoremHoeffding–Azuma (Exercises 20.6–20.7)
For adapted a.s., : Hoeffding for martingale differences. With predictable random ranges and : .
TheoremAnytime one-dimensional bound (Exercise 20.8)
If and , then for any ,
This is the method of mixtures in one dimension with a Gaussian prior of precision . SETC’s intervals in the sparse hypercube algorithm use it.
TheoremLaw of the iterated logarithm via mixtures (Exercise 20.9)
Mixing over a density on with as gives , half of the LIL. The largest fluctuations of a random walk are about , which is why anytime confidence widths need at least inside the square root.
TheoremSequential likelihood-ratio confidence sets (Exercise 20.11)
For a parametric family and any predictable estimates , let . Then is a nonnegative martingale with mean 1 (a product of likelihood ratios), so , and 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 with nonzero observations, for all with probability , with (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 spanning ; independent 1-subgaussian noise; unregularised least squares , . By (20.2), for each : (21.1). Goal: the shortest exploration sequence making this width at most for all . Exactly, that’s an intractable integer program; approximately, it’s an efficient convex problem.
DefinitionDesign, G-optimality, D-optimality
A design is a distribution on . Define
- G-optimal (“global”): minimizes , the worst-case predictive variance.
- D-optimal (“determinant”): maximizes , i.e. minimizes the volume of the confidence ellipsoid.
is the core set.
FormulaFrom a design to an exploration schedule
Play each exactly times (21.3). Then , so every has width at most with probability (each separately). The total is
So we want small and a small support.
TheoremTheorem 21.1 — Kiefer–Wolfowitz
Let be compact with . The following are equivalent:
- minimizes (G-optimal);
- maximizes (D-optimal);
- .
Moreover, some minimizer has .
ImportantEvery action can be learned at the price of dimensions
No matter how many actions (even infinitely many), a design with support at most makes every action’s predictive variance at most . That is the same as estimating unrelated means, which is where the regret of phased elimination comes from.
ProofProof (finite ): first-order optimality of a concave objective
Facts: is concave (Exercise 21.2: is concave), with gradient (21.4, Exercise 21.1, from ). And for every ,
so the average predictive variance is always , and hence .
(2)⇒(1),(3): at a maximizer of the concave , for every distribution , . Taking gives for all , so . With always, minimizes and .
(3)⇒(2): if , then , which is first-order optimality for a concave function. (1)⇒(3) now follows.
Small support: at an optimum, every in the support has (Exercise 21.5). If , the dimension of symmetric matrices, some nonzero on has . Then , so moving along keeps , hence , unchanged and stays a distribution until some weight hits zero. That gives a smaller optimal support; induct.
∎
IntuitionGeometry: the minimum-volume enclosing ellipsoid
With from a D-optimal design, the ellipsoid contains all of , with the core set on its boundary. It is the minimum-volume centred ellipsoid containing , 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 .
RemarkVariants
- Non-spanning sets (Exercise 21.6): if , some with support at most has .
- Approximate designs suffice: inflates regret bounds only by a factor .
- 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 (for example uniform). Repeat:
- (the gradient coordinate, i.e. the worst-covered action);
- move toward it: (21.7), with exact line search
From uniform, after iterations. A smarter initialisation removes the dependence and gives with core set after 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
| Object | Statement | Note |
|---|---|---|
| Fixed-design direction bound | no | |
| Covering | ||
| Linearisation | Fenchel duality | |
| Supermartingale | conditional subgaussianity | |
| Mixture | Gaussian prior | |
| Self-normalised bound | for all | anytime, adaptive designs |
| Ellipsoid radius | ||
| Unavoidable | ||
| Hoeffding–Azuma | martingale Hoeffding | |
| LIL | mixtures give the upper half | |
| Average variance | so | |
| Kiefer–Wolfowitz | G-optimal ⇔ D-optimal ⇔ ; support | |
| Samples for width ε | ||
| MVEE | John’s ellipsoid of | |
| Frank–Wolfe | move toward , step (21.8) | 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 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 . The prior pays for not knowing the worst direction. The is real: it is the mean of a , 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 , and a D-optimal design achieves maximum on a core set of at most 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 when the action set is fixed and finite.
References
- Lattimore & Szepesvári, Bandit Algorithms — Chapters 20–21
- Todd, Minimum-Volume Ellipsoids (2016) — design and MVEE algorithms
- de la Peña, Lai & Shao, Self-Normalized Processes (2008)