English

Non-Vacuous Generalisation Bounds for Shallow Neural Networks

Machine Learning 2022-10-21 v3 Machine Learning

Abstract

We focus on a specific class of shallow neural networks with a single hidden layer, namely those with L2L_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.

Keywords

Cite

@article{arxiv.2202.01627,
  title  = {Non-Vacuous Generalisation Bounds for Shallow Neural Networks},
  author = {Felix Biggs and Benjamin Guedj},
  journal= {arXiv preprint arXiv:2202.01627},
  year   = {2022}
}

Comments

19 pages, 12 figures

R2 v1 2026-06-24T09:18:01.123Z