Learning with convolution and pooling operations in kernel methods

Theodor Misiakiewicz, Song Mei

Introduction

Convolutional neural networks (CNNs) have become essential elements of the deep learning toolbox, achieving state-of-the-art performance in many computer vision tasks . CNNs are constructed by stacking convolution and pooling layers, which were shown to be paramount to their empirical success . A widely accepted hypothesis to explain their favorable properties is that these architectures successfully encode useful properties of natural images: locality and compositionality of the data, stability by local deformations, and translation invariance. While some theoretical progress has been made in studying the approximation and generalization benefits brought by convolution and pooling operations , our mathematical understanding of the interaction between network architecture, image distribution, and efficient learning remains limited.

Note that pooling and downsampling operations are often tied together in the literature. However in this work we will treat these two operations separately.

In the formula above, different values for q,ω,Δq,\omega,\Delta lead to different architectures with vastly different behaviors. For example, when q=Δ=dq=\Delta=d and ω=1\omega=1, we recover a two-layer fully-connected neural network f\mboxFC(x;a,Θ)=N−1/2∑i∈[N]aiσ(⟨wi,x⟩)f_{\mbox{\tiny\sf FC}}({\bm{x}};{\bm{a}},{\bm{\Theta}})=N^{-1/2}\sum_{i\in[N]}a_{i}\sigma(\langle{\bm{w}}_{i},{\bm{x}}\rangle) which has the universal approximation property at large NN. When ω=Δ=1\omega=\Delta=1 and q<dq<d, the network is “locally connected” f\mboxLC(x;a,Θ)=N−1/2∑i∈[N],k∈[d]aikσ(⟨wi,x(k)⟩)f_{\mbox{\tiny\sf LC}}({\bm{x}};{\bm{a}},{\bm{\Theta}})=N^{-1/2}\sum_{i\in[N],k\in[d]}a_{ik}\sigma(\langle{\bm{w}}_{i},{\bm{x}}_{(k)}\rangle), and not a universal approximator anymore: however, f\mboxLCf_{\mbox{\tiny\sf LC}} vastly outperforms f\mboxFCf_{\mbox{\tiny\sf FC}} in some cases . For ω>1\omega>1, local pooling enables learning functions that are locally invariant by translations more efficiently than without pooling. For ω=d\omega=d (global pooling), the network only fits functions fully invariant by cyclic translations.

The aim of this paper is to formalize and quantify the interplay between the target function class and the statistical efficiency brought by these different architectures. As a concrete first step in this direction, we consider kernel models that are naturally associated with the convolutional neural networks (CNN-AP-DS) through the neural tangent kernel perspective . Kernel methods have the advantage of 1) being tractable—leaving the computational issue of learning CNNs aside; 2) having well-understood approximation and generalization properties, which depends on the eigendecomposition of the kernel and the alignment between the target function and associated RKHS (see Appendices B and C for background). While kernel models only describe neural networks in the lazy training regime and miss important properties of deep learning, such as feature learning, architecture choice already plays a crucial role to learn efficiently ‘image-like’ functions in the fixed-feature regime.

Neural tangent kernels are obtained by linearizing the associated neural networks. Here we consider the tangent kernel associated to the network f\mboxCNNf_{\mbox{\tiny\sf CNN}} (c.f. Appendix A.2 for a detailed derivation):

In this paper, we will further consider a stylized setting with input signal distribution x∼Unif(\mathscrsfsQd){\bm{x}}\sim{\rm Unif}({\mathscrsfs Q}^{d}) (uniform distribution over \mathscrsfsQd:={−1,+1}d{\mathscrsfs Q}^{d}:=\{-1,+1\}^{d} the discrete hypercube in dd dimensions). This simple choice allows for a complete characterization of the eigendecomposition of Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta}, thanks to all patches having same marginal distribution x(k)∼Unif(\mathscrsfsQq){\bm{x}}_{(k)}\sim{\rm Unif}({\mathscrsfs Q}^{q}). We will be particularly interested in four specific choices of (q,ω,Δ)(q,\omega,\Delta) in (CK-AP-DS):

These kernels are respectively the neural tangent kernels of a fully-connected network f\mboxFCf_{\mbox{\tiny\sf FC}} (FC), a convolutional network f\mboxLCf_{\mbox{\tiny\sf LC}} (CK), a convolutional network followed by local average pooling (CK-AP) and a convolutional network followed by global pooling (CK-GP). We will further be interested in (CK-GP) with patch size q=dq=d, which we denote H\mboxGP\mboxFCH^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}}: this corresponds to a convolutional kernel with full-size patches q=dq=d, followed by global pooling.

Our contributions are two-fold. First, we describe the RKHS associated with the convolutional kernel (CK-AP-DS) in the stylized setting x∼Unif(\mathscrsfsQd){\bm{x}}\sim{\rm Unif}({\mathscrsfs Q}^{d}), which provides a fully explicit picture of the roles of convolution, pooling and downsampling operations in approximating specific classes of functions. Second, we provide sharp asymptotics for the generalization error of KRR in high-dimension, given any target function and one of the kernels described in the introductionNote that we modify slightly Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} to simplify the derivation of the high-dimension asymptotics. However, we believe such a simplification to be unecessary. The fixed-dimension bounds do not require such a simplification.. These asymptotics are obtained rigorously using the framework of (see Appendix C for background). For completeness, we also include bounds on the KRR test error in the classical fixed-dimension setting with capacity/source assumptions (see Appendix C for limitations of this classical approach).

We summarize our results below. Define the qq-local function class L2(\mathscrsfsQd,Locq)L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q}) and the cyclic qq-local function class L2(\mathscrsfsQd,CycLocq)L^{2}({\mathscrsfs Q}^{d},{\rm CycLoc}_{q}) (subspace of L2(\mathscrsfsQd,Locq)L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q}) consisting of cyclic-invariant functions) as follows:

When Δ≤ω\Delta\leq\omega, downsampling after average pooling leaves the low-frequency eigenspaces of Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} stable. In particular, the downsampling operation does not modify the statistical complexity of learning low-frequency functions in one-layer kernels, while being potentially beneficial in further layers in deep convolutional kernels.

There are two important model assumptions in this paper, which deserve some discussions:

One-layer convolutional kernel (CK): extra layers allow for hierarchical interactions between the patches (see for example ). However, we believe that the main insights on the approximation and statistical trade-off are already captured in the one-layer case (see for multi-layer but independent patches). Note that depth might be less important for CKs than for CNNs: the one-layer CK considered in this paper achieves 80.9%80.9\% accuracy on CIFAR-10 (versus 79.6%79.6\% in ) and 3-layers CK achieves 88.2%88.2\% accuracy (versus 90%90\% for the best multi-layer CK ). See Appendix A.5 for a discussion on how our results could be extended to 2-layers.

Data uniform on the hypercube: this choice is motivated by our goal of deriving rigorous fine-grained approximation and generalization errors, which requires to diagonalize the kernel (CK-AP-DS). More general data distributions either require strong assumptions (independent patches ), loose minmax bounds on the generalization error (e.g., classical source/capacity assumptions) or non-rigorous statistical physics heuristics .

The rest of the paper is organized as follows. We discuss related work in Section 1.2. In Section 2, we present our main results on convolutional kernels and describe precisely the roles of convolution, pooling and downsampling operations. Finally, we present a numerical simulation on synthetic data in Section 3 and conclude in Section 4. Some details and discussions are deferred to Appendix A.

2 Related work

Convolutional kernels have been considered in . In particular, they showed that these architectures achieve good results in image classification (90%90\% accuracy on Cifar10) and that pooling and downsampling were necessary for their good performance .

The generalization error of kernel ridge regression (KRR) has been well-studied in both the fixed dimension regime [47, Chap. 13], and the high-dimensional regimes . These results show that the generalization error depends on the eigenvalues and eigenfunctions of the kernel, and the alignment of the kernel with the target function.

Recently, a few theoretical work have considered the generalization properties of invariant kernels and convolutional kernels . In particular, consider convolutional kernel with global pooling and full-size patches q=dq=d, and show a gain of factor dd in sample complexity when learning cyclic functions, compared to inner-product kernels. considers additionally kernels that are stable with respect to local deformations, and similarly quantify the sample complexity gain. A concurrent work considers sharp asymptotics of the KRR test error using the framework of for certain hierarchical convolutional kernels under the strong assumption of non-overlapping patches (whereas we consider the more natural architecture of overlapping patches). They arrive at a similar trade-off between approximation and generalization power in convolutional kernels, which they call ‘eigenspace restructuring principle’: given a finite statistical budget (i.e., a sample size nn), convolutional architectures allocate the ‘eigenvalue mass’ by weighting differently the eigenspaces.

consider a one-layer convolutional kernel with and without global pooling and obtain a diagonalization similar to Proposition 1 for data uniformly distributed on the continuous cube. They further derive asymptotic rates in nn, the number of samples, in a student-teacher scenario using statistical physics heuristics and a Gaussian equivalence conjecture. In particular, they show that locality rather than translation-invariance breaks the curse of dimensionality. Here, our goal is different: we derive mathematically rigorous quantitative bounds that give separation in generalization power between different architectures. We consider classical source and capacity conditions and obtain non-asymptotic bounds on the test error that are minmax optimal in both nn and dd. We further give pointwise generalization error in a high-dimensional framework that give a separation in sample complexity for learning a given function.

See for more theoretical results on the separation between convolutional and fully connected neural networks, and for the inductive bias of pooling operations in convolutional neural networks.

Main results

We start by introducing some background on functions on the hypercube and eigendecomposition of kernel operators in Section 2.1. We first consider a kernel with a single convolution layer in Section 2.2, and characterize its eigendecomposition and generalization properties. We then show how these results are modified when applying local average pooling and downsampling in Section 2.3.

2 One-layer convolutional kernel

where we recall that x(k)=(xk,…,xk+q−1)∈\mathscrsfsQq{\bm{x}}_{(k)}=(x_{k},\ldots,x_{k+q-1})\in{\mathscrsfs Q}^{q} is the kk’th patch of the image with size qq.

Let H\mboxCKH^{\mbox{\tiny\sf CK}} be a convolutional kernel as defined in Eq. (3). Then H\mboxCKH^{\mbox{\tiny\sf CK}} admits the following eigendecomposition:

On the other hand, the RKHS associated to the fully-connected kernel H\mboxFCH^{\mbox{\tiny\sf FC}} (FC) typically contains all the functions in L2(\mathscrsfsQd)L^{2}({\mathscrsfs Q}^{d}) (under genericity assumptions on hh). The RKHS with convolution dim⁡(L2(\mathscrsfsQd,Locq))=d2q−1+1\dim(L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q}))=d2^{q-1}+1 is significantly smaller than dim⁡(L2(\mathscrsfsQd))=2d\dim(L^{2}({\mathscrsfs Q}^{d}))=2^{d}, which prompts the following question: what is the statistical advantage of using H\mboxCKH^{\mbox{\tiny\sf CK}} over H\mboxFCH^{\mbox{\tiny\sf FC}} when learning functions in L2(\mathscrsfsQd,Locq)L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q})?

We first consider the classical approach to bounding the test error of which relies on the following two standard assumptions:

