Empirical Bernstein Bounds and Sample Variance Penalization: Sample Variance Penalization Has Excess Risk at Most √(32V(f*,μ)ln(3M(n)/δ)/n) + 22ln(3M(n)/δ)/(n−1) (Theorem 15)Research Paper
Motivation
Confidence bounds for the mean of a bounded random variable are the basic tool for certifying learned hypotheses, stopping sampling procedures and running bandit algorithms. Hoeffding's inequality gives an interval of width of order whatever the variance. Bennett's and Bernstein's inequalities do better for low-variance variables, with width of order , but the variance is unknown, so the bound cannot be evaluated from data.
Maurer and Pontil (arXiv:0907.3740v1, COLT 2009) prove empirical Bernstein bounds, in which the variance is replaced by the observable sample variance at the cost of a slightly larger term. They extend these bounds uniformly over infinite function classes whose complexity is controlled by a growth function. The uniform bound then suggests a learning rule, sample variance penalization (SVP): minimize the empirical risk plus a multiple of the empirical standard deviation. The paper's final result bounds the excess risk of SVP in terms of the variance of an optimal hypothesis. An analogous empirical Bernstein bound was given by Audibert, Munos and Szepesvári for multi-armed bandits, and Mnih, Szepesvári and Audibert (2008) used such bounds for stopping rules. Variance-penalized risk minimization was later studied as a convex surrogate by Namkoong and Duchi (2017).
Setting
For a vector of reals, the sample mean and sample variance are
Let be a probability distribution on a set , and a class of hypotheses (loss functions) . For , its risk is and its variance is , with . For a sample , and are and of the vector .
The complexity of enters through the growth function. For let , and let be the least number of sup-norm -balls centred in that cover . Then
For , sample variance penalization selects
With it is empirical risk minimization.
Formalization targets
Goal: Theorem 15 (p. 6)
Fix , , and any . With probability at least in ,
The constants are the paper's. The goal is stated for an arbitrary comparator , so it covers an optimal one when one exists.
Milestones
In the order of the proof:
- the fourth-moment inequality (Lemma 8, Corollary 9);
- the self-bounding concentration inequality the paper cites (Theorem 7, lower and upper tails);
- the tails (5)–(6) of the sample variance and the standard-deviation bounds of Theorem 10;
- the comparison of (7)–(9);
- Bennett's inequality for independent variables, together with the i.i.d. Theorem 3;
- the empirical Bernstein bounds (Theorem 4 for i.i.d. variables, Theorem 11 for independent ones);
- the Lipschitz properties of the confidence functionals (Lemma 12);
- the symmetrization lemma (Lemma 13) and the swap bound (Lemma 14);
- the uniform bound, Theorem 6:
Significance
Theorem 15 separates SVP from empirical risk minimization. If an optimal hypothesis has zero variance, the excess risk of SVP decays as ; for a finite class does not depend on and the rate is . The paper gives a two-hypothesis example where empirical risk minimization cannot beat order (Section 4). No convexity of the class or special structure of the loss is needed beyond the growth-function bound. Theorems 4, 10 and 11 are reused well beyond this paper, in bandit algorithms, racing and stopping procedures, and certified estimation.
All results of the paper are proved there, apart from Theorem 7, which is cited from Maurer (2006), and Bennett's inequality for non-identical variables, which is cited from McDiarmid's survey Concentration (1998). This mission produces machine-checked statements of the chain, from the self-bounding inequality to the excess-risk bound. To our knowledge, none of these results has a machine-checked proof. On this platform, the namkoong-2017 series has an open re-derivation of Theorem 6 with other constants and a -normalized variance (VarianceRegularization.Covering.uniform_empirical_bernstein). It is related work, not the printed theorem.
Difficulty
Bennett's inequality cannot be applied directly, because its variance term is not observable. Substituting the sample variance requires a concentration inequality for itself with the right dependence on , and is a U-statistic, not a sum of independent terms: bounded-differences inequalities give only a variance-free rate, which is too weak.
The uniform extension does not follow from a union bound once the class is infinite. The classical double-sample argument compares a sample with a ghost sample through a linear functional; the empirical standard deviation is nonlinear, so that comparison does not go through unchanged. After symmetrization the coordinates are independent but not identically distributed, so i.i.d. concentration results do not apply to them.
Formalization scope
Conventions:
- The source is arXiv:0907.3740v1. It has no printed page numbers, so every page cited is the PDF page.
- Indices are
i : Fin n. - A sample is a vector
Fin n → 𝒳under the product measureMeasure.pi, with i.i.d. coordinatesMeasure.pi (fun _ => μ)or independent onesMeasure.pi μs. Variables "with values in " have laws giving zero mass outside . - "With probability at least , " is written as . For a set that is not measurable this is the outer measure.
- is the published
VarianceRegularization.Expansion.empMean. is a localsampleVarwith the normalization. is Mathlib'svariance. - The covering number is Mathlib's
Metric.coveringNumber(closed balls, centres in the set, values in ). - and are uniform averages over the sign vectors.
- is an arbitrary selector that returns, for every sample, a minimizer in of the penalized objective. Every minimizer is covered.
Added hypotheses and repaired statements, each disclosed in the item:
- The paper says "questions of measurability will be ignored throughout". This is made precise as follows:
- Theorems 6 and 15: is countable and its members are measurable.
- Lemma 13: the function is measurable.
- Theorem 7: is measurable, integrable and nonnegative, and its coordinate-wise infima are measurable.
- Theorems 6 and 15 assume . With an infinite value the printed bound is vacuous.
- Theorems 4 and 11 and Lemmas 12 and 14 add . Theorem 3 and Bennett's display add . Without these, Lean's convention makes the printed formulas false at the excluded .
- Corollary 9's "set" is read as a list, so repetitions are allowed.
Ruled-out trivializations:
- SVP is not defined as the comparator .
- is pinned to .
- is the sample variance, not the true variance.
- No measurability of the bad event or of the selector is assumed.
A complete development needs the following:
- the entropy method for self-bounding functions (Theorem 7);
- Bennett's inequality for independent non-identical bounded variables;
- product-measure symmetrization;
- covering-number bookkeeping.
The concentration results of Section 2 and the symmetrization lemma are reusable beyond this mission. Proofs of any milestone are welcome. Theorem 7 and Bennett's inequality are natural standalone contributions.
Selected references
- A. Maurer, M. Pontil, Empirical Bernstein Bounds and Sample Variance Penalization, COLT 2009; arXiv:0907.3740v1, 2009. https://arxiv.org/abs/0907.3740
- A. Maurer, Concentration inequalities for functions of independent variables, Random Structures & Algorithms 29, 121–138, 2006. https://doi.org/10.1002/rsa.20105
- C. McDiarmid, Concentration, in Probabilistic Methods for Algorithmic Discrete Mathematics, Springer, 1998.
- J.-Y. Audibert, R. Munos, Cs. Szepesvári, Exploration–exploitation trade-off using variance estimates in multi-armed bandits, Theoretical Computer Science 410(19), 2009. https://doi.org/10.1016/j.tcs.2009.01.016
- V. Mnih, Cs. Szepesvári, J.-Y. Audibert, Empirical Bernstein stopping, ICML 2008. https://doi.org/10.1145/1390156.1390241
- W. Hoeffding, Probability inequalities for sums of bounded random variables, JASA 58, 1963. https://doi.org/10.1080/01621459.1963.10500830
- H. Namkoong, J. C. Duchi, Variance-based regularization with convex objectives, NeurIPS 2017; arXiv:1610.02581. https://arxiv.org/abs/1610.02581