Skip to yearly menu bar Skip to main content


Poster

Neural network learns low-dimensional polynomials with SGD near the information-theoretic limit

Jason Lee · Kazusato Oko · Taiji Suzuki · Denny Wu

West Ballroom A-D #7201
[ ]
Fri 13 Dec 11 a.m. PST — 2 p.m. PST

Abstract: We study the problem of gradient descent learning of a single-index target function f(x)=σ(x,θ) under isotropic Gaussian data in Rd, where the unknown link function σ:RR has information exponent p (defined as the lowest degree in the Hermite expansion). Prior works showed that gradient-based training of neural networks can learn this target with ndΘ(p) samples, and such complexity is predicted to be necessary by the correlational statistical query lower bound. Surprisingly, we prove that a two-layer neural network optimized by an SGD-based algorithm (on the squared loss) learns f with a complexity that is not governed by the information exponent. Specifically, for arbitrary polynomial single-index models, we establish a sample and runtime complexity of nT=Θ(dpolylogd), where Θ() hides a constant only depending on the degree of σ; this dimension dependence matches the information theoretic limit up to polylogarithmic factors. More generally, we show that nd(p1)1 samples are sufficient to achieve low generalization error, where pp is the \textit{generative exponent} of the link function. Core to our analysis is the reuse of minibatch in the gradient computation, which gives rise to higher-order information beyond correlational queries.

Chat is not available.