When Do Neural Networks Outperform Kernel Methods?

Behrooz Ghorbani, Song Mei, Theodor Misiakiewicz, Andrea Montanari

Introduction

FNNN{\mathcal{F}}_{{\rm NN}}^{N} 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 FNNN{\mathcal{F}}_{{\rm NN}}^{N}, the random features [RR08] and the neural tangent [JGH18] classes

FRFN(W){\mathcal{F}}_{{\rm RF}}^{N}({\bm{W}}) and FNTN(W){\mathcal{F}}_{{\rm NT}}^{N}({\bm{W}}) are linear classes of functions, depending on the realization of the input-layer weights W=(wi)i≤N{\bm{W}}=({\bm{w}}_{i})_{i\leq N} (which are chosen randomly). The relation between NN and these two linear classes is given by the first-order Taylor expansion: f^NN(x;b+εa,W+εS)−f^NN(x;b,W)=εf^RF(x;a;W)+εf^NT(x;S(b);W)+O(ε2)\hat{f}_{{\rm NN}}({\bm{x}};{\bm{b}}+\varepsilon{\bm{a}},{\bm{W}}+\varepsilon{\bm{S}})-\hat{f}_{{\rm NN}}({\bm{x}};{\bm{b}},{\bm{W}})=\varepsilon\hat{f}_{{\rm RF}}({\bm{x}};{\bm{a}};{\bm{W}})+\varepsilon\hat{f}_{{\rm NT}}({\bm{x}};{\bm{S}}({\bm{b}});{\bm{W}})+O(\varepsilon^{2}), where S(b)=(bisi)i≤N{\bm{S}}({\bm{b}})=(b_{i}{\bm{s}}_{i})_{i\leq N}. A number of recent papers show that, if weights and SGD updates are suitably scaled, and the network is sufficiently wide (NN sufficiently large), then SGD converges to a function f^NN\hat{f}_{{\rm NN}} that is approximately in FRFN(W)+FNTN(W){\mathcal{F}}_{{\rm RF}}^{N}({\bm{W}})+{\mathcal{F}}_{{\rm NT}}^{N}({\bm{W}}), with W{\bm{W}} determined by the SGD initialization [JGH18, DZPS19, DLL+19, AZLS19, ZCZG18, OS20]. This was termed the ‘lazy regime’ in [COB19].

where cl( ⋅ ){\rm cl}(\,\cdot\,) denotes closure. From this point of view, RF{\rm RF} and NT{\rm NT} differ in that they correspond to slightly different choices of the kernel: hRF(x1,x2):=∫σ(⟨w,x1⟩)σ(⟨w,x2⟩)ν(dw)h_{{\rm RF}}({\bm{x}}_{1},{\bm{x}}_{2}):=\int\sigma(\langle{\bm{w}},{\bm{x}}_{1}\rangle)\sigma(\langle{\bm{w}},{\bm{x}}_{2}\rangle)\nu({\rm d}{\bm{w}}) versus hNT(x1,x2):=⟨x1,x2⟩∫σ′(wTx1)σ′(wTx2)ν(dw)h_{{\rm NT}}({\bm{x}}_{1},{\bm{x}}_{2}):=\langle{\bm{x}}_{1},{\bm{x}}_{2}\rangle\int\sigma^{\prime}({\bm{w}}^{{\mathsf{T}}}{\bm{x}}_{1})\sigma^{\prime}({\bm{w}}^{{\mathsf{T}}}{\bm{x}}_{2})\nu({\rm d}{\bm{w}}). Multi-layer fully-connected NN{\rm NN}s in the lazy regime can be viewed as randomized approximations to RKHS as well, with some changes in the kernel hh. This motivates analogous questions for H(h){\mathcal{H}}(h): can the performances of NN{\rm NN} 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 f∗f_{*} 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 f∗(x)=σ(⟨w∗,x⟩)f_{*}({\bm{x}})=\sigma(\langle{\bm{w}}_{*},{\bm{x}}\rangle), then training a neural network with one hidden neuron learns the target efficiently from approximately dlog⁡dd\log d samples [MBM18], while the corresponding RKHS has test error bounded away from zero for every sample size polynomial in dd [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: (1)(1) Target functions depending on low-dimensional projections; (2)(2) Approximately low-dimensional covariates.

As for the example of a single neuron f∗(x)=σ(⟨w∗,x⟩)f_{*}({\bm{x}})=\sigma(\langle{\bm{w}}_{*},{\bm{x}}\rangle), 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].

(2)(2) 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 f∗(x)f_{*}({\bm{x}}) to depend predominantly on the low-frequency components of image x{\bm{x}}, but the image x{\bm{x}} 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 z0,i{\bm{z}}_{0,i} the signal covariates, z1,i{\bm{z}}_{1,i} the noise covariates, and rr the covariates signal-to-noise ratio (or covariates SNR). We will take r>1r>1, so that the variance of the signal covariates z0,i{\bm{z}}_{0,i} is larger than that of the noise covariates z1,i{\bm{z}}_{1,i}. In high dimension, this model is –for many purposes– similar to an anisotropic Gaussian model xi∼N(0,(r2−1)UUT+I){\bm{x}}_{i}\sim{\sf N}(0,(r^{2}-1){\bm{U}}{\bm{U}}^{{\mathsf{T}}}+{\mathbf{I}}). As shown below, the effect of anisotropy on RKHS methods is significant only if the covariate SNR rr is polynomially large in dd. We shall therefore set r=dκ/2r=d^{\kappa/2} for a constant κ>0\kappa>0.

We will consider a more general model in Appendix C, in which the distribution of xi{\bm{x}}_{i} takes a more general product-of-uniforms form, and we assume a general f∗∈L2f_{*}\in L^{2}.

2 A sharp characterization of RKHS methods

Any RKHS method with kernel hh outputs a model of the form f^(x;a)=∑i≤naih(⟨x,xi⟩/d)\hat{f}({\bm{x}};{\bm{a}})=\sum_{i\leq n}a_{i}h(\langle{\bm{x}},{\bm{x}}_{i}\rangle/d), with RKHS norm given by ∥f^( ⋅ ;a)∥h2=∑i,j≤nh(⟨xi,xj⟩/d)aiaj\|\hat{f}(\,\cdot\,;{\bm{a}})\|_{h}^{2}=\sum_{i,j\leq n}h(\langle{\bm{x}}_{i},{\bm{x}}_{j}\rangle/d)a_{i}a_{j}. We consider kernel ridge regression (KRR) on the dataset {(yi,xi)}i≤n\{(y_{i},{\bm{x}}_{i})\}_{i\leq n} with regularization parameter λ\lambda, namely:

where H=(Hij)ij∈[n]{\bm{H}}=(H_{ij})_{ij\in[n]}, with Hij=h(⟨xi,xj⟩/d)H_{ij}=h(\langle{\bm{x}}_{i},{\bm{x}}_{j}\rangle/d). We denote the prediction error of KRR by

where h(x)=(h(⟨x,x1⟩/d),…,h(⟨x,xn⟩/d))T{\bm{h}}({\bm{x}})=(h(\langle{\bm{x}},{\bm{x}}_{1}\rangle/d),\ldots,h(\langle{\bm{x}},{\bm{x}}_{n}\rangle/d))^{\mathsf{T}}.

Recall that we assume the target function f∗(xi)=φ(UTxi)f_{*}({\bm{x}}_{i})=\varphi({\bm{U}}^{{\mathsf{T}}}{\bm{x}}_{i}). We denote P≤k:L2→L2{\mathsf{P}}_{\leq k}:L^{2}\to L^{2} to be the projection operator onto the space of degree kk orthogonal polynomials, and P>k=I−P≤k{\mathsf{P}}_{>k}={\mathbf{I}}-{\mathsf{P}}_{\leq k}. 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’, d\mboxeffd_{\mbox{\tiny\rm eff}}.

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 NN? In order to simplify the picture, we focus here on the approximation error. Equivalently, we assume the sample size to be n=∞n=\infty and consider the minimum population risk for M∈{RF,NT}{\sf M}\in\{{\rm RF},{\rm NT}\}

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 RM,N,n(f∗;W)R_{{\sf M},N,n}(f_{*};{\bm{W}}) the corresponding test error (assuming for instance ridge regression, with the optimal regularization λ\lambda). Of course the minimum population risk provides a lower bound: RM,N,n(f∗;W)≥RM,N(f∗;W)R_{{\sf M},N,n}(f_{*};{\bm{W}})\geq R_{{\sf M},N}(f_{*};{\bm{W}}). Moreover, we conjecture that the risk is minimized at infinite NN, RM,N,n(f∗;W)≳Rn(f∗;hM)R_{{\sf M},N,n}(f_{*};{\bm{W}})\gtrsim R_{n}(f_{*};h_{{\sf M}}). Altogether this implies the lower bound RM,N,n(f∗;W)≳max⁡(RM,N(f∗;W),Rn(f∗;hM))R_{{\sf M},N,n}(f_{*};{\bm{W}})\gtrsim\max(R_{{\sf M},N}(f_{*};{\bm{W}}),R_{n}(f_{*};h_{{\sf M}})). We also conjecture that this lower bound is tight, up to terms vanishing as N,n,d→∞N,n,d\to\infty.

4 Neural network models

Moreover, the quantity RNN,N(f∗)R_{{\rm NN},N}(f_{*}) is independent of κ≥0\kappa\geq 0.

