English

Non-asymptotic approximations of neural networks by Gaussian processes

Probability 2021-02-18 v1 Machine Learning Machine Learning

Abstract

We study the extent to which wide neural networks may be approximated by Gaussian processes when initialized with random weights. It is a well-established fact that as the width of a network goes to infinity, its law converges to that of a Gaussian process. We make this quantitative by establishing explicit convergence rates for the central limit theorem in an infinite-dimensional functional space, metrized with a natural transportation distance. We identify two regimes of interest; when the activation function is polynomial, its degree determines the rate of convergence, while for non-polynomial activations, the rate is governed by the smoothness of the function.

Keywords

Cite

@article{arxiv.2102.08668,
  title  = {Non-asymptotic approximations of neural networks by Gaussian processes},
  author = {Ronen Eldan and Dan Mikulincer and Tselil Schramm},
  journal= {arXiv preprint arXiv:2102.08668},
  year   = {2021}
}

Comments

18 pages