We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cond-mat.stat-mech

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Condensed Matter > Disordered Systems and Neural Networks

Title: Universal mean field upper bound for the generalisation gap of deep neural networks

Abstract: Modern deep neural networks (DNNs) represent a formidable challenge for theorists: according to the commonly accepted probabilistic framework that describes their performance, these architectures should overfit due to the huge number of parameters to train, but in practice they do not. Here we employ results from replica mean field theory to compute the generalisation gap of machine learning models with quenched features, in the teacher-student scenario and for regression problems with quadratic loss function. Notably, this framework includes the case of DNNs where the last layer is optimised given a specific realisation of the remaining weights. We show how these results -- combined with ideas from statistical learning theory -- provide a stringent asymptotic upper bound on the generalisation gap of fully trained DNN as a function of the size of the dataset $P$. In particular, in the limit of large $P$ and $N_{\textrm{out}} $ (where $N_\textrm{out}$ is the size of the last layer) and $N_\textrm{out} \ll P$, the generalisation gap approaches zero faster than $2 N_\textrm{out}/P$, for any choice of both architecture and teacher function. Notably, this result greatly improves existing bounds from statistical learning theory. We test our predictions on a broad range of architectures, from toy fully-connected neural networks with few hidden layers to state-of-the-art deep convolutional neural networks.
Comments: 13 pages, 6 figures
Subjects: Disordered Systems and Neural Networks (cond-mat.dis-nn); Statistical Mechanics (cond-mat.stat-mech)
Cite as: arXiv:2201.11022 [cond-mat.dis-nn]
  (or arXiv:2201.11022v2 [cond-mat.dis-nn] for this version)

Submission history

From: Pietro Rotondo [view email]
[v1] Wed, 26 Jan 2022 15:54:35 GMT (729kb,D)
[v2] Wed, 2 Mar 2022 15:30:04 GMT (979kb,D)

Link back to: arXiv, form interface, contact.