As a consequence of Theorem 3 and 4, there is a separation between NN and (uniformly sampled) NT models when d\mboxeff≠d0d_{\mbox{\tiny\rm eff}}\neq d_{0}, i.e., κ<1−η\kappa<1-\eta. As κ\kappa increases, the gap between NN and NT becomes smaller and smaller until κ=1−η\kappa=1-\eta.

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 nn, dimension dd, and width NN. 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 22 or 33 layers (and N=4096N=4096 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: dd is fixed, while the sample size nn diverges. This is probably unrealistic for many machine learning applications, in which dd 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 dd and nn diverge, while being polynomially related. This characterization holds for any target function f∗f_{*}, 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 d0d_{0}, ambient dimension dd, and the covariate signal-to-noise ratio rr, the model presents a continuum of different behaviors. At one extreme, the covariates are fully dd-dimensional, and RKHS methods are highly suboptimal compared to NN. At the other, covariates are close to d0d_{0}-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 x{\bm{x}} 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 lr0=10−3lr_{0}=10^{-3} and T=750T=750 is the total number of training epochs. To ensure the stability of the optimization for wide models, we use 1515 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 lr0=10−5lr_{0}=10^{-5}. The batch-size is fixed at 10410^{4} to encourage fast convergence to the minimum.

For NN{\rm NN}, RF{\rm RF} and NT{\rm NT}, 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 ui{\bm{u}}_{i} and zi{\bm{z}}_{i} are drawn i.i.d from the hyper-spheres with radii rd0r\sqrt{d_{0}} and d\sqrt{d} respectively. We choose

where dd is fixed to be 10241024 and η=25\eta=\frac{2}{5}. We change κ\kappa in the interval {0,…,0.9}\{0,…,0.9\}. For each value of κ\kappa we generate 2202^{20} training and 10410^{4} test observations.Strictly speaking, the model outlined in the main text requires zi{\bm{z}}_{i} to be generated from the hyper-sphere of radius d−d0\sqrt{d-d_{0}}. In order to work with round numbers, in our experiments we use d\sqrt{d} instead of d−d0\sqrt{d-d_{0}}. The numerical difference between these two choices is negligible.

The function φ\varphi is the sum of three orthogonal components {φi}i=13\{\varphi_{i}\}_{i=1}^{3} with ∥φi∥2=1\|\varphi_{i}\|_{2}=1. To be more specific,

This choice of φi\varphi_{i} guarantees that each φi\varphi_{i} is in the span of degree i+1i+1 spherical harmonics.

In Figure 3 of the main text, we compared the generalization performance of NTK KRR with NN{\rm NN}. We use the same training and test data as above to perform this analysis. The number of training data points, nn, takes 2424 different values ranging from 5050 to 10510^{5}. The number of test data points is always fixed at 10410^{4}.

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 xnoisy{\bm{x}}_{noisy} so that it has norm d\sqrt{d}.

This choice of F{\bm{F}} 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 R0=0.9R_{0}=0.9 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 10410^{4} training and 20002000 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 d=1024d=1024.

Figure A.5 demonstrates the evolution of the model performances as the noise intensity increases. In the noiseless regime (τ=0\tau=0), all models have comparable performances. However, as the noise level increases, the performance gap between NN{\rm NN} and RKHS methods widens. For reference, the accuracy gap between NN{\rm NN} and NT{\rm NT} KRR is only 0.6%0.6\% at τ=0\tau=0. However, at τ=3\tau=3, this gap increases to 4.5%4.5\%. The normalization factor R0=0.25R_{0}=0.25 corresponds to the risk achievable by the trivial estimator y^(x)=0.5\hat{y}({\bm{x}})=0.5.

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 d\sqrt{d}.

Let Dtrain\mathcal{D}_{train} denote the set of training images in the DCT-frequency domain. We compute the mean μ\mu and the covariance Σ\Sigma of the elements of Dtrain\mathcal{D}_{train}.

The fraction of the frequencies replaced by noise is α2/k2\alpha^{2}/k^{2}.

For neural networks trained for this experiment, we fix the number of hidden units per-layer to N=4096N=4096. This corresponds to approximately 3.2×1063.2\times 10^{6} trainable parameters for two-layer networks and 2×1072\times 10^{7} trainable parameters for three-layer networks. Both models are trained using SGD with momentum with learning rate described by \eqrefeqn:cosinerule\eqref{eqn:cosine_rule} (with lr0=10−3lr_{0}=10^{-3}). For the warm-up epochs, we use batch-size of 500500. We increase the batch-size to 10001000 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 Dtrain\mathcal{D}_{train} denote the set of training images in the DCT-frequency domain. Note that CIFAR-10 images have 33 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 μ\mu and the covariance Σ\Sigma of the elements of Dtrain\mathcal{D}_{train}.

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 N=512N=512. This corresponds to approximately 7×1067\times 10^{6} parameters. Similar to the fully-connected networks, our convolutional models are also optimized via SGD with 0.90.9 momentum (learning rate evolves as (11) with lr0=0.1lr_{0}=0.1 and T=70T=70). We fix the batch-size to 128128. 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– Qk(d)(⟨x,y⟩)Q_{k}^{(d)}(\langle{\bm{x}},{\bm{y}}\rangle) is a representation of the projector onto the subspace of degree -kk spherical harmonics

B.3 Hermite polynomials

Notice that for functions gg that are kk-weakly differentiable with g(k)g^{(k)} the kk-th weak derivative, we have

Here and below, for PP a polynomial, Coeff{P(x)}{\rm Coeff}\{P(x)\} is the vector of the coefficients of PP.

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 Pk{\mathsf{P}}_{{\bm{k}}} the orthogonal projections on VkdV^{{\bm{d}}}_{{\bm{k}}} in L2(PSd,μd)L^{2}({\rm PS}^{{\bm{d}}},\mu_{{\bm{d}}}). 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 QkdQ_{{\bm{k}}}^{{\bm{d}}} is (up to a constant) a representation of the projector onto the subspace VkdV^{{\bm{d}}}_{{\bm{k}}}

Part (a)(a) comes from the normalization property (22) of Gegenbauer polynomials,

while part (c)(c) is a direct consequence of Eq. (24).

B.6 Notations

Furthermore, f=ωd(g)f=\omega_{d}(g) will denote f(d)/g(d)→∞f(d)/g(d)\to\infty.

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 L2L^{2} 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 x{\bm{x}} lies on the product of QQ spheres,

where dq=dηqd_{q}=d^{\eta_{q}} and rq=d(ηq+κq)/2r_{q}=d^{(\eta_{q}+\kappa_{q})/2}. Let d=(d1,…,dq)=(dη1,…,dηq){\bm{d}}=(d_{1},\ldots,d_{q})=(d^{\eta_{1}},\ldots,d^{\eta_{q}}) and κ=(κ1,…,κQ){\bm{\kappa}}=(\kappa_{1},\ldots,\kappa_{Q}), where ηq>0\eta_{q}>0 and κq≥0\kappa_{q}\geq 0 for q=1,…,Qq=1,\ldots,Q. We will denote this space

Furthermore, assume that the data is generated following the uniform distribution on PSκd{\rm PS}^{\bm{d}}_{{\bm{\kappa}}}, i.e.

We will make the following assumption that will simplify the proofs. Denote

then ξ\xi is attained on only one of the sphere, whose coordinate will be denoted qξq_{\xi}, i.e. ξ=ηqξ+κqξ\xi=\eta_{q_{\xi}}+\kappa_{q_{\xi}} and ηq+κq<ξ\eta_{q}+\kappa_{q}<\xi for q≠qξq\neq q_{\xi}.

We will denote θi=Dwi{\bm{\theta}}_{i}=\sqrt{D}{\bm{w}}_{i}. Notice that the normalization in the definition of the function class insures that the scalar product ⟨x,θi⟩/R\langle{\bm{x}},{\bm{\theta}}_{i}\rangle/R is of order 11. This corresponds to normalizing the data.

We consider the approximation of ff by functions in function classes FRF(Θ){\mathcal{F}}_{{\rm RF}}({\bm{\Theta}}) and FNT(Θ){\mathcal{F}}_{{\rm NT}}({\bm{\Theta}}).

C.2 Reparametrization

It is easy to check that the variables (θ‾(1),…,θ‾(Q))(\overline{\bm{\theta}}^{(1)},\ldots,\overline{\bm{\theta}}^{(Q)}) are independent and independent of (τi(1),…,τi(Q))(\tau_{i}^{(1)},\ldots,\tau_{i}^{(Q)}), and verify

We will denote θ‾i≡(θ‾i(1),…,θ‾i(Q))\overline{\bm{\theta}}_{i}\equiv(\overline{\bm{\theta}}^{(1)}_{i},\ldots,\overline{\bm{\theta}}^{(Q)}_{i}) and τi≡(τi(1),…,τi(Q)){\bm{\tau}}_{i}\equiv(\tau^{(1)}_{i},\ldots,\tau^{(Q)}_{i}). With these notations, we have

where PSd{\rm PS}^{\bm{d}} is the ‘normalized space of product of spheres’, and

Similarly, we will denote the rescaled data x‾∈PSd\overline{\bm{x}}\in{\rm PS}^{\bm{d}},

obtained by taking x‾(q)=dqx(q)/rq=d−κq/2x(q)\overline{\bm{x}}^{(q)}=\sqrt{d_{q}}{\bm{x}}^{(q)}/r_{q}=d^{-\kappa_{q}/2}{\bm{x}}^{(q)} for each q∈[Q]q\in[Q].

The proof will proceed as follows: first, noticing that τ(q)\tau^{(q)} concentrates around 11 for every q=1,…,Qq=1,\ldots,Q, we will restrict ourselves without loss of generality to the following high probability event

where ε>0\varepsilon>0 will be chosen sufficiently small. Then, we rewrite the activation function

as a function, for a random τ{\bm{\tau}} (but close to (1,…,1)(1,\ldots,1))

given for θ=(θ‾,τ){\bm{\theta}}=(\overline{\bm{\theta}},{\bm{\tau}}) 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 d=(d1,…,dq){\bm{d}}=(d_{1},\ldots,d_{q}), κ=(κ1,…,κQ){\bm{\kappa}}=(\kappa_{1},\ldots,\kappa_{Q}), dq=dηqd_{q}=d^{\eta_{q}}, rq=d(ηq+κq)/2r_{q}=d^{(\eta_{q}+\kappa_{q})/2}, D=dη1+…+dηQD=d^{\eta_{1}}+\ldots+d^{\eta_{Q}} and R=(dη1+κ1+…+dηQ+κQ)1/2R=(d^{\eta_{1}+\kappa_{1}}+\ldots+d^{\eta_{Q}+\kappa_{Q}})^{1/2}. Let us denote ξ=max⁡q∈[Q]{ηq+κq}\xi=\max_{q\in[Q]}\{\eta_{q}+\kappa_{q}\} and qξ=arg⁡min⁡q∈[Q]{ηq+κq}q_{\xi}=\arg\min_{q\in[Q]}\{\eta_{q}+\kappa_{q}\}.

C.4 Generalization error of kernel ridge regression

We consider the Kernel Ridge Regression solution a^i\hat{a}_{i}, namely

where the kernel matrix H=(Hij)ij∈[n]{\bm{H}}=(H_{ij})_{ij\in[n]} is assumed to be given by

and y=(y1,…,yn)T=f+ε{\bm{y}}=(y_{1},\ldots,y_{n})^{\mathsf{T}}={\bm{f}}+{\bm{\varepsilon}}, with

The prediction function at location x{\bm{x}} gives

The test error of empirical kernel ridge regression is defined as

Notice that by definition m(γ)>γm(\gamma)>\gamma.

We consider sequences of problems indexed by the integer dd, and we view the problem parameters (in particular, the dimensions dqd_{q}, the radii rqr_{q}, the kernel hdh_{d}, and so on) as functions of dd.

For γ>0\gamma>0 (which is specified in the theorem), we denote L=max⁡q∈[Q]⌈γ/ηq⌉L=\max_{q\in[Q]}\lceil\gamma/\eta_{q}\rceil. We assume that hdh_{d} is LL-weakly differentiable. We assume that for 0≤k≤L0\leq k\leq L, the kk-th weak derivative verifies almost surely hd(k)(u)≤Ch_{d}^{(k)}(u)\leq C for some constants C>0C>0 independent of dd. Furthermore, we assume there exists k>Lk>L such that hd(k)(0)≥c>0h_{d}^{(k)}(0)\geq c>0 with cc independent of dd.

For γ>0\gamma>0 (which is specified in the theorem), we define

We assume that σ\sigma verifies for k≤K‾k\leq\overline{K}, hd(k)(0)≥ch_{d}^{(k)}(0)\geq c, with c>0c>0 independent of dd.

Let {fd∈L2(PSκd,μdκ)}d≥1\{f_{d}\in L^{2}({\rm PS}^{\bm{d}}_{\bm{\kappa}},\mu_{{\bm{d}}}^{\bm{\kappa}})\}_{d\geq 1} be a sequence of functions. Assume wd(dγlog⁡d)≤n≤Od(dm(γ)−δ)w_{d}(d^{\gamma}\log d)\leq n\leq O_{d}(d^{m(\gamma)-\delta}) for some γ>0\gamma>0 and δ>0\delta>0. Let {hd}d≥1\{h_{d}\}_{d\geq 1} be a sequence of functions that satisfies Assumption 1 at level γ\gamma. Let X=(xi)i∈[n]{\bm{X}}=({\bm{x}}_{i})_{i\in[n]} with (xi)i∈[n]∼Unif(PSκd)({\bm{x}}_{i})_{i\in[n]}\sim{\rm Unif}({\rm PS}^{\bm{d}}_{{\bm{\kappa}}}) independently, and yi=fd(xi)+εiy_{i}=f_{d}({\bm{x}}_{i})+\varepsilon_{i} and εi∼iidN(0,τ2)\varepsilon_{i}\sim_{iid}{\sf N}(0,\tau^{2}) for some τ2≥0\tau^{2}\geq 0. Then for any ε>0\varepsilon>0, and for any λ=Od(1)\lambda=O_{d}(1), 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 γ>0\gamma>0 (which is specified in the theorem), we denote L=max⁡q∈[Q]⌈γ/ηq⌉L=\max_{q\in[Q]}\lceil\gamma/\eta_{q}\rceil. We assume that σ\sigma is LL-weakly differentiable. Define

We assume that for K≤k≤LK\leq k\leq L, the kk-th weak derivative verifies almost surely σ(k)(u)2≤c0exp⁡(c1u2/2)\sigma^{(k)}(u)^{2}\leq c_{0}\exp(c_{1}u^{2}/2) for some constants c0>0c_{0}>0 and c1<1c_{1}<1.

Furthermore we will assume that σ\sigma is not a degree-⌊γ/ηqξ⌋\lfloor\gamma/\eta_{q_{\xi}}\rfloor polynomial where we recall that qξq_{\xi} corresponds to the unique arg⁡min⁡q∈[Q]{ηq+κq}\arg\min_{q\in[Q]}\{\eta_{q}+\kappa_{q}\}.

For γ>0\gamma>0 (which is specified in the theorem), we define

We assume that σ\sigma verifies for k≤K‾k\leq\overline{K}, μk(σ)≠0\mu_{k}(\sigma)\neq 0. Furthermore we assume that for k≤K‾k\leq\overline{K}, the kk-th weak derivative verifies almost surely σ(k)(u)2≤c0exp⁡(c1u2/2)\sigma^{(k)}(u)^{2}\leq c_{0}\exp(c_{1}u^{2}/2) for some constants c0>0c_{0}>0 and c1<1c_{1}<1.

Assume N≤od(dγ)N\leq o_{d}(d^{\gamma}) for a fixed γ>0\gamma>0. Let σ\sigma satisfy Assumptions 2.(a) and 2.(b) at level γ\gamma. Then, for any ε>0\varepsilon>0, the following holds with high probability:

where Q≡QRF(γ){\mathcal{Q}}\equiv{\mathcal{Q}}_{\rm RF}(\gamma) is defined in Equation (45).

Assume N≥wd(dγ)N\geq w_{d}(d^{\gamma}) for some positive constant γ>0\gamma>0, and σ\sigma satisfy Assumptions 2.(a) and 2.(c) at level γ\gamma. Then for any ε>0\varepsilon>0, the following holds with high probability:

where Q≡Q‾RF(γ){\mathcal{Q}}\equiv\overline{{\mathcal{Q}}}_{\rm RF}(\gamma) 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 N=dγN=d^{\gamma}, RF{\rm RF} model fits the subspace of low degree polynomials F(β,κ,γ){\mathcal{F}}({\bm{\beta}},{\bm{\kappa}},\gamma) and cannot fit Fc(β,κ,γ){\mathcal{F}}^{c}({\bm{\beta}},{\bm{\kappa}},\gamma), i.e.

Each subspace has therefore an effective dimension dq,eff≡dξ−κq=dq(ξ−κq)/ηq≍D(ξ−κq)/max⁡q∈[Q]ηqd_{q,{\rm eff}}\equiv d^{\xi-\kappa_{q}}=d_{q}^{(\xi-\kappa_{q})/\eta_{q}}\asymp D^{(\xi-\kappa_{q})/\max_{q\in[Q]}\eta_{q}}. 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 γ>0\gamma>0 (which is specified in the theorem), we denote L=max⁡q∈[Q]⌈γ/ηq⌉L=\max_{q\in[Q]}\lceil\gamma/\eta_{q}\rceil. We assume that σ′\sigma^{\prime} is LL-weakly differentiable. Define

We assume that for K−1≤k≤LK-1\leq k\leq L, the kk-th weak derivative verifies almost surely σ(k+1)(u)2≤c0exp⁡(c1u2/2)\sigma^{(k+1)}(u)^{2}\leq c_{0}\exp(c_{1}u^{2}/2) for some constants c0>0c_{0}>0 and c1<1c_{1}<1.

For γ>0\gamma>0 (which is specified in the theorem), we define

We assume that σ\sigma verifies for k≤K‾+1k\leq\overline{K}+1, μk(σ′)=μk+1(σ)≠0\mu_{k}(\sigma^{\prime})=\mu_{k+1}(\sigma)\neq 0. Furthermore we assume that for k≤K‾+1k\leq\overline{K}+1 , the kk-th weak derivative verifies almost surely σ(k+1)(u)2≤c0exp⁡(c1u2/2)\sigma^{(k+1)}(u)^{2}\leq c_{0}\exp(c_{1}u^{2}/2) for some constants c0>0c_{0}>0 and c1<1c_{1}<1.

In the Assumption 3.(b), it is useful to notice that the Hermite coefficients of x2σ′(x)x^{2}\sigma^{\prime}(x) can be computed from the ones of σ′(x)\sigma^{\prime}(x) using the relation μk(x2σ′)=μk+2(σ′)+[1+2k]μk(σ′)+k(k−1)μk−2(σ′)\mu_{k}(x^{2}\sigma^{\prime})=\mu_{k+2}(\sigma^{\prime})+[1+2k]\mu_{k}(\sigma^{\prime})+k(k-1)\mu_{k-2}(\sigma^{\prime}).

Assume N≤od(dγ)N\leq o_{d}(d^{\gamma}) for a fixed γ>0\gamma>0. Let σ\sigma satisfy Assumptions 3.(a) and 3.(b) at level γ\gamma. Then, for any ε>0\varepsilon>0, the following holds with high probability:

where Q≡QNT(γ){\mathcal{Q}}\equiv{\mathcal{Q}}_{\rm NT}(\gamma) is defined in Equation (50).

Assume N≥wd(dγ)N\geq w_{d}(d^{\gamma}) for some positive constant γ>0\gamma>0, and σ\sigma satisfy Assumptions 3.(a) and 3.(c) at level γ\gamma. Then for any ε>0\varepsilon>0, the following holds with high probability:

where Q≡Q‾NT(γ){\mathcal{Q}}\equiv\overline{{\mathcal{Q}}}_{\rm NT}(\gamma) 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 γ>0\gamma>0 such that QNT(γ)c∩Q‾NT(γ)=∅{\mathcal{Q}}_{{\rm NT}}(\gamma)^{c}\cap\overline{{\mathcal{Q}}}_{{\rm NT}}(\gamma)=\emptyset, we can decompose our functional space as

such that for N=dγN=d^{\gamma}, NT{\rm NT} model fits the subspace of low degree polynomials F(β,κ,γ){\mathcal{F}}({\bm{\beta}},{\bm{\kappa}},\gamma) and cannot fit Fc(β,κ,γ){\mathcal{F}}^{c}({\bm{\beta}},{\bm{\kappa}},\gamma) at all, i.e.

where β=ξ−min⁡q∈S(k)κq\beta=\xi-\min_{q\in S({\bm{k}})}\kappa_{q}.

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 η1=η\eta_{1}=\eta, κ1=κ\kappa_{1}=\kappa for the first sphere, and η2=1\eta_{2}=1, κ2=0\kappa_{2}=0 for the second sphere. We have ξ=max⁡(η+κ,1)\xi=\max(\eta+\kappa,1).

Let wd(dγlog⁡d)≤n≤Od(dγ+δ)w_{d}(d^{\gamma}\log d)\leq n\leq O_{d}(d^{\gamma+\delta}) with δ>0\delta>0 constant sufficiently small, then by Theorem 5 the function subspace learned by KRR is given by the polynomials of degree k1k_{1} in the first sphere coordinates and k2k_{2} in the second sphere with

We consider functions that only depend on the first sphere, i.e., k2=0k_{2}=0 and denote deff=dmax⁡(η,1−κ)d_{\rm eff}=d^{\max(\eta,1-\kappa)}. Then the subspace of approximation is given by the kk polynomials in the first sphere such that deffk≤dγd_{\rm eff}^{k}\leq d^{\gamma}. Furthermore, one can check that the Assumptions listed in Theorem 1 in the main text verifies Assumption 1.

Similarly, for wd(dγ)≤N≤Od(dγ+δ)w_{d}(d^{\gamma})\leq N\leq O_{d}(d^{\gamma+\delta}) with δ>0\delta>0 constant sufficiently small, Theorem 6 implies that the RF{\rm RF} models can only approximate kk polynomials in the first sphere such that deffk≤dγd_{\rm eff}^{k}\leq d^{\gamma}. Furthermore, Assumptions listed in Theorem 2 in the main text verifies Assumption 2.

In the case of NT{\rm NT}, we only consider k=(k1,0){\bm{k}}=(k_{1},0) and S(k)={1}S({\bm{k}})=\{1\}. We get min⁡q∈S(k)κq=κ\min_{q\in S({\bm{k}})}\kappa_{q}=\kappa. The subspace approximated is given by the kk polynomials in the first sphere such that deffk≤dγdeffd_{\rm eff}^{k}\leq d^{\gamma}d_{\rm eff}. 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 {hd}d≥1\{h_{d}\}_{d\geq 1} as functions on the product of normalized spheres: for x={x(q)}q∈[Q]{\bm{x}}=\{{\bm{x}}^{(q)}\}_{q\in[Q]} and y={y(q)}q∈[Q]∈PSκd{\bm{y}}=\{{\bm{y}}^{(q)}\}_{q\in[Q]}\in{\rm PS}^{\bm{d}}_{\bm{\kappa}}:

Consider the expansion of hdh_{{\bm{d}}} in terms of tensor product of Gegenbauer polynomials. We have

where the expectation is taken over x‾=(x‾(1),…,x‾(Q))∼μd\overline{\bm{x}}=(\overline{\bm{x}}^{(1)},\ldots,\overline{\bm{x}}^{(Q)})\sim\mu_{{\bm{d}}}.

Let {hd}d≥1\{h_{d}\}_{d\geq 1} be a sequence of kernel functions that satisfies Assumption 1. Assume wd(dγ)≤n≤od(dm(γ))w_{d}(d^{\gamma})\leq n\leq o_{d}(d^{m(\gamma)}) for some γ>0\gamma>0. Consider Q=Q‾KRR(γ){\mathcal{Q}}=\overline{{\mathcal{Q}}}_{{\rm KRR}}(\gamma) as defined in Eq. (43). Then there exists constants c,C>0c,C>0 such that for dd large enough,

where αq=dq−1/2rq2/R2=(1+od(1))dηq/2+κq−ξ\alpha_{q}=d_{q}^{-1/2}r_{q}^{2}/R^{2}=(1+o_{d}(1))d^{\eta_{q}/2+\kappa_{q}-\xi}. By Assumption 1.(a)(a), we have

Furthermore, by Assumption 1.(b)(b) and dominated convergence,

for k≥K‾k\geq\overline{K}. 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 hdh_{\bm{d}}. ∎

D.2 Proof of Theorem 5

Step 1. Rewrite the y{\bm{y}}, E{\bm{E}}, H{\bm{H}}, M{\bm{M}} matrices.

The test error of empirical kernel ridge regression gives

where E=(E1,…,En)T{\bm{E}}=(E_{1},\ldots,E_{n})^{\mathsf{T}} and M=(Mij)ij∈[n]{\bm{M}}=(M_{ij})_{ij\in[n]}, with

Let the spherical harmonics decomposition of fdf_{d} be

and the Gegenbauer decomposition of hdh_{{\bm{d}}} be

We write the decompositions of vectors f{\bm{f}}, E{\bm{E}}, H{\bm{H}}, and M{\bm{M}}. We have

The rest of the proof follows closely from [GMMM19b, Theorem 4]. We decompose the risk as follows

Further, we denote fQ{\bm{f}}_{{\mathcal{Q}}}, fQc{\bm{f}}_{{\mathcal{Q}}^{c}}, EQ{\bm{E}}_{{\mathcal{Q}}}, and EQc{\bm{E}}_{{\mathcal{Q}}^{c}},

Using Cauchy Schwarz inequality for T22T_{22}, we get

As a result, combining Eqs. (58), (60) and (59), we have

where the last equality used the fact that n≤Od(dm(γ)−δ)n\leq O_{d}(d^{m(\gamma)-\delta}) and Lemma 2. Combining Eqs. (62), (63) and (64), we get

Step 5. Terms T3,T4T_{3},T_{4} and T5T_{5}. By Lemma 6 again, we have

We decompose T5T_{5} using f=fQ+fQc{\bm{f}}={\bm{f}}_{{\mathcal{Q}}}+{\bm{f}}_{{\mathcal{Q}}^{c}},

Combining Eqs. (65), (61), (66), (67) and (68), we have

D.3 Auxiliary results

Assume that n≥wd(dγlog⁡d)n\geq w_{d}(d^{\gamma}\log d) and consider

Denote A=∑k∈RB(d,k)A=\sum_{{\bm{k}}\in{\mathcal{R}}}B({\bm{d}},{\bm{k}}) and

Notice that there exists C>0C>0 such that A≤Cmax⁡k∈R∏q∈[Q]dηqkq≤CdγA\leq C\max_{{\bm{k}}\in{\mathcal{R}}}\prod_{q\in[Q]}d^{\eta_{q}k_{q}}\leq Cd^{\gamma} (by definition of m(γ)m(\gamma) and R{\mathcal{R}}) and therefore n≥wd(Alog⁡A)n\geq w_{d}(A\log A). Integrating the tail bound (69) proves the lemma. ∎

Let σ\sigma be an activation function satisfying Assumption 1. Let wd(dγlog⁡d)≤n≤Od(dm(γ)−δ)w_{d}(d^{\gamma}\log d)\leq n\leq O_{d}(d^{m(\gamma)-\delta}) for some γ>0\gamma>0 and δ>0\delta>0. Then there exists sequences κh\kappa_{h} and κu\kappa_{u} such that

with κh=∑k∈Sλkd(hd)B(d,k)=Od(1)\kappa_{h}=\sum_{{\bm{k}}\in{\mathcal{S}}}\lambda^{\bm{d}}_{{\bm{k}}}(h_{{\bm{d}}})B({\bm{d}},{\bm{k}})=O_{d}(1). From Assumption 1.(b)(b) and a proof similar to Lemma 20, there exists k=(0,…,k,…,0){\bm{k}}=(0,\ldots,k,\ldots,0) (for k>Lk>L at position qξq_{\xi}) such that lim⁡inf⁡d→∞λkd(hd)B(d,k)>0\lim\inf_{d\to\infty}\lambda^{\bm{d}}_{{\bm{k}}}(h_{{\bm{d}}})B({\bm{d}},{\bm{k}})>0. Hence, κh=Θd(1)\kappa_{h}=\Theta_{d}(1).

From Lemma 3 we have for k∈R∩Qc{\bm{k}}\in{\mathcal{R}}\cap{\mathcal{Q}}^{c},

Let σ\sigma be an activation function satisfying Assumption 1. Assume ωd(dγlog⁡d)≤n≤Od(dm(γ)−δ)\omega_{d}(d^{\gamma}\log d)\leq n\leq O_{d}(d^{m(\gamma)-\delta}) for some γ>0\gamma>0 and δ>0\delta>0. We have

Denote B=∑k∈QB(d,k)B=\sum_{{\bm{k}}\in{\mathcal{Q}}}B({\bm{d}},{\bm{k}}), and

Then, we can use the same proof as in [GMMM19b, Lemma 13] to bound ∥T1∥op\|T_{1}\|_{{\rm op}} (recall n=Od(dm(γ)−δ)n=O_{d}(d^{m(\gamma)-\delta}))

Let σ\sigma be an activation function satisfying Assumption 1. Assume ωd(dγlog⁡d)≤n≤Od(dm(γ)−δ)\omega_{d}(d^{\gamma}\log d)\leq n\leq O_{d}(d^{m(\gamma)-\delta}) for some γ>0\gamma>0 and δ>0\delta>0. 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 Θ{\bm{\Theta}}. Hence, in the proof we will restrict the sample space to the high probability event Pε≡Pd,N,ε\mathcal{P}_{\varepsilon}\equiv\mathcal{P}_{d,N,\varepsilon} for ε>0\varepsilon>0 small enough, where

Assume N=o(dγ)N=o(d^{\gamma}) for some γ>0\gamma>0. We have for any fixed ε>0\varepsilon>0,

The tail inequality in Lemma 16 and the assumption N=o(dγ)N=o(d^{\gamma}) imply that there exists some constants C,c>0C,c>0 such that

Consider the expansion of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}} in terms of tensor product of Gegenbauer polynomials. We have