Capacity condition: we assume N(h,λ):=Tr[h/(h+λI)−1]≤Chλ−1/α\mathcal{N}(h,\lambda):={\rm Tr}[h/(h+\lambda{\mathbf{I}})^{-1}]\leq C_{h}\lambda^{-1/\alpha} withHere, hh is the integral operator and Tr[h/(h+λI)−1]=∑j≥1λjλj+λ{\rm Tr}[h/(h+\lambda{\mathbf{I}})^{-1}]=\sum_{j\geq 1}\frac{\lambda_{j}}{\lambda_{j}+\lambda} with {λj}j≥1\{\lambda_{j}\}_{j\geq 1} eigenvalues of hh. α>1\alpha>1.

Source condition: ∥h−β/2g∥L2≤B\|h^{-\beta/2}g\|_{L^{2}}\leq B withAgain, hh is the operator with h−κg=∑j≥1λj−κ⟨f,ψj⟩ψjh^{-\kappa}g=\sum_{j\geq 1}\lambda_{j}^{-\kappa}\langle f,\psi_{j}\rangle\psi_{j}, where {ψj}j≥1\{\psi_{j}\}_{j\geq 1} are the eigenvectors of hh. β>α−1α\beta>\frac{\alpha-1}{\alpha} and B≥0B\geq 0.

The capacity condition (A1) characterizes the size of the RKHS: for increasing α\alpha, the RKHS contains less and less functions. The source condition (A2) characterizes the regularity of the target function (the ‘source’) with respect to the kernel: increasing β\beta corresponds to smoother and smoother functions. See Appendix B.2 for more discussions.

Based on these two assumptions, we can apply standard bounds on the KRR test error and obtain:

Theorem 1 and results of this type suffers from several limitations: 1) they are tight only in a minmax sense; 2) they do not provide comparisons for specific subclasses of functions; 3) in order to obtain the minmax rate, the regularization parameter λ\lambda has to be carefully tuned to balance the bias and variance terms, which is in contrast to modern practice where often the model is trained until interpolation. This led several groups to consider instead the test error of KRR in a high-dimensional limit and derive exact asymptotic predictions correct up to an additive vanishing constant for any f⋆∈L2f_{\star}\in L^{2} (see Appendix C for more details).

Using the general framework in , we get the following result for q,dq,d large:

where PE≤s,ν{\mathsf{P}}_{{\mathcal{E}}_{\leq{\mathsf{s}},\nu}} is the projection on the span of YSY_{S} with either ∣S∣<s|S|<{\mathsf{s}} and S∈E∣S∣S\in{\mathcal{E}}_{|S|} or ∣S∣=s|S|={\mathsf{s}} and γ(S)≤q(1−q−ν)\gamma(S)\leq q(1-q^{-\nu}).

See Appendix C.1 for a rigorous statement. In words, when dqs−1≪n≪dqsdq^{{\mathsf{s}}-1}\ll n\ll dq^{{\mathsf{s}}}, KRR with H\mboxCKH^{\mbox{\tiny\sf CK}} only learns a degree-s{\mathsf{s}} polynomial approximation to f⋆f_{\star}.

On the other hand, when considering the standard inner-product kernel H\mboxFCH^{\mbox{\tiny\sf FC}} (FC) we get:

where P≤s{\mathsf{P}}_{\leq{\mathsf{s}}} is the projection on the subspace of degree-s{\mathsf{s}} polynomials.

This theorem was proved in . Notice that Eq. (7) does not depend on the structure of f⋆f_{\star}. Hence, when f⋆∈L2(\mathscrsfsQd,Locq)f_{\star}\in L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q}), Theorems 2 and 3 shows a clear statistical advantage of H\mboxCKH^{\mbox{\tiny\sf CK}} over H\mboxFCH^{\mbox{\tiny\sf FC}} when q≪dq\ll d (and therefore of one-layer CNNs over fully-connected neural networks in the kernel regime).

3 Local average pooling and downsampling

In many applications such as object recognition, we expect the target function to depend mildly on the absolute spatial position of an object and to be stable under small shifts of the input. To take this local invariance into account, convolution layers are often followed by a pooling operation. Here we consider local average pooling on a segment of length ω\omega and obtain the kernel

Let Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} be a convolutional kernel with local average pooling as defined in Eq. (8). Then Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} admits the following eigendecomposition:

where (denoting k+Sk+S the translated set SS by kk positions with cyclic convention in [d][d])

First notice that, as long as gcd(ω,d)=1\mathsf{gcd}(\omega,d)=1, the RKHS associated to H\mboxCKH^{\mbox{\tiny\sf CK}} contains the same set of functions as the RKHS of H\mboxCKH^{\mbox{\tiny\sf CK}}, i.e., all local functions L2(\mathscrsfsQd,Locq)L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q}). (There are gcd(ω,d)−1\mathsf{gcd}(\omega,d)-1 number of zero weights: κj=0\kappa_{j}=0 for all j∈[d−1]j\in[d-1] such that dd is a divisor of jωj\omega. See Appendix A.3 for details.) However H\mboxCKH^{\mbox{\tiny\sf CK}} will penalize different frequency components of the functions differently. Denote fj(x)f_{j}({\bm{x}}) the jj-th component of the discrete Fourier transform of the function, i.e., fj(x)=1d∑k∈[d]ρjkf(tk⋅x)f_{j}({\bm{x}})=\frac{1}{\sqrt{d}}\sum_{k\in[d]}\rho_{j}^{k}f(t_{k}\cdot{\bm{x}}) where ρj=e2iπj/d\rho_{j}=e^{2i\pi j/d} and tk⋅x=(xk+1,…,xd,x1,…,xk)t_{k}\cdot{\bm{x}}=(x_{k+1},\ldots,x_{d},x_{1},\ldots,x_{k}) is the cyclic shift by kk pixels. Then H\mboxCKH^{\mbox{\tiny\sf CK}} reweights the eigenspaces associated with fj(x)f_{j}({\bm{x}}) by a factor κj\kappa_{j}, promoting low-frequency components (κj>1\kappa_{j}>1) and penalizing the high-frequencies (κj<1\kappa_{j}<1). In words, pooling biases the learning towards low-frequency functions, which are stable by small shifts.

Let us focus on two special choices here: the pooling parameter ω=1\omega=1 and ω=d\omega=d. When ω=1\omega=1, Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} reduces to H\mboxCKH^{\mbox{\tiny\sf CK}} (κj=1\kappa_{j}=1 for all j∈[d]j\in[d]) which does not bias towards either low or high frequency components. When ω=d\omega=d, we denote such kernel Hω=d\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega=d} by H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} which corresponds to global average pooling. In this case, we have κd=d\kappa_{d}=d and κj=0\kappa_{j}=0 for j<dj<d which enforces exact invariance under the group of cyclic translations. More precisely, H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} has RKHS that contains all cyclic q-local functions f(x)=∑k∈[d]g(x(k))∈L2(\mathscrsfsQd,CycLocq)f({\bm{x}})=\sum_{k\in[d]}g({\bm{x}}_{(k)})\in L^{2}({\mathscrsfs Q}^{d},{\rm CycLoc}_{q}) (c.f. Eq. (CYC-LOC)).

We obtain a bound on the test error of KRR with Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} similar to Theorem 1, but with dd replaced by an effective dimension d\mboxeffd^{\mbox{\tiny\rm eff}}.

By Jensen’s inequality, we have d\mboxeff≤d/ω1/αd^{\mbox{\tiny\rm eff}}\leq d/\omega^{1/\alpha}. In particular, for global pooling, d\mboxeff=1d^{\mbox{\tiny\rm eff}}=1 and the bound (11) does not depend on dd at all. Adding average pooling improve by a factor ω1/α\omega^{1/\alpha} the upper bound on the sample complexity for fitting low-frequency functions. Can we confirm this statistical advantage using the predictions for KRR in high dimension? Consider first the case of global pooling:

Hence, global average pooling results in an improvement by a factor dd in statistical efficiency when fitting cyclic local functions, compared to H\mboxCKH^{\mbox{\tiny\sf CK}}. This improvement was already noticed in but in the case of q=dq=d (fully connected neural networks).

For ω<d\omega<d, a direct application of the theorems in is more challenging because of the mixing of eigenvalues. In this case, a modification of , where eigenvalues are not necessary ordered anymore would apply. However, for simplicity, we present in Appendix C.1 a simplified kernel with non-overlapping local pooling which we believe captures the statistical behavior of local pooling. In this case, we show that Theorem 5 holds with n=(d/ω)⋅qs−1+νn=(d/\omega)\cdot q^{{\mathsf{s}}-1+\nu}, which interpolates between Theorem 2 (ω=1\omega=1) and Theorem 5 (ω=d\omega=d).

Often pooling is associated with a downsampling operation, which subsample one every Δ\Delta output coordinates. In Appendix A.4, we characterize the eigendecomposition of Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} (Proposition 4) and prove for the popular choice ω=Δ\omega=\Delta, that downsampling does not modify the cyclic invariant subspace j=dj=d (Proposition 5). More generally, we conjecture and check numerically that downsampling with Δ≤ω\Delta\leq\omega leaves the low-frequency eigenspaces approximately unchanged. In particular, the statistical complexity of learning low-frequency functions is not modified by downsampling operation in the one-layer case (while downsampling is potentially beneficial in further layers).

Numerical simulations

In order to check our theoretical predictions, we perform a simple numerical experiment on simulated data. We take x∼Unif(\mathscrsfsQd){\bm{x}}\sim{\rm Unif}({\mathscrsfs Q}^{d}) with d=30d=30, and consider two target functions:

Here f\mboxLF,3f_{\mbox{\tiny\sf LF},3} is a cyclic-invariant local polynomial (f\mboxLF,3f_{\mbox{\tiny\sf LF},3} is ‘low-frequency’). The function f\mboxHF,3f_{\mbox{\tiny\sf HF},3} is a high-frequency local polynomial, and is orthogonal to the space of cyclic invariant functions. On these target functions, we compare the test error of kernel ridge regression with 5 different kernels: a standard inner-product kernel H\mboxFC(x,y)=h(⟨x,y⟩/d)H^{\mbox{\tiny\sf FC}}({\bm{x}},{\bm{y}})=h(\langle{\bm{x}},{\bm{y}}\rangle/d); a cyclic invariant kernel H\mboxGP\mboxFC(x,y)H^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}}({\bm{x}},{\bm{y}}) (convolutional kernel with global pooling and full-size patches q=dq=d); a convolutional kernel H\mboxCKH^{\mbox{\tiny\sf CK}} with patch size q=10q=10; a convolutional kernel with local pooling Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} with q=10q=10 and ω=5\omega=5; and a convolutional kernel with global pooling H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} with q=10q=10. In all these kernels, we choose a common h(t)=∑i∈0.2∗tih(t)=\sum_{i\in}0.2*t^{i} which is a degree 55-polynomial.

