A Random Matrix Approach to Neural Networks

By Cosme Louart, Zhenyu A. Liao, Romain Couillet. 2017

194 citations recorded by the source corpus.

This article studies the Gram random matrix model $G=\frac1T\Sigma^{\rm T}\Sigma$, $\Sigma=\sigma(WX)$, classically found in the analysis of random feature maps and random neural networks, where $X=[x_1,\ldots,x_T]\in{\mathbb R}^{p\times T}$ is a (data) matrix of bounded norm, $W\in{\mathbb R}^{n\times p}$ is a matrix of independent zero-mean unit variance entries, and $\sigma:{\mathbb R}\to{\mathbb R}$ is a Lipschitz continuous (activation) function --- $\sigma(WX)$ being understood entry-wise. By means of a key concentration of measure lemma arising from non-asymptotic random matrix arguments, we prove that, as $n,p,T$ grow large at the same rate, the resolvent $Q=(G+\gamma I_T)^{-1}$, for $\gamma>0$, has a similar behavior as that met in sample covariance matrix models, involving notably the moment $\Phi=\frac{T}n{\mathbb E}[G]$, which provides in passing a deterministic equivalent for the empirical spectral measure of $G$. Application-wise, this result enables the estimation of the asymptotic performance of single-layer random neural networks. This in turn provides practical insights into the underlying mechanisms into play in random neural networks, entailing several unexpected consequences, as well as a fast practical means to tune the network hyperparameters.

Subfields: Computational Theory and Mathematics, Statistics and Probability, Probabilistic & Bayesian Methods

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computational Theory and Mathematics

  1. Michael W. Mahoney
  2. Taiji Suzuki
  3. Charles H. Martin
  4. Jimmy Ba
  5. Andrew M. Saxe
  6. Madhu S. Advani
  7. David P. Woodruff
  8. Petros Drineas

In Statistics and Probability

  1. Andrew M. Saxe
  2. Michael W. Mahoney
  3. Charles H. Martin
  4. Tianzong Zhang
  5. Taiji Suzuki
  6. Murat A. Erdogdu
  7. Jimmy Ba
  8. Denny Wu

In Probabilistic & Bayesian Methods

  1. Michael W. Mahoney
  2. Taiji Suzuki
  3. Murat A. Erdogdu
  4. Charles H. Martin
  5. Jimmy Ba
  6. Andrew M. Saxe
  7. Madhu S. Advani
  8. Balaji Lakshminarayanan

Citations in the corpus