where the expectation is taken over x‾=(x‾(1),…,x‾(Q))∼μd\overline{\bm{x}}=(\overline{\bm{x}}^{(1)},\ldots,\overline{\bm{x}}^{(Q)})\sim\mu_{{\bm{d}}}.

Let σ\sigma be an activation function that satisfies Assumptions 2.(a) and 2.(b). Consider N≤od(dγ)N\leq o_{d}(d^{\gamma}) and Q=QRF(γ){\mathcal{Q}}={\mathcal{Q}}_{{\rm RF}}(\gamma) as defined in Theorem 6.(a). Then there exists ε0>0\varepsilon_{0}>0 and d0d_{0} and a constant C>0C>0 such that for d≥d0d\geq d_{0} and τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q},

Notice that by Assumption 2.(b)(b) we can apply Lemma 19 to any k∈Qc{\bm{k}}\in{\mathcal{Q}}^{c} such that ∣k∣=k1+…+kQ≤L|{\bm{k}}|=k_{1}+\ldots+k_{Q}\leq L. In particular, there exists C>0C>0, ε0′>0\varepsilon_{0}^{\prime}>0 and d0′d_{0}^{\prime} such that for any k∈Qc{\bm{k}}\in{\mathcal{Q}}^{c} with ∣k∣≤L|{\bm{k}}|\leq L, d≥d0′d\geq d_{0}^{\prime} and τ∈[1−ε0′,1+ε0′]Q{\bm{\tau}}\in[1-\varepsilon_{0}^{\prime},1+\varepsilon_{0}^{\prime}]^{Q},