In Figure 1, we report the test errors of fitting f\mboxLF,3f_{\mbox{\tiny\sf LF},3} (left) and f\mboxHF,3f_{\mbox{\tiny\sf HF},3} (right) using kernel ridge regression with these 55 kernels. We choose a small regularization parameter λ=10−6\lambda=10^{-6}, and the noise level σε=0\sigma_{\varepsilon}=0. The curves are averaged over 55 independent instances and the error bar stands for the standard deviation of these instances. The results match well our theoretical predictions. For the function f\mboxLF,3f_{\mbox{\tiny\sf LF},3}, the sample sizes required to achieve vanishing test errors are ordered as H\mboxGP\mboxCK<Hω\mboxCK<H\mboxCK<H\mboxGP\mboxFC<H\mboxFCH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf CK}}_{\omega}<H^{\mbox{\tiny\sf CK}}<H^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf FC}} and are around the predicted thresholds q2<dq2/ω<d2<dq2<d3q^{2}<dq^{2}/\omega<d^{2}<dq^{2}<d^{3} respectively. Next we look at the test error of fitting the high frequency local function f\mboxHF,3f_{\mbox{\tiny\sf HF},3}. The test errors of H\mboxCKH^{\mbox{\tiny\sf CK}} and H\mboxFCH^{\mbox{\tiny\sf FC}} are the same for f\mboxHF,3f_{\mbox{\tiny\sf HF},3} and f\mboxLF,3f_{\mbox{\tiny\sf LF},3}: this is because these kernels do not have bias towards either high-frequency or low-frequency functions. The kernel Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} perform worse on f\mboxHF,3f_{\mbox{\tiny\sf HF},3} than on f\mboxLF,3f_{\mbox{\tiny\sf LF},3}: this is because the eigenspaces of Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} are biased towards low-frequency polynomials. The kernels H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} and H\mboxGP\mboxFCH^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}} do not fit f\mboxHF,3f_{\mbox{\tiny\sf HF},3} at all (test error greater than or equal to 11): this is because the RKHS of these two kernels only contain cyclic polynomials, but f\mboxHF,3f_{\mbox{\tiny\sf HF},3} is orthogonal to the space of cyclic polynomials.

Discussion and Future Work

In this paper, we characterized in a stylized setting how convolution, average pooling and downsampling operations modify the RKHS, by restricting it to qq-local functions and then biasing the RKHS towards low-frequency components. We quantified precisely the gain in statistical efficiency of KRR using these operations. Beyond illustrating the ‘RKHS engineering’ of image-like function classes, these results can further provide intuition and a rigorous foundation for convolution and pooling operations in kernels and CNNs. A natural extension would be to study the multilayer convolutional kernels in details and consider other pooling operations such as max-pooling. Another important question is how anisotropy of the data impacts the results of this paper: in particular, it was shown that pre-processing (whitening of the patches) greatly improves the performance of convolutional kernels . A more challenging question is to study how training and feature learning can further improve the performance of CNNs outside the kernel regime.

References

Appendix A Details from the main text

A.2 Convolutional neural tangent kernel

In this section, we justify the expression of the convolutional neural tangent kernel Hw,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{{\bm{w}},\Delta} (CK-AP-DS), obtained as the tangent kernel of a neural network composed of a one convolution layer followed by local average pooling and downsampling (CNN-AP-DS).

For u,v∈\mathscrsfsQq{\bm{u}},{\bm{v}}\in{\mathscrsfs Q}^{q}, define

Computing the derivative of the convolutional neural network with respect to a=(aik0)i∈[N],k∈[d/Δ]{\bm{a}}=(a_{ik}^{0})_{i\in[N],k\in[d/\Delta]}, we have

Hence by law of large number, we have almost surely

Similarly, computing the derivative with respect to qW=(qwi0)i∈[N]\sqrt{q}{\bm{W}}=(\sqrt{q}{\bm{w}}_{i}^{0})_{i\in[N]} gives

By law of large number, using that aika_{ik} and aik′a_{ik^{\prime}} are independent of mean zero and variance 11, we get almost surely

Taking h=h(1)+h(2)h=h^{(1)}+h^{(2)} concludes the proof. ∎

A.3 Local average pooling operation

Consider a function f∈L2(\mathscrsfsQd)f\in L^{2}({\mathscrsfs Q}^{d}): we can decompose it as

where ρj=e2iπjd\rho_{j}=e^{\frac{2i\pi j}{d}} and tk⋅x=(xk+1,…,xd,x1,…,xk)t_{k}\cdot{\bm{x}}=(x_{k+1},\ldots,x_{d},x_{1},\ldots,x_{k}) is the cyclic shift of x{\bm{x}} by kk pixels. We can think about fj(x)f_{j}({\bm{x}}) as the jj-th component of the discrete Fourier transform of the function f(x)f({\bm{x}}) seen as a dd-dimensional vector {f(tk⋅x)}k∈[d]\{f(t_{k}\cdot{\bm{x}})\}_{k\in[d]} for any x∈\mathscrsfsQd{\bm{x}}\in{\mathscrsfs Q}^{d}.

Notice furthermore that if ff is a local function, i.e., ff can be decomposed as a sum of functions on patches f(x)=∑k∈[d]gk(x(k))f({\bm{x}})=\sum_{k\in[d]}g_{k}({\bm{x}}_{(k)}), then we can write

where we denoted (v∈\mathscrsfsQq{\bm{v}}\in{\mathscrsfs Q}^{q})

which shows that the jj-th frequency component fjf_{j} is in the span of {Yj,S}S⊆[q]\{Y_{j,S}\}_{S\subseteq[q]}. In particular, applying average pooling operation in the kernel will reweight this eigenspace by a factor κj\kappa_{j}.

Let us further comment on the values of κj\kappa_{j}. First, we have

In particular, the maximal eigenvalue is attained at j=dj=d with κd=ω\kappa_{d}=\omega, which corresponds to the subspace of cyclic invariant functions. Furthermore, κj=0\kappa_{j}=0 if and only if dd is a divisor of jωj\omega for j≤d−1j\leq d-1, i.e., jj is a multilple of gcd(ω,d)\mathsf{gcd}(\omega,d). There are gcd(ω,d)−1\mathsf{gcd}(\omega,d)-1 such zero eigenvalues.

where d(s)=min⁡(s,d−s)d(s)=\min(s,d-s) (the distance between k+sk+s and kk on [d][d] with cyclic convention). Note that Hτ\mboxCKH^{\mbox{\tiny\sf CK}}_{\tau} has the same eigendecomposition as Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} but with different weights κj\kappa_{j}.

A popular choice for τ\tau is the Gaussian filter τ(x)=12πσe−x22σ2\tau(x)=\frac{1}{\sqrt{2\pi}\sigma}e^{-\frac{x^{2}}{2\sigma^{2}}}. In Figure 2, we compare the eigenvalues κj\kappa_{j} for local average pooling and Gaussian filter with different value of ω\omega and σ2\sigma^{2}. Note that the eigenvalue decay controls how much high-frequencies are penalized: faster decay induces heavier penalty on the high-frequency components.

A.4 Downsampling operation

As mentioned in the main text, a downsampling operation is often added after pooling. The kernel is given by

Let us introduce the family {Mr}r∈[q]\{{\bm{M}}^{r}\}_{r\in[q]} of block-circulant matrices defined by

We can now state the eigendecomposition of Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} in terms of the eigenvalues and eigenvectors of the matrices {Mr}r∈[q]\{{\bm{M}}^{r}\}_{r\in[q]}.

Let Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} be a convolutional kernel with local average pooling and downsampling, as defined in Eq. (18). Then Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} admits the following eigendecomposition:

where ψj,SΔ(x)=∑k=1dvj,kSYk+S(x)\psi^{\Delta}_{j,S}({\bm{x}})=\sum_{k=1}^{d}v_{j,k}^{S}Y_{k+S}({\bm{x}}) with {κjS,vjS}j∈[d]\{\kappa_{j}^{S},{\bm{v}}_{j}^{S}\}_{j\in[d]} eigenvalues and eigenvectors of Mγ(S){\bm{M}}^{\gamma(S)}.

Let us make a few comments on these matrices Mγ(S){\bm{M}}^{\gamma(S)}. First because they only depend on SS through the diameter γ(S)\gamma(S), the eigenvalues and eigenvectors {κjS,vjS}j∈[d]\{\kappa_{j}^{S},{\bm{v}}_{j}^{S}\}_{j\in[d]} only depend on γ(S)\gamma(S). Second, we see that M(i+Δ)(j+Δ)γ(S)=Mijγ(S)M^{\gamma(S)}_{(i+\Delta)(j+\Delta)}=M^{\gamma(S)}_{ij} and Mijγ(S)=0M^{\gamma(S)}_{ij}=0 if d(i,j)≥ωd(i,j)\geq\omega, where d(i,j)=min⁡(∣i−j∣,d−∣i−j∣)d(i,j)=\min(|i-j|,d-|i-j|) (i.e., the distance between ii and jj on the torus [d][d]). In words Mγ(S){\bm{M}}^{\gamma(S)} is a symmetric block-circulant matrix with non-zero elements on a band of size ω−1\omega-1 on the left and right of the diagonal, and on the upper-right and lower-left corners. Furthermore, notice that

which is independent of ω,Δ,γ(S)\omega,\Delta,\gamma(S) and justify the chosen normalization. In particular, this implies that (for ξq,0=0\xi_{q,0}=0)

is also independent of the parameters (q,ω,Δ)(q,\omega,\Delta).

Take Δ=3\Delta=3, ω=5\omega=5, q=11q=11, then

The matrix Hj{\bm{H}}_{j} is Hermitian and we denote (λj,s)s∈[Δ](\lambda_{j,s})_{s\in[\Delta]} and (vj,s)s∈[Δ]({\bm{v}}_{j,s})_{s\in[\Delta]} its eigenvalues and eigenvectors. Then the eigenvalues and eigenvectors of M{\bm{M}} are given by {λj,s}j∈[m],s∈[Δ]\{\lambda_{j,s}\}_{j\in[m],s\in[\Delta]} and {γj(vj,s)}j∈[m],s∈[Δ]\{\gamma_{j}({\bm{v}}_{j,s})\}_{j\in[m],s\in[\Delta]}.

In particular, if Δ=1\Delta=1 and M=Circulant(b1,b2,…,bm){\bm{M}}=\mathsf{Circulant}(b_{1},b_{2},\ldots,b_{m}) is a circulant matrix, then the eigenvalues are simply given by

and eigenvectors vj=[1,ρj,⋯ ,ρjm−1]/m{\bm{v}}_{j}=[1,\rho_{j},\cdots,\rho_{j}^{m-1}]/\sqrt{m}.

Here we will focus on the impact of downsampling for single-layer convolutional kernels. We expect the downsampling operation to have a much more important role for the next layers: for example, increasing the scale of interactions or reducing the dimensionality of the pixel space.

To emphasize the dependency on ω,Δ\omega,\Delta, denote Mω,Δr{\bm{M}}^{r}_{\omega,\Delta} the matrix (19). We will study the change in the matrix Mω,1r{\bm{M}}^{r}_{\omega,1} when adding downsampling Δ\Delta, and consider

