Self-Regularity of Output Weights for Overparameterized Two-Layer Neural Networks
At a glance
- Citations
- 1
- References
- 106
- Comments
- 0
Abstract
We consider the problem of finding a two-layer neural network with sigmoid, rectified linear unit, or binary step activation functions that “fits” a training data set as accurately as possible as quantified by the training error; and study the following question: does a low training error guarantee that the norm of the output layer (outer norm) itself is small? We address this question for the case of non-negative output weights. Using a simple covering number argument, we establish that under quite mild distributional assumptions on the input/label pairs; any such network achieving a small training error on polynomially many data necessarily has a well-controlled outer norm. Notably, our results (a) have a good sample complexity, (b) are independent of the number of hidden units, (c) are oblivious to the training algorithm; and (d) require quite mild assumptions on the data (in particular the input vector$X$∊ Rdneed not have independent coordinates). We then show how our bounds can be leveraged to yield generalization guarantees for such networks.
Publication details
- DOI
- 10.1109/isit45174.2021.9517811
- OpenAlex
- W3198742493
- Document type
- conference-paper
- Language
- EN
- Last metadata update
Comments
Log in to join the discussion.