330

Generalization error of random features and kernel methods: hypercontractivity and kernel matrix concentration

Applied and Computational Harmonic Analysis (ACHA), 2021
Abstract

Consider the classical supervised learning problem: we are given data (yi,xi)(y_i,{\boldsymbol x}_i), ini\le n, with yiy_i a response and xiX{\boldsymbol x}_i\in {\mathcal X} a covariates vector, and try to learn a model f:XRf:{\mathcal X}\to{\mathbb R} to predict future responses. Random features methods map the covariates vector xi{\boldsymbol x}_i to a point ϕ(xi){\boldsymbol \phi}({\boldsymbol x}_i) in a higher dimensional space RN{\mathbb R}^N, via a random featurization map ϕ{\boldsymbol \phi}. We study the use of random features methods in conjunction with ridge regression in the feature space RN{\mathbb R}^N. This can be viewed as a finite-dimensional approximation of kernel ridge regression (KRR), or as a stylized model for neural networks in the so called lazy training regime. We define a class of problems satisfying certain spectral conditions on the underlying kernels, and a hypercontractivity assumption on the associated eigenfunctions. These conditions are verified by classical high-dimensional examples. Under these conditions, we prove a sharp characterization of the error of random features ridge regression. In particular, we address two fundamental questions: (1)(1)~What is the generalization error of KRR? (2)(2)~How big NN should be for the random features approximation to achieve the same error as KRR? In this setting, we prove that KRR is well approximated by a projection onto the top \ell eigenfunctions of the kernel, where \ell depends on the sample size nn. We show that the test error of random features ridge regression is dominated by its approximation error and is larger than the error of KRR as long as Nn1δN\le n^{1-\delta} for some δ>0\delta>0. We characterize this gap. For Nn1+δN\ge n^{1+\delta}, random features achieve the same error as the corresponding KRR, and further increasing NN does not lead to a significant change in test error.

View on arXiv
Comments on this paper