where we denote Aω,Δr=Mω,Δr−Mω,1r{\bm{A}}^{r}_{\omega,\Delta}={\bm{M}}^{r}_{\omega,\Delta}-{\bm{M}}^{r}_{\omega,1}. Notice that Aω,Δr{\bm{A}}^{r}_{\omega,\Delta} is a symmetric block-circulant matrix. Therefore, from Remark 1, the eigenvectors of Aω,Δr{\bm{A}}^{r}_{\omega,\Delta} are given by {γj(vj,s)}j∈[m],s∈[Δ]\{\gamma_{j}({\bm{v}}_{j,s})\}_{j\in[m],s\in[\Delta]} where d=mΔd=m\Delta and γj(vj,s)=[vj,s,ρm,jvj,s,…,ρm,jm−1vj,s]\gamma_{j}({\bm{v}}_{j,s})=[{\bm{v}}_{j,s},\rho_{m,j}{\bm{v}}_{j,s},\ldots,\rho_{m,j}^{m-1}{\bm{v}}_{j,s}] with ρm,j=e2iπjm\rho_{m,j}=e^{\frac{2i\pi j}{m}} and (vj,s)s∈[Δ]({\bm{v}}_{j,s})_{s\in[\Delta]} eigenvectors of Hj{\bm{H}}_{j} (23). The eigenvectors of Mω,1r{\bm{M}}^{r}_{\omega,1} are given by ut=[1,ρd,t,⋯ ,ρd,td−1]/d{\bm{u}}_{t}=[1,\rho_{d,t},\cdots,\rho_{d,t}^{d-1}]/\sqrt{d} with ρd,t=e2iπtd\rho_{d,t}=e^{\frac{2i\pi t}{d}}. Notice that

which is except when t≡j[m]t\equiv j[m]. Hence, we see that Aω,Δr{\bm{A}}^{r}_{\omega,\Delta} in Eq. (24) only modify the eigenspaces of Mω,1r{\bm{M}}^{r}_{\omega,1} as follows: the eigendirections {γj(vj,s)}j∈[m],s∈[Δ]\{\gamma_{j}({\bm{v}}_{j,s})\}_{j\in[m],s\in[\Delta]} coming from Hj{\bm{H}}_{j} (23) only modify the eigenspaces of Mω,1r{\bm{M}}^{r}_{\omega,1} spanned by {uam+j}a=0,…,Δ−1\{{\bm{u}}_{am+j}\}_{a=0,\ldots,\Delta-1}.

For simplicity, we will focus on the popular choice Δ=ω\Delta=\omega. Furthermore, we will only look at the impact of the eigenvalues H0{\bm{H}}_{0} on the eigenspace spanned by {uam}a=0,…,Δ−1\{{\bm{u}}_{am}\}_{a=0,\ldots,\Delta-1}, which contain the cyclic invariant direction. We show below that H0=0{\bm{H}}_{0}={\bm{0}} and therefore Aω,ωr{\bm{A}}^{r}_{\omega,\omega} does not modify the cyclic invariant eigenspace of Mω,1r{\bm{M}}^{r}_{\omega,1}:

Consider d=mωd=m\omega and the symmetric block-circulant matrix Aω,ωr=Mω,ωr−Mω,1r{\bm{A}}^{r}_{\omega,\omega}={\bm{M}}^{r}_{\omega,\omega}-{\bm{M}}^{r}_{\omega,1}. Denote Aω,ωr=Circulant(B1,B2,…,Bm){\bm{A}}^{r}_{\omega,\omega}=\mathsf{Circulant}({\bm{B}}_{1},{\bm{B}}_{2},\ldots,{\bm{B}}_{m}) and

If q+1−r≡0[ω]q+1-r\equiv 0[\omega], then Aω,ωr=0{\bm{A}}^{r}_{\omega,\omega}={\bm{0}}, and downsampling does not modify the matrix Mω,ωr=Mω,1r{\bm{M}}^{r}_{\omega,\omega}={\bm{M}}^{r}_{\omega,1}.

We have H0=0{\bm{H}}_{0}={\bm{0}} and downsampling does not modify the cyclic invariant eigenspace Aω,ωr1=0{\bm{A}}^{r}_{\omega,\omega}\bm{1}={\bm{0}}.

Let us first start by proving point (a). Consider q+1−r=pωq+1-r=p\omega. Fix i∈{0,…,Δ−1}i\in\{0,\ldots,\Delta-1\} and κ∈{0,…,ω−1}\kappa\in\{0,\ldots,\omega-1\}. Let us compute the entry (i,i+κ)(i,i+\kappa) of the matrix Mω,ωr{\bm{M}}^{r}_{\omega,\omega}: this amounts to counting the number of quadruples (k,s,s′,t)(k,s,s^{\prime},t) with k∈[d/ω]k\in[d/\omega], s,s′∈[ω]s,s^{\prime}\in[\omega] and 0≤t≤pω−10\leq t\leq p\omega-1, satisfying (kω+s+t,kω+s′+t)≡(i,i+κ)[d](k\omega+s+t,k\omega+s^{\prime}+t)\equiv(i,i+\kappa)[d]. Notice that we must have s′=s+κs^{\prime}=s+\kappa and therefore s∈{0,…,ω−1−κ}s\in\{0,\ldots,\omega-1-\kappa\}. Notice that for each interval uω≤t<(u+1)ωu\omega\leq t<(u+1)\omega with u∈{0,…,p−1}u\in\{0,\ldots,p-1\}, there are exactly ω−κ\omega-\kappa ways of choosing ss and then tt and kk to satisfy the equality. We deduce that

By symmetry of Mω,ωr{\bm{M}}^{r}_{\omega,\omega}, this concludes the proof of point (a).

Consider now point (b). First notice, because Mω,ωr{\bm{M}}^{r}_{\omega,\omega} has zero entries for min⁡(∣i−j∣,d−∣i−j∣)≥ω\min(|i-j|,d-|i-j|)\geq\omega, the only non-zero blocks are B1,B2{\bm{B}}_{1},{\bm{B}}_{2} and Bm{\bm{B}}_{m}. Furthermore, when computing H0{\bm{H}}_{0}, the diagonal entries only have one contribution from the diagonal elements of B1{\bm{B}}_{1}. The off-diagonal elements of H0{\bm{H}}_{0} have two contribution: one from B1{\bm{B}}_{1} and one from B2{\bm{B}}_{2} (if below the diagonal) or Bm{\bm{B}}_{m} (if above the diagonal), i.e.,

Let us compute first the diagonal elements: we have easily, by a similar argument as above (Mω,ωr)ii=1=(Mω,1r)ii({\bm{M}}^{r}_{\omega,\omega})_{ii}=1=({\bm{M}}^{r}_{\omega,1})_{ii}, and therefore H0{\bm{H}}_{0} has zero zero diagonal entries. For off-diagonal elements, first notice that (Mω,ωr)i(i+κ−ω)=(Mω,ωr)i(i+ω−κ)({\bm{M}}^{r}_{\omega,\omega})_{i(i+\kappa-\omega)}=({\bm{M}}^{r}_{\omega,\omega})_{i(i+\omega-\kappa)}. Then for q+1−r=pω+vq+1-r=p\omega+v, we can consider each subsegment uω≤t<(u+1)ωu\omega\leq t<(u+1)\omega separately, and by a simple counting argument, get (Mω,ωr)i(i+ω−κ)+(Mω,ωr)i(i+κ)=1−κω({\bm{M}}^{r}_{\omega,\omega})_{i(i+\omega-\kappa)}+({\bm{M}}^{r}_{\omega,\omega})_{i(i+\kappa)}=1-\frac{\kappa}{\omega}. We deduce that (H0)i(i+κ)=0({\bm{H}}_{0})_{i(i+\kappa)}=0, which by symmetry implies H0=0{\bm{H}}_{0}={\bm{0}} and concludes the proof. ∎

From the above result, we conjecture that more generally, for Δ≤ω\Delta\leq\omega, the low-frequency eigenspaces of Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} remain approximately unchanged when applying a downsampling operation. We verify this conjecture numerically in several examples. In Figure 3, we plot the eigenvalues κj\kappa_{j} with and without downsampling. On the left, we compare κj\kappa_{j} for fixed ω=25\omega=25 and increasing Δ\Delta. We notice that the eigenvalues do not change much for Δ≤ω\Delta\leq\omega, and for Δ>ω\Delta>\omega, some κj\kappa_{j} become null, as discussed above. On the right, we plot κj\kappa_{j} for Δ=1\Delta=1 (continuous line) and Δ=ω\Delta=\omega (dashed lines) for several ω\omega. As conjectured, the top eigenvalues (low-frequency) are left approximately unchanged. In Figure 4, we plot a heatmap of the eigenvectors ordered vertically from highest associated eigenvalue (bottom) to lowest (top) for a fixed ω=25\omega=25 and increasing downsampling Δ∈{1,25,40}\Delta\in\{1,25,40\}. First indeed check that the top eigenvectors correspond to low-frequency functions and the bottom eigenvectors correspond to high-frequency functions. Second, most eigenvectors are not much modified between Δ=1\Delta=1 and Δ=ω=25\Delta=\omega=25. For the case, Δ>ω\Delta>\omega, the top eigenvectors corresponds still low-frequency functions.

From these observations, we expect Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} to have the same statistical properties as Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} when learning low-frequency functions. In Figure 5, we plot the test error of kernel ridge regression for fitting cyclic qq-local polynomials (see Section A.7) on the hypercube of dimension d=30d=30. We report the test error of one realization, against the sample size nn, and choose regularization λ=10−6\lambda=10^{-6} and noise σε=0\sigma_{\varepsilon}=0. We compare kernels with and without downsampling. On the left, we consider q=10q=10 and ω=Δ=5\omega=\Delta=5, and compare the test error with Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} (continous line) and with Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} (dashed line) when learning degree 22, 33 and 44 polynomials. On the right, we fix the target function to be the cubic local cyclic polynomial and consider the test error of learning with Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} for q=10q=10, ω=10\omega=10, and Δ∈{1,3,6,10}\Delta\in\{1,3,6,10\}. As expected, we observe in both simulations that the test error is almost identical between the kernels with and without downsampling, when learning cyclic invariant functions.

In Section C.1, we further check that downsampling with Δ>ω\Delta>\omega does not improve the high-dimensional predictions for the test error of KRR.

A.5 Multilayer convolutional kernels

For completeness, we briefly discuss here some intuitions of multilayer convolutional kernels. The benefit of depth in convolutional kernels has been investigated in . In particular, observed that the top layer operation of a two-layers convolutional kernel can be replaced by a low-degree polynomial without a performance change.

As an example, we will consider a two layers convolutional kernel with patch and local average pooling sizes (q1,ω1)(q_{1},\omega_{1}) on the first layer and (q2,ω2)(q_{2},\omega_{2}) on the second layer. We consider a general inner-product kernel for the first layer:

Let us decompose this two-layers convolutional kernel in the Fourier basis. Let Ψ(x)={Ψk(x)}k∈[d]\Psi({\bm{x}})=\{\Psi_{k}({\bm{x}})\}_{k\in[d]} be the output of the first layer, with

Then denoting Ψ(k)(x)=(Ψk+1(x),…,Ψk+q2(x))\Psi_{(k)}({\bm{x}})=(\Psi_{k+1}({\bm{x}}),\ldots,\Psi_{k+q_{2}}({\bm{x}})), the two-layers convolutional kernel is given by

