How rotational invariance of common kernels prevents generalization in high dimensions
Konstantin Donhauser, Mingqi Wu, Fanny Yang
Introduction
Traditional analysis establishes good generalization properties of kernel ridge regression when the dimension is relatively small compared to the number of samples . These minimax optimal and consistency results however become less powerful for modern data sets with large close to . High-dimensional asymptotic theory aims to fill this gap by providing bounds that assume and are often much more predictive of practical observations even for finite .
While recent work establishes explicit asymptotic upper bounds for the bias and variance for high-dimensional linear regression, the results for kernel regression are less conclusive in the regime with . In particular, even though several papers show that the variance decreases with the dimensionality of the data, the bounds on the bias are somewhat inconclusive. On the one hand, Liang et al. prove asymptotic consistency for ground truth functions with asymptotically bounded Hilbert norms for neural tangent kernels (NTK) and inner product (IP) kernels. In contrast, Ghorbani et al. show that for uniform distributions on the product of two spheres, consistency cannot be achieved unless the ground truth is a low-degree polynomial. This polynomial approximation barrier can also be observed for random feature and neural tangent regression .
Notably, the two seemingly contradictory consistency results hold for different distributional settings and are based on vastly different proof techniques. While proves consistency for general input distributions including isotropic Gaussians, the lower bounds in the papers are limited to data that is uniformly sampled from the product of two spheres. Hence, it is a natural question to ask whether the polynomial approximation barrier is a more general phenomenon or restricted to the explicit settings studied in . Concretely, this paper addresses the following question
Can we overcome the polynomial approximation barrier when considering different high-dimensional input distributions, eigenvalue decay rates or scalings of the kernel function?
We unify previous distributional assumptions in one proof framework and thereby characterize how the rotational invariance property of common kernels induces a bias towards low-degree polynomials. Specifically, we show that the polynomial approximation barrier persists for
a broad range of common rotationally invariant kernels such as radial basis functions (RBF) with vastly different eigenvalue decay rates, inner product kernels and NTK of any depth .
general input distributions including anisotropic Gaussians where the degree of the polynomial depends only on the growth of and not on the specific structure of . In particular, we cover the distributions studied in previous related works .
different scalings with kernel function beyond the classical choice of .
As a result, this paper demonstrates that the polynomial approximation barrier is a general high-dimensional phenomenon for rotationally invariant kernels, restricting the set of functions for which consistency can at most be reached to low degree polynomials.
Rotationally invariant kernels are a natural choice if no prior information on the structure of the ground truth is available, as they treat all dimensions equally. Since our analysis covers a broad range of distributions, eigenvalue decays and different scalings, our results motivate future work to focus on the symmetries respectively asymmetries of the kernel incorporating prior knowledge on the structure of the high-dimensional problem, e.g. .
This paper is organized as follows. First of all, we show in Section 2.2 that the bounded norm assumption that previous consistency results rely on is violated as even for simple functions such as . We then introduce our generalized setting in Section 2 and present our main results in Section 3 where we show a lower bound on the bias that increases with the dimensionality of the data. Finally, in Section 4 we empirically illustrate how the bias dominates the risk in high dimensions and therefore limits the performance of kernel regression. As a result, we argue that it is crucial to incorporate prior knowledge of the ground truth function (such as sparsity) in high-dimensional kernel learning even in the noiseless setting and empirically verify this on real-world data.
Problem setting
In this section, we briefly introduce kernel regression estimators in reproducing kernel Hilbert spaces and subsequently introduce our assumptions on the kernel, data distribution and high-dimensional regime.
with and the minimum norm interpolator (also called the kernel \sayridgeless estimate)
that can be obtained as the limit of the ridge estimate for fixed . It is well-known that the ridge estimator can attain consistency as for some sequence of such that . Recently, some works have also analyzed the consistency behavior of ridgeless estimates motivated by the curiously good generalization properties of neural networks with zero training error.
Note that consistency in terms of as can only be reached if the bias vanishes. In this paper, we lower bound the bias B which, in turn, implies a lower bound on the risk and the inconsistency of the estimator. The theoretical results in Section 3 hold for both ridge regression and minimum norm interpolation. However, it is well known that the ridge penalty controls the bias-variance trade-off, and hence we are primarily interested in lower bounds of the bias for the minimum norm interpolant.
2 Prior work on the consistency of kernel regression
For ridge regression estimates in RKHS, a rich body of work shows consistency and rate optimality when appropriately choosing the ridge parameter both in the non-asymptotic setting, e.g. , and the classical asymptotic setting, e.g. , as .
In words, for simple sequences of sparse product functions, the Hilbert norm diverges as the dimension . The precise conditions on the kernel and sequence of induced Hilbert spaces can be found in Appendix B. Figure 1 illustrates this phenomenon for for the Laplace and exponential inner product kernel.
The discussion so far implies that generalization upper bounds that rely on the bounded Hilbert norm assumption become void even for simple ground truth functions. A natural follow-up question is hence: Do we actually fail to learn sparsely parameterized functions consistently or is it simply a loose upper bound? A recent line of work by Ghorbani et al. shows that kernel regression estimates can indeed only consistently learn polynomials of degree at most as (which we refer to as the polynomial approximation barrier). While the results provide some intuition for the behavior of kernel regression, the proofs heavily rely on significant simplifications that hold for the specific distributional assumptions on the sphere. It is a priori unclear whether they apply to more general settings including the ones considered in . In the next sections, relying on a different proof technique we show that the polynomial approximation barrier indeed holds for a broad spectrum of data distributions that also capture the distributions studied in the papers and for an entire range of eigenvalue decay rates of the kernel functions (e.g. polynomial and exponential decay rates) and choices of the scaling . As a consequence, our results suggest that the polynomial approximation barrier is strongly tied to the rotational invariance of the kernel function and not specific to the settings studied so far.
3 Our problem setting
The framework we study in this paper covers random vectors that are generated from a covariance matrix model, i.e. with vector consisting of i.i.d entries and their projections onto the -dimensional unit sphere. We now specify all relevant assumptions on the kernel and data distribution.
Throughout this paper, we focus on continuous and rotationally invariant kernels. They include a vast majority of the commonly used kernels such as fully connected NTK, RBF and inner product kernels as they only depend on the squared Euclidean norm of and and the inner product . We first present one set of assumptions on the kernel functions that are sufficient for our main results involving local regularity of around the sphere where the data concentrates in high dimensions.
Rotational invariance and local power series expansion: The kernel function is rotationally invariant and there is a function such that . Furthermore, can be expanded as a power series of the form
We show in Corollary 3.2 that the kernels for which our main results hold cover a broad range of commonly studied kernels in practice. In particular, Theorem 3.1 also holds for -exponential kernels defined as for even though we could not yet show that they satisfy Assumptions A.1-A.2. In Appendix C, we show how the proof of Theorem 3.1 crucially relies on the rotational invariance Assumption C.1 and the fact that the eigenvalues of the kernel matrix are asymptotically lower bounded by a positive constant. Both conditions are also satisfied by the -exponential kernels and the separate treatment is purely due to different proof technique used to lower bound the kernel matrix eigenvalues.
We impose the following assumptions on the data distribution.
Covariance model: We assume that the input data distribution is from one of the following sets
High dimensional regime: We assume that the effective dimension grows with the sample size s.t. for some .
and parameterize the scaling by sequence of parameters dependent on . In Section 3.1 we study the standard scaling , before discussing and respectively in Section 3.2, where we show that the polynomial approximation barrier is not a consequence of the standard scaling.
Main Results
We now present our main results that hold for a wide range of distributions and kernels and show that kernel methods can at most consistently learn low-degree polynomials. Section 3.1 considers the case while Section 3.2 provides lower bounds for the regimes and .
The bias of the kernel estimators is asymptotically almost surely lower bounded for any ,
Corollary 3.2 gives examples for kernels that satisfy the assumptions of the theorem. The almost sure statements refers to the sequence of matricies X of random vectors as , but also hold true with probability over the draws of X (see Lemma C.1 for further details).
The attentive reader might notice that Ghorbani et al. achieve a lower barrier for their specific setting which implies that our results are not tight. However, this is not the main focus in this work as we primarily intend to demonstrate that the polynomial approximation barrier persists for general covariance model data distributions (Ass. B.1-B.2) and hence asymptotic consistency in high-dimensional regimes is at most reachable for commonly used rotationally invariant kernels if the ground truth function is a low degree-polynomial. We leave tighter bounds as interesting future work. Finally, we remark that as we enter again a classical asymptotic regime where is much larger compared to and hence . This underlines the difference between classical and high-dimensional asymptotics and shows that our results are only meaningful in the latter.
We now present a short proof sketch to provide intuition for why the polynomial approximation barrier holds. The full proof can be found in Appendix C.
The proof of the main theorem is primarily based on the concentration of Lipschitz continuous functions of vectors with i.i.d entries. In particular, we show in Lemma C.1 that
where we use . Furthermore, Assumption A.1and hence the rotational invariance of the kernel function , implies that for inner product kernels where are constants,
with some constant such that that exists because . Hence, as , converge to low-degree polynomials. Using the closed form solution of based on the representer theorem we can hence conclude the first statement in Theorem 3.1 if for some constant . The result follows naturally for ridge regression with non vanishing . However for the minimum norm interpolator, we need to show that the eigenvalues of the kernel matrix themselves are asymptotically lower bounded by a positive non-zero constant. This follows from the additional assumption in Theorem 3.1 and the observation that in operator norm with being the Hadamard product. Finally, the case where depend on and requires a more careful analysis and constitute the major bulk of the proof in Appendix C. ∎
The assumptions in Theorem 3.1 cover a broad range of commonly used kernels, including the ones in previous works. The following corollary summarizes some relevant special cases
The -exponential kernel for , including Laplace () and the Gaussian () kernels
The fully-connected NTK of any depth with regular activation functions including the ReLU activation
The precise regularity conditions of the activation functions for the NTK and the proof of the corollary can be found in Appendix C.3.
In particular, the assumptions hold for instance for the -exponential kernels with (see ) and the popular Matern RBF kernels with . The proof of the theorem can be found in Appendix D.1 and is based on the flat limit literature on RBFs . Finally, we remark that Theorem 3.3 applies only for the interpolating estimator . However, because the ridge penalty is well known to regulate the bias-variance tradeoff, we argue that the bias increases with the ridge penalty and hence attains its minimum at .
3 Discussion of theoretical results
As a consequence of our unifying treatment, we cover most of the settings studied in the existing literature and show that the polynomial approximation barrier (5) is neither restricted to a few specific distributions nor to a particular choice of the scaling or the eigenvalue decay of the kernel function. We summarize our setting alongside those of previous works in Table 1.
The Assumptions B.1-B.2 allow very general distributions and include the ones in the current literature. In particular, we also cover the settings studied in the papers and hence put their optimistic conclusions into perspective. Besides, our results hold true for arbitrary covariance matrices and only depend on the growth rate of the effective dimension, but are independent of the explicit structure of the covariance matrix. This stands in contrast to linear regression where consistency can (only) be guaranteed for spiky covariance matrices .
Our results do not only apply for the standard choice of the scaling , but also apply to general RBF kernels in the flat limit scaling, i.e. where . This case is particularly important since this is where we empirically find the bias to attain its minimum in Figure 3(c). We therefore conjecture that the polynomial approximation barrier cannot be overcome with different choices of the scaling.
Furthermore, by explicitly showing that the polynomial barrier persists for all -exponential kernels with (that have vastly different eigenvalue decay rates), we provide a counterpoint to previous work that suggests consistency for . In particular, the paper proves minimax optimal rates for the Nadaraya-Watson estimators with singular kernels for fixed dimensions and empirical work by suggests that spikier kernels have more favorable performances. Our results suggest that in high dimensions, the effect of eigenvalue decay (and hence “spikiness”) may be dominated by asymptotic effects of rotationally invariant kernels. We discuss possible follow-up questions in Section 5. As a result, we can conclude that the polynomial approximation barrier is a rather general phenomenon that occurs for commonly used rotationally invariant kernels in high dimensional regimes. For ground truths that are inherently higher-degree polynomials that depend on all dimensions, our theory predicts that consistency of kernel learning with fully-connected NTK, standard RBF or inner product kernels is out of reach if the data is high-dimensional. In practice however, it is possible that not all dimensions carry equally relevant information. In Section 4.3 we show how feature selection can be used in such settings to circumvent the bias lower bound. On the other hand, for image datasets like CIFAR-10 where the ground truth is a complex function of all input dimensions, kernels that incorporate convolutional structures (such as CNTK or compositional kernels ) and hence break the rotational symmetry can perform quite well.
Experiments
In this section we describe our synthetic and real-world experiments to further illustrate our theoretical results and underline the importance of feature selection in high dimensional kernel learning.
In Figure 1, we demonstrate how the Hilbert norm of the simple sparse linear function grows with dimension . We choose the scaling and consider the Hilbert space induced by the scaled Gaussian , Laplace and exponential inner product kernels. To estimate the norm, we draw i.i.d. random samples with noiseless observations from the uniform distribution on .
2 Illustration of the polynomial approximation barrier
We now provide details for the numerical experiments in Figure 2,3 that illustrate the lower bounds on the bias in Theorem 3.1 and Theorem 3.3. For this purpose, we consider the following three different distributions that satisfy the assumptions of the theorems and are covered in previous works
While our theoretical results only show lower bounds on the bias, Figure 2(a) and 2(b) suggest that as increases, the bias in fact stepwise aligns with the best polynomial of lower and lower order. This has also been shown in for the uniform distribution from the product of two spheres. Indeed, with decreasing , for the cubic polynomial in Figure 2(b) we first learn linear functions (first descent in the curve). Since the best degree 2 polynomial approximation of around is a linear function, the curve then enters a plateau before descending to zero indicating that we successfully learn the ground truth.
3 Feature selection for high-dimensional kernel learning
In this section, we demonstrate how the polynomial approximation barrier limits the performance in real world datasets and how one may overcome this issue using feature selection. How our theory motivates feature selection can be most cleanly illustrated for complex ground truth functions that only depend on a number of covariates that is much smaller than the input dimension (which we refer to as sparse)for more general functions that are not sparse we still expect a U-curve, albeit potentially less pronounced, whenever the function is not a low-degree polynomial of order :
Based on our theoretical results we expect that for sparse ground truths, the bias follows a U-shape as dimension increases: until all relevant features are included, the bias first decreases before it then starts to decrease due to the polynomial approximation barrier that holds for large when asymptotics start to kick in. Since recent work shows that the variance vanishes in high-dimensional regimes (see e.g. ), we expect the risk to follow a U-shaped curve as well. Hence, performing feature selection could effectively yield much better generalization for sparse ground truth functions. We would like to emphasize that although the described curve may ring a familiar bell, this behavior is not due to the classical bias-variance trade-off, since the U-shaped curve can be observed even in the noiseless case where we have zero variance. We now present experiments that demonstrate the U-shape of the risk curve for both synthetic experiments on sparse ground truths and real-world data. We vary the dimensionality by performing feature selectionwe expect other approaches to incorporate sparsity such as automatic relevance determination to yield a similar effect using the algorithm proposed in the paper . In order to study the impact of high-dimensionality on the variance, we add different levels of noise to the observations.
For the real-world experiments we are not able to decompose the risk to observe separate trends of the bias and the variance. However, we can argue that the ridge estimator should perform significantly better than the minimum norm interpolator whenever the variance dominates. Vice versa, if their generalization errors are close, the effect of the variance vanishes. This is due to the fact that the ridge penalty decreases the variance and increases the bias. Hence, for real-world experiments, we use the comparison between the risks of the ridge estimator and minimum norm interpolator to deduce that the bias is in fact dominating the risk for high dimensions.
We now explore the applicability of our results on real-world data where the assumptions of the theorems are not necessarily satisfied. For this purpose we select datasets where the number of features is large compared to the number of samples. In this section we show results on the regression dataset residential housing (RH) with and to predict sales prices from the UCI website . Further datasets can be found in Appendix A.3. In order to study the effect of noise, we generate an additional dataset (RH-2) where we add synthetic i.i.d. noise drawn from the uniform distribution on to the observations. The plots in Figure 4 are then generated as follows: we increase the number of features using a greedy forward selection procedure (see Appendix A.3 for further details ). We then plot the risk achieved by the kernel ridge and ridgeless estimate using the Laplace kernel on the new subset of features.
Figure 4(b) shows that the risks of the minimum norm interpolator and the ridge estimator are identical, indicating that the risk is essentially equivalent to the bias. Hence our first conclusion is that, similar to the synthetic experiment, the bias follows a U-curve. For the dataset RB-2 in Figure 4(c), we further observe that even with additional observational noise, the ridge and ridgeless estimator converge, i.e. the bias dominates for large . We observe both trends in other high-dimensional datasets discussed in Appendix A.3 as well. As a consequence, we can conclude that even for some real-world datasets that do not necessarily satisfy the conditions of our bias lower bound, feature selection is crucial for kernel learning for noisy and noiseless observations alike. We would like to note that this conclusion does not necessarily contradict empirical work that demonstrates good test performance of RBFs on other high-dimensional data such as MNIST. In fact, the latter only suggests that linear or polynomial fitting would do just as well for these datasets which has indeed been suggested in .
Conclusion and future work
Kernel regression encourages estimators to have a certain structure by means of the RKHS norm induced by the kernel. For example, the eigenvalue decay of -exponential kernels results in estimators that tend to be smooth (i.e. Gaussian kernel) or more spiky (i.e. small ). A far less discussed fact is that many kernels implicitly incorporate additional structural assumptions. For example, rotational invariant kernels are invariant under permutations and hence treat all dimensions equally Even though rotational invariance is a natural choice when no prior information on the structure of the ground truth is available, this paper shows that the corresponding inductive bias in high dimensions is in fact restricting the average estimator to a polynomial. In particular, we show in Theorems 3.1 and 3.3 that the lower bound on the bias is simply the projection error of the ground truth function onto the space of polynomials of degree at most respectively . Apart from novel technical insights that result from our unified analysis (discussed in Sec. 3.3), our result also opens up new avenues for future research.
Modern datasets which require sophisticated methods like deep neural networks to obtain good predictions are usually inherently non-polynomial and high-dimensional. Hence, our theory predicts that commonly used rotationally invariant kernels cannot perform well for these problems due to a high bias. In particular, our bounds are independent of properties like the smoothness of the kernel function and cannot be overcome by carefully choosing the eigenvalue decay. Therefore, in order to understand why certain highly overparameterized methods generalize well in these settings, our results suggest that it is at least equally important to understand how prior information can be incorporated to break the rotational symmetry of the kernel function. Examples for recent contributions in this direction are kernels relying on convolution structures (such as CNTK or compositional kernels ) for image datasets.
Another relevant future research direction is to present a tighter non-asymptotic analysis that allows a more accurate characterization of the estimator in practice. The presented results in this paper are asymptotic statements, meaning that they do not provide explicit bounds for fixed . Therefore, for given finite it is unclear which high-dimensional regime provides the most accurate characterization of the estimator’s statistical properties. For instance, our current results do not provide any evidence whether the estimator follows the bias lower bounds for with or . We remark that the methodology used to prove the statements in this paper could also be used to derive non-asymptotic bounds, allowing us to further investigate this problem. However, we omitted such results in this paper for the sake of clarity and compactness of our theorem statements and proofs and leave this for future work.
Acknowledgments
We would like to thank the anonymous reviewers for their helpful feedback, Xiao Yang for insightful discussions on the flat limit kernel and Armeen Taeb for his comments on the manuscript.
References
Appendix A Experiments
This section contains additional experiments not shown in the main text. Our code is publicly available at https://www.github.com/DonhauserK/High-dim-kernel-paper/
In this section, we provide additional experiments that discuss Theorem 3.1. In particular, we investigate kernels beyond the Laplace kernel and study the behaviour of the bias with respect to when is fixed and varies. The experimental setting is the same as the one in Section 4.2.
Instead of comparing the bias curves for different input distribution as in Figure 2(a), Figure 5 shows the bias with respect to for the -exponential kernel, i.e. , for different choices of and hence for kernels with distinct eigenvalue decays ( results in an exponential eigenvalue decay while in a polynomial eigenvalue decay). Clearly, we can see that the curves transition at a similar value for which confirms the the discussion of Theorem 3.1 in Section 3.3 where we argue that the polynomial approximation barrier occurs independently of the eigenvalue decay.
Figure 6 shows the bias of the minimum norm interpolant normalized by for the ground truth function and the Laplace kernel as in Section 4.2 with . We observe that the asymptotics already kick in for since all curves for resemble each other. This confirms the the trend in Figure 2(b).
A.2 Feature selection - Synthetic
The goal of this experiment is to compare the bias variance trade-off of ridge regression and minimum norm interpolation. We use the same experimental setting as the ones used for Figure 4(a) (see Section 4.3). We set the bandwidth to and choose the ridge parameter using -fold cross validation. While for small dimensions , ridge regularization is crucial to achieve good performance, the bias becomes dominant as the dimension grows and the difference of the risks of both methods shrinks. This aligns well with Theorem 3.1 which predicts that the bias starts to increase with for fixed once we enter the asymptotic regime.
A.3 Feature selection - Real world
We now present details for our real world experiments to emphasize the relevance of feature selection when using kernel regression for practical applications, as discussed in Section 4.3.
The residential housing regression data set from the UCI website where we predict the sales prices and construction costs given a variety of features including the floor area of the building or the duration of construction.
The ALLAML classification data set from the ASU feature selection website where we classify patients with acute myeloid leukemia (AML) and acute lymphoblastic leukemia (ALL) based on features gained from gene expression monitoring via DNA microarrays.
The CLL_SUB_111 classification dataset from the ASU feature selection web-page where we classify genetically and clinically distinct subgroups of B-cell chronic lymphocytic leukemia (B-CLL) based on features consisting of gene expressions from high density oligonucleotide arrays. While the original dataset contains three different classes, we only use the classes 2 and 3 for our experiment to obtain a binary classification problem.
Because the number of features in the ALLAML and CLL_SUB_111 datasets massively exceed the number of samples, we run the feature selection algorithm in and pre-select the best features chosen by the algorithm. In order to reduce the computational expenses, we run the algorithm in batches of features and iteratively remove all features except for the best features chosen by the algorithm. We do this until we reduce the total number of features to and then select in a last round the final features used for the further procedure. Reducing the amount of features to is important for the computational feasibility of greedy forward features selection in our experiments. The properties of the datasets are summarized in Table 2.
Results of the experiments: The following figures present the results of our experiments on all datasets except for the ones predicting the sales prices in the residential housing dataset, which we presented in Figure 4(b),4(c) in the main text. Similar to the observations made in Section 4.3, Figure 8,9,10 show that the risk reaches its minimum around , with significant differences to the right at . In particular, this holds for both ridge regression and interpolation, which again shows that the bias becomes dominant as the dimension increases. Surprisingly, we also note that the relevance of ridge regularization seems to be much smaller for classification tasks than regression tasks.
Appendix B Bounded Hilbert norm assumption
This section gives a formal statement of Lemma 2.1. We begin with the conditions under which the lemma holds. We consider tensor product kernels of the form
For any , define . First, we note that the proof follows trivially if any of the is not contained in the RKHS induced by since this implies that the Hilbert norm . Hence, we can assume that for all , is contained in the RKHS for all . Furthermore, because is a product kernel, we can write where is the Hilbert norm induced by on . Because we are only interested to see whether the sequence of Hilbert norms diverge, without loss of generality we can assume that , and hence,
Next, by Mercer’s theorem there exists an orthonormal eigenbasis in with corresponding eigenvalues such that for any , , where . Note that because the kernel depends on , and also depend on . Next, because by assumption is contained in the RKHS, there exists such that for every , and . Furthermore,
Furthermore, there exists such that . Again, because we are only interested to see whether the sequence of Hilbert norms diverge, without loss of generality we can assume that and hence also .
Together with the fact that it then follows that has to be asymptotically lower bounded by some positive non-zero constant and hence
This contradicts the assumption that is upper bounded by some constant for every . Hence, we are only left with the case where , however, this case diverges due to Equation 9. Hence, the proof is complete. ∎
Appendix C Proof of Theorem 3.1
Before presenting the proof of the (generalized) theorem, we first state the key concentration inequalities used throughout the proof. It is an extension of Lemma A.2 in the paper , which iteself is a consequence of the concentration of Lipschitz continuous functions of i.i.d random vectors.
Then, there exists some constant such that for sufficiently large,
In particular, the event holds almost surely with respect to the sequence of data sets X as , that is the probability that for infinitely many , does not hold, is zero.
The proof of the lemma can be found in Section E.1.
The proof of the Theorem is primarily separated into two parts
We first state Theorem C.2 which shows under the weaker Assumption C.1 that the results of 3.1 hold for the ridge estimate for non-vanishing or the ridgeless estimate whenever the eigenvalues of are asymptotically lower bounded.
We finish the proof for the ridgeless estimate by invoking Theorem C.2 and showing that indeed has asymptotically lower bounded eigenvalues under the stricter assumptions A.1-A.3 imposed in Theorem 3.1.
For the clarity we denote with A.3 the -dependent assumptions in Theorem 3.1
-dependent assumptions: is -times continuously differentiable in a neighborhood of and there exists such that .
We start by introducing the following weaker assumptions that allows us to jointly treat -exponential kernels and kernels satisfying Assumption A.1-A.3 when the kernel eigenvalues are lower bounded in Theorem C.2. Note that this assumption implies that the kernel is rotationally invariant.
The kernel function is rotationally invariant and there exists a function such that . Furthermore, can be expanded as a power series of the form
with that converges in a neighborhood of the sphere for some and where is -times continuously differentiable in an neighborhood of and the remainder term is a continuous function around the point .
The bias of the kernel estimators is asymptotically lower bounded, for any ,
We can find a polynomial such that for any , there exists such that asymptotically with probability over the draws of ,
The proof of this theorem can be found in Section C.1. Theorem C.2 states Theorem 3.1 under the assumption that has asymptotically lower bounded eigenvalues and the weaker Assumption C.1. For the proof of Theorem 3.1, it remains to show that Assumptions A.1-A.3 of Theorem 3.1 and the -exponential kernel both
induce kernel matrices with almost surely asymptotically positive lower bounded eigenvalues
Point (a) is relatively simple to prove and deferred to Section E.6. The bulk of the work in fact lies in showing (b) separately for the case for A.1-A.3 and -exponential kernels with in the following two propositions, as these two cases require two different proof techniques.
where is the minimum eigenvalue of the kernel matrix .
Assume that the Assumptions B.1-B.2 hold true. Then, the minimum eigenvalue of the kernel matrix of the -exponential kernel with is lower bounded by some positive constant almost surely as .
The proof of the Propositions C.3 and C.4 can be found in the Sections C.2.1 and C.2.2 respectively which concludes the proof of the theorem.
The almost sure statement in Proposition C.4 can also be replaced with an in probability statement as in Lemma C.1 and hence also the statements in Theorem 3.1.
As a result of Lemma C.1 it is sufficient to condition throughout the rest of this proof on the intersection of the events and the event where the eigenvalues of the kernel matrix are lower bounded by a positive constant.
The idea of the proof is to decompose the analysis into the term emerging from the error in the high probability region and the error emerging from the low probability region . The proof essentially relies on the following lemma.
We can construct a polynomial of degree such that for ,
The proof of the lemma can be found in Section E.2. As a result, Equation 16 follows immediately and Equation 17 is a consequence of
The first two terms vanish due to Lemma C.6. To see that the third term vanishes, note that for sufficiently large,
Next, the lower bound for the bias. Due to Lemma C.8, we have that
and hence, for any and sufficiently large,
Thus, the result follows from the definition of the infimum. ∎
C.2 Proofs for the lower bound of the eigenvalues
We use the same notaiton as used in the proof of Theorem C.2. As a result of Lemma C.1 it is sufficent to condition on throughout the rest of this proof. The proof follows straight forwardly from the following Lemma C.7 which gives an asymptotic description of the kernel matrix based on a similar analysis as the one used in the proof of Theorem 2.1 and 2.2 in the paper . In essence, it is again a consequence of the concentration inequality from Lemma C.1 and the stronger Assumption A.1-A.3 and in particular the power series expansion of . We denote with the -times Hadamard product.
Given that the assumption in Proposition C.3 hold. For ,
where is the positive semi-definite matrix with entries .
The proof of the lemma can be found in Section E.3. The proof of Proposition C.3 then follows straight forwardly when using Schur’s product theorem which shows that
where we use that is positive semi-definite by Assumption A.1. To see that the eigenvalues are lower bounded, we thus simply need to show that . This holds because the positive semi-definiteness of implies that and hence is a sum of positive coefficients and because by Assumption A.3 there exists such that . Hence, there exists a positive constant such that . We can conclude the proof when applying Lemma C.7, which implies that as . ∎
C.2.2 Proof of Proposition C.4
We use the same notaiton as used in the proof of Theorem C.2 and define to be the matrix with entries . We separate the proof into two steps. In a first step, we decompose in the terms
The proof of the lemma can be found in Section E.4. In particular, note that we can use the same argument as used in Lemma C.1 to show that there exists almost surely over the draws of Z as an additional vector , such that for any two vectors ,
Throughout the rest of this proof, we conditioned on the event and the additional event that Equation (21) holds, and remark that the intersection of these two events holds true almost surely as . It is then straight forward to show that the eigenvalues of the matrix
where we have used that . As a result, we can see that
Next, using Lemma C.1 and the fact that , we can see that . Hence, it is sufficient to show that is positive semi-definite. This is true if and only if , which is equivalent to saying that . Using again the same argument as for , we can see that for any , which completes the proof. ∎
C.3 Proof of Corollary 3.2
First, note that the Assumption A.1-A.3 straight forwardly hold true for the exponential inner product kernel with and for the Gaussian kernel with
Next, note that the -exponential kernel with is already explicitly covered in Theorem 3.1. Hence, the only thing left to show is that Theorem 3.1 also applies to ReLU-NTK.
Assume that the activation function is -homogeneous and both the activation function and its derivative possess a Hermite-polynomial series expansion (see ) where there exits such that the -th coefficient . Then, the NTK satisfies the Assumption A.1-A.3 and hence Theorem 3.1 applies.
In fact, we can easily see that any non linear activation function which is homogeneous and both the activation function and its derivative possesses a Hermite polynomial extension satisfies the assumptions in Proposition C.9. In particular, this includes the popular ReLU activation function where the explicit expression for the Hermite polynomial extension can be found in . ∎
The only thing left to show is Assumption A.3. While we have already shown that are smooth in a neighborhood of , we still need to show that there exists such that . However, this follows from the fact that by assumption there exists such that where are the Hermite coefficients of the activation function . ∎
Appendix D Different scalings τ𝜏\tau
In this section, we present results for different choices of the scaling beyond the standard choice . In Subsection D.1, we give a proof of Theorem 3.3 describing the flat limit, i.e. the limit of the interpolant where for any fixed , . Furthermore, in order to get a more comprehensive picture, we additionally present straight forward results for other choices of in Section D.2.
First, although the limit does not exists, we can apply Theorem 3.12 in to show that the flat limit interpolator of any kernel satisfying the assumption in Theorem 3.3 exists and has the form
Furthermore, for the -exponential kernel, we use Theorem 2.1 in to show that it satisfies the assumptions imposed on the eigenvalue decay in Theorem 3.3.
The estimator is also called the polyharmonic spline interpolator. This estimator is invariant under rescalings of the input data which is also the reason why we can rescale the input data by , i.e. consider as input data points.
and . In particular, this allows us to use the block matrix inverse to show that
Furthermore, by Lemma C.1 and the fact that , we can see that for , . Hence, assuming that the absolute eigenvalues of are all upper bounded by a non-zero positive constant, we can use exactly the same argument as used in the proof of Theorem C.2 to conclude the proof.
Thus, we only need to show that the eigenvalues are upper bounded. We already know from Lemma C.8 that there exists some constant independent of , such that . Thus, we only need to show that is almost surely upper bounded. Because is a rank one matrix, we know that
where we use that which we already know from the discussion above. Because by the binomial expansion, , Lemma C.1 implies that and hence, , for any sufficiently large. Therefore, . Hence, there exists some constant independent of , such that . As a consequence,
Next, note that our assumption imply that for sufficiently large, . Therefore,
D.2 Additional results
In this section, we present some additional results for different choices of the scaling. The results presented in this section are straight forward but provide a more complete picture for different choices of the scaling . We use again the same notation as used in Appendix C.1.
First, we show the case where . We assume that is the -exponential kernel with , i.e. .
and with probability over the draws of ,
Hence, the result follows immediately from . ∎
We can also show a similar result for the case where and does not vanish.
with
We use the same notation as in Lemma D.3. Again due to Lemma C.1, we find that
where we have used that .
The only thing left to show is that does not diverge. For this, let be the inverse of . As a result of a simple computation we find that and . Hence,
Appendix E Technical lemmas
We begin with the following Lemma, which is a direct consequence of the results in Appendix A in .
i.i.d entries almost surely bounded by some constant and with zero mean and unit variance.
standard normal distributed i.i.d entries.
Let be any symmetric matrix with and let be the decomposition of into two positive semi-definite matrices and with . Then, there exists some positive constants independent of such that for any ,
Following the same argument as the one used in Corollary A.2 in , we can use Lemma E.1 to show that there exists constants such that for ,
We now make us of the Borel-Cantelli Lemma. For any , let and note that because , decays at rate and in particular, for any sufficiently large, . Hence, we can see that there exists some constant such that for any sufficiently large
which allows us to apply the Borel-Cantelli Lemma. Hence,
which concludes the first step of the proof.
Next, we already know from the previous discussion that for any sufficiently large, . Furthermore, because is independently drawn from the same distribution as , . Hence, for any sufficiently large,
First, note that the case where is clear. Let and . Since we are in the Euclidean space, the inner product is given by
Due to Equation (24), we have that and . Therefore,
The rest of the proof then follows straight forwardly. ∎
We know that is -Lipschitz continuous and hence also a -Lipschitz continuous. For the case where the entries are bounded i.i.d. random variables, we can use the simple fact that the norm is convex in order to apply Corollary 4.10 in and Proposition 1.8 in in . For the case where the entries are normally distributed we can apply Theorem \@slowromancapv@.\@slowromancapi@ in . As a result, we can see that there exists a constant independent of , such that
The proof then follows straight forwardly following line by line the proof of Lemma A.2 in .
E.2 Proof of Lemma C.6
For any , let denote the partial derivatives . Define First of all, note that due to Lemma C.1, for any and sufficiently large, for any
As a result, we can make use of Assumption C.1. We are heavily going to make use of this fact throughout the proof. The proof is separated into two steps where we first show 1. and then 2. using the expression for from the first step. Proof of the first statement We construct a polynomial using the power series expansion of from Assumption A.1 and in addition the Taylor series approximation of around the point . For any sufficiently large, we can write
where are points contained in the closed ball around the point with radius . Hence, using the fact that is -times continuously differentiable, we can see that any is almost surely upper bounded by some constant.
Let be the vector defined in Equation (25) We define the polynomial as
Note that is a linear combination of the terms with , and hence a polynomial of of degree at most . If are constant, i.e. the kernel is an inner product kernel, contains only the terms and hence is a polynomial of of degree at most .
In order to conclude the first step of the proof, we only need to show that both terms go to zero. First, we show that the term . Recall that is upper bounded as independent of . Hence, we can apply Lemma C.1 which shows that for any integers and such that ,
which holds true for any positive constant . Finally, because , , and hence
Furthermore, because is a continuous function and are contained in a closed neighborhood around of , is upper bounded by some constant independent of as . Therefore, we also have that
where we have again used Lemma C.1. Hence, we can conclude the first step of the proof when observing that we have only assumed that and hence the convergence is uniformly. Proof of the second statement We can see from the definition of and the subsequent discussion that
We can decompose for sufficiently large,
where we have used Lemma C.1 in the second inequality. The first term vanishes trivially from the concentration inequality. Indeed, for sufficiently large, we have that
For the second term, note that we can see from the proof of Lemma E.1 that there exists some constant , such that for sufficiently large,
We can now apply integration by parts to show that
Hence, combining these terms, we get the desired result
E.3 Proof of Lemma C.7
As in , we separately analyze the off and on diagonal terms of . Let be the off-diagonal matrix of , with diagonal entries and let be the diagonal matrix of with entries . We have
Similarily, decompose into its off-diagonal, , and its diagonal . We have
We begin with the first term. Note that has off-diangoal entries , and hence,
where we have the same argument as used in the proof of Lemma C.6 and the fact that the Assumptions A.1-A.3 imply Assumption C.1, as shown in Lemma E.2.
Because is a diagonal matrix, for sufficiently large,
where we have used that by assumption is -Lipschitz continuous on the restriction for some . Clearly due to Lemma C.1. Furthermore, by Assumption C.1, for any , is continuously differentiable and hence also Lipschitz continuous in a closed ball around . Thus, . Hence, it is only left to show that , which is a consequence of the following claim. Claim: For any and any ,
where is a constant only depending on . Proof of the claim: In order to prove the claim, recall that due to Lemma C.1, for every
We prove the claim by induction. The case where holds trivially with . For ,
Next, by induction, for any . Furthermore, , which shows that
which completes the induction and thus the proof ∎
E.4 Proof of Lemma C.8
which holds for all . Hence, for , we can write
E.5 Proof of Lemma C.10
We start the proof with a discussion of existing results in the literature. As shown in Appendix E.1 in , the homogeneity of allows us to write
The function is continuous in $(-1,1)$
Based on this discussion, we now prove the lemma. In a first step, we derive a closed form expression for . Based on the discussion above and particularly Equation (30) we can see that
The goal is now to show that whenever , can be expressed as a sum of the form
with . We prove by induction. In a first step, note that the case where holds trivially true. Next, assume that Equation (32) holds true for . Due to the above discussion, can be expressed as a Taylor series around with positive coefficients . Thus,
Next, note that any of the properties 1-4 from the above discussion also hold true for . Therefore, we can use exactly the same argument for to show that for any ,
with . Finally, we can conclude the proof because
and when using the same argument as used in the induction step above to show that the resulting multi-sum converges absolutely. ∎
E.6 Additional lemmas
Any kernel which satisfies Assumption A.1 and A.3 also satisfies Assumption C.1.
The only point which does not follow immediately is to show that is a continuous function. For this, write as a function of the variables , i.e.
For every , define the function . Due to the series expansion, we can make use of the theory on the Taylor expansion which implies that for any , is a smooth function in the interior of (using the definition from Assumption A.1). Hence, we can conclude that there exists a function such that
In particular, the smoothness of implies that is continuous in the interior of . Next, define and note that that the continuity of implies that is continuous everywhere except for the plane . Finally, because exists point wise, we conclude that exists and is a continuous function in the interior of . ∎
Any RBF kernel with locally analytic around satisfies Assumption C.1.
Because by assumption has a local Taylor series around , we can write
is a sum of non negative summands. Hence, we get that the sum
converges absolutely. Thus, we can arbitrarily reorder the summands: