English

Investigating Generalization by Controlling Normalized Margin

Machine Learning 2022-09-21 v3

Abstract

Weight norm w\|w\| and margin γ\gamma participate in learning theory via the normalized margin γ/w\gamma/\|w\|. Since standard neural net optimizers do not control normalized margin, it is hard to test whether this quantity causally relates to generalization. This paper designs a series of experimental studies that explicitly control normalized margin and thereby tackle two central questions. First: does normalized margin always have a causal effect on generalization? The paper finds that no -- networks can be produced where normalized margin has seemingly no relationship with generalization, counter to the theory of Bartlett et al. (2017). Second: does normalized margin ever have a causal effect on generalization? The paper finds that yes -- in a standard training setup, test performance closely tracks normalized margin. The paper suggests a Gaussian process model as a promising explanation for this behavior.

Keywords

Cite

@article{arxiv.2205.03940,
  title  = {Investigating Generalization by Controlling Normalized Margin},
  author = {Alexander R. Farhang and Jeremy Bernstein and Kushal Tirumala and Yang Liu and Yisong Yue},
  journal= {arXiv preprint arXiv:2205.03940},
  year   = {2022}
}
R2 v1 2026-06-24T11:10:49.088Z