We believe that techniques contained in this paper can be used to study kernels of the type (27) by a careful combinatorial argument and a 2-dimensional Fourier transform on the second layer (see ). We leave this problem to future work. Here we only comment on the structure of Hω1,ω22\mboxCKH^{2\mbox{\tiny\sf CK}}_{\omega_{1},\omega_{2}}:

Including a second convolutional layer allows interactions between patches. The associated RKHS, which we will denote H2\mboxCK\mathcal{H}^{2\mbox{\tiny\sf CK}}, contains all the homogeneous polynomials YSY_{S} with S=S1∪S2S=S_{1}\cup S_{2} with S1S_{1}, S2S_{2} contained on segments of size q1q_{1}, with the two segments separated by at most q2+ω2−2q_{2}+\omega_{2}-2. In words, the RKHS contains interaction between patches x(k){\bm{x}}_{(k)} and x(k′){\bm{x}}_{(k^{\prime})} that are within some distance.

The eigenvalue associated to a degree-kk homogeneous polynomials is still of order q−kq^{-k} in high-dimension. To learn functions restricted to L2(\mathscrsfsQ2,Locq)L^{2}({\mathscrsfs Q}^{2},{\rm Loc}_{q}), it is statistically more efficient to use H\mboxCKH^{\mbox{\tiny\sf CK}} (smaller degeneracy of eigenvalues). However H2\mboxCKH^{2\mbox{\tiny\sf CK}} will fit a richer class of functions with two-patch interactions, while still not being plagued by dimensionality: dim⁡(H2\mboxCK)≤q2d22q1\dim(\mathcal{H}^{2\mbox{\tiny\sf CK}})\leq q_{2}d2^{2q_{1}}. Hence we still expect H2\mboxCKH^{2\mbox{\tiny\sf CK}} to be much more statistically efficient than a standard inner-product kernel.

Local pooling on the two layers plays different roles: pooling on the first layer encourages the interactions to not depend strongly on the relative positions of the patches, while pooling on the second layer penalizes functions that depend on the global position of these interactions.

For more layers and higher degree kernels, one obtain hierarchical interactions of higher-order, with multi-scale absolute and relative local invariances brought by pooling layers.

A.6 Proofs diagonalization of convolutional kernels

In this section, we prove the diagonalization of the kernels H\mboxCKH^{\mbox{\tiny\sf CK}}, Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} and Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} introduced in Propositions 1, 2 and 4 respectively.

By the spectral theorem of compact operators, there exists an orthonormal basis (ψj)j≥1(\psi_{j})_{j\geq 1} of L2(X,τ)L^{2}({\mathcal{X}},\tau) and eigenvalues (λj)j≥1(\lambda_{j})_{j\geq 1}, with nonincreasing values λ1≥λ2≥⋯≥0\lambda_{1}\geq\lambda_{2}\geq\cdots\geq 0 and ∑j≥1λj<∞\sum_{j\geq 1}\lambda_{j}<\infty, such that

We first prove the diagonalization of Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} in Proposition 4. The case of Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} and H\mboxCKH^{\mbox{\tiny\sf CK}} then follows by setting Δ=1\Delta=1, and Δ=ω=1\Delta=\omega=1 respectively.

Using Eq. (29) and that YS(x(k))=Yk+S(x)Y_{S}({\bm{x}}_{(k)})=Y_{k+S}({\bm{x}}), we have the following decomposition of Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} in the Fourier basis

where we recall the definition of the set of indices

which concludes the proof of Proposition 4. ∎

We can now prove Propositions 1 and 2 by taking ω=Δ=1\omega=\Delta=1 and Δ=1\Delta=1 respectively.

Set Δ=ω=1\Delta=\omega=1 in Proposition 4. We get

In this case, Mγ(S){\bm{M}}^{\gamma(S)} is simply equal to identity, which concludes the proof. ∎

where d(i,j)d(i,j) is the distance between ii and jj on the torus [d][d] (i.e., if i>ji>j, d(i,j)=min⁡(i−j,d+j−i)d(i,j)=\min(i-j,d+j-i)). Hence, Mγ(S){\bm{M}}^{\gamma(S)} is a circulant matrix independent of γ(S)\gamma(S), which has well known explicit formula for eigenvalues and eigenvectors (see for example Remark 1). ∎

A.7 Additional numerical simulations

Here, we consider a numerical experiment similar to Figure 1. We consider x∼Unif(\mathscrsfsQd){\bm{x}}\sim{\rm Unif}({\mathscrsfs Q}^{d}) with d=30d=30 and consider three cyclic invariant target functions:

We consider a higher order polynomial kernel h(x)=∑k∈0.2⋅xkh(x)=\sum_{k\in}0.2\cdot x^{k} than in Figure 1, which should lead to higher self-induced regularization. We consider the same kernels as before, with q=10q=10 and ω=5\omega=5.

In Figure 6, we report the test errors of fitting f2f_{2} (top), f3f_{3} (middle) and f4f_{4} (bottom) using kernel ridge regression with the 55 kernels of interests in the main text. We choose a small regularization parameter λ=10−6\lambda=10^{-6}, and the noise level σε=0\sigma_{\varepsilon}=0. The curves are averaged over 55 independent instances and the error bar stands for the standard deviation of these instances. The results again match with our overall theoretical predictions. We report the predicted thresholds for the three functions:

For f2f_{2} target: q<d<dq/ω<dq<d2q<d<dq/\omega<dq<d^{2} for H\mboxGP\mboxCK<H\mboxGP\mboxFC<Hω\mboxCK<H\mboxCK<H\mboxFCH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf CK}}_{\omega}<H^{\mbox{\tiny\sf CK}}<H^{\mbox{\tiny\sf FC}}.

For f3f_{3} target: q2<dq2/ω<d2<dq2<d3q^{2}<dq^{2}/\omega<d^{2}<dq^{2}<d^{3} for H\mboxGP\mboxCK<Hω\mboxCK<H\mboxCK<H\mboxGP\mboxFC<H\mboxFCH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf CK}}_{\omega}<H^{\mbox{\tiny\sf CK}}<H^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf FC}}.

For f4f_{4} target: q3<dq3/ω<d3<dq3<d4q^{3}<dq^{3}/\omega<d^{3}<dq^{3}<d^{4} for H\mboxGP\mboxCK<Hω\mboxCK<H\mboxCK<H\mboxGP\mboxFC<H\mboxFCH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf CK}}_{\omega}<H^{\mbox{\tiny\sf CK}}<H^{\mbox{\tiny\sf FC}}_{\mbox{\tiny\sf GP}}<H^{\mbox{\tiny\sf FC}}.

We see that the kernels, especially for f4f_{4}, perform much better than their theoretical high-dimension predictions: this can be explained by the low-dimensionality of the experiment where q=10q=10.

Appendix B Generalization error of kernel methods in fixed dimension

We first consider the case of a Lipschitz bounded loss and uniform convergence, and make a few simple remarks on the connection between generalization error and eigendecomposition in kernel methods.

The generalization error of f^B\hat{f}_{B} has the following standard bound on the Rademacher complexity of the kernel class {f:∥f∥H≤B}\{f:\|f\|_{\mathcal{H}}\leq B\} : with probability 1−δ1-\delta,

Note that instead of a constraint on the norm in Eq. (33), one might find more convenient to use a penalty. In that case, there exists an equivalent to the bound (34) , but we focus here on the constrained formulation for simplicity.

Consider Hω,Δ\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega,\Delta} as in Eq. (8) and assume ξq,0=0\xi_{q,0}=0. From the normalization choice of the kernel (see Eq. (22)), we have

Consider now for simplicity Δ=1\Delta=1. From the eigendecomposition in Proposition 2, the RKHS norm of f∈L2(\mathscrsfsQd,Locq)f\in L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q}) is given by

We make the following two remarks on this bound:

It depends on ∥g∥h\|g\|_{h}, which is a RKHS norm on \mathscrsfsQq{\mathscrsfs Q}^{q} instead of \mathscrsfsQd{\mathscrsfs Q}^{d}, which has potentially much lower dimension and contain less smooth function for balls of same radius.

There is a factor κj\kappa_{j} gain in sample complexity when learning functions that have jj-th frequency with κj>1\kappa_{j}>1. In particular, for j=dj=d (cyclic invariant functions), κj=ω\kappa_{j}=\omega, and we need ω\omega less samples to get the same (upper) bound on the generalization error. On the contrary, when κj<1\kappa_{j}<1, i.e., high-frequency oscillatory functions, the generalization bound becomes worse.

B.2 Generalization error of KRR in the classical regime

We consider here the regression setting which allows for finer results. Several works have considered bounding the generalization error of kernel ridge regression (KRR) , [47, Theorem 13.17]. In this section, we consider the following fully-explicit upper bound from .

which has the following analytical formula:

where H=(H(xi,xj))ij∈[n]{\bm{H}}=(H({\bm{x}}_{i},{\bm{x}}_{j}))_{ij\in[n]} is the empirical kernel matrix, h(x)=[H(x,x1),…,H(x,xn)]{\bm{h}}({\bm{x}})=[H({\bm{x}},{\bm{x}}_{1}),\ldots,H({\bm{x}},{\bm{x}}_{n})] and y=(y1,…,yn){\bm{y}}=(y_{1},\ldots,y_{n}). The risk is taken to be the test error with squared error loss

[3, Theorem 7.2] Assume H(x,x)≤R2H({\bm{x}},{\bm{x}})\leq R^{2} almost surely and let the regularization parameter λ≤R2\lambda\leq R^{2}. If n≥5R2λ(1+log⁡R2λ)n\geq\frac{5R^{2}}{\lambda}\left(1+\log\frac{R^{2}}{\lambda}\right), then

Let us comment on the upper-bound in Eq. (36). The first term corresponds to an upper bound on the variance: N(H,λ)\mathcal{N}(H,\lambda) is sometimes called the degrees of freedom or the effective dimension of the kernel HH. The second term bounds the bias term and corresponds to an approximation error. In particular, for any r>0r>0,

From the above discussion, it is natural to consider the following two assumptions on HH and f⋆f_{\star}, that are standard in the kernel literature:

Capacity condition: N(H,λ)≤CHλ−1/α\mathcal{N}(H,\lambda)\leq C_{H}\lambda^{-1/\alpha} with α>1\alpha>1.

Intuitively, the capacity condition (B1) characterizes the size of the RKHS: for increasing α\alpha, the RKHS contains less and less functions. It is verified when the eigenvalues λj\lambda_{j}’s of HH decay at the rate j−αj^{-\alpha}. For example, taking the Matern kernel of order s>d/2s>d/2, whose RKHS is the Sobolev space of order ss (i.e., functions with bounded ss-order derivatives), we have α=2s/d\alpha=2s/d (e.g., see ). The source condition (B2) characterizes the regularity of the target function (the ‘source’) with respect to the kernel: β=1\beta=1 is equivalent to f⋆∈Hf_{\star}\in\mathcal{H}, while β>1\beta>1 corresponds to f⋆f_{\star} more smooth (and β<1\beta<1 less smooth f⋆f_{\star}).

Assuming (B1) and (B2) in Theorem 6, we get the bound

