When Do Neural Networks Outperform Kernel Methods?
Behrooz Ghorbani, Song Mei, Theodor Misiakiewicz, Andrea Montanari
Introduction
is a non-linearly parametrized class of functions: while nonlinearity poses a challenge to theoreticians, it is often claimed to be crucial in order to learn rich representation of the data. Recent efforts to understand NN have put the spotlight on two linearizations of , the random features [RR08] and the neural tangent [JGH18] classes
and are linear classes of functions, depending on the realization of the input-layer weights (which are chosen randomly). The relation between NN and these two linear classes is given by the first-order Taylor expansion: , where . A number of recent papers show that, if weights and SGD updates are suitably scaled, and the network is sufficiently wide ( sufficiently large), then SGD converges to a function that is approximately in , with determined by the SGD initialization [JGH18, DZPS19, DLL+19, AZLS19, ZCZG18, OS20]. This was termed the ‘lazy regime’ in [COB19].
where denotes closure. From this point of view, and differ in that they correspond to slightly different choices of the kernel: versus . Multi-layer fully-connected s in the lazy regime can be viewed as randomized approximations to RKHS as well, with some changes in the kernel . This motivates analogous questions for : can the performances of be achieved by RKHS methods?
Recent work addressed the separation between NN and RKHS from several points of view, without providing a unified answer. Some empirical studies on various datasets showed that networks can be replaced by suitable kernels with limited drop in performances [ADL+20, LWY+19, LXS+19, NXB+19, LSdP+18, DMHR+18, GARA19, SFG+20]. At least two studies reported a larger gap for convolutional networks and the corresponding kernels [ADH+19, GSJW19]. On the other hand, theoretical analysis provided a number of separation examples, i.e. target functions that can be represented and possibly efficiently learnt using neural networks, but not in the corresponding RKHS [YS19, Bac17, GMMM19b, GMMM19a, AZL19, AZL20]. For instance, if the target is a single neuron , then training a neural network with one hidden neuron learns the target efficiently from approximately samples [MBM18], while the corresponding RKHS has test error bounded away from zero for every sample size polynomial in [YS19, GMMM19b]. Further even in the infinite width limit, it is known that two-layers neural networks can actually capture a richer class of functions than the associated RKHS, provided SGD training is scaled differently from the lazy regime [MMN18, CB18, RVE18, SS18, CB20].
Can we reconcile empirical and theoretical results?
In this paper we introduce a stylized scenario – which we will refer to as the spiked covariates model – that can explain the above seemingly divergent observations in a unified framework. The spiked covariates model is based on two building blocks: Target functions depending on low-dimensional projections; Approximately low-dimensional covariates.
As for the example of a single neuron , we expect RKHS to suffer from a curse of dimensionality in learning functions of low-dimensional projections. Indeed, this is well understood in low dimension or for isotropic covariates [Bac17, GMMM19b].
Approximately low-dimensional covariates. RKHS behave well on certain image classification tasks [ADH+19, LWY+19, NXB+19], and this seems to contradict the previous point. However, the example of image classification naturally brings up another important property of real data that helps to clarify this puzzle. Not only we expect the target function to depend predominantly on the low-frequency components of image , but the image itself to have most of its spectrum concentrated on low-frequency components (linear denoising algorithms exploit this very observation).
2 Notations and outline
In section 2, we introduce the spiked covariates model and characterize the performance of KRR, RF, NT, and NN models. Section 3 presents numerical experiments with real and synthetic data. Section 4 discusses our results in the context of earlier work.
Rigorous results for kernel methods and NT, RF NN expansions
We call the signal covariates, the noise covariates, and the covariates signal-to-noise ratio (or covariates SNR). We will take , so that the variance of the signal covariates is larger than that of the noise covariates . In high dimension, this model is –for many purposes– similar to an anisotropic Gaussian model . As shown below, the effect of anisotropy on RKHS methods is significant only if the covariate SNR is polynomially large in . We shall therefore set for a constant .
We will consider a more general model in Appendix C, in which the distribution of takes a more general product-of-uniforms form, and we assume a general .
2 A sharp characterization of RKHS methods
Any RKHS method with kernel outputs a model of the form , with RKHS norm given by . We consider kernel ridge regression (KRR) on the dataset with regularization parameter , namely:
where , with . We denote the prediction error of KRR by
where .
Recall that we assume the target function . We denote to be the projection operator onto the space of degree orthogonal polynomials, and . Our next theorem shows that the impact of the low-dimensional latent structure on the generalization error of KRR is characterized by a certain ‘effective dimension’, .
3 RF and NT models
How do the results of the previous section generalize to finite-width approximations of the RKHS? In particular, how do the RF and NT models behave at finite ? In order to simplify the picture, we focus here on the approximation error. Equivalently, we assume the sample size to be and consider the minimum population risk for
The next two theorems characterize the asymptotics of the approximation error for RF and NT models. We give generalizations of these statements to other settings and under weaker assumptions in Appendix C.
Finally, it is natural to ask what are the behaviors of RF and NT models at finite sample size. Denote by the corresponding test error (assuming for instance ridge regression, with the optimal regularization ). Of course the minimum population risk provides a lower bound: . Moreover, we conjecture that the risk is minimized at infinite , . Altogether this implies the lower bound . We also conjecture that this lower bound is tight, up to terms vanishing as .
4 Neural network models
Moreover, the quantity is independent of .
As a consequence of Theorem 3 and 4, there is a separation between NN and (uniformly sampled) NT models when , i.e., . As increases, the gap between NN and NT becomes smaller and smaller until .
Further numerical experiments
We carried out extensive numerical experiments on synthetic data to check our predictions for RF, NT, RKHS methods at finite sample size , dimension , and width . We simulated two-layers fully-connected NN in the same context in order to compare their behavior to the behavior of the previous models. Finally, we carried out numerical experiments on FMNIST and CIFAR-10 data to test whether our qualitative predictions apply to image datasets. Throughout we use ReLU activations.
The basic qualitative insight of our work can be summarized as follows. Kernel methods are effective when a low-dimensional structure in the target function is aligned with a low-dimensional structure in the covariates. In image data, both the target function and the covariates are dominated by the low-frequency subspace. In Figure 1 we tested this hypothesis by removing the low-dimensional structure of the covariate vectors: we simply added noise to the high-frequency part of the image. In Figure 4 we try the opposite, by removing the component of the target function that is localized on low-frequency modes. We decompose each images into a low-frequency and a high-frequency part. We leave the high-frequency part unchanged, and replace the low-frequency part by Gaussian noise with the first two moments matching the empirical moments of the data.
In the left frame, we consider FMNIST data and compare fully-connected NNs with or layers (and nodes at each hidden layer) with the corresponding NT KRR model (infinite width). In the right frame, we use CIFAR-10 data and compare a Myrtle-5 network (a lightweight convolutional architecture [Pag18, SFG+20]) with the corresponding NT KRR. We observe the same behavior as in Figure 1. While for the original data NT is comparable to NN, as the proportion of perturbed Fourier modes increases, the performance of NT deteriorates much more rapidly than the one of NN.
Discussion
However, these classical statistical results have some limitations. First, they focus on the low-dimensional regime: is fixed, while the sample size diverges. This is probably unrealistic for many machine learning applications, in which is at least of the order of a few hundreds. Second, classical lower bounds are typically established for the minimax risk, and hence they do not necessarily apply to specific functions.
To bridge these gaps, we developed a sharp characterization of the test error in the high-dimensional regime in which both and diverge, while being polynomially related. This characterization holds for any target function , and expresses the limiting test error in terms of the polynomial decomposition. We also present analogous results for finite-width RF and NT models.
Depending on the relation between signal dimension , ambient dimension , and the covariate signal-to-noise ratio , the model presents a continuum of different behaviors. At one extreme, the covariates are fully -dimensional, and RKHS methods are highly suboptimal compared to NN. At the other, covariates are close to -dimensional and RKHS methods are instead more competitive with NN.
Finally, the Fourier decomposition of images is a simple proxy for the decomposition of the covariate vector into its low-dimensional dominant component (low frequency) and high-dimensional component (high frequency) [YLS+19].
Acknowledgements
This work was partially supported by the NSF grants CCF-1714305, IIS-1741162, DMS-1418362, DMS-1407813 and by the ONR grant N00014-18-1-2729.
References
Appendix A Details of numerical experiments
where and is the total number of training epochs. To ensure the stability of the optimization for wide models, we use linear warm-up epochs in the beginning.
In order to use CG, we first implement a function to perform Hessian-vector products in TensorFlow [ABC+16]. The function handle is then passed to scipy.sparse.cg for CG. Our Hessian-vector product code uses tensor manipulation utilities implemented by [GKX19].
Unfortunately, scipy.sparse.cg does not support one-hot encoded labels. To avoid running CG for each class separately, when the labels are one-hot encoded, we use Adam optimizer [KB14] instead. When using Adam, the learning-rate still evolves as (11) with . The batch-size is fixed at to encourage fast convergence to the minimum.
For , and , the training is primary done in TensorFlow (v1.12) [ABC+16]. For KRR, we generate the kernel matrix first and directly fit the model in regular python. The kernels associated with two-layer models are calculated analytically. For deeper models, the kernels are computed using neural-tangents library in JAX [BFH+18, NXH+20].
A.2 Synthetic data experiments
The synthetic data follows the distribution outlined in the main text. In particular,
where and are drawn i.i.d from the hyper-spheres with radii and respectively. We choose
where is fixed to be and . We change in the interval . For each value of we generate training and test observations.Strictly speaking, the model outlined in the main text requires to be generated from the hyper-sphere of radius . In order to work with round numbers, in our experiments we use instead of . The numerical difference between these two choices is negligible.
The function is the sum of three orthogonal components with . To be more specific,
This choice of guarantees that each is in the span of degree spherical harmonics.
In Figure 3 of the main text, we compared the generalization performance of NTK KRR with . We use the same training and test data as above to perform this analysis. The number of training data points, , takes different values ranging from to . The number of test data points is always fixed at .
A.3 High-frequency noise experiment on FMNIST
In effort to make the distribution of the covariates more isotropic, in this experiment, we add high-frequency noise to both the training and test data.
Finally, we normalize the so that it has norm .
This choice of mirrors the average frequency domain representation of FMNIST images (see Figure A.1 for a comparison). Figure A.2 shows the eigenvalues of the empirical covariance of the dataset for various noise levels. As discussed in the main text, the distribution of the covariates becomes more isotropic as more and more high-frequency noise is added to the images.
Figure A.3 shows the normalized squared loss and the classification accuracy of the models as more and more high-frequency noise is added to the data. The normalization factor corresponds to the risk achievable by the (trivial) predictor \bigg{[}\hat{y}_{j}({\bm{x}})\bigg{]}_{1\leq j\leq 10}=0.1.
A.4 High-frequency noise experiment on CIFAR-2
We perform a similar experiment on a subset of CIFAR-10. We choose two classes (airplane and cat) from the ten classes of CIFAR-10. This choice provides us with training and test data points. Given that the number of training observations is not very large, we reduce the covariate dimension by converting the images to grayscale. This transformation reduces the covariate dimension to .
Figure A.5 demonstrates the evolution of the model performances as the noise intensity increases. In the noiseless regime (), all models have comparable performances. However, as the noise level increases, the performance gap between and RKHS methods widens. For reference, the accuracy gap between and KRR is only at . However, at , this gap increases to . The normalization factor corresponds to the risk achievable by the trivial estimator .
A.5 Low-frequency noise experiments on FMNIST
To examine the ability of NN and RKHS methods in learning the information in low-variance components of the covariates, we replace the low-frequency components of the image with Gaussian noise. To be specific, we follow the following steps to generate the noisy datasets:
We normalize all images to have mean zero and norm .
Let denote the set of training images in the DCT-frequency domain. We compute the mean and the covariance of the elements of .
The fraction of the frequencies replaced by noise is .
For neural networks trained for this experiment, we fix the number of hidden units per-layer to . This corresponds to approximately trainable parameters for two-layer networks and trainable parameters for three-layer networks. Both models are trained using SGD with momentum with learning rate described by (with ). For the warm-up epochs, we use batch-size of . We increase the batch-size to after the warm-up stage. The regularization grids used for training our models are presented in Table A.3.
A.6 Low-frequency noise experiments on CIFAR-10
To test whether our insights are valid for convolutional models, we repeat the same experiment for CNNs trained on CIFAR-10. The noisy data is generated as follows:
Let denote the set of training images in the DCT-frequency domain. Note that CIFAR-10 images have channels. To convert the images to frequency domain, we apply two-dimensional Discrete Cosine Transform (DCT-II orthogonal) to each channel separately. We compute the mean and the covariance of the elements of .
We normalize the noisy data to have zero per-channel mean and unit per-channel standard deviation. The normalization statistics are computed using only the training data.
We use Myrtle-5 architecture for our analysis. The Myrtle family is a collection of simple light-weight high-performance purely convolutional models. The simplicity of these models coupled with their good performance makes them a natural candidate for our analysis. Figure A.7 describes the details of this architecture. We fix the number of channels in all convolutional layer to be . This corresponds to approximately parameters. Similar to the fully-connected networks, our convolutional models are also optimized via SGD with momentum (learning rate evolves as (11) with and ). We fix the batch-size to . To keep the experimental setting as simple as possible, we do not use any data augmentation for training the network.
Appendix B Technical background on function spaces on the sphere
The dimension of each subspace is given by
B.2 Gegenbauer polynomials
We will use the following properties of Gegenbauer polynomials
Note in particular that property 2 implies that –up to a constant– is a representation of the projector onto the subspace of degree - spherical harmonics
B.3 Hermite polynomials
Notice that for functions that are -weakly differentiable with the -th weak derivative, we have
Here and below, for a polynomial, is the vector of the coefficients of .
B.4 Tensor product of spherical harmonics
We will consider in this paper the product space
The dimension of each subspace is given by
We have the following orthonormalization property
We denote by the orthogonal projections on in . This can be written in terms of spherical harmonics as
B.5 Tensor product of Gegenbauer polynomials
We will use the following properties of the tensor product of Gegenbauer polynomials:
Notice that Lemma 1.(c) implies that is (up to a constant) a representation of the projector onto the subspace
Part comes from the normalization property (22) of Gegenbauer polynomials,
while part is a direct consequence of Eq. (24).
B.6 Notations
Furthermore, will denote .
Appendix C General framework and main theorems
In this section, define a more general model than the model considered in the main text. In the general model, we will assume the covariate vectors will follow a product of uniform distributions on the sphere, and assume a target function in space. We establish more general versions of Theorems 1, 2, 3 on the two-spheres cases in the main text as Theorems 5, 6, 7. We will prove Theorems 5, 6, 7 in the following sections. At the end of this section, we will show that Theorems 5, 6, 7 will imply Theorems 1, 2, 3 in the main text.
Assume that the data lies on the product of spheres,
where and . Let and , where and for . We will denote this space
Furthermore, assume that the data is generated following the uniform distribution on , i.e.
We will make the following assumption that will simplify the proofs. Denote
then is attained on only one of the sphere, whose coordinate will be denoted , i.e. and for .
We will denote . Notice that the normalization in the definition of the function class insures that the scalar product is of order . This corresponds to normalizing the data.
We consider the approximation of by functions in function classes and .
C.2 Reparametrization
It is easy to check that the variables are independent and independent of , and verify
We will denote and . With these notations, we have
where is the ‘normalized space of product of spheres’, and
Similarly, we will denote the rescaled data ,
obtained by taking for each .
The proof will proceed as follows: first, noticing that concentrates around for every , we will restrict ourselves without loss of generality to the following high probability event
where will be chosen sufficiently small. Then, we rewrite the activation function
as a function, for a random (but close to )
given for by
We can therefore apply the algebra of tensor product of spherical harmonics and use the machinery developed in [GMMM19b].
C.3 Notations
Recall the definitions , , , , and . Let us denote and .
C.4 Generalization error of kernel ridge regression
We consider the Kernel Ridge Regression solution , namely
where the kernel matrix is assumed to be given by
and , with
The prediction function at location gives
The test error of empirical kernel ridge regression is defined as
Notice that by definition .
We consider sequences of problems indexed by the integer , and we view the problem parameters (in particular, the dimensions , the radii , the kernel , and so on) as functions of .
For (which is specified in the theorem), we denote . We assume that is -weakly differentiable. We assume that for , the -th weak derivative verifies almost surely for some constants independent of . Furthermore, we assume there exists such that with independent of .
For (which is specified in the theorem), we define
We assume that verifies for , , with independent of .
Let be a sequence of functions. Assume for some and . Let be a sequence of functions that satisfies Assumption 1 at level . Let with independently, and and for some . Then for any , and for any , with high probability we have
See Section D for the proof of this Theorem.
C.5 Approximation error of the random features model
We consider the minimum population error for the random features model
For (which is specified in the theorem), we denote . We assume that is -weakly differentiable. Define
We assume that for , the -th weak derivative verifies almost surely for some constants and .
Furthermore we will assume that is not a degree- polynomial where we recall that corresponds to the unique .
For (which is specified in the theorem), we define
We assume that verifies for , . Furthermore we assume that for , the -th weak derivative verifies almost surely for some constants and .
Assume for a fixed . Let satisfy Assumptions 2.(a) and 2.(b) at level . Then, for any , the following holds with high probability:
where is defined in Equation (45).
Assume for some positive constant , and satisfy Assumptions 2.(a) and 2.(c) at level . Then for any , the following holds with high probability:
where is defined in Equation (46).
See Section E for the proof of the lower bound (48), and Section F for the proof of the upper bound (49).
such that for , model fits the subspace of low degree polynomials and cannot fit , i.e.
Each subspace has therefore an effective dimension . This can be understood intuitively as follows,
C.6 Approximation error of the neural tangent model
We consider the minimum population error for the random features model
For (which is specified in the theorem), we denote . We assume that is -weakly differentiable. Define
We assume that for , the -th weak derivative verifies almost surely for some constants and .
For (which is specified in the theorem), we define
We assume that verifies for , . Furthermore we assume that for , the -th weak derivative verifies almost surely for some constants and .
In the Assumption 3.(b), it is useful to notice that the Hermite coefficients of can be computed from the ones of using the relation .
Assume for a fixed . Let satisfy Assumptions 3.(a) and 3.(b) at level . Then, for any , the following holds with high probability:
where is defined in Equation (50).
Assume for some positive constant , and satisfy Assumptions 3.(a) and 3.(c) at level . Then for any , the following holds with high probability:
where is defined in Equation (51).
See Section G for the proof of lower bound, and Section H for the proof of upper bound.
This theorems shows that each for each such that , we can decompose our functional space as
such that for , model fits the subspace of low degree polynomials and cannot fit at all, i.e.
where .
C.7 Connecting to the theorems in the main text
Let us connect the above general results to the two-spheres setting described in the main text. We consider two spheres with , for the first sphere, and , for the second sphere. We have .
Let with constant sufficiently small, then by Theorem 5 the function subspace learned by KRR is given by the polynomials of degree in the first sphere coordinates and in the second sphere with
We consider functions that only depend on the first sphere, i.e., and denote . Then the subspace of approximation is given by the polynomials in the first sphere such that . Furthermore, one can check that the Assumptions listed in Theorem 1 in the main text verifies Assumption 1.
Similarly, for with constant sufficiently small, Theorem 6 implies that the models can only approximate polynomials in the first sphere such that . Furthermore, Assumptions listed in Theorem 2 in the main text verifies Assumption 2.
In the case of , we only consider and . We get . The subspace approximated is given by the polynomials in the first sphere such that . Furthermore, Assumptions listed in Theorem 3 in the main text verifies Assumption 3.
Appendix D Proof of Theorem 5
The proof follows closely the proof of [GMMM19b, Theorem 4].
Let us rewrite the kernel functions as functions on the product of normalized spheres: for and :
Consider the expansion of in terms of tensor product of Gegenbauer polynomials. We have
where the expectation is taken over .
Let be a sequence of kernel functions that satisfies Assumption 1. Assume for some . Consider as defined in Eq. (43). Then there exists constants such that for large enough,
where . By Assumption 1., we have
Furthermore, by Assumption 1. and dominated convergence,
for . The lemma then follows from the same proof as in Lemma 9 and Lemma 10, where we adapt the proofs of Lemma 19 and 20 to . ∎
D.2 Proof of Theorem 5
Step 1. Rewrite the , , , matrices.
The test error of empirical kernel ridge regression gives
where and , with
Let the spherical harmonics decomposition of be
and the Gegenbauer decomposition of be
We write the decompositions of vectors , , , and . We have
The rest of the proof follows closely from [GMMM19b, Theorem 4]. We decompose the risk as follows
Further, we denote , , , and ,
Using Cauchy Schwarz inequality for , we get
As a result, combining Eqs. (58), (60) and (59), we have
where the last equality used the fact that and Lemma 2. Combining Eqs. (62), (63) and (64), we get
Step 5. Terms and . By Lemma 6 again, we have
We decompose using ,
Combining Eqs. (65), (61), (66), (67) and (68), we have
D.3 Auxiliary results
Assume that and consider
Denote and
Notice that there exists such that (by definition of and ) and therefore . Integrating the tail bound (69) proves the lemma. ∎
Let be an activation function satisfying Assumption 1. Let for some and . Then there exists sequences and such that
with . From Assumption 1. and a proof similar to Lemma 20, there exists (for at position ) such that . Hence, .
From Lemma 3 we have for ,
Let be an activation function satisfying Assumption 1. Assume for some and . We have
Denote , and
Then, we can use the same proof as in [GMMM19b, Lemma 13] to bound (recall )
Let be an activation function satisfying Assumption 1. Assume for some and . We have
This lemma can be deduced directly from [GMMM19b, Lemma 14], by noticing that
Appendix E Proof of Theorem 6.(a): lower bound for the RF model
In the theorems, we show our results in high probability with respect to . Hence, in the proof we will restrict the sample space to the high probability event for small enough, where
Assume for some . We have for any fixed ,
The tail inequality in Lemma 16 and the assumption imply that there exists some constants such that
Consider the expansion of in terms of tensor product of Gegenbauer polynomials. We have
where the expectation is taken over .
Let be an activation function that satisfies Assumptions 2.(a) and 2.(b). Consider and as defined in Theorem 6.(a). Then there exists and and a constant such that for and ,
Notice that by Assumption 2. we can apply Lemma 19 to any such that . In particular, there exists , and such that for any with , and ,
Furthermore, using that , there exists such that for with ,
where we used in the last inequality implies by definition.
Furthermore, from Assumption 2 and Lemma 17., there exists , and , such that
In particular, for , we have
Combining Eqs (79) and (80) yields the result. ∎
E.2 Proof of Theorem 6.(a): Outline
Define the random vectors , , , with
Define the random matrix , with
In what follows, we write for the random features risk, omitting the dependence on the weights . By the definition and a simple calculation, we have
where the last inequality used the fact that
The Theorem follows from the following two claims
This is achieved by the Proposition 1 and 2 stated below.
Let be an activation function satisfying Assumptions 2.(a) and 2.(b) for a fixed . Denote . Let and define by
Then there exists a constant and (depending only on the constants of Assumptions 2.(a) and 2.(b)) such that for sufficiently large,
The proofs of these two propositions are provided in the next sections.
Proposition 1 shows that there exists such that
Hence, by Markov’s inequality, we get for any ,
where we used Lemma 8. By assumption, we have , hence Eq. (86) is verified. Furthermore Eq. (87) follows simply from Proposition 2. This proves the theorem.
E.3 Proof of Proposition 1
such that is a function on the normalized product of spheres (Note that we defined the unambiguous polynomial approximation of with polynomial of degree ). We have
We recall the expansion of in terms of tensor product of Gegenbauer polynomials
Using Eq. (39) to get the following property
Let be a constant as specified in Lemma 9. We consider
From Lemma 9, there exists a constant such that for sufficiently large, we have for any ,
E.4 Proof of Proposition 2
Step 1. Construction of the activation functions , .
Without loss of generality, we will assume that . From Assumption 2., is not a degree -polynomial. This is equivalent to having such that . Let us denote
Recall the expansion of in terms of product of Gegenbauer polynomials
Step 2. The kernel functions , and .
Let , and be defined by
We immediately have . Note that all three correspond to positive semi-definite kernels.
Since , we immediately have . In the following, we will lower bound .
By the decomposition of in terms of Gegenbauer polynomials (93), we have
From Assumption 2. and Lemma 20 applied to coefficient , as well as the assumption that , there exists and such that for large enough,
We restrict ourselves to the event defined in Eq. (76), which happens with high probability (Lemma 8). Hence from Eqs. (94) and (95), we deduce that with high probability
Appendix F Proof of Theorem 6.(b): upper bound for RF model
The first inequality comes simply from Assumption 2. and Lemma 17.. For the second inequality, notice that by Assumption 2. we can apply Lemma 19 to any . Hence (using that and we can choose sufficiently small), we deduce that there exists , and such that for any , and ,
Furthermore, using that , there exists such that for any ,
where we used in the last inequality implies by definition. ∎
F.2 Properties of the limiting kernel
Similarly to the proof of [GMMM19b, Theorem 1.], we construct a limiting kernel which is used as a proxy to upper bound the RF risk.
We recall the decomposition of in terms of tensor product of Gegenbauer polynomials
F.3 Proof of Theorem 6.(b)
Without loss of generality, let us assume that are polynomials contained in , i.e. .
Let be defined as in Lemma 10 and consider the expectation over of the RF risk (in particular, are well defined):
We can expand the squared loss at as
The second term of the expansion (LABEL:eq:expansion_squared_loss_RF) around verifies
Let us consider the third term in the expansion (LABEL:eq:expansion_squared_loss_RF) around : the non diagonal term verifies
For and and , we have (for large enough)
for large enough (using Lemma 10). Furthermore
Combining Eq. (99), Eq. (100) and Eq. (101), we get
By Markov’s inequality, we get for any and large enough,
The assumption and Lemma 8 conclude the proof.
Appendix G Proof of Theorem 7.(a): lower bound for NT model
Consider the expansion of in terms of product of Gegenbauer polynomials. We have
where the expectation is taken over .
with and , and
with the convention . Then there exists constants and such that for large enough, we have for any and ,
where we recall is the subset of indices corresponding to the non zero integers .
Let us fix an integer such that . We will denote for simplicity. Following the same proof as in Lemma 9, there exists , and such that for any and , we have for any ,
while for , we get
Injecting this bound in the formula (104) of , we get for , and any : if ,
where we used that for , there exists a constant such that and . Similarly, we get for
G.2 Proof of Theorem 7.(a): Outline
The structure of the proof for the NT model is the same as for the RF case, however some parts of the proof requires more work.
Proceeding as for the RF model, we obtain
To show this result, we will need the following two propositions.
where the expectation is taken with respect to . Then,
with block diagonal. Furthermore, and verifies the following properties:
For any , there exists constants such that we have with high probability
For any , we have
The proofs of these two propositions are provided in the next sections.
From Proposition 4, we can upper bound Eq. (106) as follows
Let us fix as prescribed in Lemma 11. We decompose the vector where
Hence, using the upper bounds on in Lemma 11, we get for with :
where we used that and (we have and therefore by definition). Similarly for with :
where we used that by definition of we have and . We deduce that
and therefore by Markov’s inequality that
Combining Eq. (111) and Eq. (113) yields Eq. (106). This proves the theorem.
G.3 Proof of Proposition 3
Let us consider as prescribed in Lemma 11. We have for
where we denoted the kernel given by
where is given in Lemma 12. Hence we get
Then, we have the following decomposition in terms of product of Gegenbauer polynomials,
with and , and
Recall the decomposition of in terms of tensor product of Gegenbauer polynomials,
Injecting this decomposition into the definition of yields
By the recurrence relationship for Gegenbauer polynomials (25), we have
where (we use the convention )
where we get by matching the coefficients,
with and . ∎
G.4 Proof of Proposition 4
where the coefficients are given recursively: denoting , if ,
where we recall the notations and .
We recall the following two formulas for (see Section B.2):
Furthermore, we have , and therefore therefore . Similarly to the proof of [GMMM19b, Lemma 6], we insert these expressions in the expansion of the function . Matching the coefficients of the expansion yields the result. ∎
We have the following lemma which is a generalization of [GMMM19b, Lemma 7], that shows essentially the same decomposition of the matrix as by integration by part if we had .
Denote . Let us rotate each sphere such that
Let us start with . For clarity, we will denote (in the rotated basis (115))
Then it is easy to show that we can rewrite
Case (a): .
where (we dropped the dependency on for clarity)
is invertible almost surely (for and ).
Case (b): .
Similarly, for some fixed and , we define
Step 2: for .
Case (a): .
where is given by
which is invertible almost surely (for and ).
Case (b): .
G.4.2 Proof of Proposition 4
Step 1. Construction of the activation function .
Recall the definition of in Eq. (102) and its expansion in terms of tensor product of Gegenbauer polynomials:
We recall the definition of . Let be two indices that satisfy the conditions of Assumption 3. and we define ( at position ) and ( at position ). Using the Gegenbauer coefficients of , we define a new activation function by
for some that we will fix later (with ).
Step 2. The functions and .
Let and be the matrix-valued functions associated respectively to and
From Lemma 14, there exists functions and (for ), which decompose and along and vectors. We define . Then we have the same decomposition for for .
Step 3. Construction of the kernel matrices.
Note that we have . By Eq. (122) and (120), it is easy to see that . Then we have . In the following, we would like to lower bound matrix .
Let us start with for . Denoting , we get, from Eq. (116),
We get similar expressions for with replaced by . Because we defined and by only modifying the -th and -th coefficients, we get
Recalling that only depend on and , and on , and , (Lemma 13), we get
where we used the convention if one of the coordinates verifies .
From Lemma 13, Lemma 19 and Lemma 20, we get for and :
while for and ,
From Lemma (26), we recall that the coefficients of the -th Gegenbauer polynomial satisfy
Plugging the estimates (130) and (134) into Eqs. (128) and (129), we obtain that
We deduce from (135), (126) and (136) that for ,
As a result, combining Eq. (137) with Eq. (123) in the expression of given in Lemma 14, we get
By the expression of given by (125), we conclude that
Step 5. Checking the properties of matrix .
By Lemma 14, we can express as a block matrix with
Let us first focus on the sphere. Using Eqs. (128) and (129) with the expressions (131) and (132), we get the following convergence in probability (using that concentrates on ),
where we denoted (where first appears in the definition of in Eq. (117), and till now are still not determined) and, similarly to the proof of [GMMM19b, Proposition 5] and letting , we have
Following the same reasoning as in [GMMM19b, Proposition 5], we can verify that under Assumption 3., we have and . We can therefore find such that , . Furthermore,
Similarly, we get for from Eqs. (128) and (129) with the expressions (130) (recalling that concentrates on ),
We deduce that for and ,
which finishes to prove properties (107) and (108).
Appendix H Proof of Theorem 7.(b): upper bound for NT model
with and , and
Then there exists constants and such that for large enough, we have for any ,
From Assumptions 3. and 3. and Lemma 19, there exists and such that for any and ,
Hence for , we get , and for , we get . Carefully injecting these bounds in Eq. (144) yields the lemma. ∎
H.2 Proof of Theorem 7.(b): outline
In this proof, we will consider sub-classes of functions corresponding to the NT model restricted to the -th sphere:
We define similarly the risk associated to this sub-model
where is defined in Equation (145).
From the proof of Theorem 7., we have a matching lower bound for .
Denote , such that for any . Furthermore, notice that by definition for any and ,
H.3 Proof of Theorem 8
Similarly to the proof of Theorem 6., we construct a limiting kernel which is used as a proxy to upper bound the risk.
We recall the decomposition of in terms of tensor product of Gegenbauer polynomials:
Following the same computations as in Lemma 12, we get
with and , and convention ,
H.3.2 Proof of Theorem 8
Let us assume that is contained in , i.e. .
We can expand the squared loss at as
The second term of the expansion (LABEL:eq:expansion_squared_loss_NT) around verifies
Let us consider the third term in the expansion (LABEL:eq:expansion_squared_loss_NT) around : the non diagonal term verifies
For and and , we have
Hence from Lemma 17 and for small enough, there exists such that for large enough
Combining Eq. (149), Eq. (150) and Eq. (151), we get
By Markov’s inequality, we get for any and large enough,
The assumption that and Lemma 8 conclude the proof.
Appendix I Proof of Theorem 4 in the main text
Note that can be regarded as a function in and , this implies that
It is easy to see that, when , we have
Step 3. Show that is independent of .
Appendix J Convergence of the Gegenbauer coefficients
In this section, we prove a string of lemmas that are used to show convergence of the Gegenbauer coefficients.
There exists constants such that for any ,
Let us first consider with . The are sub-exponential random variables with
From standard sub-exponential concentration inequality, we get
Hence, for , we have
In the case of , applying (154) with shows that
Combining the above bounds into (153) yields for ,
Notice that and we only need to consider . We conclude that for any , we have
We will denote in the rest of this section for . Notice in particular that where we recall that . Without loss of generality, we will assume that the (unique) maximum is attained on the first sphere, i.e. and for .
A simple calculation shows that as , and hence . Therefore for , we have
Recalling the definition of , with and . Hence for any , uniformly on , we have for and . Hence if we choose , there exists such that for sufficiently large and for any
Finally, for part , without loss of generality we will take so that . From part , there exists and such that
Consider and an arbitrary coupling between and . For any we can choose bounded continuous so that for any and ,
It is therefore sufficient to prove the claim for . Letting independently for each and independent of , we construct the coupling via
where we set for each . We thus have almost surely, hence the limit superior of Eq. (160) is by weak convergence bounded by for any arbitrary . Furthermore, noticing that uniformly on for , we have by bounded convergence
We further have . Hence, by bounded convergence,
Combining Eq. (160) with the coupling (161) and Eqs (162) and (163) yields the result. ∎
Consider the expansion of in terms of tensor product of Gegenbauer polynomials. We have
with the expectation taken over . We will need the following lemma, which is direct consequence of Rodrigues formula, to get the scaling of the Gegenbauer coefficents of .
where and
where we used the definition (36) of tensor product of Gegenbauer polynomials.
Consider the integration with respect to . Denote for ease of notations . We use the Rodrigues formula for the Gegenbauer polynomials (see Eq. (26)):
Iterating Eq. (167) over and Eq. (166) yield the desired formula (164).
which converges to when . We deduce that
J.2 Proof of convergence in probability of the Gegenbauer coefficients
Then for any , there exists and such that for any and ,
Recall with and . Hence, we have
We can apply Lemma 17 to the activation function . In particular part of the lemma implies that there exists such that for sufficiently large, we have for any ,
Then for any , there exists and such that for any and ,
Recall the correspondence (31) between Gegenbauer and Hermite polynomials. Note for any monomial , we can apply Lemma 17. to and find a coupling such that for any , there exists and
Using the asymptotic correspondence between Gegenbauer polynomials and Hermite polynomials (31)
and Eq. (173), we get for any , there exists such that for sufficiently large, we have for any ,
Appendix K Bound on the operator norm of Gegenbauer polynomials
For each , we consider where with
Then, defining , we have
For sufficiently large, there exists such that for any :
Hence, there exists constant , such that for large , we have
Recalling that , and , we deduce
Let us now consider . We will denote . Then it is easy to check (recall the diagonal elements of are equal to one) that for any
where denotes the Hadamard product, or entrywise product, . We recall the following inequality on the operator norm of Hadamard product of two matrices, with positive definite:
Furthermore, is finite and from Proposition 5, we directly get
Combining bounds (175) and (176) yields the result.
The proof follows closely the proof of the uniform case presented in [GMMM19b]. For completeness, we copy here the relevant lemmas.
Step 1. Bounding operator norm by moments.
Denote . We define for each , and . Then it is easy to check (recall the diagonal elements of are equal to one)
where denotes the Hadamard product, or entrywise product, . For any sequence of integers , we have
To prove the proposition, it suffices to show that for any sequence , we have
where we used that and are independent for .
We will denote for any , define for each
Similarly, we define associated to ,
To calculate these quantities, we will apply repeatedly the following identity, which is an immediate consequence of Eq. (23). For any distinct, we have
Throughout the proof, we will denote by constants that may depend on but not on . The value of these constants is allowed to change from line to line.
Step 2. The induced graph and equivalence of index sequences.
For any index sequence , we defined an undirected multigraph associated to index sequence . The vertex set is the set of distinct elements in . The edge set is formed as follows: for any we add an edge between and (with convention ). Notice that this could be a self-edge, or a repeated edge: will be –in general– a multigraph. We denote to be the number of vertices of , and to be the number of edges (counting multiplicities). In particular, for . We define
For any two index sequences , we say they are equivalent , if the two graphs and are isomorphic, i.e. there exists an edge-preserving bijection of their vertices (ignoring vertex labels). We denote the equivalent class of to be
We define the quotient set by
The following Lemma was proved in [GMMM19b, Proposition 3]
The following properties holds for all sufficiently large and :
For any equivalent index sequences , we have .
For any index sequence , we have .
For any index sequence , the degree of any vertex in must be even.
The number of equivalent classes .
Recall that denotes the number of distinct elements in . Then, for any , the number of elements in the corresponding equivalence class satisfies .
In view of property in the last lemma, given an equivalence class , we will write for the corresponding value.
It is easy to see that the outcome of this process is independent of the order in which we select vertices.
For the above skeletonization process, the following properties hold
If , then . That is, the skeletons of equivalent index sequences are equivalent.
For any , and , we have
For any , its skeleton is either formed by a single element, or an index sequence whose graph has the property that every vertex has degree greater or equal to .
Given an index sequence , we say is of type 1, if contains only one index. We say is of type 2 if is not empty (so that by Lemma 22, can only contain vertices with degree greater or equal to ). Denote the class of type 1 index sequence (respectively type 2 index sequence) by (respectively ). We also denote by , the set of equivalence classes of sequences in . This definition makes sense since the equivalence class of the skeleton of a sequence only depends on the equivalence class of the sequence itself.
Recall that is the number of vertices in , and is the number of edges in (which coincides with the length of ). We consider . Since for , every edge of must be at most a double edge. Indeed, if had multiplicity larger than in , neither nor could be deleted during the skeletonization process, contradicting the assumption that contains a single vertex. Therefore, we must have . According the Lemma 22., for every , we have
Note by Lemma 21., the number of elements in the equivalence class of is . Hence we get
Therefore, denoting ,
where in the last step we used Lemma 21 and the fact that for , for some .
We have the following simple lemma bounding , copied from [GMMM19b, Proposition 3]. This bound is useful when is a skeleton.
For any , there exists constants and depending uniquely on such that, for any , and any index sequence with , we have
Suppose , and denote to be the number of vertices in . We have, for a sequence , and each
Here holds by Lemma 22.; by Lemma 23, and the fact that , together by ; because ; by Lemma 22., implying that for , each vertex of has degree greater or equal to , so that (notice that for we can assume ). Finally, follows since , and the definition of implying .
Note by Lemma 21., the number of elements in equivalent class . Since depends only on the equivalence class of , we will write, with a slight abuse of notation . Notice that the number of equivalence classes with is upper bounded by the number multi-graphs with vertices and edges, which is at most . Denoting , we have
Define . We will assume hereafter that is selected such that
By calculus and condition (185), the function is maximized over at , whence
Using Eqs. (181) and (186), we have, for any satisfying Eq. (185), we have
Finally setting and , this yields
Appendix L Technical lemmas
We put here one technical lemma that is used in the proof of Theorem 7.(a).
For any , there exists such that we have with high probability
Then for any , we have
Let us show the result recursively on the integer . Note that the case is direct.
Assume that verifies Eq. (191). Denote
From the two by two blockmatrix inversion, we have:
For completeness, we reproduce in this section lemmas proven in [GMMM19b].
For any fixed , let be the -th Gegenbauer polynomial. We expand