Memory capacity of neural networks with threshold and ReLU activations
Machine Learning
2020-06-04 v2 Information Theory
Neural and Evolutionary Computing
math.IT
Machine Learning
Abstract
Overwhelming theoretical and empirical evidence shows that mildly overparametrized neural networks -- those with more connections than the size of the training data -- are often able to memorize the training data with accuracy. This was rigorously proved for networks with sigmoid activation functions and, very recently, for ReLU activations. Addressing a 1988 open question of Baum, we prove that this phenomenon holds for general multilayered perceptrons, i.e. neural networks with threshold activation functions, or with any mix of threshold and ReLU activations. Our construction is probabilistic and exploits sparsity.
Keywords
Cite
@article{arxiv.2001.06938,
title = {Memory capacity of neural networks with threshold and ReLU activations},
author = {Roman Vershynin},
journal= {arXiv preprint arXiv:2001.06938},
year = {2020}
}
Comments
26 pages. Minor inaccuracies corrected, discussion of prior work expanded