Furthermore, using that B(d,k)=Θ(d1k1d2k2…dQkQ)B({\bm{d}},{\bm{k}})=\Theta(d_{1}^{k_{1}}d_{2}^{k_{2}}\ldots d_{Q}^{k_{Q}}), there exists C′>0C^{\prime}>0 such that for k∈Qc{\bm{k}}\in{\mathcal{Q}}^{c} with ∣k∣≤L|{\bm{k}}|\leq L,

where we used in the last inequality k∉QRF(γ){\bm{k}}\not\in{\mathcal{Q}}_{{\rm RF}}(\gamma) implies (ξ−κ1)k1+…+(ξ−κQ)kQ≥γ(\xi-\kappa_{1})k_{1}+\ldots+(\xi-\kappa_{Q})k_{Q}\geq\gamma by definition.

Furthermore, from Assumption 2 and Lemma 17.(b)(b), there exists ε0′′>0\varepsilon_{0}^{\prime\prime}>0, d0′′d_{0}^{\prime\prime} and C<∞C<\infty, such that

In particular, for ∣k∣=k1+…+kQ>L=max⁡q∈[Q]⌈γ/ηq⌉|{\bm{k}}|=k_{1}+\ldots+k_{Q}>L=\max_{q\in[Q]}\lceil\gamma/\eta_{q}\rceil, we have

Combining Eqs (79) and (80) yields the result. ∎

E.2 Proof of Theorem 6.(a): Outline

Define the random vectors V=(V1,…,VN)T{\bm{V}}=(V_{1},\ldots,V_{N})^{\mathsf{T}}, VQ=(V1,Q,…,VN,Q)T{\bm{V}}_{{\mathcal{Q}}}=(V_{1,{\mathcal{Q}}},\ldots,V_{N,{\mathcal{Q}}})^{\mathsf{T}}, VQc=(V1,Qc,…,VN,Qc)T{\bm{V}}_{{\mathcal{Q}}^{c}}=(V_{1,{\mathcal{Q}}^{c}},\ldots,V_{N,{\mathcal{Q}}^{c}})^{\mathsf{T}}, with

Define the random matrix U=(Uij)i,j∈[N]{\bm{U}}=(U_{ij})_{i,j\in[N]}, with

In what follows, we write RRF(fd)=RRF(fd,W)=RRF(fd,Θ/D)R_{{\rm RF}}(f_{d})=R_{{\rm RF}}(f_{d},{\bm{W}})=R_{{\rm RF}}(f_{d},{\bm{\Theta}}/\sqrt{D}) for the random features risk, omitting the dependence on the weights W=Θ/D{\bm{W}}={\bm{\Theta}}/\sqrt{D}. 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 σ\sigma be an activation function satisfying Assumptions 2.(a) and 2.(b) for a fixed γ>0\gamma>0. Denote Q=QRF(γ){\mathcal{Q}}={\mathcal{Q}}_{{\rm RF}}(\gamma). Let ε>0\varepsilon>0 and define EQc,ε{\mathcal{E}}_{{\mathcal{Q}}^{c},\varepsilon} by

Then there exists a constant C>0C>0 and ε0>0\varepsilon_{0}>0 (depending only on the constants of Assumptions 2.(a) and 2.(b)) such that for dd sufficiently large,

The proofs of these two propositions are provided in the next sections.

Proposition 1 shows that there exists ε0>0\varepsilon_{0}>0 such that

Hence, by Markov’s inequality, we get for any ε>0\varepsilon>0,

where we used Lemma 8. By assumption, we have N=od(dγ)N=o_{d}(d^{\gamma}), 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 f‾\overline{f} is a function on the normalized product of spheres PSd{\rm PS}^{\bm{d}} (Note that we defined Pkfd(x)≡Pkf‾d(x‾){\mathsf{P}}_{\bm{k}}f_{d}({\bm{x}})\equiv{\mathsf{P}}_{{\bm{k}}}\overline{f}_{d}(\overline{\bm{x}}) the unambiguous polynomial approximation of fdf_{d} with polynomial of degree k{\bm{k}}). We have

We recall the expansion of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}} in terms of tensor product of Gegenbauer polynomials

Using Eq. (39) to get the following property

Let ε0>0\varepsilon_{0}>0 be a constant as specified in Lemma 9. We consider

From Lemma 9, there exists a constant C>0C>0 such that for dd sufficiently large, we have for any k∈Qc{\bm{k}}\in{\mathcal{Q}}^{c},

E.4 Proof of Proposition 2

Step 1. Construction of the activation functions σ^\hat{\sigma}, σˉ\bar{\sigma}.

Without loss of generality, we will assume that qξ=1q_{\xi}=1. From Assumption 2.(b)(b), σ\sigma is not a degree ⌊γ/η1⌋\lfloor\gamma/\eta_{1}\rfloor-polynomial. This is equivalent to having m≥⌊γ/η1⌋+1m\geq\lfloor\gamma/\eta_{1}\rfloor+1 such that μm(σ)≠0\mu_{m}(\sigma)\neq 0. Let us denote

Recall the expansion of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}} in terms of product of Gegenbauer polynomials

Step 2. The kernel functions udu_{d}, u^d\hat{u}_{d} and uˉd\bar{u}_{d}.

Let udu_{d}, u^d\hat{u}_{d} and uˉd\bar{u}_{d} be defined by

We immediately have udτ1,τ2=u^dτ1,τ2+uˉdτ1,τ2u_{{\bm{d}}}^{{\bm{\tau}}_{1},{\bm{\tau}}_{2}}=\hat{u}_{{\bm{d}}}^{{\bm{\tau}}_{1},{\bm{\tau}}_{2}}+\bar{u}_{{\bm{d}}}^{{\bm{\tau}}_{1},{\bm{\tau}}_{2}}. Note that all three correspond to positive semi-definite kernels.

Since U^=U−Uˉ⪰0\hat{\bm{U}}={\bm{U}}-\bar{\bm{U}}\succeq 0, we immediately have U⪰Uˉ{\bm{U}}\succeq\bar{\bm{U}}. In the following, we will lower bound Uˉ\bar{\bm{U}}.

By the decomposition of Uˉ\bar{\bm{U}} in terms of Gegenbauer polynomials (93), we have

From Assumption 2.(a)(a) and Lemma 20 applied to coefficient m{\bm{m}}, as well as the assumption that μm(σ)≠0\mu_{m}(\sigma)\neq 0, there exists ε0>0\varepsilon_{0}>0 and C,c>0C,c>0 such that for dd large enough,

We restrict ourselves to the event Pε0\mathcal{P}_{\varepsilon_{0}} 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.(a)(a) and Lemma 17.(b)(b). For the second inequality, notice that by Assumption 2.(c)(c) we can apply Lemma 19 to any k∈Q{\bm{k}}\in{\mathcal{Q}}. Hence (using that μk(σ)2>0\mu_{k}(\sigma)^{2}>0 and we can choose δ\delta sufficiently small), we deduce that there exists c>0c>0, ε0>0\varepsilon_{0}>0 and d0d_{0} such that for any d≥d0d\geq d_{0}, τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q} and k∈Q{\bm{k}}\in{\mathcal{Q}},

Furthermore, using that B(d,k)=Θ(d1k1d2k2…dQkQ)B({\bm{d}},{\bm{k}})=\Theta(d_{1}^{k_{1}}d_{2}^{k_{2}}\ldots d_{Q}^{k_{Q}}), there exists c′>0c^{\prime}>0 such that for any k∈Q{\bm{k}}\in{\mathcal{Q}},

where we used in the last inequality k∈Q‾RF(γ){\bm{k}}\in\overline{{\mathcal{Q}}}_{{\rm RF}}(\gamma) implies (ξ−κ1)k1+…+(ξ−κQ)kQ≤γ(\xi-\kappa_{1})k_{1}+\ldots+(\xi-\kappa_{Q})k_{Q}\leq\gamma by definition. ∎

F.2 Properties of the limiting kernel

Similarly to the proof of [GMMM19b, Theorem 1.(b)(b)], we construct a limiting kernel which is used as a proxy to upper bound the RF risk.

We recall the decomposition of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}} in terms of tensor product of Gegenbauer polynomials

F.3 Proof of Theorem 6.(b)

