2021/09/20 by Zhichao Wang, Yizhe Zhu, Wang, Zhichao +1 · 5 citations
Computer Science · #60B20 (Primary) 68T07 #62J07 (Secondary) #FOS: Computer and information sciences #FOS: Mathematics #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Matrix Theory and Algorithms #Neural Networks and Applications #Probability (math.PR) #Statistics Theory (math.ST) #Stochastic Gradient Optimization Techniques
paper · pdf · doi:10.48550/arxiv.2109.09304
openalex publication_date 2021/09/20 · openalex created_date 2025/10/10 · openalex updated_date 2026/08/01
In this paper, we investigate a two-layer fully connected neural network of the form f(X)=(1)/(√(d1))\boldsymbola^\top σ(WX), where X∈ℝd0× n is a deterministic data matrix, W∈ℝd1× d0 and \boldsymbola∈ℝd1 are random Gaussian weights, and σ is a nonlinear activation function. We study the limiting spectral distributions of two empirical kernel matrices associated with f(X): the empirical conjugate kernel (CK) and neural tangent kernel (NTK), beyond the linear-width regime (d1\asymp n). We focus on the ultra-wide regime, where the width d1 of the first layer is much larger than the sample size n. Under appropriate assumptions on X and σ, a deformed semicircle law emerges as d1/n→∞ and n→∞. We first prove this limiting law for generalized sample covariance matrices with some dependency. To specify it for our neural network model, we provide a nonlinear Hanson-Wright inequality that is suitable for neural networks with random weights and Lipschitz activation functions. We also demonstrate non-asymptotic concentrations of the empirical CK and NTK around their limiting kernels in the spectral norm, along with lower bounds on their smallest eigenvalues. As an application, we show that random feature regression induced by the empirical kernel achieves the same asymptotic performance as its limiting kernel regression under the ultra-wide regime. This allows us to calculate the asymptotic training and test errors for random feature regression using the corresponding kernel regression.