Learnability, Stability and Uniform Convergence III: For an ERM, Leave-One-Out Stability, Universal Consistency and Universal Generalization Are EquivalentResearch Paper
Motivation
Algorithmic stability asks how much the output of a learning algorithm changes when its training sample is perturbed. Since Devroye and Wagner (IEEE Trans. Inf. Theory 1979) it has served as a route to generalization bounds that does not go through the complexity of the hypothesis class. Bousquet and Elisseeff (JMLR 2002) popularised uniform stability, and Mukherjee, Niyogi, Poggio and Rifkin (Adv. Comput. Math. 2006) showed that for empirical risk minimisation in supervised learning, a leave-one-out type of stability is necessary and sufficient for consistency.
Shalev-Shwartz, Shamir, Srebro and Sridharan (JMLR 11, 2010) study stability in Vapnik's General Learning Setting, where uniform convergence can fail even though the problem is learnable. In Appendix A.2 they compare replace-one and leave-one-out (LOO) stability. For an empirical risk minimiser they prove that LOO stability is equivalent to consistency and to generalization, provided each property holds with one rate for all distributions (Theorem 31, p. 2667). This mission formalizes that theorem and the lemmas of Section 5.3 on which its proof rests.
Timeline:
- 1979, Devroye–Wagner: leave-one-out estimates for local rules.
- 2002, Bousquet–Elisseeff: uniform stability implies generalization.
- 2002, Kutin–Niyogi (UAI 2002): a taxonomy of stability notions.
- 2006, Mukherjee et al.: LOO stability characterises consistency of ERM in supervised learning.
- 2010, Shalev-Shwartz et al.: in the General Learning Setting, for ERMs, LOO stability, universal consistency and universal generalization are equivalent (Theorem 31). Universally consistent AERMs need not be LOO stable (Example 6).
Setting
A learning problem consists of an instance space with a -algebra, a nonempty hypothesis class , and an objective with for all . For a probability measure on :
- the risk is and the optimal risk is ;
- for a sample of i.i.d. draws, the empirical risk is , and denotes the minimal empirical risk;
- a learning rule maps each sample of size to a hypothesis . It is an ERM if for every sample. It is an AERM with rate if ;
- is consistent with rate if . It generalizes with rate if , and it on-average generalizes if ;
- writing for with removed, is LOO stable with rate (Definition 29) if
A rate is a sequence that is non-increasing and tends to . A property holds universally if it holds under every with one and the same rate.
Formalization targets
Goal: Theorem 31
For an ERM :
The statement has no rates. Each side asserts that some rate exists and serves every distribution.
Milestones
In the order the proof uses them:
- Utility Lemma 12: for the mean of i.i.d. variables bounded by .
- Utility Lemma 13: a.s. implies .
- Lemma 14: an AERM that on-average generalizes with rate generalizes with rate .
- Lemma 15: under the same hypotheses the rule is consistent with rate .
- Lemma 16 (Main Converse Lemma): in a learnable problem, for .
- Lemma 17: Eq. (12), together with an AERM that is consistent, gives generalization with rate .
- First display of the proof of Theorem 31: a generalizing ERM is LOO stable with rate .
- Second display: a LOO stable ERM on-average generalizes on samples of size with rate .
Significance
Theorem 31 shows that for exact ERMs, LOO stability is not only sufficient but necessary for consistency. It transfers the supervised-learning characterisation of Mukherjee et al. to the General Learning Setting, where uniform convergence is no longer available as an intermediate. The hypothesis is sharp in one direction: Example 6 of the paper gives a universally consistent AERM that is not LOO stable. The equivalence therefore depends on exact minimisation, and an asymptotic minimiser does not suffice.
The lemmas are useful on their own. Lemmas 14 and 15 are the standard bridges between on-average generalization, generalization and consistency. Lemma 16 says that the minimal empirical risk estimates consistently in every learnable problem, even when no ERM learns. Lemma 16 also underlies Theorem 7, the paper's main characterisation of learnability, which is the subject of mission I of this series.
The results are proved in the paper. As far as is known they have not been formalized in any proof assistant. The platform has the textbook side of this framework (Shalev-Shwartz and Ben-David, Understanding Machine Learning, Chapter 13). Those statements are in and use a replace-one stability notion; they do not cover leave-one-out stability or the General Learning Setting.
Difficulty
The implications between stability and generalization change the sample size: has points, so a statement about the rule at size has to be compared with the rule at size under the marginal law of the reduced sample. The step from universal consistency to generalization needs Lemma 16. That lemma estimates from a sample on which the ERM itself may be inconsistent, and it is the only place where universality of the consistency rate is used. Per-distribution consistency of an ERM does not imply generalization (Example 1 of the paper). An argument that fixes throughout therefore cannot succeed.
Formalization scope
Samples are tuples with law Measure.pi (the i.i.d. product), and is Fin.removeNth i S. A learning rule is a family . Its value at is never used, and LOO stability is required only for . is the infimum and no minimiser is chosen. An ERM is a rule attaining this infimum at every sample. A rate is non-increasing on and tends to . Universal properties are stated as "there exists with IsRate ε such that for every probability measure …", with the rate chosen before the distribution. The bound is any bound on ; the paper's is , and all its rates increase with .
Measurability is not discussed in the paper. The formalization assumes that each is measurable and that the rule is measurable in the sense that is jointly measurable. Lemmas 14–17 also assume that is measurable. For a measurable ERM this holds automatically, so Theorem 31 makes no such assumption. Without these assumptions Lean's integral of a non-measurable function is and every rate bound would hold trivially. For the same reason Utility Lemma 13 assumes integrable. The ERM hypothesis of the goal must not be weakened to an AERM: the statement would then be false (Example 6).
One statement corrects the printed text. In the second display of the proof of Theorem 31 (p. 2668), the chain adds and then drops it. The milestone states the bound the argument proves, . The rate-free Theorem 31 is unaffected.
The development needs product measures, independence and variance bounds, all available in Mathlib, and the marginals of Measure.pi under removal of a coordinate. The definitions of risks, rules and stability notions can be reused by other stability results. Proofs of any milestone are welcome, and so are alternative proofs of the goal.
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
- O. Bousquet, A. Elisseeff, Stability and Generalization, Journal of Machine Learning Research 2 (2002) 499–526. https://www.jmlr.org/papers/v2/bousquet02a.html
- S. Mukherjee, P. Niyogi, T. Poggio, R. Rifkin, Learning theory: stability is sufficient for generalization and necessary and sufficient for consistency of empirical risk minimization, Advances in Computational Mathematics 25 (2006) 161–193. https://doi.org/10.1007/s10444-004-7634-z
- S. Kutin, P. Niyogi, Almost-everywhere algorithmic stability and generalization error, UAI 2002. https://arxiv.org/abs/1301.0579
- L. Devroye, T. Wagner, Distribution-free performance bounds for potential function rules, IEEE Transactions on Information Theory 25(5) (1979) 601–604. https://doi.org/10.1109/TIT.1979.1056087