where in the second line, we balanced the two terms by taking λ∗:=(CHσε2Bf⋆2n)ααβ+1\lambda_{*}:=\left(\frac{C_{H}\sigma_{\varepsilon}^{2}}{B_{f_{\star}}^{2}n}\right)^{\frac{\alpha}{\alpha\beta+1}}. Note that in order to use Theorem 6, we need further to constrain n≥5R2λ(1+log⁡R2λ)n\geq\frac{5R^{2}}{\lambda}\left(1+\log\frac{R^{2}}{\lambda}\right). For simplicity, we will choose r>α−1αr>\frac{\alpha-1}{\alpha}, so that this condition is verified for nn sufficiently large.

The rate in nn in Eq. (38) is minmax optimal over all functions that verify assumptions (A1) and (A2) . However, for large dd, the RKHS is composed of very smooth functions (e.g., Sobolev spaces of order ss are RKHS if and only if s>d/2s>d/2, i.e., if the order of the bounded derivatives grows with the dimension dd) and β\beta will be small, such that βα≈κ/d\beta\alpha\approx\kappa/d for functions with bounded derivatives up to order κ\kappa. In that case, the risk decreases at the rate n−O(κd)n^{-O(\frac{\kappa}{d})}: KRR suffers from the curse of dimensionality when κ\kappa does not scale with dd. As a consequence, the bound (38) is vacuous when nn does not scale exponentially in dd, which led several groups to derive finer bounds on KRR in the high dimensional regime (see Section C).

Let us now apply Theorem 6 and Eq. (38) to our convolutional kernels to show Theorems 1 and 4.

First notice that H\mboxCK(x,x)=h(1)=:R2H^{\mbox{\tiny\sf CK}}({\bm{x}},{\bm{x}})=h(1)=:R^{2} and we can therefore apply Theorem 6. The effective dimension of H\mboxCKH^{\mbox{\tiny\sf CK}} is bounded by

Injecting the two above bounds in Eq. (38), we deduce that there exists constants C1,C2,C3C_{1},C_{2},C_{3} that only depends on the constants in (A1) and (A2), and h(1),σε2h(1),\sigma^{2}_{\varepsilon} (but independent of dd), such that taking n≥C1max⁡(∥f⋆∥L∞2,d)n\geq C_{1}\max(\|f_{\star}\|^{2}_{L^{\infty}},d) and λ∗=C2d(d/n)ααβ+1\lambda_{*}=\frac{C_{2}}{d}(d/n)^{\frac{\alpha}{\alpha\beta+1}}, we get

The proof is similar to the proof of Theorem 1. Notice that Hω\mboxCK(x,x)≤h(1)H^{\mbox{\tiny\sf CK}}_{\omega}({\bm{x}},{\bm{x}})\leq h(1), and that the effective dimension of Hω\mboxCKH^{\mbox{\tiny\sf CK}}_{\omega} is bounded by

where we used condition (A1). Denoting d\mboxeff=∑j=1d(κj/ω)1/αd_{\mbox{\tiny\rm eff}}=\sum_{j=1}^{d}(\kappa_{j}/\omega)^{1/\alpha}, the rest of the proof follows from the proof of Theorem 1 with dd replaced by d\mboxeffω1/αd_{\mbox{\tiny\rm eff}}\omega^{1/\alpha} and B2B^{2} replaced by ωβB2\omega^{\beta}B^{2}. ∎

Appendix C Generalization error of KRR in high dimension

In Section B.2, we considered upper bounds on the test error of KRR using the standard capacity and source conditions. However, these results suffer from several limitations:

They only provide an upper bound on the test error. While the decay rate with respect to nn is minmax optimal (see ), this is not strong enough to show, for example, a statistical advantage of using local average pooling, which appears as a prefactor d\mboxeffd_{\mbox{\tiny\rm eff}}, and which would require a lower bound matching the upper bound within a constant factor.

As mentioned in Remark 2, the bound is of order n−1/O(d)n^{-1/O(d)}, except when the target function has smoothness order increasing with dd. This bound is non-vacuous only if n=exp⁡(O(d))n=\exp(O(d)) which is impractical in modern image datasets where typically d≥100d\geq 100. This motivates a new type of question: given n≍dαn\asymp d^{\alpha}, what is the prediction error achieved by KRR for a given function?

In order to achieve the bound Eq. (38), one need to carefully balance the bias and the variance terms by setting the regularization parameter. This is in contrast with modern practice which usually train until interpolation (which corresponds to setting λ→0\lambda\to 0).

Given the above limitations, several recent works have instead considered a high-dimensional setting where the number of samples scales with dd, and derived asymptotic test errors, exact up to a vanishing additive error . In addition to these works, several papers have derived general estimates for the test error using non-rigorous methods that are believe to be correct in the high dimensional limit and which show great agreement with numerical experiments. The picture that emerges in this regime is much more precise than in the classical regime: KRR approximately acts as a shrinkage operator on the target function (not assumed to be in a particular space anymore), with shrinkage parameter that scales as a self-induced regularization parameter over the number of samples.

for some δ>0\delta>0. Then, assuming some additional conditions insuring that the kernel HdH_{d} is ‘spread-out’ and well behaved, the KRR solution

is equal up to a vanishing additive L2L^{2}-error (as d→∞d\to\infty) to the following effective ridge regression estimator

The solution of Eq. (40) admits an explicit solution in terms of a shrinkage operator in the basis (ψd,j)j≥1(\psi_{d,j})_{j\geq 1} of eigenfunctions of HdH_{d}:

Hence, KRR will fit better the target function along eigendirections associated to larger eigenvalues of HH. If λd,j≫λ\mboxeff/n\lambda_{d,j}\gg\lambda_{\mbox{\tiny\rm eff}}/n, KRR fits perfectly f⋆f_{\star} along the eigendirection ψd,j\psi_{d,j}, while if λd,j≪λ\mboxeff/n\lambda_{d,j}\ll\lambda_{\mbox{\tiny\rm eff}}/n, KRR does not fit this eigendirection at all. This phenomena has been referred as the spectral bias and task-kernel alignment of kernel ridge regression in several works.

Finally, notice from Eq. (41) that the minimum test error is achieved for the regularization parameter λ=0\lambda=0, which corresponds to the KRR estimator fitting perfectly the training data. In other words, the interpolating solution is optimal for kernel ridge regression in high dimension.

we first consider a vanilla one-layer convolutional kernel H\mboxCKH^{\mbox{\tiny\sf CK}} as defined in Eq. (3). We will assume that the kernels {hq}q≥1\{h_{q}\}_{q\geq 1} verify the following ‘genericity’ condition.

Assumption 1 will be verified by standard kernels, e.g., the Gaussian kernel. We discuss this assumption in Section C.2 and present sufficient conditions on the activation function σ\sigma for its associated CNTK to verify Assumption 1.

for any u,v∈\mathscrsfsQq{\bm{u}},{\bm{v}}\in{\mathscrsfs Q}^{q}. The following result is a consequence of the general theorem on the generalization error of KRR in .

Let {fd∈L2(\mathscrsfsQd,Locq)}q≥1\{f_{d}\in L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q})\}_{q\geq 1} be a sequence of local functions. Let (xi)i∈[n(d)]∼\mboxi.i.d.Unif(\mathscrsfsQd)({\bm{x}}_{i})_{i\in[n(d)]}\sim_{\mbox{\tiny\sf i.i.d.}}{\rm Unif}({\mathscrsfs Q}^{d}) and yi=fd(xi)+εiy_{i}=f_{d}({\bm{x}}_{i})+\varepsilon_{i} with εi∼\mboxi.i.d.N(0,σε2)\varepsilon_{i}\sim_{\mbox{\tiny\sf i.i.d.}}{\sf N}(0,\sigma_{\varepsilon}^{2}). Assume d⋅qs−1+δ≤n≤d⋅qs−δd\cdot q^{{\mathsf{s}}-1+\delta}\leq n\leq d\cdot q^{{\mathsf{s}}-\delta} for some δ>0\delta>0 and let {hq}q≥1\{h_{q}\}_{q\geq 1} be a sequence of activation functions satisfying Assumption 1 at level s{\mathsf{s}}. Consider {H\mboxCK,d}q≥1\{H^{\mbox{\tiny\sf CK},d}\}_{q\geq 1} the sequence of convolutional kernels associated to {hq}q≥1\{h_{q}\}_{q\geq 1} as defined in Eq. (3). Then the following holds for the solution f^λ\hat{f}_{\lambda} of KRR with kernels {H\mboxCK,d}q≥1\{H^{\mbox{\tiny\sf CK},d}\}_{q\geq 1}.

For any regularization parameter λ≥0\lambda\geq 0, define the effective regularization λ\mboxeff:=λ+hq,>s(1)\lambda_{\mbox{\tiny\rm eff}}:=\lambda+h_{q,>{\mathsf{s}}}(1). Then for any η>0\eta>0, we have

The proof of Theorem 7 is deferred to Section C.4.

Let us expound on the predictions of Theorem 7. First, recall that f^λ\mboxeff\mboxeff\hat{f}_{\lambda_{\mbox{\tiny\rm eff}}}^{\mbox{\tiny\rm eff}} is given explicitly in Eq. (41) by a shrinkage operator with parameter λ\mboxeff\lambda_{\mbox{\tiny\rm eff}}. From Assumption 1 and taking λ=0\lambda=0, the shrinkage operator is of order 11

KRR fits the eigendirections corresponding to the homogeneous polynomials of degree s−1{\mathsf{s}}-1 and less, and of degree s{\mathsf{s}} for subsets SS such that γ(S)≪q−q1−α\gamma(S)\ll q-q^{1-\alpha}.

KRR does not fit at all the eigendirections correpsonding to homogeneous polynomials of degree s+1{\mathsf{s}}+1 and larger, and degree s{\mathsf{s}} for subsets SS such that γ(S)≫q−q1−α\gamma(S)\gg q-q^{1-\alpha}.

In words, for d⋅qs−1≪n≪d⋅qsd\cdot q^{{\mathsf{s}}-1}\ll n\ll d\cdot q^{{\mathsf{s}}}, KRR fits at least a degree-(s−1)({\mathsf{s}}-1) polynomial approximation to f⋆f_{\star} and at most a degree-s{\mathsf{s}} polynomial approximation. As nn increases from d⋅qs−1d\cdot q^{{\mathsf{s}}-1} to d⋅qsd\cdot q^{{\mathsf{s}}}, KRR first fits degree-s{\mathsf{s}} homogeneous polynomials that have smaller diameter γ(S)\gamma(S) (i.e., ‘more localized’).

Test error of CK with global average pooling:

we consider the kernel H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} given by a convolutional layer followed by global average pooling:

In addition to the genericity condition, we will assume that the kernels {hq}q≥1\{h_{q}\}_{q\geq 1} verify the following differentiability condition.

where we denoted hq,>vh_{q,>v} the truncated inner-product kernel hqh_{q} as in Eq. (45).

Assumption 2 is used to extend the following theorem to non-polynomial kernel hqh_{q} (in particular, it is trivially verified for polynomial kernels by taking vv larger than the degree of hqh_{q}). This assumption is difficult to check in practice, however we provide some examples where it holds in Appendix C.2.

