×

The generalization error of random features regression: precise asymptotics and the double descent curve. (English) Zbl 07513415

Summary: Deep learning methods operate in regimes that defy the traditional statistical mindset. Neural network architectures often contain more parameters than training samples, and are so rich that they can interpolate the observed labels, even if the latter are replaced by pure noise. Despite their huge complexity, the same architectures achieve small generalization error on real data.
This phenomenon has been rationalized in terms of a so-called ‘double descent’ curve. As the model complexity increases, the test error follows the usual \(\bigcup\)-shaped curve at the beginning, first decreasing and then peaking around the interpolation threshold (when the model achieves vanishing training error). However, it descends again as model complexity exceeds this threshold. The global minimum of the test error is found above the interpolation threshold, often in the extreme overparametrization regime in which the number of parameters is much larger than the number of samples. Far from being a peculiar property of deep neural networks, elements of this behavior have been demonstrated in much simpler settings, including linear regression with random covariates.
In this paper we consider the problem of learning an unknown function over the \(d\)-dimensional sphere \(\mathbb{S}^{d-1}\), from \(n\) i.i.d. samples \((x_i, y_i) \in \mathbb{S}^{d-1} \times \mathbb{R}\), \(i \leq n\). We perform ridge regression on \(N\) random features of the form \(\sigma(w^T_a x)\), \(a \leq N\). This can be equivalently described as a two-layer neural network with random first-layer weights. We compute the precise asymptotics of the test error, in the limit \(N, n, d \rightarrow \infty\) with \(N/d\) and \(n/d\) fixed. This provides the first analytically tractable model that captures all the features of the double descent phenomenon without assuming ad hoc misspecification structures. In particular, above a critical value of the signal-to-noise ratio, minimum test error is achieved by extremely overparametrized interpolators, i.e., networks that have a number of parameters much larger than the sample size, and vanishing training error.

MSC:

68-XX Computer science
92-XX Biology and other natural sciences
PDFBibTeX XMLCite
Full Text: DOI arXiv