Without loss of generality, let us assume that {fd}\{f_{d}\} are polynomials contained in VQdV^{\bm{d}}_{{\mathcal{Q}}}, i.e. f‾d=PQf‾d\overline{f}_{d}={\mathsf{P}}_{{\mathcal{Q}}}\overline{f}_{d}.

Let ε0>0\varepsilon_{0}>0 be defined as in Lemma 10 and consider the expectation over Pε0\mathcal{P}_{\varepsilon_{0}} of the RF risk (in particular, a∗=(a1∗,…,aN∗){\bm{a}}^{*}=(a_{1}^{*},\ldots,a_{N}^{*}) are well defined):

We can expand the squared loss at a∗{\bm{a}}^{*} as

The second term of the expansion (LABEL:eq:expansion_squared_loss_RF) around a∗{\bm{a}}^{*} verifies

Let us consider the third term in the expansion (LABEL:eq:expansion_squared_loss_RF) around a∗{\bm{a}}^{*}: the non diagonal term verifies

For k∈Q{\bm{k}}\in{\mathcal{Q}} and s∈[B(d,k)]{\bm{s}}\in[B({\bm{d}},{\bm{k}})] and τ1,τ2∈[1−ε0,1+ε0]Q{\bm{\tau}}^{1},{\bm{\tau}}^{2}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q}, we have (for dd large enough)

for dd large enough (using Lemma 10). Furthermore

Combining Eq. (99), Eq. (100) and Eq. (101), we get

By Markov’s inequality, we get for any ε>0\varepsilon>0 and dd large enough,

The assumption N=ωd(dγ)N=\omega_{d}(d^{\gamma}) and Lemma 8 conclude the proof.

Appendix G Proof of Theorem 7.(a): lower bound for NT model

Consider the expansion of σd,τ′\sigma_{{\bm{d}},{\bm{\tau}}}^{\prime} in terms of product of Gegenbauer polynomials. We have

where the expectation is taken over x‾=(x‾(1),…,x‾(Q))∼μd\overline{\bm{x}}=(\overline{\bm{x}}^{(1)},\ldots,\overline{\bm{x}}^{(Q)})\sim\mu_{{\bm{d}}}.

with kq+=(k1,…,kq+1,…,kQ){\bm{k}}_{q+}=(k_{1},\ldots,k_{q}+1,\ldots,k_{Q}) and kq−=(k1,…,kq−1,…,kQ){\bm{k}}_{q-}=(k_{1},\ldots,k_{q}-1,\ldots,k_{Q}), and

with the convention td,−1=0t_{d,-1}=0. Then there exists constants ε0>0\varepsilon_{0}>0 and C>0C>0 such that for dd large enough, we have for any τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q} and k∈QNT(γ)c{\bm{k}}\in{\mathcal{Q}}_{\rm NT}(\gamma)^{c},

where we recall S(k)⊂[Q]S({\bm{k}})\subset[Q] is the subset of indices corresponding to the non zero integers kq>0k_{q}>0.

Let us fix an integer MM such that Q⊂[M]Q{\mathcal{Q}}\subset[M]^{Q}. We will denote Q≡QNT(γ){\mathcal{Q}}\equiv{\mathcal{Q}}_{{\rm NT}}(\gamma) for simplicity. Following the same proof as in Lemma 9, there exists ε0>0\varepsilon_{0}>0, d0d_{0} and C>0C>0 such that for any d≥d0d\geq d_{0} and τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q}, we have for any k∈Qc∩[M]Q{\bm{k}}\in{\mathcal{Q}}^{c}\cap[M]^{Q},

while for k∉[M]Q{\bm{k}}\not\in[M]^{Q}, we get

Injecting this bound in the formula (104) of Aτ,k(q){\bm{A}}^{(q)}_{{\bm{\tau}},{\bm{k}}}, we get for d≥d0d\geq d_{0}, τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q} and any k∈Qc∩[M]Q{\bm{k}}\in{\mathcal{Q}}^{c}\cap[M]^{Q}: if kq>0k_{q}>0,

where we used that for kq∈[M]k_{q}\in[M], there exists a constant c>0c>0 such that sdq,kq≤cd−ηqs_{d_{q},k_{q}}\leq cd^{-\eta_{q}} and tdq,kq≤ct_{d_{q},k_{q}}\leq c. Similarly, we get for k∉[M]Q{\bm{k}}\not\in[M]^{Q}

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 x=(x(1),…,x(Q))∼μdκ{\bm{x}}=({\bm{x}}^{(1)},\ldots,{\bm{x}}^{(Q)})\sim\mu_{\bm{d}}^{\bm{\kappa}}. Then,

with D=diag(Dii){\bm{D}}=\text{{\rm diag}}({\bm{D}}_{ii}) block diagonal. Furthermore, D{\bm{D}} and Δ{\bm{\Delta}} verifies the following properties:

For any q∈[Q]q\in[Q], there exists constants cq,Cq>0c_{q},C_{q}>0 such that we have with high probability

For any q≠q′∈[Q]q\neq q^{\prime}\in[Q], 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 ε0>0\varepsilon_{0}>0 as prescribed in Lemma 11. We decompose the vector Vi,Qc=(Vi,Qc(q))q∈[Q]{\bm{V}}_{i,{\mathcal{Q}}^{c}}=({\bm{V}}^{(q)}_{i,{\mathcal{Q}}^{c}})_{q\in[Q]} where

Hence, using the upper bounds on Aτ,k(q)A_{{\bm{\tau}},{\bm{k}}}^{{(q)}} in Lemma 11, we get for k∈Qc{\bm{k}}\in{\mathcal{Q}}^{c} with kq>0k_{q}>0:

where we used that N=od(dγ)N=o_{d}(d^{\gamma}) and κq≥min⁡q∈S(k)κq\kappa_{q}\geq\min_{q\in S({\bm{k}})}\kappa_{q} (we have kq>0k_{q}>0 and therefore q∈S(k)q\in S({\bm{k}}) by definition). Similarly for k∈Qc{\bm{k}}\in{\mathcal{Q}}^{c} with kq=0k_{q}=0:

where we used that by definition of ξ\xi we have ηq+κq≤ξ\eta_{q}+\kappa_{q}\leq\xi and min⁡q∈S(k)κq≤ξ\min_{q\in S({\bm{k}})}\kappa_{q}\leq\xi. 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 ε0>0\varepsilon_{0}>0 as prescribed in Lemma 11. We have for q∈[Q]q\in[Q]

where we denoted Hτ(q)H^{(q)}_{\bm{\tau}} the kernel given by

where Aτ,k(q)A_{{\bm{\tau}},{\bm{k}}}^{{(q)}} is given in Lemma 12. Hence we get

Then, we have the following decomposition in terms of product of Gegenbauer polynomials,

with kq+=(k1,…,kq+1,…,kQ){\bm{k}}_{q+}=(k_{1},\ldots,k_{q}+1,\ldots,k_{Q}) and kq−=(k1,…,kq−1,…,kQ){\bm{k}}_{q-}=(k_{1},\ldots,k_{q}-1,\ldots,k_{Q}), and

Recall the decomposition of σ′\sigma^{\prime} in terms of tensor product of Gegenbauer polynomials,

Injecting this decomposition into the definition of Hτ(q)H^{(q)}_{{\bm{\tau}}} yields

By the recurrence relationship for Gegenbauer polynomials (25), we have

where (we use the convention tdq,−1=0t_{d_{q},-1}=0)

where we get by matching the coefficients,

with kq+=(k1,…,kq+1,…,kQ){\bm{k}}_{q+}=(k_{1},\ldots,k_{q}+1,\ldots,k_{Q}) and kq−=(k1,…,kq−1,…,kQ){\bm{k}}_{q-}=(k_{1},\ldots,k_{q}-1,\ldots,k_{Q}). ∎

G.4 Proof of Proposition 4

where the coefficients λkd,i(ψ)\lambda_{{\bm{k}}}^{{\bm{d}},{\bm{i}}}(\psi) are given recursively: denoting iq+=(i1,…,iq+1,…,iQ){\bm{i}}_{q+}=(i_{1},\ldots,i_{q}+1,\ldots,i_{Q}), if kq=0k_{q}=0,

where we recall the notations kq+=(k1,…,kq+1,…,kQ){\bm{k}}_{q+}=(k_{1},\ldots,k_{q}+1,\ldots,k_{Q}) and kq−=(k1,…,kq−1,…,kQ){\bm{k}}_{q-}=(k_{1},\ldots,k_{q}-1,\ldots,k_{Q}).

We recall the following two formulas for k≥1k\geq 1 (see Section B.2):

Furthermore, we have Q0(d)(x)=1Q^{(d)}_{0}(x)=1, Q1(d)(x)=x/dQ^{(d)}_{1}(x)=x/d and therefore therefore xQ0(d)(x)=dQ1(d)(x)xQ^{(d)}_{0}(x)=dQ^{(d)}_{1}(x). Similarly to the proof of [GMMM19b, Lemma 6], we insert these expressions in the expansion of the function ψ\psi. 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 u(θ1,θ2){\bm{u}}({\bm{\theta}}_{1},{\bm{\theta}}_{2}) as by integration by part if we had x∼N(0,I){\bm{x}}\sim{\sf N}(0,{\mathbf{I}}).

Denote γ(q)=⟨θ‾1(q),θ‾2(q)⟩/dq\gamma^{(q)}=\langle\overline{\bm{\theta}}^{(q)}_{1},\overline{\bm{\theta}}^{(q)}_{2}\rangle/d_{q}. Let us rotate each sphere q∈[Q]q\in[Q] such that

Let us start with u(qq){\bm{u}}^{(qq)}. For clarity, we will denote (in the rotated basis (115))

Then it is easy to show that we can rewrite

Case (a): θ1(q)≠θ2(q){\bm{\theta}}^{(q)}_{1}\neq{\bm{\theta}}^{(q)}_{2}.

where (we dropped the dependency on (θ1,θ2)({\bm{\theta}}_{1},{\bm{\theta}}_{2}) for clarity)

is invertible almost surely (for τ1(q),τ2(q)≠0\tau^{(q)}_{1},\tau^{(q)}_{2}\neq 0 and γ(q)≠1\gamma^{(q)}\neq 1).

Case (b): θ1(q)=θ2(q){\bm{\theta}}^{(q)}_{1}={\bm{\theta}}^{(q)}_{2}.

Similarly, for some fixed α\alpha and β\beta, we define

Step 2: u(qq′){\bm{u}}^{(qq^{\prime})} for q≠q′q\neq q^{\prime}.

Case (a): θ1(q)≠θ2(q){\bm{\theta}}^{(q)}_{1}\neq{\bm{\theta}}^{(q)}_{2}.

where M(qq′){\bm{M}}^{(qq^{\prime})} is given by

which is invertible almost surely (for τ1(q),τ2(q)≠0\tau^{(q)}_{1},\tau^{(q)}_{2}\neq 0 and γ(q)≠1\gamma^{(q)}\neq 1).

Case (b): θ1(q)=θ2(q){\bm{\theta}}^{(q)}_{1}={\bm{\theta}}^{(q)}_{2}.

G.4.2 Proof of Proposition 4

Step 1. Construction of the activation function σ^\hat{\sigma}.

Recall the definition of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}} in Eq. (102) and its expansion in terms of tensor product of Gegenbauer polynomials:

We recall the definition of qξ=arg⁡max⁡q∈[Q]{ηq+κq}q_{\xi}=\arg\max_{q\in[Q]}\{\eta_{q}+\kappa_{q}\}. Let l2>l1≥2L+5l_{2}>l_{1}\geq 2L+5 be two indices that satisfy the conditions of Assumption 3.(b)(b) and we define l1=(0,…,0,l1,0,…,0){\bm{l}}_{1}=(0,\ldots,0,l_{1},0,\ldots,0) (l1l_{1} at position qξq_{\xi}) and l2=(0,…,0,l2,0,…,0){\bm{l}}_{2}=(0,\ldots,0,l_{2},0,\ldots,0) (l2l_{2} at position qξq_{\xi}). Using the Gegenbauer coefficients of σ′\sigma^{\prime}, we define a new activation function σ^′\hat{\sigma}^{\prime} by

for some δ1,δ2\delta_{1},\delta_{2} that we will fix later (with ∣δt∣≤1|\delta_{t}|\leq 1).

Step 2. The functions u,u^{\bm{u}},\hat{\bm{u}} and uˉ\bar{\bm{u}}.