Let {fd∈L2(\mathscrsfsQd,CycLocq)}q≥1\{f_{d}\in L^{2}({\mathscrsfs Q}^{d},{\rm CycLoc}_{q})\}_{q\geq 1} be a sequence of convolutional functions. Assume qs−1+δ≤n≤qs−δq^{{\mathsf{s}}-1+\delta}\leq n\leq q^{{\mathsf{s}}-\delta} for some δ>0\delta>0 and let {hq}q≥1\{h_{q}\}_{q\geq 1} be a sequence of activation functions satisfying Assumptions 1 and 2 at level s{\mathsf{s}}. Consider {H\mboxGP\mboxCK,d}q≥1\{H^{\mbox{\tiny\sf CK},d}_{\mbox{\tiny\sf GP}}\}_{q\geq 1} the sequence of convolutional kernels with global pooling associated to {hq}q≥1\{h_{q}\}_{q\geq 1} as defined in Eq. (47). Then the solution f^λ\hat{f}_{\lambda} of KRR with kernels {H\mboxGP\mboxCK,d}q≥1\{H^{\mbox{\tiny\sf CK},d}_{\mbox{\tiny\sf GP}}\}_{q\geq 1} verifies Eq. (46) with λ\mboxeff:=λ+hq,>s(1)\lambda_{\mbox{\tiny\rm eff}}:=\lambda+h_{q,>{\mathsf{s}}}(1).

The proof of Theorem 8 is deferred to Section C.5.

The predictions of Theorem 8 are similar to the ones of Theorem 7 but with a factor dd gain in statistical efficiency: this is due to the eigenvalues of H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} being a factor dd larger than for H\mboxCKH^{\mbox{\tiny\sf CK}}. Therefore, with global average pooling, for qs−1≪n≪qsq^{{\mathsf{s}}-1}\ll n\ll q^{{\mathsf{s}}}, KRR fits at least a degree-(s−1)({\mathsf{s}}-1) invariant polynomial approximation to f⋆f_{\star} and at most a degree-s{\mathsf{s}} invariant polynomial approximation. As nn increases from qs−1q^{{\mathsf{s}}-1} to qsq^{{\mathsf{s}}}, KRR first degree-s{\mathsf{s}} invariant homogeneous polynomials with increasing diameter γ(S)\gamma(S).

Test error of CK with local average pooling:

Assume q≤ω/2q\leq\omega/2 and ω\omega is a divisor of dd. Denote x(kω)=(xkω+1,…,xkω+ω){\bm{x}}^{(k\omega)}=(x_{k\omega+1},\ldots,x_{k\omega+\omega}) the kk-th segment of length ω\omega in [d][d] and x(i)(kω)=(xkω+i,…,xkω+q+i){\bm{x}}_{(i)}^{(k\omega)}=(x_{k\omega+i},\ldots,x_{k\omega+q+i}) the patch of size qq with cyclic convention in {kω+1,…,kω+ω}\{k\omega+1,\ldots,k\omega+\omega\}. Consider the following convolutional kernel with ‘non-overlapping’ average pooling:

In words, Hω\mboxCK,\mboxNOH^{\mbox{\tiny\sf CK},\mbox{\tiny\sf NO}}_{\omega} is the combination of d/ωd/\omega non-overlapping convolutional kernels with global average pooling on images of size ω\omega:

where ψk,S(x)=1ω∑i∈[ω]Yi+S(x(kω))\psi_{k,S}({\bm{x}})=\frac{1}{\sqrt{\omega}}\sum_{i\in[\omega]}Y_{i+S}({\bm{x}}^{(k\omega)}) where i+Si+S is the translated set with cyclic convention in [ω][\omega].

Denote L2(\mathscrsfsQd,LocCycLocq)L^{2}({\mathscrsfs Q}^{d},{\rm Loc}{\rm CycLoc}_{q}) the RKHS associated to Hω\mboxCK,\mboxNOH^{\mbox{\tiny\sf CK},\mbox{\tiny\sf NO}}_{\omega}, which contains functions that are locally convolutions on segments of size ω\omega. For this simplified model, the proof of Theorem 8 can be easily adapted and we obtain the following result:

Let {fd∈L2(\mathscrsfsQd,LocCycLocq)}q≥1\{f_{d}\in L^{2}({\mathscrsfs Q}^{d},{\rm Loc}{\rm CycLoc}_{q})\}_{q\geq 1} be a sequence of local convolutional functions. Assume (d/ω)⋅qs−1+δ≤n≤(d/ω)⋅qs−δ(d/\omega)\cdot q^{{\mathsf{s}}-1+\delta}\leq n\leq(d/\omega)\cdot q^{{\mathsf{s}}-\delta} for some δ>0\delta>0 and let {hq}q≥1\{h_{q}\}_{q\geq 1} be a sequence of activation functions satisfying Assumptions 1 and 2 at level s{\mathsf{s}}. Consider {Hω\mboxCK,\mboxNO,d}q≥1\{H^{\mbox{\tiny\sf CK},\mbox{\tiny\sf NO},d}_{\omega}\}_{q\geq 1} the sequence of convolutional kernels with non-overlapping pooling associated to {hq}q≥1\{h_{q}\}_{q\geq 1} as defined in Eq. (48). Then the solution f^λ\hat{f}_{\lambda} of KRR with kernels {Hω\mboxCK,\mboxNO,d}q≥1\{H^{\mbox{\tiny\sf CK},\mbox{\tiny\sf NO},d}_{\omega}\}_{q\geq 1} verifies Eq. (46) with λ\mboxeff:=λ+dωhq,>s(1)\lambda_{\mbox{\tiny\rm eff}}:=\lambda+\frac{d}{\omega}h_{q,>{\mathsf{s}}}(1).

Corollary 1 shows that Hω\mboxCK,\mboxNOH^{\mbox{\tiny\sf CK},\mbox{\tiny\sf NO}}_{\omega} enjoys a factor ω\omega gain in statistical efficiency compared to H\mboxCKH^{\mbox{\tiny\sf CK}}, due to a factor ω\omega smaller effective ridge regularization. Therefore, with (non-overlapping) local average pooling, for (d/ω)⋅qs−1≪n≪(d/ω)⋅qs(d/\omega)\cdot q^{{\mathsf{s}}-1}\ll n\ll(d/\omega)\cdot q^{{\mathsf{s}}}, KRR fits degree-(s−1)({\mathsf{s}}-1) locally invariant polynomials and none of the polynomials of degree-(s+1)({\mathsf{s}}+1) and larger. Heuristically, we see that this yields the same statistical efficiency than H\mboxCKH^{\mbox{\tiny\sf CK}} for ω=1\omega=1 and H\mboxGP\mboxCKH^{\mbox{\tiny\sf CK}}_{\mbox{\tiny\sf GP}} for ω=d\omega=d, and interpolates between the two cases for 1<ω<d1<\omega<d.

Test error of convolutional kernels with downsampling:

We consider adding a downsampling operation to the previous kernels. Let Δ\Delta be a constant and a divisor of dd and ω\omega and consider the following ‘downsampled’ kernels:

We can easily adapt the proofs of Theorems 7 and 8, and Corollary 1 to these kernels. In particular, their conclusions do not change (for any constant Δ\Delta) and downsampling do not provide a statistical advantage.

C.2 Checking the assumptions

In this section, we discuss Assumptions 1 and 2 and present sufficient conditions for them to be verified.

The genericity assumption amounts to: 1) A universality condition in Eqs. (42) and (43): if Pkh(⟨1,⋅⟩/q)=0P_{k}h(\langle\bm{1},\cdot\rangle/q)=0, then hh does not learn degree-kk homogeneous polynomials; 2) A constant order scaling of the self-induced regularization hq,>s(1)h_{q,>{\mathsf{s}}}(1), from hq(1)≤Ch_{q}(1)\leq C and Eq. (43) with s′{\mathsf{s}}^{\prime}, i.e., hq,>s(1)≤hq(1)=Oq(1)h_{q,>{\mathsf{s}}}(1)\leq h_{q}(1)=O_{q}(1) and hq,>s(1)≥ξq,s′B(q,s′)=Ωq(1)h_{q,>{\mathsf{s}}}(1)\geq\xi_{q,{\mathsf{s}}^{\prime}}B(q,{\mathsf{s}}^{\prime})=\Omega_{q}(1); 3) The last eigenvalues decay sufficiently fast in Eq. (44) in order to avoid pathological cases.

The function σq\sigma_{q} is differentiable and there exists c0>0c_{0}>0 and c1<1c_{1}<1 independent of qq, such that ∣σq(x)∣,∣σq′(x)∣≤c0exp⁡(c1x2/2)|\sigma_{q}(x)|,|\sigma_{q}^{\prime}(x)|\leq c_{0}\exp(c_{1}x^{2}/2).

where e∈\mathscrsfsQq{\bm{e}}\in{\mathscrsfs Q}^{q} is arbitrary.

Differentiability assumption:

C.3 Proof of Proposition 6

Let us decompose both functions σq\sigma_{q} and σq′\sigma_{q}^{\prime} in the Gegenbauer polynomial on the hypercube basis:

From the definition of hq(1)h_{q}^{(1)} in Eq. (54) and the eigendecomposition (60), we have

Similarly, from the definition of hq(2)h_{q}^{(2)} in Eq. (55), the eigendecomposition (61) and using Lemma 1 stated below, we get

such that the NT kernel (53) can be written as the kernel of the effective activation σeff,q\sigma_{{\rm eff},q}:

Recall that the sequence {σq}q≥1\{\sigma_{q}\}_{q\geq 1} satisfies Assumption 3 at level s{\mathsf{s}}. From Assumption 3.(a)(a) (for example by adapting the proof of Lemma C.1 in to the hypercube), there exists C>0C>0 such that

Furthermore, by Assumption 3.(b)(b), using that χq,k2=B(\mathscrsfsQq;k)−1∥Pkσq∥L2(\mathscrsfsQq)2\chi_{q,k}^{2}=B({\mathscrsfs Q}^{q};k)^{-1}\|{\mathsf{P}}_{k}\sigma_{q}\|_{L^{2}({\mathscrsfs Q}^{q})}^{2} and ξq,k2≥χq,k2\xi^{2}_{q,k}\geq\chi^{2}_{q,k}, we get

In particular, this implies that ∥σeff,d,>s∥L2(\mathscrsfsQq)2≥∥Ps′σq∥L2(\mathscrsfsQq)2=Ωq(1)\|\sigma_{{\rm eff},d,>{\mathsf{s}}}\|_{L^{2}({\mathscrsfs Q}^{q})}^{2}\geq\|{\mathsf{P}}_{{\mathsf{s}}^{\prime}}\sigma_{q}\|_{L^{2}({\mathscrsfs Q}^{q})}^{2}=\Omega_{q}(1). ∎

where we recall the definition of the homogeneous polynomial YS(x)=xS=∏i∈SxiY_{S}({\bm{x}})={\bm{x}}^{S}=\prod_{i\in S}x_{i}. We have

with the convention Q−1(q)=Qq+1(q)=0Q^{(q)}_{-1}=Q^{(q)}_{q+1}=0.

C.4 Proof of Theorem 7

