about
Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descent (arxiv.org)
2 points by gballan on Feb 20, 2019 | hide | past | pdf | discuss on HN

In plain words: When a neural network is made very wide, training it with gradient descent behaves like fitting a simple straight-line model built from its starting weights. Its predictions matched the real network's closely even at ordinary sizes, across many architectures, optimizers, and loss functions.

Abstract

A longstanding goal in deep learning research has been to precisely characterize training and generalization. However, the often complex loss landscapes of neural networks have made a theory of learning dynamics elusive. In this work, we show that for wide neural networks the learning dynamics simplify considerably and that, in the infinite width limit, they are governed by a linear model obtained from the first-order Taylor expansion of the network around its initial parameters. Furthermore, mirroring the correspondence between wide Bayesian neural networks and Gaussian processes, gradient-based training of wide neural networks with a squared loss produces test set predictions drawn from a Gaussian process with a particular compositional kernel. While these theoretical results are only exact in the infinite width limit, we nevertheless find excellent empirical agreement between the predictions of the original network and those of the linearized version even for finite practically-sized networks. This agreement is robust across different architectures, optimization methods, and loss functions.

Jaehoon Lee, Lechao Xiao, Samuel S. Schoenholz, Yasaman Bahri, Roman Novak, Jascha Sohl-Dickstein, Jeffrey Pennington
arXiv:1902.06720 · stat.ML, cs.LG · submitted Feb 18, 2019 · updated Dec 8, 2019
abstract · pdf · html · 12+16 pages; open-source code available at https://github.com/google/neural-tangents; accepted to NeurIPS 2019

add comment on HN
Also discussed: Feb 2019 (3 points, 0 comments)