The Sample Complexity of Pattern Classification with Neural Networks: The Size of the Weights is More Important than the Size of the Network II: Fat-Shattering Bound for Bounded-Weight NetworksResearch Paper
Motivation
In the mid-1990s, neural networks trained by gradient descent were observed to generalize well even when the number of weights far exceeded the number of training examples. The classical theory could not explain this: VC-dimension bounds for networks grow with the number of parameters, so for large networks they are vacuous. Bartlett's paper (IEEE Trans. Inform. Theory 44 (1998)) showed that, for classification with a margin, what controls generalization is the size of the weights, not the size of the network. Its two main technical results are a margin bound in terms of the fat-shattering dimension (Theorem 2, the subject of mission I of this series) and a bound on the fat-shattering dimension of networks with bounded weights (Theorem 17, the subject of this mission). The same idea of weight-norm capacity control underlies much of the later theory of margins, boosting and kernel methods.
Timeline. Kearns and Schapire (1994) introduced the fat-shattering dimension. Alon, Ben-David, Cesa-Bianchi and Haussler (1997) bounded ℓ∞ covering numbers by it. Bartlett, Kulkarni and Posner (1997) gave the matching lower bound on ℓ1 covering numbers used here as Lemma 19. Maurey's approximation lemma (reported by Pisier, 1981) was used by Jones (1992) and Barron (1993) for approximation by networks, and by Lee, Bartlett and Williamson (1996) for covering numbers of convex hulls. Bartlett (1998) combined these into Theorem 17.
Setting
Let be a set and a class of functions . For , a sequence is -shattered by if there is such that for every some satisfies for all . The fat-shattering dimension is
A cover of a class at scale for a pseudometric on functions is a set of functions such that every has some with ; is the least size of a cover. For a sample the pseudometrics , , are the maximum, the mean, and the root mean square of over , and the uniform covering numbers are .
The hidden units form a nonempty class of functions . For the two-layer network class with ℓ1-bounded output weights is
In Lean these are BartlettNN.Margin.fat, BartlettNN.Margin.coverNum and BartlettNN.Margin.Ninf (shared with mission I), and BartlettNN.FatNet.N1, BartlettNN.FatNet.N2 and BartlettNN.FatNet.combos F A.
Formalization targets
Goal: Theorem 17
There is a universal constant such that for every , , , and with ,
The constant is left unspecified, as in the paper, so that the goal survives any improvement of the numerical constants.
Milestones (in the order the proof uses them)
- Lemma 19 (cited from Bartlett–Kulkarni–Posner): for -valued with , .
- Lemma 20, (5): for and , .
- Lemma 21 (Maurey): in a Hilbert space, a point of the closed convex hull of a set of norm at most is within of an average of points of the set, for every .
- Lemma 22: .
- Inequality (6): if with , then .
Significance
Theorem 17 bounds the capacity of a network class without reference to the number of hidden units . With Theorem 2 (mission I) it gives misclassification bounds for networks with small weights that hold for networks of any size, and by iteration it yields the bounds for deep sigmoid networks of Theorem 28 (mission III). Its method — upper-bound ℓ2 covering numbers through Maurey's lemma and compare with a lower bound in terms of fat-shattering — is a template for bounding the fat-shattering dimension of convex hulls in general.
The results are proved in the paper (Lemmas 19 and 21 by citation). None of them is formalized: the platform has no fat-shattering dimension, no uniform sample covering numbers of function classes, and only a finite-dimensional, diameter-based form of Maurey's lemma (HighDimProb.Appetizer.approx_caratheodory), which is not Lemma 21. This mission produces machine-checked statements of all five ingredients and of the theorem.
Difficulty
The obvious route would bound through a VC-type count of the network's parameters, which fails because is unbounded. The paper's route needs a lower bound on covering numbers by the fat-shattering dimension (Lemma 19, a combinatorial packing argument not proved in the paper), an upper bound by the fat-shattering dimension at a finer scale (Lemma 20, which goes through the Alon et al. scale-sensitive Sauer lemma and a quantization argument), and a probabilistic approximation argument in the empirical space (Lemmas 21 and 22). The final step solves a transcendental inequality (6) for , with care at the boundary where the logarithm is small.
Formalization scope
- Functions are maps
X → ℝ;fatis valued inℕ∞, so an unbounded shattering is , not a junk . Labels areBoolread throughpm(true ↦ 1); sequences are indexed byFin m(0-based). - Covers are external (finite sets of arbitrary functions ) with the strict inequality of Definition 3; the covering number is when no finite cover exists. The ℓ1 and ℓ2 distances carry the factor . Mathlib's
Metric.coveringNumber(closed balls, metric types) is not used. - Bounds of the form "" are stated for every finite value of , and where the paper's bound implies finiteness (Lemmas 20, 22, Theorem 17) finiteness is part of the conclusion.
- The constant of Theorem 17 is quantified before , , , , and ; a constant chosen after them would make the statement trivially true.
- Corrections of the printed statement. Theorem 17 is stated for (printed , under which the bound is meaningless) and (printed , under which is undefined). Implicit positivity (, , ) in Lemmas 20, 22 and (6) is stated as hypotheses. Inequality (6) is copied as printed, with .
- In Theorem 17 the logarithm is natural (the base is absorbed by ); (5) and (6) use .
- Lemma 21 is stated in a complete real inner product space, with "convex closure" read as the closure of the convex hull.
Contributions welcome: proofs of the five milestones and of the goal, and reusable infrastructure on fat-shattering and covering numbers of function classes.
Selected references
- P. L. Bartlett, The Sample Complexity of Pattern Classification with Neural Networks: The Size of the Weights is More Important than the Size of the Network, IEEE Trans. Inform. Theory 44(2), 1998, 525–536. https://doi.org/10.1109/18.661502
- N. Alon, S. Ben-David, N. Cesa-Bianchi, D. Haussler, Scale-sensitive dimensions, uniform convergence, and learnability, J. ACM 44(4), 1997, 615–631. https://doi.org/10.1145/263867.263927
- P. L. Bartlett, S. R. Kulkarni, S. E. Posner, Covering numbers for real-valued function classes, IEEE Trans. Inform. Theory 43(5), 1997, 1721–1724. https://doi.org/10.1109/18.623181
- M. J. Kearns, R. E. Schapire, Efficient distribution-free learning of probabilistic concepts, J. Comput. Syst. Sci. 48(3), 1994, 464–497. https://doi.org/10.1016/S0022-0000(05)80062-5
- W. S. Lee, P. L. Bartlett, R. C. Williamson, Efficient agnostic learning of neural networks with bounded fan-in, IEEE Trans. Inform. Theory 42(6), 1996, 2118–2132. https://doi.org/10.1109/18.556601
- A. R. Barron, Universal approximation bounds for superpositions of a sigmoidal function, IEEE Trans. Inform. Theory 39(3), 1993, 930–945. https://doi.org/10.1109/18.256500