Let u{\bm{u}} and u^\hat{\bm{u}} be the matrix-valued functions associated respectively to σ′\sigma^{\prime} and σ^′\hat{\sigma}^{\prime}

From Lemma 14, there exists functions u1ab,u2,1ab,u2,2ab,u3,1ab,u3,2abu^{ab}_{1},u^{ab}_{2,1},u^{ab}_{2,2},u^{ab}_{3,1},u^{ab}_{3,2} and u^1ab,u^2,1ab,u^2,2ab,u^3,1ab,u^3,2ab\hat{u}^{ab}_{1},\hat{u}^{ab}_{2,1},\hat{u}^{ab}_{2,2},\hat{u}^{ab}_{3,1},\hat{u}^{ab}_{3,2} (for a,b∈[Q]a,b\in[Q]), which decompose u{\bm{u}} and u^\hat{\bm{u}} along θ1{\bm{\theta}}_{1} and θ2{\bm{\theta}}_{2} vectors. We define uˉ=u−u^\bar{\bm{u}}={\bm{u}}-\hat{\bm{u}}. Then we have the same decomposition for uˉk,jab=uk,jab−u^k,jab\bar{u}^{ab}_{k,j}=u^{ab}_{k,j}-\hat{u}^{ab}_{k,j} for a,b∈[Q],k=1,2,3,j=1,2a,b\in[Q],k=1,2,3,j=1,2.

Step 3. Construction of the kernel matrices.

Note that we have U=U^+Uˉ{\bm{U}}=\hat{\bm{U}}+\bar{\bm{U}}. By Eq. (122) and (120), it is easy to see that U^⪰0\hat{\bm{U}}\succeq 0. Then we have U⪰Uˉ{\bm{U}}\succeq\bar{\bm{U}}. In the following, we would like to lower bound matrix Uˉ\bar{\bm{U}}.

Let us start with u‾(qq)\overline{\bm{u}}^{(qq)} for q∈[Q]q\in[Q]. Denoting γij(q)=⟨θ‾i(q),θ‾j(q)⟩/dq<1\gamma_{ij}^{(q)}=\langle\overline{\bm{\theta}}^{(q)}_{i},\overline{\bm{\theta}}^{(q)}_{j}\rangle/d_{q}<1, we get, from Eq. (116),

We get similar expressions for U^ij\hat{\bm{U}}_{ij} with λkd(σd,τ′)\lambda^{{\bm{d}}}_{{\bm{k}}}(\sigma_{{\bm{d}},{\bm{\tau}}}^{\prime}) replaced by λkd(σ^d,τ′)\lambda^{{\bm{d}}}_{{\bm{k}}}(\hat{\sigma}_{{\bm{d}},{\bm{\tau}}}^{\prime}). Because we defined σ′\sigma^{\prime} and σ^′\hat{\sigma}^{\prime} by only modifying the l1{\bm{l}}_{1}-th and l2{\bm{l}}_{2}-th coefficients, we get

Recalling that λkd,1q\lambda^{{\bm{d}},{\bm{1}}_{q}}_{{\bm{k}}} only depend on λk−1qd\lambda^{{\bm{d}}}_{{\bm{k}}-{\bm{1}}_{q}} and λk+1qd\lambda^{{\bm{d}}}_{{\bm{k}}+{\bm{1}}_{q}}, and λkd,2q\lambda^{{\bm{d}},{\bm{2}}_{q}}_{{\bm{k}}} on λk−2qd\lambda^{{\bm{d}}}_{{\bm{k}}-{\bm{2}}_{q}}, λkd\lambda_{{\bm{k}}}^{\bm{d}} and λk+2qd\lambda^{{\bm{d}}}_{{\bm{k}}+{\bm{2}}_{q}}, (Lemma 13), we get

where we used the convention λkd(σd,τ′)=0\lambda^{\bm{d}}_{{\bm{k}}}(\sigma_{{\bm{d}},{\bm{\tau}}}^{\prime})=0 if one of the coordinates verifies kq<0k_{q}<0.

From Lemma 13, Lemma 19 and Lemma 20, we get for t=1,2t=1,2 and q≠qξq\neq q_{\xi}:

while for q=qξq=q_{\xi} and u∈{−1,1}u\in\{-1,1\},

From Lemma (26), we recall that the coefficients of the kk-th Gegenbauer polynomial Qk(d)(x)=∑s=0kpk,s(d)xsQ_{k}^{(d)}(x)=\sum_{s=0}^{k}p^{(d)}_{k,s}x^{s} satisfy

Plugging the estimates (130) and (134) into Eqs. (128) and (129), we obtain that

We deduce from (135), (126) and (136) that for a∈,b∈a\in,b\in,

As a result, combining Eq. (137) with Eq. (123) in the expression of u‾(qq)\overline{u}^{(qq)} given in Lemma 14, we get

By the expression of Δ{\bm{\Delta}} given by (125), we conclude that

Step 5. Checking the properties of matrix D{\bm{D}}.

By Lemma 14, we can express Uˉii\bar{\bm{U}}_{ii} as a block matrix with

Let us first focus on the q=qξq=q_{\xi} sphere. Using Eqs. (128) and (129) with the expressions (131) and (132), we get the following convergence in probability (using that {τi(q)}i∈[N]\{\tau_{i}^{(q)}\}_{i\in[N]} concentrates on 11),

where we denoted δ=(δ1,δ2){\bm{\delta}}=(\delta_{1},\delta_{2}) (where δ1,δ2\delta_{1},\delta_{2} first appears in the definition of σ^\hat{\sigma} in Eq. (117), and till now δ1,δ2\delta_{1},\delta_{2} are still not determined) and, similarly to the proof of [GMMM19b, Proposition 5] and letting μk≡μk(σ′)\mu_{k}\equiv\mu_{k}(\sigma^{\prime}), we have

Following the same reasoning as in [GMMM19b, Proposition 5], we can verify that under Assumption 3.(b)(b), we have ∇F1(0),∇F2(0)≠0\nabla F_{1}({\bm{0}}),\nabla F_{2}({\bm{0}})\neq{\bm{0}} and det⁡(∇F1(0),∇F2(0))≠0\det(\nabla F_{1}({\bm{0}}),\nabla F_{2}({\bm{0}}))\neq 0. We can therefore find δ=(δ1,δ2){\bm{\delta}}=(\delta_{1},\delta_{2}) such that F1(δ)>0F_{1}({\bm{\delta}})>0, F2(δ)>0F_{2}({\bm{\delta}})>0. Furthermore,

Similarly, we get for q≠qξq\neq q_{\xi} from Eqs. (128) and (129) with the expressions (130) (recalling that {τi(q)}i∈[N]\{\tau_{i}^{(q)}\}_{i\in[N]} concentrates on 11),

We deduce that for q≠qξq\neq q_{\xi} and q≠q′q\neq q^{\prime},

which finishes to prove properties (107) and (108).

Appendix H Proof of Theorem 7.(b): upper bound for NT model

with kq+=(k1,…,kq+1,…,kQ){\bm{k}}_{q+}=(k_{1},\ldots,k_{q}+1,\ldots,k_{Q}) and kq−=(k1,…,kq−1,…,kQ){\bm{k}}_{q-}=(k_{1},\ldots,k_{q}-1,\ldots,k_{Q}), and

Then there exists constants ε0>0\varepsilon_{0}>0 and C>0C>0 such that for dd large enough, we have for any τ,τ′∈[1−ε0,1+ε0]Q{\bm{\tau}},{\bm{\tau}}^{\prime}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q},

From Assumptions 3.(a)(a) and 3.(c)(c) and Lemma 19, there exists c>0c>0 and ε0>0\varepsilon_{0}>0 such that for any τ,τ′∈[1−ε0,1+ε0]Q{\bm{\tau}},{\bm{\tau}}^{\prime}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q} and k∈Q{\bm{k}}\in{\mathcal{Q}},

Hence for kq>0k_{q}>0, we get λkd(σd,τ′)λkd(σd,τ′′)≥cd−γ−ξ+κq\lambda_{{\bm{k}}}^{\bm{d}}(\sigma_{{\bm{d}},{\bm{\tau}}}^{\prime})\lambda_{{\bm{k}}}^{\bm{d}}(\sigma_{{\bm{d}},{\bm{\tau}}^{\prime}}^{\prime})\geq cd^{-\gamma-\xi+\kappa_{q}}, and for kq=0k_{q}=0, we get λkd(σd,τ′)λkd(σd,τ′′)≥cd−γ+ξ−ηq−κq\lambda_{{\bm{k}}}^{\bm{d}}(\sigma_{{\bm{d}},{\bm{\tau}}}^{\prime})\lambda_{{\bm{k}}}^{\bm{d}}(\sigma_{{\bm{d}},{\bm{\tau}}^{\prime}}^{\prime})\geq cd^{-\gamma+\xi-\eta_{q}-\kappa_{q}}. Carefully injecting these bounds in Eq. (144) yields the lemma. ∎

H.2 Proof of Theorem 7.(b): outline

In this proof, we will consider QQ sub-classes of functions corresponding to the NT model restricted to the qq-th sphere:

We define similarly the risk associated to this sub-model

where Q≡Q‾NT(q)(γ){\mathcal{Q}}\equiv\overline{{\mathcal{Q}}}_{{\rm NT}^{(q)}}(\gamma) is defined in Equation (145).

From the proof of Theorem 7.(a)(a), we have a matching lower bound for FNT(q){\mathcal{F}}_{{\rm NT}^{(q)}}.

Denote qk=arg⁡min⁡q∈S(k)κqq_{{\bm{k}}}=\arg\min_{q\in S({\bm{k}})}\kappa_{q}, such that k∈Q‾NT(qk){\bm{k}}\in\overline{{\mathcal{Q}}}_{{\rm NT}^{(q_{{\bm{k}}})}} for any k∈Q‾NT(γ){\bm{k}}\in\overline{{\mathcal{Q}}}_{\rm NT}(\gamma). Furthermore, notice that by definition for any f∈L2(PSκd,μdκ)f\in L^{2}({\rm PS}^{\bm{d}}_{\bm{\kappa}},\mu_{{\bm{d}}}^{\bm{\kappa}}) and q∈[Q]q\in[Q],

H.3 Proof of Theorem 8

Similarly to the proof of Theorem 6.(b)(b), we construct a limiting kernel which is used as a proxy to upper bound the NT(q){\rm NT}^{(q)} risk.

We recall the decomposition of σd,τ′\sigma_{{\bm{d}},{\bm{\tau}}}^{\prime} in terms of tensor product of Gegenbauer polynomials:

Following the same computations as in Lemma 12, we get

with kq+=(k1,…,kq+1,…,kQ){\bm{k}}_{q+}=(k_{1},\ldots,k_{q}+1,\ldots,k_{Q}) and kq−=(k1,…,kq−1,…,kQ){\bm{k}}_{q-}=(k_{1},\ldots,k_{q}-1,\ldots,k_{Q}), and convention tdq,−1=0t_{d_{q},-1}=0,

H.3.2 Proof of Theorem 8

Let us assume that {fd}\{f_{d}\} is contained in ⨁k∈QVkd\bigoplus_{{\bm{k}}\in{\mathcal{Q}}}{\bm{V}}^{\bm{d}}_{{\bm{k}}}, i.e. f‾d=PQf‾d\overline{f}_{d}={\mathsf{P}}_{{\mathcal{Q}}}\overline{f}_{d}.

We can expand the squared loss at a{\bm{a}} as

The second term of the expansion (LABEL:eq:expansion_squared_loss_NT) around a∗{\bm{a}}^{*} verifies

Let us consider the third term in the expansion (LABEL:eq:expansion_squared_loss_NT) around a∗{\bm{a}}^{*}: the non diagonal term verifies

For k∈Q{\bm{k}}\in{\mathcal{Q}} and s∈[B(d,k)]{\bm{s}}\in[B({\bm{d}},{\bm{k}})] and τ1,τ2∈[1−ε0,1+ε0]Q{\bm{\tau}}^{1},{\bm{\tau}}^{2}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q}, we have

Hence from Lemma 17 and for ε0\varepsilon_{0} small enough, there exists C>0C>0 such that for dd large enough

Combining Eq. (149), Eq. (150) and Eq. (151), we get

By Markov’s inequality, we get for any ε>0\varepsilon>0 and dd large enough,

The assumption that N=ωd(dγ)N=\omega_{d}(d^{\gamma}) and Lemma 8 conclude the proof.

