For one-hidden-layer equivariant networks, generalization bounds depend only on filter norms and the sample size, while suitable weight sharing can match equivariance and locality adds an extra gain.
Non-Vacuous Generalisation Bounds for Shallow Neural Networks
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
We focus on a specific class of shallow neural networks with a single hidden layer, namely those with $L_2$-normalised data and either a sigmoid-shaped Gaussian error function ("erf") activation or a Gaussian Error Linear Unit (GELU) activation. For these networks, we derive new generalisation bounds through the PAC-Bayesian theory; unlike most existing such bounds they apply to neural networks with deterministic rather than randomised parameters. Our bounds are empirically non-vacuous when the network is trained with vanilla stochastic gradient descent on MNIST and Fashion-MNIST.
fields
cs.LG 1years
2024 1verdicts
ACCEPT 1representative citing papers
citing papers explorer
-
On the Sample Complexity of One Hidden Layer Networks with Equivariance, Locality and Weight Sharing
For one-hidden-layer equivariant networks, generalization bounds depend only on filter norms and the sample size, while suitable weight sharing can match equivariance and locality adds an extra gain.