Learnability, Stability and Uniform Convergence II: Tikhonov-Regularized ERM Learns Convex Lipschitz Stochastic Optimization in Hilbert Space with High ProbabilityResearch Paper
Motivation
Statistical learning theory asks when a rule that sees only an i.i.d. sample from an unknown distribution can return a hypothesis whose expected loss is close to the best possible. In supervised classification the classical answer is uniform convergence: learnability holds exactly when empirical risks converge to expected risks uniformly over the hypothesis class, and then empirical risk minimization (ERM) learns. Shalev-Shwartz, Shamir, Srebro and Sridharan (JMLR 11, 2010) showed that in Vapnik's broader General Learning Setting this picture breaks down. Their motivating example is stochastic convex optimization in a Hilbert space: minimizing an expected convex, Lipschitz objective over a bounded convex set from samples. This problem underlies regularized linear prediction, kernel methods and online-to-batch conversions, and the paper shows (§4.1) that in infinite dimension uniform convergence can fail and the plain empirical minimizer can fail to converge, while the problem is still learnable.
This mission formalizes the positive half of that example: Tikhonov-regularized ERM learns every such problem, with an explicit bound holding with probability (Theorem 3, p. 2644), through the stability of strongly convex empirical minimization (Theorem 2).
Setting
Let be a measurable space of instances and a real Hilbert space. A stochastic convex optimization problem consists of a nonempty, closed, convex, bounded set and an objective such that for every the map is convex and -Lipschitz on , each is measurable, and on . For a distribution on define the risk and optimal risk
and for a sample the empirical risk . A function is -strongly convex on if is convex there. The regularized empirical minimizer is
For the general part, a learning rule maps samples to hypotheses; it is an AERM with rate if , consistent with rate if , and uniform-RO stable with rate if replacing any one sample point changes the loss at any test point by at most on average over the replaced index (Definition 4).
Formalization targets
Goal: Theorem 3
If on , , , and , then with probability at least over
The constants are the paper's.
Milestones, in the order the proof uses them
- Quadratic growth at a minimizer of a -strongly convex : (§4.2, p. 2644).
- Eq. (6): if is -strongly convex and -Lipschitz, empirical minimizers of and of satisfy for all (p. 2645).
- Theorem 8: a uniform- or average-RO stable AERM is consistent with rate and generalizes with rate (p. 2649).
- for the strongly convex empirical minimizer (p. 2645).
- Theorem 2: with probability , (p. 2644).
- Theorem 2 applied to : with probability , (p. 2645).
Significance
The result. Theorem 3 shows that every convex, Lipschitz, bounded stochastic optimization problem over a bounded subset of a Hilbert space is learnable at rate , with no dimension dependence and no uniform convergence. Together with the counterexamples of §4.1 it separates learnability from uniform convergence and from ERM, and it motivates the paper's general characterization: a problem is learnable if and only if it admits a uniform-RO stable asymptotic empirical risk minimizer (Theorem 7). Theorem 8 is the sufficiency half of that characterization and is reused wherever stability arguments give generalization bounds.
Formalizing it. The results are proved in the paper; to our knowledge none has a machine-checked proof. The closest platform material is the textbook treatment in Understanding Machine Learning, chapter 13 (Shalev-Shwartz and Ben-David): Corollary 13.9 (UnderstandingML.convex_lipschitz_bounded_learnable), Corollary 13.6 (rlm_lipschitz_stable) and Lemma 13.5 (strongly_convex_lemma). Those are stated in , bound the risk in expectation, use the regularizer over all of , and have different constants; the present mission works in an arbitrary Hilbert space, over a constraint set , with high-probability bounds and the paper's constants. Its definitions of learning rules, AERM, consistency and replace-one stability in the General Learning Setting are reusable by the other missions of this series.
Difficulty
The obvious route, bounding , is unavailable: §4.1 exhibits problems of exactly this type in which that supremum stays bounded away from zero for every sample size. Any successful argument therefore has to rely on a property of the learning rule rather than of the class , and the plain empirical minimizer does not have it: §4.1 shows it can stay a constant away from at every sample size. A second difficulty is purely formal: the regularization parameter depends on and , so the regularized minimizer changes with them, and all expectations involve a data-dependent hypothesis in a possibly non-separable Hilbert space, where measurability is not automatic.
Formalization scope
Lean conventions, fixed for every item:
- is a real inner product space with
CompleteSpace E, never assumed finite-dimensional; isHset : Set E, and all infima, suprema, strong convexity and Lipschitz conditions are taken onHsetonly. is⨅ h : Hset, F h. - Samples are
Fin m → Z, isMeasure.pi, isFunction.update S i z', and throughout. - The paper's standing loss bound (p. 2637) is named , because Theorem 3 uses for the norm bound . and are implicit in Theorem 3's choice of and are stated.
- Strong convexity is Mathlib's
StrongConvexOn Hset λ, which is the paper's definition. - Minimizers are selections satisfying the minimization property; the theorems hold for every such selection, hence for the minimizer, which is unique by strong convexity.
- Measurability, not discussed in the paper, is the series' single standing convention: each is measurable and the selection makes jointly measurable; for Theorem 8, the rule is measurable in the same sense and is measurable.
- "With probability at least " is the bound with .
No statement of the paper is corrected: all printed constants were checked against the proofs and are reproduced exactly.
A formalization in which the expected excess risk is a Bochner integral of a non-measurable or non-integrable function, or in which is an infimum over all of or over an unbounded family, would make the bounds trivially true; the measurability hypotheses, the bound and the infimum over the nonempty set rule this out.
Contributions welcome: proofs of the milestones in order, and in particular a reusable replace-one identity under Measure.pi, and Markov's inequality in the form used for high-probability bounds.
Selected references
- S. Shalev-Shwartz, O. Shamir, N. Srebro, K. Sridharan, Learnability, Stability and Uniform Convergence, Journal of Machine Learning Research 11 (2010) 2635–2670. https://jmlr.org/papers/v11/shalev-shwartz10a.html
- S. Shalev-Shwartz, O. Shamir, N. Srebro, K. Sridharan, Stochastic Convex Optimization, COLT 2009. https://www.cs.mcgill.ca/~colt2009/papers/018.pdf
- O. Bousquet, A. Elisseeff, Stability and Generalization, Journal of Machine Learning Research 2 (2002) 499–526. https://jmlr.org/papers/v2/bousquet02a.html
- S. Shalev-Shwartz, S. Ben-David, Understanding Machine Learning: From Theory to Algorithms, Cambridge University Press, 2014, chapter 13. https://doi.org/10.1017/CBO9781107298019
- V. N. Vapnik, Statistical Learning Theory, Wiley, 1998.