Appendix I Proof of Theorem 4 in the main text

Note that g^N\hat{g}_{N} can be regarded as a function in FNN2N{\mathcal{F}}_{{\rm NN}}^{2N} and f^NT,N∈FNNN(W)\hat{f}_{{\rm NT},N}\in{\mathcal{F}}_{{\rm NN}}^{N}({\bm{W}}), this implies that

It is easy to see that, when f∗(x)=φ(UTx)f_{*}({\bm{x}})=\varphi({\bm{U}}^{\mathsf{T}}{\bm{x}}), we have

Step 3. Show that RNN,N(f∗)R_{{\rm NN},N}(f_{*}) is independent of κ\kappa.

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 c,C>0c,C>0 such that for any ε>0\varepsilon>0,

Let us first consider NdqN_{d_{q}} with ε∈(0,2]\varepsilon\in(0,2]. The Gi2G_{i}^{2} are sub-exponential random variables with

From standard sub-exponential concentration inequality, we get

Hence, for ε∈(0,2]\varepsilon\in(0,2], we have

In the case of NDN_{D}, applying (154) with ε/(2+2ε)≤1\varepsilon/(2+2\varepsilon)\leq 1 shows that

Combining the above bounds into (153) yields for ε≥0\varepsilon\geq 0,

Notice that ∣τ(q)−1∣≤D/dq−1|\tau^{(q)}-1|\leq\sqrt{D/d_{q}}-1 and we only need to consider ε∈[0,D/dq−1]\varepsilon\in[0,\sqrt{D/d_{q}}-1]. We conclude that for any ε≥0\varepsilon\geq 0, we have

We will denote in the rest of this section αq=τ(q)rq/R\alpha_{q}=\tau^{(q)}r_{q}/R for q=1,…,Qq=1,\ldots,Q. Notice in particular that αq∝dηq+κq−ξ\alpha_{q}\propto d^{\eta_{q}+\kappa_{q}-\xi} where we recall that ξ=max⁡q∈[Q]{ηq+κq}\xi=\max_{q\in[Q]}\{\eta_{q}+\kappa_{q}\}. Without loss of generality, we will assume that the (unique) maximum is attained on the first sphere, i.e. ξ=η1+κ1\xi=\eta_{1}+\kappa_{1} and ξ>ηq+κq\xi>\eta_{q}+\kappa_{q} for q≥2q\geq 2.

A simple calculation shows that Cn→(2π)−1/2C_{n}\to(2\pi)^{-1/2} as n→∞n\to\infty, and hence sup⁡nCn≤C‾<∞\sup_{n}C_{n}\leq\overline{C}<\infty. Therefore for τ∈[1−ε,1+ε]Q{\bm{\tau}}\in[1-\varepsilon,1+\varepsilon]^{Q}, we have

Recalling the definition of αq=τ(q)rq/R\alpha_{q}=\tau^{(q)}r_{q}/R, with rq=d(ηq+κq)/2r_{q}=d^{(\eta_{q}+\kappa_{q})/2} and R=dξ/2(1+od(1))R=d^{\xi/2}(1+o_{d}(1)). Hence for any ε>0\varepsilon>0, uniformly on τ∈[1−ε,1+ε]Q{\bm{\tau}}\in[1-\varepsilon,1+\varepsilon]^{Q}, we have αq→0\alpha_{q}\to 0 for q≥2q\geq 2 and lim⁡sup⁡d→∞∣α1−1∣≤ε\lim\sup_{d\to\infty}|\alpha_{1}-1|\leq\varepsilon. Hence if we choose ε0<c1−1−1\varepsilon_{0}<c_{1}^{-1}-1, there exists c>0c>0 such that for dd sufficiently large M⪰cIQ{\bm{M}}\succeq c{\mathbf{I}}_{Q} and for any τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q}

Finally, for part (c)(c), without loss of generality we will take w(q)=e1(q){\bm{w}}^{(q)}={\bm{e}}^{(q)}_{1} so that ⟨w(q),x‾(q)⟩=x‾1(q)\langle{\bm{w}}^{(q)},\overline{\bm{x}}^{(q)}\rangle=\overline{x}^{(q)}_{1}. From part (b)(b), there exists ε>0\varepsilon>0 and d0d_{0} such that

Consider G∼N(0,IQ){\bm{G}}\sim{\sf N}(0,{\mathbf{I}}_{Q}) and an arbitrary coupling between x‾\overline{\bm{x}} and G{\bm{G}}. For any M>0M>0 we can choose σM\sigma_{M} bounded continuous so that for any dd and τ∈[1−ε,1+ε]Q{\bm{\tau}}\in[1-\varepsilon,1+\varepsilon]^{Q},

It is therefore sufficient to prove the claim for σM\sigma_{M}. Letting ξq∼N(0,Idq−1){\bm{\xi}}_{q}\sim{\sf N}(0,{\mathbf{I}}_{d_{q}-1}) independently for each q∈[Q]q\in[Q] and independent of G{\bm{G}}, we construct the coupling via

where we set x‾(q)=(x‾1(q),x‾−1(q))\overline{\bm{x}}^{(q)}=(\overline{x}^{(q)}_{1},\overline{\bm{x}}_{-1}^{(q)}) for each q∈[Q]q\in[Q]. We thus have (x‾1(q),x‾−1(q))→G(\overline{x}^{(q)}_{1},\overline{\bm{x}}_{-1}^{(q)})\to{\bm{G}} almost surely, hence the limit superior of Eq. (160) is by weak convergence bounded by 1/M1/M for any arbitrary MM. Furthermore, noticing that αq→0\alpha_{q}\to 0 uniformly on τ∈[1−ε,1+ε]Q{\bm{\tau}}\in[1-\varepsilon,1+\varepsilon]^{Q} for q≥2q\geq 2, we have by bounded convergence

We further have lim⁡(d,τ(1))→(∞,1)α1=1\lim_{(d,\tau^{(1)})\to(\infty,1)}\alpha_{1}=1. Hence, by bounded convergence,

Combining Eq. (160) with the coupling (161) and Eqs (162) and (163) yields the result. ∎

Consider the expansion of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}} in terms of tensor product of Gegenbauer polynomials. We have

with the expectation taken over x‾=(x‾(1),…,x‾(Q))∼μd≡Unif(PSd)\overline{\bm{x}}=(\overline{\bm{x}}^{(1)},\ldots,\overline{\bm{x}}^{(Q)})\sim\mu_{{\bm{d}}}\equiv{\rm Unif}({\rm PS}^{\bm{d}}). We will need the following lemma, which is direct consequence of Rodrigues formula, to get the scaling of the Gegenbauer coefficents of σd,τ\sigma_{{\bm{d}},{\bm{\tau}}}.

where x‾∼Unif(PSd)\overline{\bm{x}}\sim{\rm Unif}({\rm PS}^{\bm{d}}) and

where we used the definition (36) of tensor product of Gegenbauer polynomials.

Consider the integration with respect to x‾(Q)\overline{\bm{x}}^{(Q)}. Denote for ease of notations u=α1x‾1(1)+…+αQ−1x‾1(Q−1)u=\alpha_{1}\overline{x}^{(1)}_{1}+\ldots+\alpha_{Q-1}\overline{x}^{(Q-1)}_{1}. We use the Rodrigues formula for the Gegenbauer polynomials (see Eq. (26)):

Iterating Eq. (167) over q∈[Q]q\in[Q] and Eq. (166) yield the desired formula (164).

which converges to 11 when dq→∞d_{q}\to\infty. We deduce that

J.2 Proof of convergence in probability of the Gegenbauer coefficients

Then for any δ>0\delta>0, there exists ε0∈(0,1)\varepsilon_{0}\in(0,1) and d0d_{0} such that for any d≥d0d\geq d_{0} and τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q},

Recall αq=τ(q)rq/R\alpha_{q}=\tau^{(q)}r_{q}/R with rq=d(κq+ηq)/2r_{q}=d^{(\kappa_{q}+\eta_{q})/2} and R=dξ/2(1+od(1))R=d^{\xi/2}(1+o_{d}(1)). Hence, we have

We can apply Lemma 17 to the activation function σ(∣k∣)\sigma^{(|{\bm{k}}|)}. In particular part (c)(c) of the lemma implies that there exists ε0∈(0,1)\varepsilon_{0}\in(0,1) such that for dd sufficiently large, we have for any τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q},

Then for any δ>0\delta>0, there exists ε0=ε0(c1,δ)\varepsilon_{0}=\varepsilon_{0}(c_{1},\delta) and d0=d0(c1,δ)d_{0}=d_{0}(c_{1},\delta) such that for any d≥d0d\geq d_{0} and τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q},

Recall the correspondence (31) between Gegenbauer and Hermite polynomials. Note for any monomial ml(x)=xkm_{l}(x)=x^{k}, we can apply Lemma 17.(c)(c) to ml(x‾1(qξ))σm_{l}(\overline{x}_{1}^{(q_{\xi})})\sigma and find a coupling such that for any η>0\eta>0, there exists ε0>0\varepsilon_{0}>0 and

Using the asymptotic correspondence between Gegenbauer polynomials and Hermite polynomials (31)

and Eq. (173), we get for any δ>0\delta>0, there exists ε0>0\varepsilon_{0}>0 such that for dd sufficiently large, we have for any τ∈[1−ε0,1+ε0]Q{\bm{\tau}}\in[1-\varepsilon_{0},1+\varepsilon_{0}]^{Q},

Appendix K Bound on the operator norm of Gegenbauer polynomials

For each q∈[Q]q\in[Q], we consider Δ(q)=Wk(q)−In{\bm{\Delta}}^{(q)}={\bm{W}}^{(q)}_{k}-{\mathbf{I}}_{n} where Wk(q)=((Wk(q))ij)i,j∈[n]{\bm{W}}^{(q)}_{k}=(({\bm{W}}^{(q)}_{k})_{ij})_{i,j\in[n]} with

Then, defining γq≡γ/ηq\gamma_{q}\equiv\gamma/\eta_{q}, we have

For dd sufficiently large, there exists C>0C>0 such that for any p≥m≡⌈2γq+3⌉p\geq m\equiv\lceil 2\gamma_{q}+3\rceil:

Hence, there exists constant C′C^{\prime}, such that for large dd, we have

Recalling that B(dq,m)=Θd(dηqm)=ωd(d2γ)B(d_{q},m)=\Theta_{d}(d^{\eta_{q}m})=\omega_{d}(d^{2\gamma}), and n=od(dγ)n=o_{d}(d^{\gamma}), we deduce

Let us now consider Δ=Wk−In{\bm{\Delta}}={\bm{W}}_{{\bm{k}}}-{\mathbf{I}}_{n}. We will denote Δ(q)=Wkq(q)−In{\bm{\Delta}}^{(q)}={\bm{W}}^{(q)}_{k_{q}}-{\mathbf{I}}_{n}. Then it is easy to check (recall the diagonal elements of Wkq(dq){\bm{W}}^{(d_{q})}_{k_{q}} are equal to one) that for any q∈[Q]q\in[Q]

where A⊙B{\bm{A}}\odot{\bm{B}} denotes the Hadamard product, or entrywise product, (A⊙B)i,j∈[n]=(AijBij)i,j∈[n]({\bm{A}}\odot{\bm{B}})_{i,j\in[n]}=(A_{ij}B_{ij})_{i,j\in[n]}. We recall the following inequality on the operator norm of Hadamard product of two matrices, with A{\bm{A}} positive definite:

Furthermore, I∩Q{\mathcal{I}}\cap{\mathcal{Q}} 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 Δ=W−In{\bm{\Delta}}={\bm{W}}-{\mathbf{I}}_{n}. We define for each q∈[Q]q\in[Q], Wkq(dq)=(Qkq(dq)(⟨x‾i(q),x‾j(q)⟩))ij∈[n]{\bm{W}}^{(d_{q})}_{k_{q}}=(Q_{k_{q}}^{(d_{q})}(\langle\overline{\bm{x}}_{i}^{(q)},\overline{\bm{x}}_{j}^{(q)}\rangle))_{ij\in[n]} and Δ(q)=Wkq(dq)−In{\bm{\Delta}}^{(q)}={\bm{W}}^{(d_{q})}_{k_{q}}-{\mathbf{I}}_{n}. Then it is easy to check (recall the diagonal elements of Wkq(dq){\bm{W}}^{(d_{q})}_{k_{q}} are equal to one)