Let {d(q)}q≥1\{d(q)\}_{q\geq 1} be a sequence of integers with 2q≤d(q)≤q1/δ2q\leq d(q)\leq q^{1/\delta} for some δ>0\delta>0. We will denote d=d(q)d=d(q) for simplicity. Consider x∼Unif(\mathscrsfsQd){\bm{x}}\sim{\rm Unif}({\mathscrsfs Q}^{d}), dqs−1+δ≤n≤dqs−δdq^{{\mathsf{s}}-1+\delta}\leq n\leq dq^{{\mathsf{s}}-\delta} for some δ>0\delta>0 and a sequence of inner-product kernels {hq}q≥1\{h_{q}\}_{q\geq 1} that satisfies Assumption 1 at level s{\mathsf{s}}. We consider the vanilla one-layer convolutional kernel

Theorem 7 is a consequence of Theorem 4 in where we take Xd=\mathscrsfsQd{\mathcal{X}}_{d}={\mathscrsfs Q}^{d}, νd=Unif(Xd)\nu_{d}={\rm Unif}({\mathcal{X}}_{d}) and Dd=L2(\mathscrsfsQd,Locq)⊂L2(\mathscrsfsQd){\mathcal{D}}_{d}=L^{2}({\mathscrsfs Q}^{d},{\rm Loc}_{q})\subset L^{2}({\mathscrsfs Q}^{d}). The proof amounts to checking that {H\mboxCK,d}q≥1\{H^{\mbox{\tiny\sf CK},d}\}_{q\geq 1} verifies the kernel concentration properties and eigenvalue condition (see Section 3.2 in ). We borrow some of the notations introduced in and we refer the reader to their Section 2.1.

Step 1. Diagonalization of the kernel and choosing m=m(q){\mathsf{m}}={\mathsf{m}}(q).

From Proposition 1, we have the following diagonalization of H\mboxCK,dH^{\mbox{\tiny\sf CK},d}:

Step 2. Diagonal elements of the truncated kernel.

Define the truncated kernel Hd,>mH_{d,>{\mathsf{m}}} to be

The diagonal elements of the truncated kernel are given by: for any x∈\mathscrsfsQd{\bm{x}}\in{\mathscrsfs Q}^{d},

Hence using that ξq,s=Od(q−s)\xi_{q,{\mathsf{s}}}=O_{d}(q^{-{\mathsf{s}}}), we have

Let s′{\mathsf{s}}^{\prime} be chosen as in Assumption 1, i.e., such that ξq,s′B(\mathscrsfsQq;s′)=Ωq(1)\xi_{q,{\mathsf{s}}^{\prime}}B({\mathscrsfs Q}^{q};{\mathsf{s}}^{\prime})=\Omega_{q}(1). We have

Step 4. Checking the kernel concentration property at level {(n(q),m(q))}q≥1\{(n(q),{\mathsf{m}}(q))\}_{q\geq 1}.

Let us check the kernel concentration property at level (n,m)(n,{\mathsf{m}}) with the sequence of integers {u(q)}q≥1\{u(q)\}_{q\geq 1} defined in the previous step (Assumption 4 in ):

(Hypercontractivity of finite eigenspaces) The subspace spanned by the top eigenvectors {ψq,j}j∈[u]\{\psi_{q,j}\}_{j\in[u]} is contained in the subspace of polynomials of degree less or equal to s′−1{\mathsf{s}}^{\prime}-1 on the hypercube. The hypercontractivity of this subspace is a consequence of a classical result due to Beckner, Bonami and Gross (see Lemma 4 in Section D).

(Properly decaying eigenvalues.) From step 3 and recalling that s′≥1/δ+2s+3{\mathsf{s}}^{\prime}\geq 1/\delta+2{\mathsf{s}}+3 where δ>0\delta>0 verifies q≥dδq\geq d^{\delta}, we have

for δ′>0\delta^{\prime}>0 sufficiently small. Similarly,

for δ′>0\delta^{\prime}>0 chosen sufficiently small.

(Concentration of the diagonal elements of the kernel) From Eqs. (66) and (67), the diagonal elements of the kernel are constant and the assumption is automatically verified.

Step 5. Checking the eigenvalue condition at level {(n(q),m(q))}q≥1\{(n(q),{\mathsf{m}}(q))\}_{q\geq 1}.

Let us now check the eigenvalue condition at level {(n(q),m(q))}q≥1\{(n(q),{\mathsf{m}}(q))\}_{q\geq 1} which corresponds to Assumption 5 in ):

for δ>0\delta>0 sufficiently small. Similarly,

This is a direct consequence of Eq. (65).

We can therefore apply Theorem 4 in , which concludes the proof. ∎

C.5 Proof of Theorem 8

Consider qs−1+δ≤n≤qs−δq^{{\mathsf{s}}-1+\delta}\leq n\leq q^{{\mathsf{s}}-\delta} for some δ>0\delta>0 and a sequence of inner-product kernels {hq}q≥1\{h_{q}\}_{q\geq 1} that satisfies Assumptions 1 and 2 at level s{\mathsf{s}}. We consider the one-layer convolutional kernel with global average pooling

Again, the proof of Theorem 8 will amount to checking that the conditions of Theorem 4 in hold.

Step 1. Diagonalization of the kernel and choosing m=m(q){\mathsf{m}}={\mathsf{m}}(q).

From Proposition 2 with ω=d\omega=d, we have the following diagonalization of Hd,qdH_{d,q}^{d}:

Step 2. Diagonal elements of the truncated kernel.

Define the truncated kernel Hd,>mH_{d,>{\mathsf{m}}} to be

The diagonal elements of the truncated kernel are given by: for any x∈\mathscrsfsQd{\bm{x}}\in{\mathscrsfs Q}^{d},

Step 4. Checking the kernel concentration property at level {(n(q),m(q))}q≥1\{(n(q),{\mathsf{m}}(q))\}_{q\geq 1}.

The kernel concentration property at level (n,m)(n,{\mathsf{m}}) hold with the sequence {u(q)}q≥1\{u(q)\}_{q\geq 1} as defined in step 3. The hypercontractivity of finite eigenspaces and the properly decaying eigenvalues are obtained as in step 4 of the proof of Theorem 7, while the concentration of the diagonal elements of the kernel is given by Eq. (71).

Step 5. Checking the eigenvalue condition at level {(n(q),m(q))}q≥1\{(n(q),{\mathsf{m}}(q))\}_{q\geq 1}.

This is obtained similarly as in step 5 of the proof of Theorem 7.

C.6 Auxiliary results

where hq,>sh_{q,>{\mathsf{s}}} is the inner-product kernel where the s+1{\mathsf{s}}+1 first Gegenbauer coefficients are set to .

Then for n=Oq(qp)n=O_{q}(q^{p}) for some fixed pp, letting (xi)i∈[n]∼Unif(\mathscrsfsQd)({\bm{x}}_{i})_{i\in[n]}\sim{\rm Unif}({\mathscrsfs Q}^{d}), we have

Following the same proof as Proposition 8 in , notice that for the integer vv in Assumption 2, by Lemma 2 stated below, we have

By Assumption 2, there exists C>0C>0 such that for any γ∈\gamma\in,

and ∣hq,>v(r)(0)∣≤Cq−(v+1−r)/2|h_{q,>v}^{(r)}(0)|\leq Cq^{-(v+1-r)/2} for r≤vr\leq v. Moreover, by Hanson-Wright inequality as in Lemma 3, using n=Oq(qp)n=O_{q}(q^{p}) (at most polynomial in qq) and a union bound, we have for any η>0\eta>0,

Therefore, injecting these bounds in Eq. (74), we get

which concludes the proof of the first bound.

Then, by Lemma 2, we get for any u≥su\geq{\mathsf{s}},

We conclude following the same argument as in the proof of Proposition 9 in . ∎

Let n≤qpn\leq q^{p} for some fixed pp. Then, for (xi)i∈[n]∼i.i.d.Unif(\mathscrsfsQd)({\bm{x}}_{i})_{i\in[n]}{\stackrel{{\scriptstyle i.i.d.}}{{\sim}}}{\rm Unif}({\mathscrsfs Q}^{d}), we have

Let us bound the right hand side. We have

where B1=i+SB_{1}=i+S, B2=j+SB_{2}=j+S, B3=i′+S′B_{3}=i^{\prime}+S^{\prime} and B4=j′+S′B_{4}=j^{\prime}+S^{\prime}, and we denoted

Notice that ω(B1,B2,B3,B4)=1\omega(B_{1},B_{2},B_{3},B_{4})=1 if B1ΔB2=B3ΔB4B_{1}\Delta B_{2}=B_{3}\Delta B_{4} (the symmetric difference) and otherwise. In other words, every elements in B1∪B2∪B3∪B4B_{1}\cup B_{2}\cup B_{3}\cup B_{4} appears exactly in 2 or 4 of these sets.

where we used that r(S′)≤qr(S^{\prime})\leq q.

Using Markov’s inequality and taking mm sufficiently small yield Eq. (77).

Further notice that following the same computation as in Eq. (69), we get

where we recall that x(k)=(xk,…,xk+q−1){\bm{x}}_{(k)}=(x_{k},\ldots,x_{k+q-1}).

Taking the union bound over k≠lk\neq l concludes the proof. ∎

Appendix D Technical background of function spaces on the hypercube

Fourier analysis on the hypercube is a well studied subject . The purpose of this section is to introduce some notations and objects that are useful in the statement and proofs in the main text.

It is easy to verify that (notice that xik=xix_{i}^{k}=x_{i} if kk is odd and xik=1x_{i}^{k}=1 if kk is even)

D.2 Hypercubic Gegenbauer

Notice that the right hand side only depends on ⟨x,y⟩\langle{\bm{x}},{\bm{y}}\rangle and therefore these polynomials are well defined. In particular,

Furthermore, Eq. (86) imply 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 polynomials

By permutation invariance, the space VkV_{k} of homogeneous polynomials of degree kk is an eigenspace of \mathscrsfsHd\mathscrsfs{H}_{d}, and we will denote the corresponding eigenvalue by ξd,k(hd)\xi_{d,k}(h_{d}). In other words \mathscrsfsHdf(x)≡∑k=0qξd,k(hd)Pkf\mathscrsfs{H}_{d}f({\bm{x}})\equiv\sum_{k=0}^{q}\xi_{d,k}(h_{d}){\mathsf{P}}_{k}f. The eigenvalues can be computed via

D.3 Hermite polynomials

Here and below, for PP a polynomial, Coeff{P(x)}{\rm Coeff}\{P(x)\} is the vector of the coefficients of PP. As a consequence, for any fixed integer kk, we have

where μk(σ)\mu_{k}(\sigma) and ξd,k(σ)\xi_{d,k}(\sigma) are given in Eq. (92) and (88).

D.4 Hypercontractivity of uniform distributions on the hypercube

By Holder’s inequality, we have ∥f∥Lp≤∥f∥Lq\|f\|_{L^{p}}\leq\|f\|_{L^{q}} for any ff and any p≤qp\leq q. The reverse inequality does not hold in general, even up to a constant. However, for some measures, the reverse inequality will hold for some sufficiently nice functions. These measures satisfy the celebrated hypercontractivity properties .