where A⊙B{\bm{A}}\odot{\bm{B}} denotes the Hadamard product, or entrywise product, (A⊙B)i,j∈[n]=(AijBij)i,j∈[n]({\bm{A}}\odot{\bm{B}})_{i,j\in[n]}=(A_{ij}B_{ij})_{i,j\in[n]}. For any sequence of integers p=p(d)p=p(d), we have

To prove the proposition, it suffices to show that for any sequence Ad→∞A_{d}\to\infty, we have

where we used that x‾(q)\overline{\bm{x}}^{(q)} and x‾(q′)\overline{\bm{x}}^{(q^{\prime})} are independent for q≠q′q\neq q^{\prime}.

We will denote for any i=(i1,…,ik)∈[n]k{\bm{i}}=(i_{1},\ldots,i_{k})\in[n]^{k}, define for each q∈[Q]q\in[Q]

Similarly, we define MiM_{{\bm{i}}} associated to Δ{\bm{\Delta}},

To calculate these quantities, we will apply repeatedly the following identity, which is an immediate consequence of Eq. (23). For any i1,i2,i3i_{1},i_{2},i_{3} distinct, we have

Throughout the proof, we will denote by C,C′,C′′C,C^{\prime},C^{\prime\prime} constants that may depend on kk but not on p,d,np,d,n. 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 i=(i1,i2,…,i2p)∈[n]2p{\bm{i}}=(i_{1},i_{2},\ldots,i_{2p})\in[n]^{2p}, we defined an undirected multigraph Gi=(Vi,Ei)G_{\bm{i}}=(V_{\bm{i}},E_{\bm{i}}) associated to index sequence i{\bm{i}}. The vertex set ViV_{\bm{i}} is the set of distinct elements in i1,…,i2pi_{1},\ldots,i_{2p}. The edge set EiE_{{\bm{i}}} is formed as follows: for any j∈[2p]j\in[2p] we add an edge between iji_{j} and ij+1i_{j+1} (with convention 2p+1≡12p+1\equiv 1). Notice that this could be a self-edge, or a repeated edge: Gi=(Vi,Ei)G_{\bm{i}}=(V_{\bm{i}},E_{\bm{i}}) will be –in general– a multigraph. We denote v(i)=∣Vi∣v({\bm{i}})=|V_{\bm{i}}| to be the number of vertices of GiG_{\bm{i}}, and e(i)=∣Ei∣e({\bm{i}})=|E_{\bm{i}}| to be the number of edges (counting multiplicities). In particular, e(i)=ke({\bm{i}})=k for i∈[n]k{\bm{i}}\in[n]^{k}. We define

For any two index sequences i1,i2{\bm{i}}_{1},{\bm{i}}_{2}, we say they are equivalent i1≍i2{\bm{i}}_{1}\asymp{\bm{i}}_{2}, if the two graphs Gi1G_{{\bm{i}}_{1}} and Gi2G_{{\bm{i}}_{2}} are isomorphic, i.e. there exists an edge-preserving bijection of their vertices (ignoring vertex labels). We denote the equivalent class of i{\bm{i}} to be

We define the quotient set Q(p){\mathcal{Q}}(p) by

The following Lemma was proved in [GMMM19b, Proposition 3]

The following properties holds for all sufficiently large nn and dd:

For any equivalent index sequences i=(i1,…,i2p)≍j=(j1,…,j2p){\bm{i}}=(i_{1},\ldots,i_{2p})\asymp{\bm{j}}=(j_{1},\ldots,j_{2p}), we have Mi(q)=Mj(q)M_{{\bm{i}}}^{(q)}=M_{{\bm{j}}}^{(q)}.

For any index sequence i∈[n]2p∖T⋆(p){\bm{i}}\in[n]^{2p}\setminus{\mathcal{T}}_{\star}(p), we have Mi=0M_{{\bm{i}}}=0.

For any index sequence i∈T⋆(p){\bm{i}}\in{\mathcal{T}}_{\star}(p), the degree of any vertex in GiG_{\bm{i}} must be even.

The number of equivalent classes ∣Q(p)∣≤(2p)2p|{\mathcal{Q}}(p)|\leq(2p)^{2p}.

Recall that v(i)=∣Vi∣v({\bm{i}})=|V_{\bm{i}}| denotes the number of distinct elements in i{\bm{i}}. Then, for any i∈[n]2p{\bm{i}}\in[n]^{2p}, the number of elements in the corresponding equivalence class satisfies ∣C(i)∣≤v(i)v(i)⋅nv(i)≤ppnv(i)|{\mathcal{C}}({\bm{i}})|\leq v({\bm{i}})^{v({\bm{i}})}\cdot n^{v({\bm{i}})}\leq p^{p}n^{v({\bm{i}})}.

In view of property (a)(a) in the last lemma, given an equivalence class C=C(i){\mathcal{C}}={\mathcal{C}}({\bm{i}}), we will write MC=MiM_{{\mathcal{C}}}=M_{{\bm{i}}} 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 i≍j∈[n]p{\bm{i}}\asymp{\bm{j}}\in[n]^{p}, then sk(i)≍sk(j){\rm sk}({\bm{i}})\asymp{\rm sk}({\bm{j}}). That is, the skeletons of equivalent index sequences are equivalent.

For any i=(i1,…,ik)∈[n]k{\bm{i}}=(i_{1},\ldots,i_{k})\in[n]^{k}, and q∈[Q]q\in[Q], we have

For any i∈T⋆(p)⊂[n]2p{\bm{i}}\in{\mathcal{T}}_{\star}(p)\subset[n]^{2p}, 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 44.

Given an index sequence i∈T⋆(p)⊂[n]2p{\bm{i}}\in{\mathcal{T}}_{\star}(p)\subset[n]^{2p}, we say i{\bm{i}} is of type 1, if sk(i){\rm sk}({\bm{i}}) contains only one index. We say i{\bm{i}} is of type 2 if sk(i){\rm sk}({\bm{i}}) is not empty (so that by Lemma 22, Gsk(i)G_{{\rm sk}({\bm{i}})} can only contain vertices with degree greater or equal to 44). Denote the class of type 1 index sequence (respectively type 2 index sequence) by T1(p){\mathcal{T}}_{1}(p) (respectively T2(p){\mathcal{T}}_{2}(p)). We also denote by T~a(p)\widetilde{\mathcal{T}}_{a}(p), a∈{1,2}a\in\{1,2\} the set of equivalence classes of sequences in Ta(p){\mathcal{T}}_{a}(p). 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 v(i)v({\bm{i}}) is the number of vertices in GiG_{\bm{i}}, and e(i)e({\bm{i}}) is the number of edges in GiG_{\bm{i}} (which coincides with the length of i{\bm{i}}). We consider i∈T1(p){\bm{i}}\in{\mathcal{T}}_{1}(p). Since for i∈T1(p){\bm{i}}\in{\mathcal{T}}_{1}(p), every edge of GiG_{\bm{i}} must be at most a double edge. Indeed, if (u1,u2)(u_{1},u_{2}) had multiplicity larger than 22 in GiG_{{\bm{i}}}, neither u1u_{1} nor u2u_{2} could be deleted during the skeletonization process, contradicting the assumption that sk(i){\rm sk}({\bm{i}}) contains a single vertex. Therefore, we must have min⁡i∈T1v(i)=p+1\min_{{\bm{i}}\in{\mathcal{T}}_{1}}v({\bm{i}})=p+1. According the Lemma 22.(b)(b), for every i∈T1(p){\bm{i}}\in{\mathcal{T}}_{1}(p), we have

Note by Lemma 21.(e)(e), the number of elements in the equivalence class of i{\bm{i}} is ∣C(i)∣≤pp⋅nv(i)|{\mathcal{C}}({\bm{i}})|\leq p^{p}\cdot n^{v({\bm{i}})}. Hence we get

Therefore, denoting K=∑q∈[Q]ηqkqK=\sum_{q\in[Q]}\eta_{q}k_{q},

where in the last step we used Lemma 21 and the fact that for q∈[Q]q\in[Q], B(dq,kq)≥C0dqkqB(d_{q},k_{q})\geq C_{0}d_{q}^{k_{q}} for some C0>0C_{0}>0.

We have the following simple lemma bounding MiM_{\bm{i}}, copied from [GMMM19b, Proposition 3]. This bound is useful when i{\bm{i}} is a skeleton.

For any q∈[Q]q\in[Q], there exists constants CC and d0d_{0} depending uniquely on kqk_{q} such that, for any d≥d0(kq)d\geq d_{0}(k_{q}), and any index sequence i∈[n]m{\bm{i}}\in[n]^{m} with 2≤m≤dq/(4kq)2\leq m\leq d_{q}/(4k_{q}), we have

Suppose i∈T2(p){\bm{i}}\in{\mathcal{T}}_{2}(p), and denote v(i)v({\bm{i}}) to be the number of vertices in GiG_{\bm{i}}. We have, for a sequence p=od(d)p=o_{d}(d), and each q∈[Q]q\in[Q]

Here (1)(1) holds by Lemma 22.(b)(b); (2)(2) by Lemma 23, and the fact that sk(i)∈[n]e(sk(i)){\rm sk}({\bm{i}})\in[n]^{e({\rm sk}({\bm{i}}))}, together by B(dq,kq)≥C0dqkqB(d_{q},k_{q})\geq C_{0}d_{q}^{k_{q}}; (3)(3) because e(sk(i))≤2pe({\rm sk}({\bm{i}}))\leq 2p; (4)(4) by Lemma 22.(c)(c), implying that for i∈T2(p){\bm{i}}\in{\mathcal{T}}_{2}(p), each vertex of Gsk(i)G_{{\rm sk}({\bm{i}})} has degree greater or equal to 44, so that v(sk(i))≤e(sk(i))/2v({\rm sk}({\bm{i}}))\leq e({\rm sk}({\bm{i}}))/2 (notice that for d≥d0(kq)d\geq d_{0}(k_{q}) we can assume Cp/dq<1Cp/d_{q}<1). Finally, (5)(5) follows since r(i),v(sk(i))≤v(i)r({\bm{i}}),v({\rm sk}({\bm{i}}))\leq v({\bm{i}}), and (6)(6) the definition of r(i)r({\bm{i}}) implying r(i)=v(i)−v(sk(i))r({\bm{i}})=v({\bm{i}})-v({\rm sk}({\bm{i}})).

Note by Lemma 21.(e)(e), the number of elements in equivalent class ∣C(i)∣≤pv(i)⋅nv(i)|{\mathcal{C}}({\bm{i}})|\leq p^{v({\bm{i}})}\cdot n^{v({\bm{i}})}. Since v(i)v({\bm{i}}) depends only on the equivalence class of i{\bm{i}}, we will write, with a slight abuse of notation v(i)=v(C(i))v({\bm{i}})=v({\mathcal{C}}({\bm{i}})). Notice that the number of equivalence classes with v(C)=vv({\mathcal{C}})=v is upper bounded by the number multi-graphs with vv vertices and 2p2p edges, which is at most v4pv^{4p}. Denoting α=max⁡q∈[Q]{1/ηq}\alpha=\max_{q\in[Q]}\{1/\eta_{q}\}, we have

Define ε=Cnpα(K+1)/dK\varepsilon=Cnp^{\alpha(K+1)}/d^{K}. We will assume hereafter that pp is selected such that

By calculus and condition (185), the function F(v)=v4pεvF(v)=v^{4p}\varepsilon^{v} is maximized over v∈[2,2p]v\in[2,2p] at v=2v=2, whence

Using Eqs. (181) and (186), we have, for any p=od(d)p=o_{d}(d) satisfying Eq. (185), we have

Finally setting n=dKe−2Alog⁡dn=d^{K}e^{-2A\sqrt{\log d}} and p=(K/A)log⁡dp=(K/A)\sqrt{\log d}, this yields

Appendix L Technical lemmas

We put here one technical lemma that is used in the proof of Theorem 7.(a).

For any q∈[Q]q\in[Q], there exists cq,Cq>0c_{q},C_{q}>0 such that we have with high probability

Then for any q≠q′∈[Q]q\neq q^{\prime}\in[Q], we have

Let us show the result recursively on the integer QQ. Note that the case Q=1Q=1 is direct.

Assume that A−1{\bm{A}}^{-1} 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 kk, let Qk(d)(x)Q_{k}^{(d)}(x) be the kk-th Gegenbauer polynomial. We expand