Gradient-Based Empirical Risk Minimization Using Local Polynomial Regression

Published Online:https://doi.org/10.1287/stsy.2022.0003

Abstract

In this paper, we consider the widely studied problem of empirical risk minimization (ERM) of strongly convex and smooth loss functions using iterative gradient-based methods. A major goal of the existing literature has been to compare different prototypical algorithms, such as batch gradient descent (GD) or stochastic gradient descent (SGD), by analyzing their rates of convergence to ϵ-approximate solutions with respect to the number of gradient computations, which is also known as the oracle complexity. For example, the oracle complexity of GD is O(n log(pϵ1)), where n is the number of training samples and p is the parameter space dimension. When n is large, this can be prohibitively expensive in practice, and SGD is preferred due to its oracle complexity of O(pϵ1). Such standard analyses only utilize the smoothness of the loss function in the parameter being optimized. In contrast, we demonstrate that when the loss function is smooth in the data, we can learn the oracle at every iteration and beat the oracle complexities of GD, SGD, and their variants in important regimes. Specifically, at every iteration, our proposed algorithm, Local Polynomial Interpolation-based Gradient Descent (LPI-GD), first performs local polynomial regression with a virtual batch of data points to learn the gradient of the loss function and then estimates the true gradient of the ERM objective function. We establish that the oracle complexity of LPI-GD is O˜((pϵ1)d/(2η)), where d is the data space dimension, and the gradient of the loss function is assumed to belong to an η-Hölder class with respect to the data. Our proof extends the analysis of local polynomial regression in nonparametric statistics to provide supremum norm guarantees for interpolation in multivariate settings and also exploits tools from the inexact GD literature. Unlike the complexities of GD and SGD, the complexity of our method depends on d. However, our algorithm outperforms GD, SGD, and their variants in oracle complexity for a broad range of settings where d is small relative to n. For example, with typical loss functions (such as squared or cross-entropy loss), when p=O(nβ) for any β>0 and ϵ=Θ(n1/2) is at the statistical limit, our method can be made to require O(nδ) oracle calls for any δ>0, while SGD and GD require O(nβ+(1/2)) and O(n log(n)) oracle calls, respectively.

Funding: This work was supported in part by the Office of Naval Research [Grant N000142012394], in part by the Army Research Office [Multidisciplinary University Research Initiative Grant W911NF-19-1-0217], and in part by the National Science Foundation [Transdisciplinary Research In Principles Of Data Science, Foundations of Data Science].

1. Introduction

Empirical risk minimization (ERM) is one of the mainstays of contemporary machine learning. Indeed, training tasks, such as classification, regression, or representation learning using deep neural networks, can all be formulated as specific instances of ERM. In this paper, we consider gradient-based iterative optimization methods that are used to perform ERM, such as batch gradient descent (GD) (Nesterov 2004), stochastic gradient descent (SGD) (Nemirovski et al. 2009), and their refinements (e.g., Nesterov 1983, Dekel et al. 2012, Johnson and Zhang 2013). Because the primary computational bottleneck in such settings is gradient computation, a significant portion of this literature is concerned with analyzing the convergence rates of different iterative algorithms and determining which ones have smaller oracle complexities—that is, smaller numbers of gradient queries required to obtain approximate minimizers—under various levels of convexity, smoothness, sampling, and other assumptions (see, e.g., Bubeck 2015, Jain and Kar 2017, Netrapalli 2019, and the references therein). We focus on the widely studied setting of strongly convex loss functions. In this case, neglecting any implicit dependence on dimension, the (first-order) oracle complexity of GD is known to be O(n log(ϵ1)) (Nesterov 2004, theorem 2.1.15), where n is the number of training samples in the ERM objective function and ϵ>0 is the desired approximation accuracy of the solution. In comparison, it is well-known that SGD has an oracle complexity of O(ϵ1) (Nemirovski et al. 2009, Bottou et al. 2018), again neglecting any implicit dependence on dimension. It turns out that by carefully accounting for dependence on the dimension p of the parameter of interest in such an ERM, the oracle complexities of GD and SGD are O(n log(pϵ1)) and O(pϵ1), respectively (see Propositions A.1 and A.2 in Appendix A). Typically, when performing ERM, the objective is to learn the true risk minimizer as accurately as possible. However, the empirical risk is statistically inaccurate, generally, up to a scaling of order n1/2. Hence, the typical regime of interest is ϵ=Ω(n1/2) when n is large. For small p (i.e., pn), this makes SGD, at least theoretically, preferable to GD. This is corroborated by the ubiquitous use of SGD and its variants for ERM in practice. Our goal is to improve upon the oracle complexities of GD, SGD, and their variants by exploiting additional information about the loss functions that has heretofore been neglected in the literature.

Recall that the general unconstrained ERM problem is usually abstracted as (cf. Bubeck 2015, section 1, and Hastie et al. 2009):

minθRp1ni=1nfi(θ),(1)
where each fi:RpR represents the loss function corresponding to a different sample of training data. (Note that this formulation, modulo the extra scaling n1, is also utilized in distributed optimization, where each fi represents the local objective function corresponding to an individual agent in a network of n agents (Nedić 2015).) As suggested by the form of (1), typical analysis of iterative gradient-based algorithms, such as GD or SGD, makes no assumptions about fi as a function of i. However, the loss function often varies quite smoothly with the data in ERM. For example, given training data {(x(i),yi)Rp×R:i[n]} (with [n]{1,,n}), the classical least-squares formulation of linear regression with regularization is (see, e.g., Hastie et al. 2009, chapter 3):
minθRp1ni=1n(yiθTx(i))2+R(θ),(2)
where the objective is to find the optimal coefficients θRp, and R(θ) is some regularization term—for example, ridge or lasso. Letting f(x,y;θ)=(yθTx)2+R(θ) for x,θRp and yR, we see that each fi(θ)=f(x(i),yi;θ) in formulation (1) for this problem. Clearly, f is very smooth as a function of the data (x, y), regardless of the regularization R(θ). This observation that loss functions are often smooth in the data holds in more complex learning scenarios, too. For instance, when training a deep neural network with a smooth loss—for example, cross-entropy loss—and smooth activation functions—for example, logistic (sigmoid) functions—the induced loss function for the ERM objective function is infinitely differentiable in both the data and the network’s weight parameters (Bishop 2006, chapter 5). Because such smoothness in the data has not been exploited by modern analysis of iterative gradient-based optimization algorithms for (1) (see Section 1.3), the main thrust of this work is to speed up algorithms for ERM using this additional smoothness in data when the data dimension is small relative to n.

To this end, in the remainder of this work, we consider the following problem formulation of ERM. For d,nN, assume that we are given n samples of d-dimensional training data D={x(i)[h,1h]d:i[n]}, where h>0 is an arbitrarily small parameter (to be determined later; see (25)), and each training sample belongs to the hypercube [h,1h]d.1 Note that when d2, the first d − 1 elements of a sample can be perceived as a (d1)-dimensional feature vector, and the last element of the sample can be construed as the corresponding continuous-valued label (as in the linear regression example above). For a given pN, fix the (parametrized) loss function f:[0,1]d×RpR,f(x;θ), where θ=(θ1,,θp)Rp is a p-dimensional parameter vector. Now, define the ERM objective function F:RpR as follows:

θRp,F(θ)1ni=1nf(x(i);θ),(3)
which is the empirical risk—that is, the empirical expectation of the loss function with respect to the training data D. With this objective function, we consider the unconstrained minimization problem:
F*infθRp F(θ),(4)
where F*R denotes the minimum value. In contrast to the canonical formulation (1), our formulation in (3) and (4) explicitly states the dependence of the loss function on the data. The next subsection delineates the main idea of our new inexact gradient descent algorithm to solve this problem, which exploits the smoothness of loss functions in the data. This is followed by discussions of our main contributions, related literature, and an outline of the rest of the paper.

1.1. Main Idea: Local Polynomial Interpolation-Based Gradient Descent

To present the main idea of our algorithm, it is useful to introduce the inexact gradient descent framework, where inexact gradients are used with bounded 2-norm error guarantees (cf. Friedlander and Schmidt 2012, section 1.1).2 Given the objective function F(θ), a sequence of step sizes {αt>0:tN}, and an initial parameter vector θ(0)Rp, an inexact gradient descent algorithm proceeds via the iterations

θ(t)=θ(t1)αtF^(t),(5)
where θ(t) is the parameter vector generated by the tth iteration, and for each tN,F^(t) denotes an approximation of the true gradient θF(θ(t1)). Moreover, the inexact gradient approximations possess 2-norm error guarantees of the form
F^(t)θF(θ(t1))22Δt,(6)
for each tN, where {Δt0:tN} is the sequence of error bounds. Note that for standard GD, Δt=0 for tN. On the other hand, for stochastic optimization algorithms, Δt bounds the expected values of squared 2-norm errors. These error guarantees are useful for the convergence analysis of inexact gradient descent methods (e.g., Friedlander and Schmidt 2012, lemma 2.1).

In this work, we propose an inexact gradient descent method that uses local polynomial interpolation to produce the approximations of true gradients over iterations. Specifically, we first construct a (finite) uniform grid G[0,1]d over the data space [0,1]d. Then, for every training data sample x(j)D, we precompute the (scalar-valued) interpolation weights {wy*(x(j)):yG} such that

yGwy*(x(j))fθi(y;θ)fθi(x(j);θ),(7)
for all i and any θRp—that is, linear combinations of partial derivatives evaluated at grid points with the aforementioned weights are “good” estimators of the partial derivatives at training samples (this will be rigorized in Section 3). These weights are calculated using basic linear algebraic operations so that local polynomial regression analysis ensures that the above approximation holds; we defer the details of local polynomial regression to Sections 3.1 and 3.2.1. Notably, these linear weights need only be computed once for a specific choice of bounded kernel function, and calculating them does not require any gradient computation of the loss function.

After these precomputation stages, we run inexact gradient descent using a pertinent choice of fixed step size αt=α>0 for an appropriately chosen number of iterations TN. In particular, at iteration tN, due to the finite sum structure of the ERM objective function in (3), we construct the following approximation of the true gradient θF(θ(t1)) using the previous iteration’s parameter vector θ(t1):

F^(t)=1nj=1n[yGwy*(x(j))fθ1(y;θ(t1))yGwy*(x(j))fθp(y;θ(t1))].(8)

Constructing this local polynomial interpolation-based approximation F^(t) requires us to compute |G| gradients of the loss function {θf(y;θ(t1)):yG}. Then, we compute the standard inexact gradient descent update

θ(t)=θ(t1)αF^(t),(9)
as shown in (5). We refer to this new approach as the Local Polynomial Interpolation-based Gradient Descent (LPI-GD) algorithm. This algorithm and its analysis are presented in detail in Section 3.

We note that the inexact gradients F^(t) are “good” approximations of θF(θ(t1)), and we can derive 2-norm error guarantees Δt as long as (7) is a “good” approximation. Intuitively, when the inexact gradients in LPI-GD are good approximations of the true gradients and the grid size |G| is much smaller than the number of training samples |D|=n, we expect LPI-GD to outperform GD and related methods in oracle complexity. As we will illustrate in the sequel, sufficient smoothness of loss functions f in the data yields accurate approximations of true gradients in LPI-GD and small grid sizes |G| when the data dimension d is also small with respect to n. This small data dimension requirement, at least theoretically, restricts the scope of problems where LPI-GD can be applied. However, LPI-GD could be useful in settings where the data dimension is inherently small or can be reduced using dimensionality-reduction techniques (as discussed in Section 1.3).

1.2. Main Contributions

Having conveyed the main idea of our approach, we now delineate the main contributions of this work. Note that our theoretical results hold under the smoothness and strong convexity assumptions stated in Section 2.2.

  1. We propose a new inexact gradient descent algorithm, LPI-GD, to compute approximate solutions of (4) in Section 3.2.1 (see Algorithm 1). As noted in Section 1.1, the main innovation of this algorithm is to use the smoothness of the loss function f in the data to learn the gradient oracle of f at every iteration by performing local polynomial regression based on oracle queries over a set of virtual data points. Interestingly, our estimators for gradients are linear, and the corresponding weights must only be computed once in the algorithm.

  2. We derive the iteration and oracle complexities of this algorithm in Proposition 2 and Theorem 2, respectively, in Section 3.2.2. In particular, we show that the oracle complexity scales like O˜((pϵ1)d/(2η)) (neglecting subdominant factors), where η>0 denotes the Hölder exponent determining the smoothness of the gradient of the loss function with respect to the data (see Section 2.2). Although we focus on strongly convex loss functions to leverage simple convergence results on inexact gradient descent from the literature, analogous analysis could be carried out for larger classes of loss functions—for example, when f(x;·):RpR is nonconvex.

  3. Furthermore, we show in Propositions 3 and 4 in Section 3.2.2 that if the data dimension is sufficiently small—for example, d=O(log log(n)) (as in healthcare data analytics or control applications; see Section 1.3)—and the loss function f exhibits modest smoothness in the data—that is, η=Θ(d)—then the oracle complexity for our method beats the oracle complexities (or, more precisely, the oracle complexity bounds) of GD, SGD, and their variance reduced and mini-batch variants for any p=O(poly(n)) and any ϵ=Θ(poly(n1)). In particular, when we let ϵ=Θ(n1/2) have the smallest scaling of interest (as explained earlier) and p=O(nβ) for any β>0, Table 1 depicts that the oracle complexities of GD, SGD, and LPI-GD are O(n log(n)),O(nβ+(1/2)), and O(nδ) for any small δ>0 (when the loss function is smooth enough), respectively. This demonstrates that smoothness of loss functions in the data can be successfully exploited in gradient-based algorithms for ERM to significantly lower their oracle complexity when the data dimension is small.

  4. Finally, in order to analyze the convergence rate of our algorithm, we also generalize the technique of local polynomial regression, or, more precisely, local polynomial interpolation (because there is no noise in our problem), in nonparametric statistics to the multivariate setting in Theorem 1 in Section 3.1. Although it is implied in many standard expositions that univariate local polynomial regression can be generalized to multiple variables (see, e.g., Fan and Gijbels 1996, Tsybakov 2009, or Wasserman 2019), the multivariate case requires significantly more care than the univariate case. Consequently, a detailed and rigorous nonasymptotic analysis of local polynomial interpolation is difficult to find in the literature. Therefore, we believe that Theorem 1 may be of independent interest in statistics.

Table

Table 1. Oracle Complexities of GD, SGD, and Our LPI-GD Algorithm for Strongly Convex Loss Functions

Table 1. Oracle Complexities of GD, SGD, and Our LPI-GD Algorithm for Strongly Convex Loss Functions

AlgorithmOracle complexity
GDO(n log(n)) (Nesterov 2004)
SGDO(nβ+(1/2)) (Bottou et al. 2018)
LPI-GDO(nδ) (Theorem 2)


Notes. Here, we assume that the approximation accuracy ϵ=Θ(n1/2), the parameter dimension p=O(nβ) for any constant β>0, and the data dimension d=O(log log(n)), where n is the number of training samples. Moreover, δ>0 is a very small constant of our choice (when the loss function is commensurately smooth with respect to the data).

1.3. Related Literature

We next discuss relevant existing work to contextualize the contributions of this work. There is an enormous literature concerning the analysis of convergence rates of exact and inexact gradient-based algorithms that perform the optimization in (1). It is not feasible to survey this entire literature here, and, instead, we only mention some closely related works. For example, in our strongly convex setting of interest, the convergence analysis of GD can be found in the standard text Nesterov (2004), and corresponding convergence analysis of SGD can be found in Nemirovski et al. (2009) (also see Jain and Kar 2017, Bottou et al. 2018, Netrapalli 2019, and the references therein). Moreover, various refinements and improvements of these basic iterative algorithms have been analyzed in the literature. Notable examples include the accelerated or momentum-based GD (Polyak 1964, Nesterov 1983), mini-batch SGD methods (Dekel et al. 2012), and variance reduced SGD methods (Le Roux et al. 2012, Johnson and Zhang 2013, Shalev-Shwartz and Zhang 2013, Schmidt et al. 2017, Schmidt et al. 2017). We also refer readers to Bubeck (2015) for a unified treatment of many of the aforementioned approaches. More generally, inexact gradient descent methods for ERM have been analyzed extensively as well—see, for example, Friedlander and Schmidt (2012), Devolder et al. (2013), So and Zhou (2017), and the references therein for work on the strongly convex case. (Note that SGD can be perceived as an example of an inexact GD method, which uses unbiased estimates of the true gradient of the ERM objective function at every iteration.) This paper and most of the aforementioned references focus on obtaining upper bounds on oracle complexity. Conversely, several authors have also established fundamental lower bounds; cf. Agarwal et al. (2009), Carmon et al. (2021), and the references therein. As mentioned earlier, to our knowledge, all of this literature concerns the formulation in (1). Hence, these methods do not exploit smoothness of loss functions in data by learning gradients in every iteration as in our proposed method, LPI-GD.

On the other hand, motivated by the utility of gradient information for various tasks like variable selection and determining coordinate covariations (cf. Mukherjee and Zhou 2006 and Mukherjee and Wu 2006), the problem of learning gradients in the context of supervised learning problems, such as classification and regression, has received a lot of attention from the machine learning community. Specifically, most of this literature aims to simultaneously learn a classification or regression function along with its gradient from training data. For example, the text Fan and Gijbels (1996) presents the classical theory of gradient estimation using local polynomial fitting, and De Brabanter et al. (2013) present a gradient estimation approach that uses local polynomial regression, but does not estimate the regression function. In a related vein, Delecroix and Rosa (1996) study kernel methods for gradient estimation. Various other methods for learning gradients have also been analyzed, such as reproducing kernel Hilbert space methods (Mukherjee and Wu 2006, Mukherjee and Zhou 2006), regression splines (Zhou and Wolfe 2000), and nearest-neighbor methods (Ausset et al. 2021). However, the vast majority of these techniques have not been considered in the context of optimization for machine learning.

Two recent exceptions where gradient estimation has been applied to optimization are Wang et al. (2018) and Ausset et al. (2021). In particular, Wang et al. (2018) illustrate that learning gradients leads to desirable convergence rates for derivative-free optimization algorithms, which only use zeroth order oracles (i.e., these oracles only output function values, not gradient values). Similarly, Ausset et al. (2021) portray the effectiveness of their nearest-neighbor-based gradient estimator for derivative-free optimization. It is worth noting that both of these works analyze optimization of a single objective function (rather than a sum, as in (1)). Hence, the results derived in these papers are not particularly insightful in our ERM setting in (3) and (4), where we seek to learn gradients based on a first-order oracle by exploiting the smoothness of f(x;θ) in the data x.

Lastly, as we mentioned in Section 1.2, we establish in Propositions 3 and 4 that the oracle complexity of our LPI-GD algorithm beats GD, SGD, and their variants for a range of values of p and ϵ, if the data dimension d is sufficiently small—for example, d=O(log log(n)). Because we utilize multivariate local polynomial regression in our algorithm, a “small d assumption” is theoretically unavoidable due to the curse of dimensionality (cf. Fan and Gijbels 1996, section 7.1). This restriction on d can be a limitation for a variety of ERM applications where d is very large, and LPI-GD is a suitable choice primarily when data can be restricted to live in small dimensions. We close this discussion of related literature by delineating some application domains where LPI-GD may be suitable. Indeed, among the multitude of contexts where ERM is used, certain contexts have problems where the number of data attributes is inherently small. For example, in the healthcare data analytics space, electronic health records often have very few features per patient—for example, temperature, blood pressure, initial diagnosis, and medications—but machine learning techniques are still widely used to build models for diagnosing diseases or predicting risk (cf. Choi et al. 2018 and its references). As another example, in the problem of system identification for robotics and control applications, neural networks have been used to build nonlinear state-space models (cf. Nechyba and Xu 1994, Nelles 2001, and their references). The number of inputs of these networks is often very small when the model order is known or assumed to be low. On the other hand, in certain applications with high-dimensional “raw data,” data scientists believe that the useful information in the data lives in a low-dimensional manifold. This belief is the basis of a myriad of dimensionality reduction techniques, such as principal component analysis (Pearson 1901, Hotelling 1933), canonical correlation analysis (Hotelling 1936), Laplacian eigenmaps (Belkin and Niyogi 2001) or diffusion maps (Coifman and Lafon 2006), and modal decompositions (or correspondence analysis), cf. Huang et al. (2024), Makur (2019, chapter 4), and the references therein. As a result, certain high-dimensional data analysis pipelines proceed by first preprocessing the data using pertinent dimensionality reduction techniques and then using the extracted low-dimensional features for other learning tasks. Our contributions in this work could be of utility in such high-dimensional settings where a dimensionality reduction procedure is used before ERM. In applications where the data dimension is large and dimensionality reduction cannot be applied, however, LPI-GD is likely not a suitable algorithm.

1.4. Outline

Finally, the outline of the remainder of this paper is as follows. We introduce required notation and definitions, formally state key assumptions, and briefly elaborate on the standard framework of oracle complexity in optimization theory in Section 2. In Section 3, we provide a detailed development of the LPI-GD algorithm and present our main results. We prove our results on local polynomial regression in Section 4 and carry out the convergence analysis of LPI-GD in Section 5. Finally, we conclude our discussion and propose future research directions in Section 6. Furthermore, we present known oracle complexity results for GD, SGD, and their variants in Appendix A and discuss the key assumptions used in this work in Appendix B (cf. Jadbabaie et al. 2023, section II-C, appendix I).

2. Formal Setup

In this section, we present relevant notation in Section 2.1, important assumptions and definitions in Section 2.1, and the notion of oracle complexity in Section 2.3. These notions will be used throughout this work.

2.1. Notation

We briefly introduce some notation that is utilized in this work. Let N{1,2,3,} and Z+{0,1,2,3,} denote the sets of natural numbers and nonnegative integers, respectively. As mentioned earlier, for any nN, let [n]{1,,n} be the set of positive integers less than n + 1. In the sequel, we analyze a d-variate problem setting with dN. So, we briefly introduce the standard multi-index notation. For any d-tuple of nonnegative integers s=(s1,,sd)Z+d, we let |s|=s1++sd,s!=s1!sd!, and xs=x1s1xdsd for xRd. Moreover, given a continuously differentiable function f:RdR,f(x1,,xd), we write sf=|s|f/(x1s1xdsd) to denote its sth partial derivative, and xf=[fx1fxd]T to denote its gradient. We will sometimes find it convenient to index certain vectors or matrices using the set {sZ+d:|s|l}. In these cases, to help with calculations, we will assume that these vectors or matrices are written out with the indices in lexicographical order (i.e., for any r,sZ+d with rs, we say that r < s if rj < sj for the first position j[d], where rjsj).

Furthermore, for any vector xRn with nN, we let xi be the ith entry of x, where i[n] (or more generally, i belongs to an appropriate index set), and we let xp be the usual p-norm of x with p[1,+]. Similarly, for any matrix ARm×n with m,nN, we let Ai,j be the (i, j)th element of A, where i[m] and j[n] (or, more generally, i and j belong to appropriate index sets), and we let AF be the Frobenius norm of A. In addition, AT represents the transpose of A, and when m = n, A1 represents the inverse of A if it is invertible, tr(A) represents the trace of A, det(A) represents the determinant of A, and λmin(A) represents the minimum eigenvalue of A. We also let In be the n × n identity matrix. For any pair of symmetric matrices A,BRn×n, we use to denote the Löwner partial order and write AB to mean that AB is positive semidefinite.

Finally, we introduce some useful miscellaneous notation. We let · denote the ceiling function, 𝟙{·} denote the indicator function that equals one if its input proposition is true and equals zero otherwise, exp(·) and log(·) denote the natural exponential and logarithm functions with base e, respectively, and E[·] denote the expectation operator, where the underlying probability measure can be easily inferred from context. Throughout this work, we will also utilize standard Bachmann-Landau asymptotic notation—that is, O(·),Θ(·),ω(·), etc.—where the underlying limiting variable (that usually tends to infinity) will be clear from context. Moreover, we will let poly(n) and polylog(n) denote polynomial and polylogarithmic scaling in the variable n, respectively.

2.2. Smoothness Assumptions and Approximate Solutions

In order to perform tractable convergence analysis of gradient-based iterative algorithms that solve (4), we must formally define the problem class by specifying any smoothness conditions on the loss function, the appropriate notion of “approximate solutions,” and the oracle model of computational complexity used (Nesterov 2004, section 1.1). This section and Section 2.3 precisely define our problem class.

Recall our formulation where we are given n samples of d-dimensional training data D and a loss function f:[0,1]d×RpR,f(x,θ), and we consider the ERM problem in (3) and (4). Let ϵ>0 be any small constant representing the desired level of approximation accuracy. We begin by imposing the following assumptions on our loss function (also see Jadbabaie et al. 2023, section II-B):

  1. Smoothness in parameter: There exists L1>0 such that for all fixed x[0,1]d, the gradient of f with respect to θ, denoted θf(x;·):RpRp, exists and is L1-Lipschitz continuous as a function of the parameter vector:

    θ1,θ2Rp,θf(x;θ1)θf(x;θ2)2L1θ1θ22.(10)

  2. Smoothness in data: There exist η>0 and L2>0 such that for all fixed ϑRp and for all i[p], the ith partial derivative of f (with respect to θi) at ϑ, denoted gi(·;ϑ)fθi(·;ϑ):[0,1]dR, belongs to the (η,L2)-Hölder class as a function of the data vector x (cf. Tsybakov 2009, definition 1.2). By definition, this means that each gi(·;ϑ):[0,1]dR is l=η1 times differentiable, and for every s=(s1,,sd)Z+d such that |s|=l, we have

    y1,y2[0,1]d,|sgi(y1;ϑ)sgi(y2;ϑ)|L2y1y21ηl.(11)

  3. Strong convexity: There exists μ>0 such that for all fixed x[0,1]d, the map f(x;·):RpR is μ-strongly convex as a function of the parameter vector:

    θ1,θ2Rp,f(x;θ1)f(x;θ2)+θf(x;θ2)T(θ1θ2)+μ2θ1θ222.(12)

    Moreover, as additional notation, we let σL1/μ represent the condition number of the problem.

Under these assumptions, we define an ϵ-approximate solution to the optimization problem (4) as any parameter vector θ*Rp that satisfies

F(θ*)F*ϵ.(13)

We make two pertinent remarks at this point. Firstly, it can be argued based on the above strong convexity assumption (as shown in Section 5.1) that F* is finite and achievable by a unique global minimizer of F; in particular, the infimum in (4) may be replaced by a minimum. Secondly, both the first and third assumptions are standard in the optimization literature (cf. Nesterov 2004 and Bubeck 2015). As noted earlier, our main idea is the observation that the second assumption, that gradients of loss functions are also smooth in the data, holds in many machine learning applications. For example, in the regularized linear regression example in (2), θf is clearly smooth with respect to the data (assuming R is differentiable). We refer readers to Appendix B, which is taken from (Jadbabaie et al. 2023, section II-C, appendix I), for further discussion of these standard assumptions within the optimization and statistics contexts, where examples of (2-regularized) linear and logistic regression satisfying these assumptions are also provided.

2.3. Oracle Complexity

Finally, we briefly introduce the formalism of “first-order oracle complexity,” upon which our analysis will be based. The notion of oracle complexity was first introduced in Nemirovskiĭ and Yudin (1983) to circumvent the difficulties faced by vanilla computational complexity theory style analysis of convex programming (cf. Agarwal et al. 2009, section 1). As elucidated in Nesterov (2004, section 1.1), although “arithmetical complexity,” which accounts for the running time of all basic arithmetic operations executed by an algorithm, is a more realistic measure of computational complexity, the idea of oracle (or “analytical”) complexity—that is, the total number of oracle calls made by an algorithm—is a more convenient abstraction for the analysis of iterative optimization methods. Indeed, the running time of oracle queries is often the main bottleneck in arithmetical complexity.

Observe that any iterative algorithm that approximately solves optimization problems belonging to the class of ERM problems defined in (3) and (4), under the assumptions in Section 2.2, requires two inputs: (a) the training data D, and (b) a “description” of the specific loss function f under consideration. To fulfill the latter requirement, we will assume that such algorithms have access to a local first-order oracle O:[0,1]d×RpRp, defined as:

x[0,1]d,ϑRp,O(x,ϑ)θf(x;ϑ),(14)
which returns the value of the gradient θf with respect to θ for any input query pair (x,ϑ). (Note that sometimes the oracle is defined to produce values of both the loss function f and its gradient θf at (x,ϑ) (cf. Nesterov 2004, section 1.1.2).) This oracle O can be construed as a black box “description” of the gradient of the loss function f with respect to θ.

Given any particular iterative algorithm, A=A(D,O), that solves our class of ERM problems, its first-order oracle complexity (or just oracle complexity) Γ(A) is formally defined as the minimum number of oracle calls made by the algorithm A to obtain an ϵ-approximate solution in the sense of (13) (Nesterov 2004, section 1.1). Thus, Γ(A) is typically a function of the notable problem variables n,d,p,η,ϵ, and the fixed constants L1,L2,μ. In this paper, following the paradigm of the optimization literature, we will focus on determining the first-order oracle complexity of our proposed optimization method.

3. Main Results and Discussion

In this section, we first present a generalization of local polynomial regression to the multivariate case in Section 3.1 and then describe our proposed algorithm LPI-GD and its convergence analysis in Section 3.2.

3.1. Local Polynomial Regression

We commence by introducing the essentials of local polynomial regression specialized to our noiseless setting. Local polynomial regression was originally introduced in Stone (1977) and has experienced much subsequent development. Most references on the subject only present the univariate (i.e., d = 1) framework—for example, Tsybakov (2009, section 1.6), and Wasserman (2019, chapter “Nonparametric Regression”). However, local polynomial regression has also been developed and analyzed in multivariate settings (i.e., for general d). We refer readers to Fan and Gijbels (1996) for a standard reference on local polynomial regression from classical statistics, Cleveland and Loader (1996, section 2) for more classical references and a brief history of smoothing using local regression, and Zhang and Chan (2011) for a succinct, but reasonably complete, review of classical work on multivariate local polynomial regression. These sources and the references therein convey that whereas the univariate local polynomial regression literature contains nonasymptotic analyses (in the spirit of modern high-dimensional statistics, where both dimension and number of samples are finite, e.g., the presentation in Tsybakov 2009, section 1.6), the multivariate local polynomial regression literature primarily presents asymptotic results (with a more classical feel). For example, Masry (1996, 1999) analyzes multivariate local polynomial regression and proves asymptotic normality results (i.e., central limit theorems) for estimators of the regression function under various assumptions. Because we will require a nonasymptotic conditioning analysis of multivariate local polynomial regression in our subsequent analysis, we rigorously carry out the much more involved analysis for the multivariate case by carefully generalizing the univariate development in Tsybakov (2009, section 1.6).

Consider a function g:[0,1]dR that belongs to the (η,L2)-Hölder class, namely, g is l=η1 times differentiable and satisfies (11) for every sZ+d with |s|=l. In order to present the local polynomial regression-based approximation of this function, we first state some required definitions and notation. Let K:[1,1][b,c] be a bounded kernel function, where cb>0 are some fixed constants, and we will use the extension of this kernel (to the domain R), K:R[0,c], in the sequel with the assumption that the support of K is [1,1]. For any fixed mN, define the uniform grid

Gm{u[0,1]d:i[d],uim[m]},(15)
which has cardinality |Gm|=md. Finally, define the vector(-valued function) U:RdRD as:
uRd,U(u)[uss!:sZ+d,|s|l]T,(16)
which has dimension Ddim(U)=k=0l(k+d1d1)=(l+dd) (via the hockey-stick identity). Then, for any arbitrarily small value of the bandwidth parameter h(0,12), we can write the following weighted regression problem:
x[h,1h]d,Φ^(x)arg minϕRDyGm(g(y)ϕTU(yxh))2j=1dK(yjxjh),(17)
where Φ^(x)=[Φ^1(x)Φ^D(x)]T. We refer to the first entry ϕ^(x)Φ^1(x)=U(0)TΦ^(x) of Φ^(x) as the local polynomial interpolator of order l for g:[h,1h]dR (which is g:[0,1]dR restricted to the domain [h,1h]d).

To intuitively understand the problem in (17), let us temporarily specialize to the setting where d = 1, η = 2, and the kernel K(u)=𝟙{1u1}. Then, g:[0,1]R belonging to the (2,L2)-Hölder class means that g is continuously differentiable and g:[0,1]R is L2-Lipschitz continuous. Moreover, the uniform grid has the form Gm={1m,2m,,m1m,1} and U:RR2 is given by U(u)=[1u]T. Hence, the problem in (17) becomes:

x[h,1h],(Φ^1(x),Φ^2(x))=arg min(ϕ1,ϕ2)R2yGm(g(y)ϕ1ϕ2(yxh))2𝟙{1yxh1}=arg min(ϕ1,ϕ2)R2i[m]:xhimx+h(g(im)ϕ1ϕ2(imxh))2.(18)

For any fixed x, the problem in (18) performs linear regression to approximate g in a local neighborhood of x using a subset of the samples {(y,g(y)):yGm}, which live in the neighborhood. Because the mean value theorem and L2-Lipschitz continuity of g imply the first-order Taylor approximation g(y)=g(x)+g(x)(yx)+O((yx)2) for all y in the neighborhood of x (cf. Rudin 1976, theorem 5.10), we can intuitively interpret the (scaled) optimal regression coefficients Φ^1(x) and Φ^2(x)/h as estimators of the first-order Taylor approximation coefficients g(x) and g(x), respectively. This is why ϕ^(x)=Φ^1(x) is a reasonable approximation of g(x). We refer readers to standard texts, such as Tsybakov (2009), for further intuition.

Returning to our general setting in (17), we proceed to showing that the interpolator ϕ^ provides a “good” approximation of the function g in the supremum (or Chebyshev) norm. Toward this end, for any x[h,1h]d, construct the D × D symmetric matrix

B(x)1(mh)dyGmU(yxh)U(yxh)Tj=1dK(yjxjh),(19)
which is also positive semidefinite due to the nonnegativity of the kernel. We next present a key technical proposition, which generalizes (Tsybakov 2009, lemma 1.5) to the d-variate setting and demonstrates that B(x) is actually positive definite. To present this proposition, we also define the (d, l)-dependent constant
Λ(d,l)(πd8e2)d(l+dd)d((l+dd)d1)(l+dd)d11(l+d)3l(e(l+d)d)d,(20)
which will be used in subsequent results as well for crisper exposition.

Proposition 1

(Löwner Lower Bound). Suppose that ld19. Then, for all mN and h(0,12) such that mh4l(3e)d/Λ(d,l) and all x[h,1h]d, we have

B(x)bdΛ(d,l)2ID,
where the constant Λ(d,l)>0 is defined in (20).

Proposition 1 is established in Section 4.1 by exploiting several basic ideas from the theory of orthogonal polynomials and matrix perturbation theory. The proof is much more sophisticated than Tsybakov (2009, lemma 1.5) because explicit quantitative estimates are derived to deal with general d. Because Proposition 1 shows that B(x) is invertible, a straightforward exercise in calculus conveys that the unique solution Φ^(x) of the weighted regression problem in (17) can be written as:

x[h,1h]d,Φ^(x)=yGmg(y)wy(x),(21)
where the optimal vector-valued weights wy(x)RD are given by:
yGm,x[h,1h]d,wy(x)=1(mh)d(j=1dK(yjxjh))B(x)1U(yxh).(22)

Then, defining the first coordinates of these vectors as the real-valued interpolation weights:

yGm,x[h,1h]d,wy*(x)=[wy(x)]1,(23)
the local polynomial interpolator of order l for g:[h,1h]dR can be written as the following weighted sum of the interpolation points {g(y):yGm}:
x[h,1h]d,ϕ^(x)=yGmg(y)wy*(x).(24)

In Section 4.2, we present some important properties of the interpolation weights given in (23) based on the key estimate in Proposition 1.

The ensuing theorem conveys that local polynomial regression can be used to uniformly approximate a function g (in the supremum norm sense) for sufficiently large uniform grids Gm.

Theorem 1

(Supremum Norm Interpolation). Fix any constant δ(0,1), and consider the function g:[0,1]dR, which belongs to the (η,L2)-Hölder class. Suppose that l=η1d19 and that the grid size |Gm|=md is sufficiently large so that

m110(2L2+1)cb(d(3e)dΛ(d,l)2)(1δ)1/η,
where cb>0 are the bounds on our kernel, and Λ(d,l) is defined in (20). Then, the supremum norm approximation error of g is bounded by δ, viz.
supx[h,1h]d|ϕ^(x)g(x)|δ,
where ϕ^ is the local polynomial interpolator of order l for g given in (24), and the bandwidth h(0,12) is chosen to be
h=4l(3e)dmΛ(d,l)2b55(2L2+1)c(Λ(d,l)ld).

Theorem 1 is proved in Section 4.3 using the pertinent properties of the interpolation weights in (23) derived in Section 4.2. We note that the lower bound on m is expected to scale at least exponentially with the dimension d due to the curse of dimensionality (cf. Fan and Gijbels 1996, section 7.1). However, it may be possible to improve the dependence we have above using a different analysis. In the sequel, we will exploit Theorem 1 to perform inexact gradient descent.

3.2. Inexact Gradient Descent Using Local Polynomial Interpolation

We now turn to describing and analyzing a new inexact gradient descent algorithm that performs local polynomial interpolation at every iteration. In particular, we delineate the algorithm in Section 3.2.1 and derive convergence rates in Section 3.2.2.

3.2.1. Description of Algorithm.

Under the setup and assumptions described in Section 1, suppose we seek to solve the optimization problem (4) in the ϵ-approximate sense of (13) based on some training data D with the parameter h>0 given by

h=2b55(2L2+1)c(Λ(d,l)ld)=O(Λ(d,l)ld),(25)
which is the upper bound on the bandwidth h in Theorem 1.3 We first present our proposed algorithm, dubbed Local Polynomial Interpolation-based Gradient Descent (LPI-GD), in general, and then specialize the values of different problem variables with judicious choices later on.

Let θ(t)Rp denote the “approximate solution” produced by LPI-GD at iteration tZ+, and let δt(0,1) denote the maximum allowable supremum norm approximation error at iteration tN. Fix any bounded kernel K:[1,1][b,c] with cb>0 as in Section 3.1, and corresponding to each δt at iteration tN, consider the uniform grid Gmt, where mtN is given by Theorem 1:

mt=110(2L2+1)cb(d(3e)dΛ(d,l)2)(1δt)1/η,(26)
and the bandwidth ht(0,12) is also given by Theorem 1:
ht=4l(3e)dmtΛ(d,l)h.(27)

We initialize θ(0)Rp to be any arbitrary vector in Rp at iteration t = 0. Then, at iteration tN, we make |Gmt|=mtd first-order oracle calls at all virtual data points in the uniform grid Gmt with the approximate solution at iteration t − 1:

yGmt,O(y,θ(t1))=θf(y;θ(t1)).(28)

For every i[p], these oracle calls give us information about the partial derivatives {gi(y;θ(t1)):yGmt} with respect to θi. Hence, akin to (17) and (24) in Section 3.1, we can construct the local polynomial interpolator ϕ^i(t):[h,1h]dR of order l=η1 for the (η,L2)-Hölder class function gi(·;θ(t1))=fθi(·;θ(t1)):[h,1h]dR as follows:

x[h,1h]d,ϕ^i(t)(x)=yGmtgi(y;θ(t1))wy,t*(x),(29)
where the interpolation weights wy,t*(x)R are defined analogously to (23); in particular, they are the first elements of the weight vectors wy,t(x)RD, which are defined as in (22), but with Gm replaced by the grid Gmt and h replaced by the bandwidth ht. Using the learned interpolators in (29) at iteration t, we next approximate the true gradient of the ERM objective function θF(θ(t1)) at θ(t1) as follows. For every i[p] and every j[n], we first estimate the partial derivative gi(x(j);θ(t1)) at the data sample x(j) by computing ϕ^i(t)(x(j)). Then, because we have θF(θ(t1))=1nj=1nθf(x(j);θ(t1)) due to (3), we can estimate θF(θ(t1)) by computing
F^(t)=1nj=1n[ϕ^i(t)(x(j)):i[p]]​T,(30)
where [ϕ^i(t)(x(j)):i[p]]TRp is our approximation of θf(x(j);θ(t1)) at the data sample x(j). Finally, we perform the inexact gradient descent update
θ(t)=θ(t1)1L1F^(t),(31)
which produces the approximate solution for iteration t. We run the above procedure for TN iterations, and the LPI-GD algorithm returns θ(T)Rp as the ϵ-approximate solution for the optimization problem (4). The analysis in Section 3.2.2 elucidates the specific values taken by the constants δt and the total number of iterations T in terms of the other problem parameters. In particular, we will see that the approximation errors δt, grid sizes mt, bandwidths ht, and uniform grids Gmt all will not change over iterations. A pseudocode summary of the LPI-GD algorithm with pertinent δt and T (selected as in Section 3.2.2) is presented in Algorithm 1; the algorithm uses the training data D and a first-order oracle O to generate an ϵ-approximate solution that solves (4). Notably, the interpolation weights only need to be computed once in the algorithm, and our local polynomial interpolators for gradients are linear estimators, which use these weights.

We remark that Algorithm 1 portrays a “theoretical” algorithm because it assumes that problem parameters, such as the Lipschitz constant L1, the strong convexity parameter μ, the minimum value F*, and the Hölder exponent η and constant L2, are known to the practitioner. When some or all of these parameters are unknown, we can use appropriate bounds and estimates based on the loss function f:[0,1]d×RpR. For example, we may use any upper bound on L1, any lower bound on μ, and any universal lower bound on f (which yields a lower bound on F*). Moreover, for smooth functions f, we can choose any desired level of differentiability η and any upper bound on L2. When such bounds or estimates are also unavailable, the best we can do is to select T, δt, mt, and ht to have the correct scaling with respect to ϵ, d, p, and η. It is worth mentioning that our objective in this paper is to demonstrate that the oracle complexities of GD, SGD, and their variants can be theoretically beaten by the proposed algorithm. So, we do not delve further into considerations regarding its practical implementation here.

Algorithm 1

(Local Polynomial Interpolation-Based Gradient Descent (LPI-GD) Algorithm to Approximately Solve (4))

Input: n samples of training data D={x(i)[h,1h]d:i[n]}

Input: first order oracle O:[0,1]d×RpRp defined in (14)

Input: approximation accuracy level ϵ>0

Output: ϵ-approximate solution θ*Rp satisfying (13)

  Step 1: Initialization

  • 1: Set arbitrary initial parameter vector θ(0)Rp

  • 2: Set number of iterations TN according to (32)

  • 3: Set supremum norm approximation error δ=(1σ1)T/2     ⊳ This is the same for all iterations; see (33)

  • 4: Set grid size mN according to (26)      ⊳ Substitute mt = m and δt=δ in (26)

  • 5: Set bandwidth h=4l(3e)d/(mΛ(d,l))      ⊳ See (27)

      Step 2: Calculation of Interpolation Weights

  • 6: Construct uniform grid Gm according to (15)

  • 7: for j = 1 to n do     ⊳ Iterate over training data

  • 8:    Compute interpolation weights {wy*(x(j)):yGm} via (22) and (23) using kernel K:[1,1][b,c]

  • 9: end for

      Step 3: Inexact Gradient Descent

  • 10: for t = 1 to T do

  • 11:   Make oracle calls {O(y,θ(t1)):yGm} to get partial derivatives {fθi(y;θ(t1)):yGm,i[p]}

         Step 3a: Local Polynomial Interpolation

  • 12:   for j = 1 to n do      ⊳ Iterate over training data

  • 13:    Compute local polynomial interpolators {ϕ^i(t)(x(j)):i[p]} according to (29) using the oracle queries {fθi(y;θ(t1)):yGm,i[p]} and precomputed interpolation weights {wy*(x(j)):yGm}

  • 14:   end for

  • 15:   Construct estimate of gradient F^(t) using interpolators {ϕ^i(t)(x(j)):i[p],j[n]} as shown in (30)

         Step 3b: Standard GD Update

  • 16:   Update: θ(t)θ(t1)1L1F^(t)      ⊳ See (31)

  • 17: end for

  • 18: return θ*=θ(T)

Before proceeding to our analysis, it is worth juxtaposing Algorithm 1 with standard batch GD and SGD. At each iteration, GD exactly computes gradients for all data samples using n oracle calls. Because this can be computationally expensive, inexact gradient descent methods attempt to reduce the number of oracle calls per iteration at the cost of increasing the number of iterations. For example, SGD makes only one oracle call per iteration for a randomly chosen data sample and uses this queried gradient value as a proxy for the true gradient. In contrast, at each iteration, our approach evaluates the gradients at certain representative virtual points in the data space and uses these oracle calls to learn a model for the gradient oracle. This learnt model is then used to approximate the true gradient. Intuitively, when the loss function is a sufficiently smooth function of the data and the data dimension d is small, we can learn an accurate model with a reasonably small number of virtual points. It is also worth mentioning that because we perform regressions to learn our models, our proposed method is likely to be robust to noise in the oracle queries. However, we do not analyze this in the present work.

3.2.2. Analysis of Algorithm.

To compute the oracle complexity of the LPI-GD algorithm, we must first compute the minimum number of iterations it takes to obtain an ϵ-approximate solution to (4). The ensuing proposition chooses appropriate values of δt, which turn out to be constant with respect to t, and develops an upper bound on the minimum number of iterations T.

Proposition 2

(Iteration Complexity of LPI-GD). Suppose that l=η1d19 and the assumptions of Section 2.2 hold. For any (small) accuracy ϵ>0, define the number of iterations TN as

T=(log(σσ1))1log(F(θ(0))F*+p2μϵ),(32)
and the maximum allowable supremum norm approximation errors {δt(0,1):t[T]} via:
t[T],δt2=(11σ)T=Θ(ϵp),(33)
where F:RpR is the ERM objective function in (3), and σ=L1/μ>1 is the condition number defined in Section 2.2. Then, after T iterations of the LPI-GD algorithm described in Algorithm 1 in Section 3.2.1 with arbitrary initialization θ(0)Rp and updates (31), the parameter vector θ(T)Rp is an ϵ-approximate solution to the optimization problem in (4) in the sense of (13):
F(θ(T))F*ϵ.

Proposition 2 is proved in Section 5.1 using Theorem 1 and some known ideas from the analysis of inexact gradient descent methods in Friedlander and Schmidt (2012). Using Proposition 2, the ensuing theorem bounds the first-order oracle complexity of the LPI-GD algorithm in Algorithm 1, denoted Γ(LPI-GD), which is the minimum number of oracle calls made by Algorithm 1 to obtain an ϵ-approximate solution (as defined in Section 2.3). To succinctly present the ensuing theorem, we also define the constant

Cμ,L1,L2,b,c(d,l)(σ+2(L1μ)(L1μ)log(σσ1))(220(2L2+1)cb)d(dd(3e)d2Λ(d,l)2d),(34)
where Λ(d,l) is defined in (20), and μ,L1,L2,b,c are constants defined in Sections 2.2 and 3.1.

Theorem 2

(Oracle Complexity of LPI-GD). Suppose that l=η1d19 and the assumptions of Section 2.2 hold. For any (small) accuracy 0<ϵ(L1μ)p2L1μ, consider the LPI-GD algorithm described in Algorithm 1 in Section 3.2.1 with arbitrary initialization θ(0)Rp, number of iterations TN as in (32), maximum allowable approximation errors {δt(0,1):t[T]} as in (33), and updates (31). Then, the first-order oracle complexity of this algorithm (to obtain an ϵ-approximate solution in the sense of (13)) is upper bounded by

Γ(LPI-GD)Cμ,L1,L2,b,c(d,l)(p+2μ(F(θ(0))F*)ϵ)d/(2η)log(p+2μ(F(θ(0))F*)2μϵ),
where the constant Cμ,L1,L2,b,c(d,l) is defined in (34).

Theorem 2 is established in Section 5.2. Unlike the “dimension-free” oracle complexity bounds for batch GD and SGD, our bound in Theorem 2 explicitly depends on d and p. However, even though it may not be immediately obvious, the oracle complexity of LPI-GD scales much better than the oracle complexities of GD or SGD in important regimes. We next elucidate a scaling law for the bound in Theorem 2 in the setting where n,d,p,η,ϵ1 and all other problem parameters are constant.

As in Appendix A, let θ*Rp be the unique global minimizer of the empirical risk F:RpR in (3) so that F*=F(θ*). Assume that the parameter θ belongs to an -ball with constant radius (as in many applications). Then, in the aforementioned setting, we have using Bubeck (2015, lemma 3.4) that

F(θ(0))F*L12θ(0)θ*22=O(p),(35)
where the inequality uses the facts that θF:RpRp is L1-Lipschitz continuous (see the proof of Proposition 2 in Section 5.1) and θF(θ*)=0 (which is the zero vector), and the equality holds because θ(0)θ*22=O(p) (see Appendix A). Hence, the first-order oracle complexity of the LPI-GD algorithm in Theorem 2 scales as
Γ(LPI-GD)=O(Cμ,L1,L2,b,c(d,l)(pϵ)d/(2η)log(pϵ)).(36)

For comparison, recall from Proposition A.1 in Appendix A (cf. Nesterov 2004, theorem 2.1.15) that when the empirical risk is strongly convex and has Lipschitz continuous gradient, the first-order oracle complexity of GD is bounded by

Γ(GD)Γ*(GD)Θ(n log(pϵ)).(37)

Similarly, recall from Proposition A.2 in Appendix A (cf. Nemirovski et al. 2009 and Bottou et al. 2018, theorem 4.7) that when the empirical risk is strongly convex and has Lipschitz continuous gradient, and certain universal second moment bounds are also satisfied on stochastic gradients, the first-order oracle complexity of SGD is bounded by

Γ(SGD)Γ*(SGD)Θ(pϵ).(38)

Note that the expressions in (37) and (38) do not depend on d,η and neglect all other constant parameters.

To compare the first-order oracle complexities in (36), (37), and (38), we will assume for simplicity that the quantities d,p,η,ϵ1 are all functions of the number of training samples n and study the asymptotics as n. The ensuing proposition presents a broad scaling regime where the oracle complexity of LPI-GD beats the oracle complexities (or, more precisely, the oracle complexity bounds) of both GD and SGD shown above.

Proposition 3

(Comparison with GD and SGD). Fix any constants α>0,β>0,τ>max{1,(α+β)1}, and γ>max{1,τ(α+β)/2}. Suppose that ϵ=Θ(nα),p=Θ(nβ),dlog log(n)4 log(e(γ+1)), and η=Θ(d) such that ητ(α+β)d/2 and dl=η1γd. Furthermore, suppose that μ,L1,L2,b,c are Θ(1) with respect to n. Then, we have

Γ(LPI-GD)=O(log(n) exp(2log(n))n1/τ),Γ*(GD)=Θ(n log(n)),Γ*(SGD)=Θ(nα+β).

This implies the following results:

limnΓ(LPI-GD)Γ*(GD)=limnΓ(LPI-GD)Γ*(SGD)=0,
as well as the stronger results:
lim supnlog(Γ(LPI-GD))log(Γ*(GD))1τ,lim supnlog(Γ(LPI-GD))log(Γ*(SGD))1τ(α+β).

Proposition 3 is proved in Section 5.3. Under its assumptions, the LPI-GD algorithm not only beats the oracle complexities of both GD and SGD, but the stronger bounds on the limits of log-oracle-complexity-ratios show that it beats them in the exponent (because τ1,τ1(α+β)1<1). In fact, under the above assumptions, the LPI-GD algorithm beats any Θ(poly(n)) oracle complexity when τ is chosen to be large enough (i.e., when there is sufficient smoothness). Furthermore, as explained in Appendix A, comparisons of oracle complexities of GD and SGD in the literature typically neglect the factors of p in (37) and (38) (see our discussion at the outset of Section 1). In contrast, Proposition 3 performs comparisons with these additional factors because they stem from a reasonable assumption that the parameter lives in an -ball. However, because the LPI-GD algorithm can beat any Θ(poly(n)) oracle complexity, the oracle complexity of LPI-GD would scale better than the oracle complexities of GD or SGD in the regime outlined in Proposition 3, even if we neglected the factors of p in GD and SGD.

We can also calculate interesting regimes where LPI-GD scales better than more refined gradient-based algorithms. In particular, there are a spectrum of algorithms interpolating between GD and SGD that utilize more oracle calls per iteration than SGD, but less than GD. Two notable classes of such algorithms include variance reduced methods (Le Roux et al. 2012, Johnson and Zhang 2013, Shalev-Shwartz and Zhang 2013, Schmidt et al. 2017), which periodically compute full gradients while running SGD to reduce the variance of gradient estimates over iterations, and mini-batch SGD (Dekel et al. 2012, Bottou et al. 2018), where each iteration uses a randomly chosen mini-batch to approximate the full gradient. We next show that the LPI-GD algorithm can also beat mini-batch SGD and stochastic variance reduced gradient (SVRG)—a representative variance reduced SGD method (Johnson and Zhang 2013). Note that the known first-order oracle complexities of SVRG and mini-batch SGD are presented in Propositions A.3 and A.4 in Appendix A, respectively. As before, we assume that the quantities d,p,η,ϵ1 are all functions of the number of training samples n and study the asymptotics as n.

Proposition 4

(Comparison with SVRG and Mini-Batch SGD). As in Proposition 3, fix any constants α>0,β>0,τ>max{1,(α+β)1}, and γ>max{1,τ(α+β)/2}. Suppose that ϵ=Θ(nα),p=Θ(nβ),dlog log(n)4 log(e(γ+1)), and η=Θ(d) such that ητ(α+β)d/2 and dl=η1γd. Furthermore, suppose that μ,L1,L2,b,c are Θ(1) with respect to n. Then, the existing upper bounds in Propositions A.3 and A.4 on the first-order oracle complexities of SVRG and mini-batch SGD satisfy

Γ(SVRG)Γ*(SVRG)=Θ(n log(n)),Γ(Mini-batchSGD)Γ*(Mini-batchSGD)=Θ(nα+β).

This implies the following results:

limnΓ(LPI-GD)Γ*(SVRG)=limnΓ(LPI-GD)Γ*(Mini-batchSGD)=0,
as well as the stronger results:
lim supnlog(Γ(LPI-GD))log(Γ*(SVRG))1τ,lim supnlog(Γ(LPI-GD))log(Γ*(Mini-batchSGD))1τ(α+β).

Proposition 4 is proved in Section 5.4. It shows that LPI-GD can beat various algorithms interpolating between GD and SGD under the assumptions of Proposition 3.

We close this section by making several remarks. Firstly, we reiterate that although Propositions 3 and 4 illustrate regimes where LPI-GD beats GD, SGD, and their variants in oracle complexity, these results require the assumption that d=O(log log(n)), which stems from the curse of dimensionality. This small data dimension assumption can be restrictive for applications where d is large. However, because ERM is used in a diverse array of contexts, LPI-GD may be useful in settings where data dimension is inherently small or can be restricted to live in low-dimensional spaces via dimensionality reduction methods (see Section 1.3). We discuss potential avenues to improve this data dimension requirement in the last remark.

Secondly, we note that many refined gradient-based methods exist that lead to improvements in the dependence of oracle complexity bounds on the condition number—for example, SVRG (Johnson and Zhang 2013) and accelerated GD (Nesterov 1983) (which have similar oracle complexity expressions as GD, but with better dependence on condition number). We do not consider or optimize the trade-off with condition number in our analysis; this is, in many ways, an orthogonal question. As indicated at the outset of this paper, our focus has been on beating GD, SGD, and their variants’ oracle complexities with regard to their dependence on n, ϵ, d, and p.

Thirdly, although methods like SGD, SVRG, and mini-batch SGD match or improve upon GD’s oracle complexity (in terms of its trade-off between n and ϵ, neglecting implicit dependencies on p), they are stochastic optimization algorithms that only generate ϵ-approximate solutions on expectation. In contrast, the LPI-GD algorithm can beat GD and is completely deterministic.

Fourthly, because the LPI-GD algorithm has to learn the oracle at every iteration, the computational (or “arithmetical”) complexity of its iterations may seem to be higher than usual iterative methods. However, the motivation for focusing on the concept of oracle complexity in optimization theory is partly that computing gradients can be extremely hard for certain functions. So, the computational complexity of the oracle usually dominates any other reasonable computation done during iterations. We, therefore, adhere to this canonical paradigm in this work (Nesterov 2004). If the precise running time of the first-order oracle O is known, then the complete computational complexity of the LPI-GD algorithm can be derived from Proposition 2 and Theorem 2 by accounting for the running time of the D × D matrix inversions used to compute the interpolation weights. Note that, as mentioned earlier and shown in Algorithm 1, these weights only need to be precomputed once before running iterations of inexact GD.

Fifthly, we emphasize that Proposition 3 holds in the important regime where α=12 and β>1. As noted in Section 1, α=12 is typically the smallest accuracy of interest because it produces the scaling of the statistical error of the empirical risk. (Table 1 presents the oracle complexities computed in Proposition 3 in this regime.) The latter condition, β>1, ensures that the number of parameters p can be much larger than the number of training samples n, which corresponds to the popular overparametrized regime considered in the context of deep neural networks (cf. Poggio et al. 2020).

Finally, we note that the dependence of the bounds in Theorem 2 and Propositions 3 and 4 on both p and d could potentially be improved. The dependence on p arises because Theorem 1 holds for functions with codomain R. However, the proof of Proposition 2 in Section 5.1 needs to bound the 2-norm error between (30) and the true gradient θF(θ(t1)) for tN. Translating the coordinate-wise guarantee to an 2-norm guarantee in this proof introduces the dependence on p (see (60)). If some version of Theorem 1 could be proved for functions with codomain Rp, then we could possibly improve the (explicit) dependence of Theorem 2 on p.

On the other hand, the dependence on d could be improved by tightening the minimum eigenvalue lower bound in Lemma 1, as explained in Appendix C. Nevertheless, this eigenvalue lower bound must decay at least exponentially (and, correspondingly, the grid size parameter m must grow at least exponentially) in d, as noted after Theorem 1 in Section 3.1. Thus, it is intuitively straightforward to see that the best scaling of d we can hope for, so that the conclusions of Propositions 3 and 4 remain true, is d=O(log(n)). This logarithmic scaling of data dimension d with number of training samples n is significant. Indeed, for several data science applications of interest, the useful information in a dataset of n high-dimensional feature vectors is contained within the pairwise (Euclidean) distances between different feature vectors. In these scenarios, standard low-dimensional approximate isometric embedding theorems—for example, the Johnson-Lindenstrauss lemma (cf. Dasgupta and Gupta 2003 and references therein)—ensure that the feature vectors can be mapped to a d=O(log(n)) dimensional space such that the pairwise distances are preserved. We leave the development of our results along this direction of improving the conditioning analysis of local polynomial regression as future work. Moreover, we note that to beat a d=O(log(n)) scaling, one would need to apply dimensionality reduction techniques, such as principal component analysis (Pearson 1901, Hotelling 1933) or other methods listed in Section 1.3. Theoretically analyzing dimensionality reduction simultaneously with nonparametric regression can be challenging. So, another possible future research direction is to incorporate and analyze (kernelized) principal component regression (Jolliffe 1982, Agarwal et al. 2021) in the context of gradient descent.

4. Proofs for Local Polynomial Interpolation

In this section, we provide proofs for the results presented in Section 3.1. Specifically, we first establish Proposition 1, then portray some properties of the interpolation weights in (23), and, finally, prove Theorem 1.

4.1. Proof of Proposition 1

To prove Proposition 1, we require the following intermediate lemma, which generalizes Tsybakov (2009, lemma 1.4) to our multivariate setting.

Lemma 1

(Auxiliary Löwner Lower Bound). When ld19, the D × D symmetric matrix

B[1,1]dU(u)U(u)Tdu,
satisfies the Löwner lower bound
BΛ(d,l)ID,
where U:RdRD is defined in (16), and the constant Λ(d,l)>0 is defined in (20).

Proof.

We break this proof into several claims for clarity. For convenience in the sequel, we begin by establishing some simple facts about B in Claim 1 and Claim 2. Specifically, Claim 1 shows that B is positive definite.

Claim 1

(Positive Definiteness of B). B is positive definite.

Proof of Claim 1.

Following the proof of Tsybakov (2009, lemma 1.4) mutatis mutandis, note that B is positive semidefinite:

vRD,vTBv=[1,1]d(vTU(u))2du0.

Suppose for the sake of contradiction that there exists a nonzero vector vRD such that

vTBv=[1,1]d(vTU(u))2du=0.

Then, we must have vTU(u)=0 almost everywhere (with respect to the Lebesgue measure) on [1,1]d. However, because the nonzero polynomial uvTU(u) is analytic, its set of roots has Lebesgue measure zero (see, e.g., Krantz and Parks 2002, section 4.1). This is a contradiction. So, we actually have vTBv>0 for all nonzero vectors vRD. Thus, B is positive definite. □

From hereon, our proof of Lemma 1 deviates greatly from that of Tsybakov (2009, lemma 1.4). We next compute the entries of B in Claim 2.

Claim 2

(Entries of B). For every r,sZ+d with |r|,|s|l, we have

Br,s=𝟙{r+sisevenentry-wise}2dr!s!j=1d1rj+sj+1.

Proof of Claim 2.

Because the rows and columns of B are indexed by {sZ+d:|s|l}, for any r,sZ+d with |r|,|s|l, we have

Br,s=[1,1]dur+sr!s!du=2dr!s![1,1]d12dj=1dujrj+sjdu=2dr!s!j=1dE[Xjrj+sj]=𝟙{r+sisevenentry-wise}2dr!s!j=1d1rj+sj+1,
where X1,,Xd[1,1] are independent and identically distributed (i.i.d.) uniform random variables, and the kth moment of X1 is E[X1k]=0 for all odd kZ+ and E[X1k]=(k+1)1 for all even kZ+ (cf. Forbes et al. 2011, chapter 40). □

Thus, Claim 1 and Claim 2 demonstrate that B is a doubly nonnegative matrix with all rational entries. To establish Lemma 1, we need to lower bound the strictly positive minimum eigenvalue of B. To this end, using Claim 2, we first upper bound the trace of B in Claim 3, compute the determinant of B in Claim 4, and then lower bound the determinant of B in Claim 5.

Claim 3

(Upper Bound on Trace of B).

tr(B)(2e)d.

Proof of Claim 3.

To upper bound tr(B), observe that

tr(B)=sZ+d:|s|lBs,s=2dsZ+d:|s|l1(s!)2j=1d12sj+12dk=0l1k!sZ+d:|s|=kk!s!=2dk=0ldkk!(2e)d,
where the second equality uses Claim 2, the fourth equality follows from the multinomial theorem, and the fifth inequality uses the Maclaurin series of the exponential function. □

Note that the exponential scaling of tr(B) in Claim 3 is tight, because tr(B)B(0,,0),(0,,0)=2d. To lower bound det(B), we will first compute it exactly in the next claim.

Claim 4

(Determinant of B).

det(B)=sZ+d:|s|l2d4|s|(s!)2i=1d12si+1(2sisi)2.

Proof of Claim 4.

To prove this, we introduce the Legendre polynomials, {Lk:[1,1]R|kZ+}, which are defined by Rodrigues’ formula (Chihara 2011, chapter V, equation (2.1)):

t[1,1],Lk(t)k+122kk!dkdtk(t21)k,(39)
where Lk is a polynomial with degree k.4 It is well-known that the family of Legendre polynomials satisfies the orthonormality property (Chihara 2011, chapter I, exercise 1.6):
j,kZ+,11Lj(t)Lk(t)dt=𝟙{j=k}.(40)

Moreover, for each kZ+, the leading coefficient (of tk) of Lk(t) is known to be (Chihara 2011, chapter V, equation (2.7)):

k+122k(2kk).(41)

Next, for any rZ+d, define the Legendre product polynomial:

u[1,1]d,Lr(u)j=1dLrj(uj),(42)
which has degree r (with abuse of notation). It is straightforward to verify that these product polynomials also satisfy an orthonormality condition:
r,sZ+d,[1,1]dLr(u)Ls(u)du=j=1d11Lrj(uj)Lsj(uj)duj=𝟙{r=s},(43)
where we utilize (40). Moreover, for each rZ+d, the leading coefficient (of ur) of Lr(u) is known to be
j=1drj+122rj(2rjrj)=12|r|j=1drj+12(2rjrj),(44)
where we utilize (41). Because of (43), we know that for any rZ+d, the set {Ls:sZ+d,sr} is linearly independent and spans the set of all d-variate polynomials with degree at most r. (Note that we use the lexicographical order over indices in Z+d here. Furthermore, it is straightforward to verify that all d-variate monomials with nonzero coefficients that appear in Ls have degree at most s.) As a result, defining a vector(-valued) function L:RdRD as:
u[1,1]d,L(u)[Ls(u):sZ+d,|s|l]T,(45)
which is indexed in lexicographical order, we have the linear relation:
u[1,1]d,U(u)=RL(u),(46)
for some fixed matrix RRD×D (that does not depend on u). It is evident that this matrix R is lower triangular due to the lexicographical order that we have imposed. Furthermore, it is easy to derive the diagonal entries of R. Indeed, for any index sZ+d with |s|l, equating the leading coefficients on both sides of (46) produces
1s!=Rs,s12|s|j=1d2sj+12(2sjsj),
using (44). Rearranging this, we get:
sZ+dwith|s|l,Rs,s=2|s|2d/2s!j=1d12sj+1(2sjsj)1.(47)

Now, notice that

B=[1,1]dU(u)U(u)Tdu=R([1,1]dL(u)L(u)Tdu)RT=RIDRT=RRT,(48)
where the second equality follows from (46), and the third equality follows from (43). We recognize (48) as the Cholesky decomposition of B (Horn and Johnson 2013, corollary 7.2.9). Thus, using (47) and (48), we can write
det(B)=sZ+d:|s|lRs,s2=sZ+d:|s|l4|s|i=1d(2sisi)21·2d(s!)2j=1d12sj+1=Bs,s=sZ+d:|s|l2d4|s|(s!)2i=1d12si+1(2sisi)2,
which is the desired result. □

We next establish a lower bound on the determinant of B using Claim 4.

Claim 5

(Lower Bound on Determinant of B).

det(B)(πd4e)dD1(l+d)3lD.

Proof of Claim 5.

Observe from Claim 4 that we can write

det(B)=sZ+d:|s|l2d4|s|(s!)2i=1d12si+1(2sisi)2sZ+d:|s|l(π/2)d4|s|(s!)2j=1d4sj+12sj+11(π2)dD(sZ+d:|s|l12|s|s!)2=(π2)dD14i=0li(i+d1d1)(j=0lsZ+d:|s|=j1s1!sd!)2(π2e2)dD(e24)i=0li(i+d1d1)j=0lsZ+d:|s|=j1(s1+1)2s1+1(sd+1)2sd+1(π4e)dDj=0lsZ+d:|s|=j1(s1+1)2s1+1(sd+1)2sd+1(π4e)dDj=0l(dd(j+d)d(j+1)2j)(j+d1d1)(πd4e)dD1(l+d)j=0l(2j+d)(j+d1d1)(πd4e)dD1(l+d)3lD,
where the second inequality follows from the following bound (on Catalan numbers; see Dutton and Brigham 1986, theorem):
jZ+,(2jj)2<42j+1π(4j+1),
the fourth equality follows from adding the exponents |s| of two over all sZ+d with |s|l and then squaring, the fifth inequality follows from the Stirling’s formula bound (see, e.g., Feller 1968, chapter II, section 9, equation (9.15)):5
jZ+,j!(j+1)j+12ej+1,
the sixth inequality follows from the straightforward bound:
i=0li(i+d1d1)l(l+d1d1)=dll+d(l+dd)d2(l+dd)=dD2,
which uses the fact that ld, the seventh inequality uses the following basic facts and calculations:
  1. The hypercube maximizes the volume over all d-orthotopes with an isoperimetric constraint:6

    maxs1++sd=js1,,sd0:(s1+1)(sd+1)=(jd+1)d,
    where jZ+ and the maximum is achieved by s1==sd=jd;

  2. For any jZ+ and any sZ+d with |s|=j, the vector (j,0,,0)Z+d majorizes the vector s, and because the map 0t2t log(t+1) is convex, Karamata’s majorization inequality yields (cf. Marshall et al. 2011, chapter 1, section A)

    i=1d2si log(si+1)2j log(j+1),
    which implies that
    i=1d(si+1)2si(j+1)2j,
    using the monotonicity of the exponential function;

    and the ninth inequality follows from the simple bound:

    j=0l(2j+d)(j+d1d1)(2l+d)j=0l(j+d1d1)=(2l+d)D3lD,
    which uses the fact that ld. □

Finally, we can complete the proof of Lemma 1 using Claim 1, Claim 3, and Claim 5. Recall the following well-known result (cf. Merikoski and Virtanen 1997, lemma 1):

λmin(B)det(B)(D1tr(B))D1,(49)
which utilizes the positive definiteness of B established earlier in Claim 1.7 Combining Claim 3, Claim 5, and (49), we obtain
λmin(B)(πd4e)dD(D1)D1(l+d)3lD(2e)d(D1)(πd8e2)dD(D1)D1(l+d)3lD(πd8e2)d(l+dd)d((l+dd)d1)(l+dd)d11(l+d)3l(e(l+d)d)d=Λ(d,l),(50)
where the third inequality holds because d19, the map 1ttt is strictly increasing, and we utilize the ensuing standard bounds on binomial coefficients:
(l+dd)dD=(l+dd)(e(l+d)d)d,
and the final equality follows from (20). Because Bλmin(B)ID, applying the lower bound in (50) completes the proof of Lemma 1. □

We make three pertinent remarks at this point. Firstly, we note that when dlγd for some constant γ>1, one can verify that

Λ(d,l)(πd8e2)d2d(2d1)2d1((γ+1)d)3γd(e(γ+1))d.(51)

Hence, Λ(d,l)1 scales double exponentially in d:

1Λ(d,l)=O(exp(eΘ(poly(d)))).(52)

This observation will be used in the proof of Proposition 3 in Section 5.3. Secondly, we also conjecture that the scaling in (52) can be significantly improved, as shown in Appendix C. Thirdly, we remark that the condition d19, which appears in many of the results in Section 3, originates from the proof of Lemma 1 (see (50)). This condition is simply stated in our results for precision; it becomes immaterial when d scales with n in, for example, Proposition 3.

We are now in a position to establish Proposition 1 using Lemma 1.

Proof of Proposition 1.

We take inspiration from the proof technique of Tsybakov (2009, lemma 1.5), but the details of our analysis are more involved and quite different. Observe that for all mN, for all x[h,1h]d, and for all vRD with v2=1, we have

vTB(x)v=vT(1(mh)dyGmU(yxh)U(yxh)Tj=1dK(yjxjh))v=vT(1(mh)dx+hzGmz[1,1]d:U(z)U(z)Tj=1dK(zj))vbdvTB˜(x)v,
where the first equality follows from (19), the second equality follows from the substitution y=x+hz and the fact that the kernel K has support [1,1], the third inequality holds because the kernel K is lower bounded by b > 0 on [1,1], and we define B˜(x) as the D × D symmetric positive semidefinite matrix
B˜(x)x+hzGmz[1,1]d:U(z)U(z)T1(mh)d.

Then, applying the Courant-Fischer-Weyl min-max theorem (Horn and Johnson 2013, theorem 4.2.6), we obtain

λmin(B(x))bdλmin(B˜(x)).(53)

We proceed to lower bounding λmin(B˜(x)). Fix any x[h,1h]d, and any r,sZ+d with |r|,|s|l, and consider the Riemann sum

B˜(x)r,s=1r!s!x+hzGmz[1,1]d:zr+s(mh)d.

Because x[h,1h]d, it is easy to verify that the points {z[1,1]d:x+hzGm} partition the hypercube [1,1]d into a grid of d-cells {Iz[1,1]d:z[1,1]d,x+hzGm} with side lengths (mh)1 and volume (mh)d each, where Iz is the d-cell with tag z. Moreover, because the polynomial function γ:[1,1]dR,γ(u)=ur+s is Riemann integrable, if h=ω(m1) with all other problem parameters fixed, then the above Riemann sum converges to an entry of the matrix B defined in Lemma 1 (cf. Apostol 1974, section 14.3):

limm1r!s!x+hzGmz[1,1]d:zr+s(mh)d=[1,1]dur+sr!s!du=Br,s.

Therefore, we have shown that limmB˜(x)r,s=Br,s when h=ω(m1). We will require an explicit upper bound on the rate of this convergence in the sequel. To this end, let us first compute the Lipschitz constant of γ with respect to the -norm. For any i[d] and any fixed (u1,,ui1,ui+1,,ud)[1,1]d1, notice that for every ui,u˜i[1,1],

|γ(u)γ(u1,,ui1,u˜i,ui+1,,ud)|=|u1r1+s1ui1ri1+si1ui+1ri+1+si+1udrd+sd||uiri+siu˜iri+si|(ri+si)|uiu˜i|.

Hence, via repeated application of the triangle inequality, we have

u,u˜[1,1]d,|γ(u)γ(u˜)|=|ur+su˜r+s|i=1d(ri+si)|uiu˜i|(|r|+|s|)uu˜.(54)

Now, observe that for all r,sZ+d with |r|,|s|l,

|B˜(x)r,sBr,s|=1r!s!|x+hzGmz[1,1]d:zr+s(mh)dIzur+sdu|1r!s!x+hzGmz[1,1]d:|zr+s(mh)dIzur+sdu|1r!s!x+hzGmz[1,1]d:1(mh)d(supuIz ur+sinfu˜Iz u˜r+s)|r|+|s|mhr!s!x+hzGmz[1,1]d:1(mh)d|r|+|s|mhr!s!(2+1mh)d3d(|r|+|s|)mhr!s!,
where the second inequality follows from the triangle inequality, the fourth inequality follows from (54), the fifth inequality holds because |{z[1,1]d:x+hzGm}|(2mh+1)d (because the interval [1,1] can have at most 2mh+1 points at distance (mh)1 from each other), and the sixth inequality holds because we assume that mN and h(0,12) are such that mh4l(3e)d/Λ(d,l)1. This implies that for all such m and h,
B˜(x)BFr,sZ+d:|r|,|s|l9d(|r|+|s|)2m2h2(r!s!)22l3dmhrZ+d:|r|l1(r!)22l3dmhj=0l1j!rZ+d:|r|=jj!r!=2l3dmhj=0ldjj!2l(3e)dmh,(55)
where the fourth equality follows from the multinomial theorem, and the fifth inequality uses the Maclaurin series of the exponential function. Next, recall the following consequence of the Wielandt-Hoffman-Mirsky inequality (cf. Horn and Johnson 2013, corollary 6.3.8):
|λmin(B˜(x))λmin(B)|B˜(x)BF.(56)

Then, combining (55) and (56) produces the lower bound

λmin(B˜(x))λmin(B)2l(3e)dmh,
and employing Lemma 1 to this bound yields
λmin(B˜(x))Λ(d,l)2l(3e)dmh,(57)
where Λ(d,l) is defined in (20).

Finally, because m and h satisfy mh4l(3e)d/Λ(d,l), or, equivalently,

2l(3e)dmhΛ(d,l)2,
using (53) and (57), we get
λmin(B(x))bdλmin(B˜(x))bdΛ(d,l)2,
for all x[h,1h]d. Because B(x)λmin(B(x))ID, the above lower bound on λmin(B(x)) completes the proof. □

4.2. Properties of Local Polynomial Interpolation Weights

To prove Theorem 1, we will require two useful lemmata regarding the interpolation weights in (23). The first lemma generalizes Tsybakov (2009, proposition 1.12) to the d-variate setting and demonstrates the intuitively pleasing property that polynomials are reproduced exactly by the local polynomial interpolator given in (24).

Lemma 2

(Polynomial Reproduction (Tsybakov 2009)). Consider any d-variate polynomial g:[0,1]dR with total degree at most l (i.e., for every monomial of g with nonzero coefficient, the sum of the exponents of the d variables is at most l). Then, for every mN and every x[h,1h]d such that B(x) is positive definite, we have

ϕ^(x)=yGmg(y)wy*(x)=g(x),
where the interpolation weights are given in (23).

Proof.

We follow the proof of Tsybakov (2009, proposition 1.12) mutatis mutandis. Fix any mN and x[h,1h]d such that B(x) is positive definite, and define the vector q(x)RD as

q(x)[h|s|sg(x):sZ+d,|s|l]​T.

Then, for every yGm, we have

g(y)=sZ+d:|s|lsg(x)s!(yx)s=q(x)TU(yxh),
where the first equality uses Taylor’s theorem (cf. Apostol 1974, theorem 12.14) and the fact that g is a polynomial with total degree at most l, and U is defined in (16). This implies that the solution to the weighted regression problem in (17) satisfies
Φ^(x)=arg minϕRDyGm(q(x)ϕ)TU(yxh)U(yxh)T(q(x)ϕ)j=1dK(yjxjh)=arg minϕRD(q(x)ϕ)TB(x)(q(x)ϕ)=q(x),
where the second equality follows from (19), and the third equality holds because B(x) is positive definite. Therefore, by considering the first coordinate, we get
ϕ^(x)=yGmg(y)wy*(x)=g(x),
where the first equality uses (24). This completes the proof. □

The second lemma generalizes Tsybakov (2009, lemma 1.3) to the d-variate setting and portrays an 1-norm bound on the sequence of interpolation weights in (23).

Lemma 3

(1-Norm Bound). Suppose that ld19. Then, for all mN and h(0,12) such that mh4l(3e)d/Λ(d,l) and all x[h,1h]d, we have

yGm|wy*(x)|2Λ(d,l)(3ecb)d,
where cb>0 are the bounds on our kernel, and Λ(d,l) is defined in (20).

Proof.

Once again, we follow the proof of Tsybakov (2009, lemma 1.3) mutatis mutandis. First, notice that B(x) satisfies the Löwner lower bound in Proposition 1 because the conditions of the proposition are satisfied. Then, observe that for every x[h,1h]d, we have

yGm|wy*(x)|yGmwy(x)2=1(mh)dyGmB(x)1U(yxh)2j=1dK(yjxjh)cd(mh)dyGmB(x)1U(yxh)2𝟙{j[d],|yjxj|h}2cd(mhb)dΛ(d,l)yGmU(yxh)2𝟙{j[d],|yjxj|h}2Λ(d,l)(ecmhb)dyGm𝟙{j[d],|yjxj|h}2Λ(d,l)(ecb)d(2mh+1mh)d2Λ(d,l)(3ecb)d,
where the first inequality follows from (23) and the Cauchy-Schwarz inequality, the second equality follows from (22), the third inequality holds because the kernel K is upper bounded by c > 0 on the support [1,1], the fourth inequality follows from Proposition 1, the fifth inequality follows from the bound
U(yxh)22=sZ+d:|s|l((yx)sh|s|s!)2sZ+d:|s|l1(s!)2i=0l1i!sZ+d:|s|=ii!s!=i=0ldii!ed,
which uses (16), the fact that |yjxj|h for all j[d], the multinomial theorem, and the Maclaurin series of the exponential function, the sixth inequality holds because the interval [h,h] can have at most 2mh+1 points at distance m1 from each other, and the seventh inequality follows from the assumption that mh4l(3e)d/Λ(d,l)1. This completes the proof. □

4.3. Proof of Theorem 1

Finally, we are in a position to prove the supremum norm interpolation guarantee in Theorem 1.

Proof of Theorem 1.

Suppose that mN and h(0,12) are such that mh4l(3e)d/Λ(d,l). Then, observe that for every x[h,1h]d, we have

ϕ^(x)g(x)=yGmg(y)wy*(x)g(x)=yGmwy*(x)(g(y)g(x))=yGmwy*(x)sZ+d:1|s|lsg(x)s!(yx)s+wy*(x)sZ+d:|s|=lsg(x+τ(yx))sg(x)s!(yx)s=yGmwy*(x)sZ+d:|s|=lsg(x+τ(yx))sg(x)s!(yx)s,
where the first equality follows from (24), the second equality holds because
yGmwy*(x)=1,
which follows from Lemma 2 applied to the constant unit polynomial, the third equality follows from Taylor’s theorem with Lagrange remainder term (Apostol 1974, theorem 12.14):
g(y)g(x)=sZ+d:1|s|lsg(x)s!(yx)s+sZ+d:|s|=lsg(x+τ(yx))sg(x)s!(yx)s,
which holds for some τ(0,1), and the fourth equality holds because
yGm(yx)swy*(x)=0,
for all sZ+d with 1|s|l, which follows from Lemma 2 applied to the polynomials u(ux)s. This implies that for every x[h,1h]d,
|ϕ^(x)g(x)|yGm|wy*(x)|sZ+d:|s|=l|sg(x+τ(yx))sg(x)|s!|(yx)s|L2yGm|wy*(x)|yx1ηlsZ+d:|s|=l1s!|(yx)s|=L2l!yGm|wy*(x)|yx1η=L2l!yGm|wy*(x)|yx1η𝟙{j[d],|yjxj|h}2L2(dh)ηl!Λ(d,l)(3ecb)d=2L2l!Λ(d,l)(3ecb)d(4dl(3e)dmΛ(d,l))η,
where the first inequality follows from the triangle inequality, the second inequality follows from the Hölder class assumption (see (11)) and the fact that τ(0,1), the third equality follows from the multinomial theorem, the fourth equality follows from (22) and (23) because the kernel K has support [1,1], the fifth inequality follows from Lemma 3, and the sixth equality holds because we set
h=4l(3e)dmΛ(d,l),(58)
which minimizes the upper bound in the fifth inequality and achieves equality in the condition imposed on m and h at the outset of the proof. Lastly, to ensure that supx[h,1h]d|ϕ^(x)g(x)|δ, it suffices to impose the condition
2L2l!Λ(d,l)(3ecb)d(4dl(3e)dmΛ(d,l))ηδ,
which is equivalent to
m4dl(3e)dΛ(d,l)(2L2l!Λ(d,l))1/η(3ecb)d/η(1δ)1/η,
which, in turn, is implied by the condition
m12e(2L2+1)cb(dl(3e)d(l!)1/ηΛ(d,l)2)(1δ)1/η,(59)
where we use the facts that η>ld19 (by assumption) and Λ(d,l)(η+1)/ηΛ(d,l)2 (because Λ(d,l)1), and we slacken the Λ(d,l) term because this does not change its double exponential nature. Moreover, we may further simplify (59) by noting that
(l!)1/η(2π)1/(2η)l(l/η)+(1/(2η))el/ηlel1l+1l2e,
where we use the Stirling’s formula bound (cf. Feller 1968, chapter II, section 9, equation (9.15)) and the fact that l1/(l+1)2 for l19. Applying this bound, we obtain the sufficient condition for (59) presented in the theorem statement:
m110(2L2+1)cb(d(3e)dΛ(d,l)2)(1δ)1/η,
where we use the fact that 24ee110. Finally, plugging this bound into (58) yields the upper bound on h in the theorem statement. This completes the proof. □

We note that this argument is inspired by the proof of Tsybakov (2009, proposition 1.13), but the details are quite different.

5. Proofs of Convergence Analysis

In this section, we prove Propositions 24 and Theorem 2 from Section 3.2.2.

5.1. Proof of Proposition 2

To establish Proposition 2, we require the following lemma from the literature, Friedlander and Schmidt (2012, lemma 2.1) (also see So and Zhou 2017, theorem 2, equation (19)), that upper bounds the difference between the ERM objective function (3) evaluated at θ(T), which is the output of our proposed algorithm, and the true infimum in (4). Although Friedlander and Schmidt (2012, lemma 2.1) holds for general inexact gradient descent methods, we unwind the recursion in Friedlander and Schmidt (2012, lemma 2.1) and adapt it to our setting below for convenience.

Lemma 4

(Inexact Gradient Descent Bound (Friedlander and Schmidt 2012)). Suppose the ERM objective function F:RpR is continuously differentiable and μ-strongly convex, and its gradient θF:RpRp is L1-Lipschitz continuous. Then, for any TN, we have

0F(θ(T))F*(11σ)T(F(θ(0))F*)+12L1t=1T(11σ)TtF^(t)θF(θ(t1))22,
where θ(t) for tZ+ are the inexact gradient descent updates given by (31) with an arbitrary initialization θ(0),F^(t) for tN are our local polynomial interpolation-based approximations of the true gradients θF(θ(t1)) shown in (30), and σ=L1/μ1 is the condition number defined in Section 2.2.

We remark that Friedlander and Schmidt (2012, lemma 2.1) additionally assumes that the infimum F* in (4) can be achieved. However, this condition need not be imposed in Lemma 4, because if F is strongly convex on Rm, its infimum F* is finite and achieved by some unique global minimizer θ*Rp. We next prove Proposition 2 using Lemma 4 and Theorem 1.

Proof of Proposition 2.

We first verify that the conditions of Lemma 4 hold. Recall the ERM objective function F:RpR given in (3):

θRp,F(θ)=1ni=1nf(x(i);θ),
where {x(i)[h,1h]d:i[n]} is some fixed training data, and f:[0,1]d×RpR is a given loss function satisfying the assumptions in Section 2.2. Specifically, because f(x(i);·):RpR is μ-strongly convex for all i[n], F is also μ-strongly convex due to Nesterov (2004, lemma 2.1.4). Likewise, because θf(x(i);·):RpRp is L1-Lipschitz continuous for all i[n], F is continuously differentiable and θF:RpRp is also L1-Lipschitz continuous. Indeed, observe that for all θ1,θ2Rp,
θF(θ1)θF(θ2)21ni=1nθf(x(i);θ1)θf(x(i);θ2)2L1ni=1nθ1θ22=L1θ1θ22,
where we use the triangle inequality.

Next, we apply Lemma 4 and get

F(θ(T))F*(11σ)T(F(θ(0))F*)+12L1t=1T(11σ)TtF^(t)θF(θ(t1))22,
where θ(t) for tZ+ are our updates in (31), and F^(t) for tN are our approximations of the true gradients in (30). To upper bound the squared 2-norms on the right-hand side of this inequality, notice that for any tN, we have
F^(t)θF(θ(t1))21nj=1ni=1p(ϕ^i(t)(x(j))fθi(x(j);θ(t1)))2i=1psupx[h,1h]d|ϕ^i(t)(x)fθi(x;θ(t1))|2pδt,(60)
where the first inequality follows from (30) and the triangle inequality, ϕ^i(t)(x) denotes the output of our local polynomial interpolator for fθi(·;θ(t1)) when evaluated at x (cf. (29)), and the third inequality follows from Theorem 1 and our choices of grid sizes (26) and bandwidths (27) in Section 3.2.1 (where (25) ensures that hth). Hence, we obtain
F(θ(T))F*(11σ)T(F(θ(0))F*)+p2L1t=1T(11σ)Ttδt2.(61)

We now select appropriate values for the constants {δt(0,1):tN}. For every tN, choose

δt=(11σ)T2,
as indicated in (33). This produces
F(θ(T))F*(11σ)T(F(θ(0))F*+p2L1t=1T(11σ)Tt)(11σ)T(F(θ(0))F*+p2L1t=0(11σ)t)=(11σ)T(F(θ(0))F*+pσ2L1),
where the last equality computes the value of the previous geometric series.

Finally, to ensure that θ(T) is an ϵ-approximate solution of (4), viz.

F(θ(T))F*ϵ,
as defined in (13), it suffices to impose the condition
(11σ)T(F(θ(0))F*+pσ2L1)ϵ.

By taking logarithms of both sides, and rearranging and simplifying the resulting inequality, we obtain the equivalent inequality

T(log(σσ1))1 log(F(θ(0))F*+p2μϵ),
where we also use the fact that σ=L1/μ. Therefore, after running the LPI-GD algorithm for TN iterations with T given by (32), the updated parameter vector θ(T) forms an ϵ-approximate solution to (4). We also note that for every tN,
δt=(11σ)T2=Θ(ϵp),
by substituting (32) into our choice of δt above. This yields (33). Hence, the proof is complete. □

5.2. Proof of Theorem 2

We next establish the oracle complexity bound in Theorem 2 using Proposition 2.

Proof of Theorem 2.

Because we choose the maximum allowable approximation errors {δt(0,1):t[T]} to be constant with respect to t as in (33), a single common uniform grid Gm is used in all iterations of the LPI-GD algorithm, where m = mt is given by (26) (and is also constant with respect to t). The size of this grid is upper bounded by

|Gm|=md(110(2L2+1)cb(d(3e)dΛ(d,l)2)(σσ1)T/(2η)+1)d(110(2L2+1)cb(d(3e)dΛ(d,l)2)(σσ1)1/(2η)(F(θ(0))F*+p2μϵ)1/(2η)+1)d(220(2L2+1)cb)d(dd(3e)d2Λ(d,l)2d)(1+σ2(L1μ))d/(2η)(p+2μ(F(θ(0))F*)ϵ)d/(2η)(1+σ2(L1μ))(220(2L2+1)cb)d(dd(3e)d2Λ(d,l)2d)(p+2μ(F(θ(0))F*)ϵ)d/(2η),
where the second inequality follows from (26) and (33) in Proposition 2, the third inequality follows from (32) in Proposition 2, the fourth inequality follows from straightforward manipulations and the fact that σ=L1/μ, and the fifth inequality holds because we have assumed that η>d. The LPI-GD algorithm makes |Gm| oracle calls in each of its T iterations and produces an ϵ-approximate solution θ(T)Rp (in the sense of (13)), as argued in Proposition 2. Hence, the oracle complexity of this algorithm is upper bounded by
Γ(LPI-GD)T|Gm|(σ+2(L1μ)(L1μ)log(σσ1))log(p+Δ2μϵ)(220(2L2+1)cb)d(dd(3e)d2Λ(d,l)2d)(p+Δϵ)d/(2η)=Cμ,L1,L2,b,c(d,l)(p+Δϵ)d/(2η)log(p+Δ2μϵ),
where we let Δ=2μ(F(θ(0))F*) for convenience, the second inequality uses (32) and the fact that ϵ(L1μ)p2L1μ (which is equivalent to σσ1p2μϵ), and the third equality uses (34). This completes the proof. □

We remark that the proofs of Proposition 2 and Theorem 2 can be combined to determine the optimal t-dependent approximation error parameters {δt(0,1):tN} and number of iterations TN such that θ(T) is an ϵ-approximate solution in the sense of (13) with as few oracle calls as possible. However, this does not qualitatively improve the dominant term in the scaling law of the derived oracle complexity bound in Theorem 2.

5.3. Proof of Proposition 3

We next derive Proposition 3 from Theorem 2.

Proof of Proposition 3.

We first establish the scaling of Cμ,L1,L2,b,c(d,l) in (34). Because d=O(log log(n)), observe that

(σ+2(L1μ)(L1μ) log(σσ1))(220(2L2+1)cb)d=O(polylog(n)),(62)
and because ld, observe that
dd(3e)d2Λ(d,l)2d=O((l+d)6dl(e(l+d)d)d),
as d grows, where we use (20). To simplify the second scaling expression above, notice that for all sufficiently large dN, we have
log log(dd(3e)d2Λ(d,l)2d)log log(l+d)+log(dl)+d log(e(l+d)d)+Θ(1)=Θ(log(d))+d log(e(l+d)d)Θ(log(d))+d log(e(γ+1))2 log(e(γ+1))d,
where the second equality follows from the fact that η=Θ(d), and the third inequality holds because lγd. This implies that for all sufficiently large dN,
dd(3e)d2Λ(d,l)2dexp(e2 log(e(γ+1))d).(63)

Hence, because dlog log(n)4 log(e(γ+1)), we obtain

Cμ,L1,L2,b,c(d,l)O(polylog(n))·exp(e12log log(n))=O(polylog(n))·exp(log(n))exp(2log(n)),
for all sufficiently large nN, using (34), (62), and (63). Next, applying (36), which follows from Theorem 2, we have that for all sufficiently large nN,
Γ(LPI-GD)c1 exp(2log(n))(pϵ)d/(2η)log(pϵ)c1 exp(2log(n))(c2nα+β)d/(2η)log(c2nα+β)c3log(n) exp(2log(n))n1/τ,(64)
where c1,c2,c3>0 are some positive constants, the second inequality follows from the assumptions that ϵ=Θ(nα) and p=Θ(nβ), and the third inequality holds because ητ(α+β)d/2. This proves the scaling of Γ(LPI-GD) with respect to n.

Finally, from the relations (37) and (38) and the assumptions that ϵ=Θ(nα) and p=Θ(nβ), we get Γ*(GD)=Θ(n log(n)) and Γ*(SGD)=Θ(nα+β). Therefore, for some constants c4,c5>0, we obtain

limnΓ(LPI-GD)Γ*(GD)limnc3exp(2log(n))n1/τc4n=0,limnΓ(LPI-GD)Γ*(SGD)limnc3 log(n)exp(2log(n))n1/τc5nα+β=0,
which use the fact that τ>max{1,(α+β)1}. Furthermore, we also obtain the stronger results
lim supnlog(Γ(LPI-GD))log(Γ*(GD))limnlog(c3)+log log(n)+2log(n)+τ1log(n)log(c4)+log(n)+log log(n)=1τ<1,lim supnlog(Γ(LPI-GD))log(Γ*(SGD))limnlog(c3)+log log(n)+2log(n)+τ1log(n)log(c5)+(α+β)log(n)=1τ(α+β)<1,
which imply the previous limits.8 This completes the proof. □

5.4. Proof of Proposition 4

Lastly, we derive Proposition 4 from Theorem 2.

Proof of Proposition 4.

This proof closely follows the argument for Proposition 3. Under the assumptions of the proposition statement, for some positive constant c3>0 and all sufficiently large nN, we have from (64) that

Γ(LPI-GD)c3 log(n) exp(2log(n))n1/τ.

Next, from the existing first-order oracle complexity results in Propositions A.3 and A.4 in Appendix A, we define the upper bounds:

Γ(SVRG)Γ*(SVRG)Θ(n log(pϵ)),Γ(Mini-batchSGD)Γ*(Mini-batchSGD)Θ(pϵ).

Using the assumptions ϵ=Θ(nα) and p=Θ(nβ), we obtain that Γ*(SVRG)=Θ(n log(n)) and Γ*(Mini-batchSGD)=Θ(nα+β). Therefore, as in the proof of Proposition 3, we have

limnΓ(LPI-GD)Γ*(SVRG)=limnΓ(LPI-GD)Γ*(Mini-batchSGD)=0.

Furthermore, we also obtain the stronger results

lim supnlog(Γ(LPI-GD))log(Γ*(SVRG))1τ<1,lim supnlog(Γ(LPI-GD))log(Γ*(Mini-batchSGD))1τ(α+β)<1.

This completes the proof. □

6. Conclusion

We conclude by summarizing our main contributions and reiterating some directions of future research. In order to exploit the smoothness of loss functions in data for optimization purposes, we proposed the LPI-GD algorithm (see Algorithm 1) to compute approximate solutions of the ERM problem in (3) and (4), where the loss function is strongly convex and smooth in both the parameter and the data. We then derived the iteration and oracle complexities of this algorithm and illustrated that its oracle complexity beats the oracle complexities of GD, SGD, and their variants for a broad range of dependencies between the parameter dimension p, the approximation accuracy ϵ, and the number of training samples n, when the data dimension d is sufficiently small. Finally, in the course of our convergence analysis, we also provided a careful and detailed analysis of multivariate local polynomial interpolation with supremum norm guarantees, which may be of independent interest in nonparametric statistics.

We close our discussion by stating four avenues for future work. First, as stated at the end of Section 3 and in Appendix C, it would be very valuable to improve the dependence of Theorem 2 on the data dimension d by, for example, sharpening the minimum eigenvalue lower bound in the key technical estimate in Lemma 1 (see Conjecture C.1 in Appendix C). Second, the explicit dependence of Theorem 2 on the parameter dimension p could be improved by establishing new interpolation guarantees for approximating smooth Rp-valued functions using local polynomial regression or other methods. Third, as noted in Section 1, although our work focuses on the strongly convex loss function setting, similar analyses could be carried out for other conventional scenarios—for example, when f(x;·):RpR is nonconvex and ϵ-approximate solutions to (4) are defined via approximate first-order stationary points. Last, the LPI-GD method presented in Algorithm 1 is really a theoretical method because it depends on parameters that are often unknown in practice, such as Lipschitz and strong convexity constants. It would therefore be interesting to create a practical version of this method and empirically investigate any average-case gains it provides in running time.

Acknowledgments

Note: The author ordering is alphabetical.

Appendix A. Oracle Complexity of Gradient Descent and its Variants

In this appendix, we present the well-known first-order oracle complexities of GD, SGD, and variants, such as variance reduced SGD and mini-batch SGD, from the literature. Recall the setup in Section 2.2. To illustrate the scaling of these complexities, we assume that n,d,p,ϵ1 and all other problem parameters are constant. As shown in the proof of Proposition 2 in Section 5.1, the strong convexity and Lipschitz continuous gradient assumptions on the loss function in Section 2.2 imply that the ERM objective function F:RpR in (3) is μ-strongly convex and has L1-Lipschitz continuous gradient. Let θ*Rp be the unique global minimizer of the empirical risk F:RpR so that F*=F(θ*) and θF(θ*)=0 (which is the zero vector). Furthermore, assume that the parameter θ belongs to an -ball with constant radius—for example, [0,1]p—and all iterates of LPI-GD, GD, SGD, and their variants also live in this -ball. (This is very often the case in applications because unbounded values cannot be easily represented in computers.) The 2-diameter of such an -ball scales as O(p), which means that the 2-distance between any two parameter vectors is O(p). Under these assumptions, the next proposition presents the first-order oracle complexity of GD with appropriately chosen constant step size (Nesterov 2004, theorem 2.1.15).

Proposition A.1

(Oracle Complexity of GD (Nesterov 2004)). Under the aforementioned assumptions, the first-order oracle complexity of GD is bounded by

Γ(GD)n2(log(σ+1σ1))1log(L1θ(0)θ*222ϵ)=O(n log(pϵ)),
where θ(0)Rp is the initial parameter vector of GD and θ(0)θ*22=O(p).

Proposition A.1 follows from the facts that: (1) Each iteration of GD makes n first-order oracle queries, and (2) it takes O(log(pϵ1)) iterations for GD to produce an ϵ-approximate solution in the sense of (13) (Nesterov 2004, theorem 2.1.15). When certain universal second moment bounds are additionally satisfied on stochastic gradients, the next proposition conveys the first-order oracle complexity of SGD with linearly diminishing step sizes (cf. Nemirovski et al. 2009 and Bottou et al. 2018, theorem 4.7).

Proposition A.2

(Oracle Complexity of SGD (Bottou et al. 2018)). Under the aforementioned assumptions, the first-order oracle complexity of SGD is bounded by

Γ(SGD)2σ max{supx,ϑθf(x;ϑ)22,μ(F(θ(0))F*)}μϵ=O(pϵ),
where θ(0)Rp is the initial parameter vector of SGD, supx,ϑθf(x;ϑ)22=O(p), and F(θ(0))F*=O(p).

Note that because the iterates of SGD are random variables, the notion of an ϵ-approximate solution in (13) is defined in expectation for SGD. So, Proposition A.2 presents the number of first-order oracle queries required to obtain an ϵ-approximate solution in expectation. Furthermore, we remark that the scaling of supx,ϑθf(x;ϑ)22 is obtained as follows:

supx,ϑθf(x;ϑ)22=supx,ϑθf(x;ϑ)θf(x;θx)22L12supx,ϑϑθx22=O(p),(A.1)
where θxRp denotes the unique global minimizer of the strongly convex function f(x;·):RpR for any x[0,1]d,θf(x;θx)=0 (i.e., the zero vector) at the minimum point, the inequality follows from the smoothness in parameter assumption in Section 2.2, and supx,ϑϑθx22=O(p) because the parameter belongs to an -ball with constant radius. Similarly, the scaling of F(θ(0))F* is obtained as follows (also see Jadbabaie et al. 2023, footnote 11):
F(θ(0))F*=F(θ(0))F(θ*)=θF(τθ(0)+(1τ)θ*)T(θ(0)θ*)=(θF(τθ(0)+(1τ)θ*)θF(θ*))​T(θ(0)θ*)θF(τθ(0)+(1τ)θ*)θF(θ*)2θ(0)θ*2L1θ(0)θ*22=O(p),(A.2)
where the second equality holds due to the mean value theorem for some τ(0,1) (cf. Apostol 1974, theorem 12.14), the third equality holds because θF(θ*)=0, the penultimate inequality uses the Cauchy-Schwarz inequality, and the last inequality follows from the L1-Lipschitz continuity of θF.

Several variants of the basic GD and SGD procedures have been developed and studied in the literature. One such variant is the family of variance reduced methods, which periodically compute full gradients while running SGD to reduce the variance of gradient estimates over iterations. There are several known variance reduced iterative descent methods, such as the stochastic average gradient (SAG) (Le Roux et al. 2012, Schmidt et al. 2017), the stochastic dual coordinate ascent (SDCA) (Shalev-Shwartz and Zhang 2013), and the stochastic variance reduced gradient (Johnson and Zhang 2013). Although the specific details of these methods vary, it is known that they exhibit the same convergence rate for strongly convex loss functions with Lipschitz continuous gradients. So, the next proposition states the first-order oracle complexity of SVRG as one representative of the family of variance reduced methods. Specifically, we consider the regime where SVRG has been shown to be effective, with learning rate proportional to 1/L1—for example, 0.1/L1—and number of (variance reduced) SGD iterations per epoch proportional to the condition number σ—for example, 50σ (Johnson and Zhang 2013, theorem 1).

Proposition A.3

(Oracle Complexity of SVRG (Johnson and Zhang 2013)). Under the aforementioned assumptions, the first-order oracle complexity of SVRG is bounded by

Γ(SVRG)(n+100σ)log2(F(θ(0))F*ϵ)=O(n log(pϵ)),
where θ(0)Rp is the initial parameter vector of SVRG and F(θ(0))F*=O(p).

Proposition A.3 follows from combining the convergence analysis in Johnson and Zhang (2013, theorem 1) with the number of oracle calls in each epoch of the algorithm. Note that as with SGD, the notion of an ϵ-approximate solution in (13) is defined in expectation for SVRG. On the other hand, the scaling of the oracle complexity of SVRG is similar to GD, despite SVRG being a stochastic optimization method.

Another variant of batch GD and SGD that has been studied in the literature is the notion of mini-batch SGD, where each iteration uses a randomly chosen mini-batch of size BN to approximate the full gradient (instead of a batch of size n, which corresponds to GD, or size 1, which corresponds to SGD). As expounded in Bottou et al. (2018, section 4.2), the analysis of mini-batch SGD can be carried out in two important regimes: Bn or Bn. The latter regime reduces to a GD-like setting, so we only consider the Bn regime here. In this setting, under universal second moment bounds on stochastic gradients and linearly diminishing step sizes as with SGD, the ensuing proposition presents the first-order oracle complexity of mini-batch SGD. This result is known in the literature, but seldom stated explicitly (Dekel et al. 2012; also see Nemirovski et al. 2009 and Bottou et al. 2018, theorem 4.7).

Proposition A.4

(Oracle Complexity of Mini-Batch SGD). Under the aforementioned assumptions, the first-order oracle complexity of mini-batch SGD is bounded by

Γ(Mini-batchSGD)B2σ max{B1 supx,ϑθf(x;ϑ)22,μ(F(θ(0))F*)}μϵ=O(pϵ),
where θ(0)Rp is the initial parameter vector of mini-batch SGD, supx,ϑθf(x;ϑ)22=O(p),F(θ(0))F*=O(p), and we let B=O(1) to obtain the scaling of the oracle complexity.

As before, the notion of an ϵ-approximate solution in (13) is defined in expectation here. Proposition A.4 demonstrates that mini-batch SGD does not offer any oracle complexity benefit over SGD when (canonical) sequential computation is used (cf. Dekel et al. 2012, Bubeck 2015, and Bottou et al. 2018); indeed, it is well-known that benefits are only seen when computational tricks like parallelization are exploited. However, because we do not analyze the parallel computation setting in this work, we do not consider mini-batch SGD with parallel computation here.

Finally, it is worth mentioning that standard comparisons between the oracle complexities of GD, SGD, and their variants in the literature (as we described at the outset of Section 1) often assume that the factors θ(0)θ*22,F(θ(0))F*, and supx,ϑθf(x;ϑ)22 are constants with respect to the parameter dimension p. However, we perform our comparisons between oracle complexities by allowing these factors to scale as O(p), which is reasonable when the parameter belongs to an -ball with constant radius.

Appendix B. Discussion of Assumptions

In this appendix, the text from here up to the beginning of Section B.1 is taken almost verbatim from section II-C of Jadbabaie et al. (2023), and the text in Section B.2 is taken almost verbatim from appendix I of Jadbabaie et al. (2023). On the other hand, the technical example in Section B.1 does not appear in Jadbabaie et al. (2023), but the text in Section B.1 follows the template of Section B.2.

In the following, we discuss the smoothness and convexity assumptions in Section 2.2 and contextualize them within the optimization and statistics literatures.

Firstly, smoothness in parameter is a standard assumption in the optimization for machine learning literature (see, e.g., Nesterov 2004, Bubeck 2015, and the references therein). This assumption implies that the ERM objective function F(θ) has L1-Lipschitz continuous gradient (which is used in the proofs of Proposition 2 and Theorem 2 in Section 5). In particular, when the gradient of F(θ) is L1-Lipschitz continuous, it can be shown that each iteration of GD reduces the objective value:

F(θ1L1θF(θ))F(θ)12L1θF(θ)22,(B.1)
for all θRp (Bubeck 2015, equation (3.5)). The inequality (B.1) illustrates the greedy nature in which GD decreases its objective value and is fundamental in the analysis of iterative descent methods. For this reason, this assumption has been used extensively in the optimization literature to prove convergence guarantees in a variety of settings—for example, Nemirovski et al. (2009), Friedlander and Schmidt (2012), Dekel et al. (2012), Le Roux et al. (2012), Johnson and Zhang (2013), Devolder et al. (2013), Schmidt et al. (2017), Bottou et al. (2018), etc. We also impose this assumption because the analysis to obtain Lemma 4 in Section 5.1, which is a restatement of Friedlander and Schmidt (2012, lemma 2.1), requires it.

Similarly, strong convexity in parameter is another standard assumption in the optimization for machine learning literature (see, e.g., Nesterov 2004, Bubeck 2015, and the references therein). As before, this assumption implies that the empirical risk F(θ) is strongly convex (which is used in the proofs of Proposition 2 and Theorem 2 in Section 5 (cf. Nesterov 2004, lemma 2.1.4)). As explained in Bubeck (2015), the main utility of imposing strong convexity of the ERM objective function F(θ) is a notable and provable speed-up in the convergence rate of gradient-based iterative methods. Specifically, GD can achieve linear convergence rate (i.e., exponentially fast convergence) when the objective function is strongly convex (Nesterov 2004, Bubeck 2015). In addition, the strong convexity assumption often leads to far simpler proofs of convergence and oracle complexity and easier illustration of the benefits of improvements like momentum (or heavy-ball method) (Polyak 1964) and acceleration (Nesterov 1983) on oracle complexity (cf. Nesterov 2004 and Bottou et al. 2018). For these reasons, this assumption has also been used extensively in the optimization literature to prove convergence guarantees in a variety of settings—for example, Nemirovski et al. (2009), Friedlander and Schmidt (2012), Le Roux et al. (2012), Johnson and Zhang (2013), Devolder et al. (2013), Schmidt et al. (2017), Bottou et al. (2018), etc. We also assume strong convexity for these reasons; indeed, strong convexity is needed for Lemma 4 in Section 5.1 to hold, and the first term in the bound in Lemma 4 illustrates the aforementioned exponential convergence.

Smoothness in data is a standard assumption in the nonparametric estimation literature (see, e.g., Tsybakov 2009, Chen and Shah 2018, Xu 2018, Wasserman 2019, Agarwal et al. 2021, and the references therein). Intuitively, we can think of a function in a Hölder class with positive integer parameter η as an (η1)-times differentiable function with Lipschitz continuous (η1)th partial derivatives (or even an η-times differentiable function with bounded ηth partial derivatives). Hence, the size of the parameter η controls the degree of smoothness of functions in a Hölder class. Such Hölder class assumptions are used in various nonparametric regression techniques—for example, kernel regression such as Nadaraya-Watson estimators (Nadaraya 1964, Watson 1964, Tsybakov 2009), local polynomial regression (Cleveland and Loader 1996, Fan and Gijbels 1996, Tsybakov 2009), and nearest-neighbor methods (Fix and Hodges 1951, Cover and Hart 1967, Chen and Shah 2018). At a high level, the basic idea of imposing smoothness assumptions for nonparametric regression is the following: If an unknown function living in a “large” class is known to be smoother, then it can be better estimated from samples using Taylor polynomials.

In a related vein, in a recent sequence of works in high-dimensional statistics, it has been shown that matrices defined by bivariate graphon functions (or latent variable models) are approximately low rank when the function satisfies Hölder class assumptions on one of the variables (Chatterjee 2015, Xu 2018, Agarwal et al. 2021, Jadbabaie et al. 2023). Because partial derivatives of loss functions gi can be construed as bivariate functions of the data x and parameter θ, our Hölder class smoothness in data assumption can also be construed as an approximate low rank property of gi. The close connection between smoothness of bivariate functions and approximate low rank structure may seem peculiar at first glance. Perceiving a bivariate function as a kernel of an integral operator, the results in Chatterjee (2015), Xu (2018), Agarwal et al. (2021), and Jadbabaie et al. (2023) suggest that smoothness of the kernel is closely related to approximate low rankness of the associated integral operator. This relation has its roots in more familiar ideas from harmonic analysis. Indeed, in the setting of difference (or convolution) kernels that are defined by a univariate function, increasing the smoothness of the function is essentially equivalent to faster decay of its Fourier transform (Körner 1988, theorem 60.2) (also see Stein and Shakarchi 2003a and b). This, in turn, can be interpreted as approximate low rankness of the associated convolution operator because the Fourier transform is precisely the spectrum of the convolution operator.

A litany of recent empirical investigations in the literature have demonstrated various forms of this alternative manifestation of smoothness—namely, approximate low rank structure—in large-scale machine learning tasks. For instance, the empirical risk and loss functions corresponding to deep neural network training problems display approximate low rank structure in their gradients (which live in low-dimensional subspaces) or Hessians (which have decaying eigenvalues, thereby inducing low rank gradients via Taylor approximation arguments) (Gur-Ari et al. 2018, Sagun et al. 2018, Papyan 2019, Cui et al. 2020, Singh et al. 2021, Wu et al. 2021). In particular, Gur-Ari et al. (2018) show that when training deep neural networks for K-ary classification, the computed gradients belong to K-dimensional eigenspaces of the Hessian corresponding to its dominant eigenvalues. On a related front, low rankness has also been observed in weight matrices of deep neural networks (Le and Jegelka 2022, Galanti et al. 2023), neural collapse has been demonstrated when training deep neural networks (Papyan et al. 2020, Fang et al. 2021, Hui et al. 2022), and training memory for neural networks has been saved using low rank parameterizations of gradient weight matrices (Gooneratne et al. 2020).

While the above discussion elucidates why the assumptions in Section 2.2 are necessary for theoretical analysis and presents some of the literature surrounding these assumptions, we next delineate two examples of commonly used loss functions in machine learning that satisfy the smoothness and strong convexity assumptions in Section 2.2. Specifically, Section B.1 provides an example of 2-regularized linear regression, and Section B.2 provides an example of 2-regularized logistic regression.

B.1. Example: Ridge Regression

In this subsection, we consider 2-regularized linear regression or ridge regression. Let the parameter space be [1,1]d1 so that any parameter vector θ[1,1]d1, and each labeled training data sample (x,y)[0,1]d1×[0,1]. The compactness of the parameter space is reasonable because training usually takes place in a compact subset of the Euclidean space Rd1. Consider the 2-regularized squared loss function:

fsq(x,y;θ)(yθTx)2+μ2θ22,(B.2)
where μ>0 is a hyperparameter that determines the level of regularization, and we assume for convenience that the bias parameter in ridge regression is zero. Given a training data set (x(1),y1),,(x(n),yn)[0,1]d1×[0,1], the associated empirical risk is
F(θ)=1ni=1nfsq(x(i),yi;θ)=1ni=1n(yiθTx(i))2+μ2θ22,(B.3)
which indeed corresponds to the well-known setting of ridge regression (Hastie et al. 2009, section 3.4.1). We next verify that fsq in (B.2) satisfies the smoothness and strong convexity assumptions in Section 2.2:
  1. Smoothness in parameter: Observe that θfsq(x,y;θ)=2(θTxy)x+μθ, which means that for all x, y and all θ(1),θ(2)Rd1,

    θfsq(x,y;θ(1))θfsq(x,y;θ(2))22|(θ(1)θ(2))Tx|x2+μθ(1)θ(2)22x22θ(1)θ(2)2+μθ(1)θ(2)2(2d+μ2)θ(1)θ(2)2,(B.4)
    where the first inequality follows from the triangle inequality, the second inequality follows from the Cauchy-Schwarz inequality, and the final inequality follows from the bound x22d1. Hence, the Lipschitz constant L1=2d+μ2.

    Although L1 is usually assumed to be constant in theoretical analyses in the optimization for machine learning literature (see the above discussion and references on smoothness in parameter), such dependence of L1 on d does not affect our analysis in this work. Indeed, note that L1 appears in the oracle complexity bound in Theorem 2 through the constant Cμ,L1,L2,b,c(d,l) in (34). When the condition number σ,

    σ+2(L1μ)(L1μ)log(σσ1)=O(σ),(B.5)
    which implies that Cμ,L1,L2,b,c(d,l) grows linearly in σ. Hence, when L1=O(d) as in this example, and, consequently, the condition number σ=O(d),Cμ,L1,L2,b,c(d,l) picks up an additional O(d) factor. However, this does not change the result in Proposition 3, as d=O(log log(n)) (see the proof in Section 5.3).

  2. Smoothness in data: Observe that for any fixed ϑ[1,1]d1 and any i[d1], the ith partial derivative gi(x,y;ϑ)=2(ϑTxy)xi+μϑi. Let us arbitrarily choose η = 2 for the purposes of illustration. Then, for any s=(0,,0,1,0,,0)Z+d with a value of one at the jth position and |s|=η1=1, we have for all x[0,1]d1 and y[0,1],

    x,ysgi(x,y;ϑ)={2fsqθixi(x,y;ϑ)=2(ϑTxϑixiy),j=i,2fsqθixj(x,y;ϑ)=2xiϑj,j[d1],ji,2fsqθiy(x,y;ϑ)=2xi,j=d.(B.6)

    Hence, for all x(1),x(2)[0,1]d1 and y1,y2[0,1], we obtain

    |x,ysgi(x(1),y1;ϑ)x,ysgi(x(2),y2;ϑ)|{4x(1)x(2)1+2|y1y2|,j=i,2|xi(1)xi(2)|,j[d1],ji,2|xi(1)xi(2)|,j=d,(B.7)
    where the second and third cases follow from direct calculation and the fact that ϑ1, and the first case holds because
    |x,ysgi(x(1),y1;ϑ)x,ysgi(x(2),y2;ϑ)|=2|(ϑTx(1)ϑixi(1)y1)(ϑTx(2)ϑixi(2)y2)|2(|ϑT(x(1)x(2))|+|xi(1)xi(2)|+|y1y2|)2(x(1)x(2)1+|xi(1)xi(2)|+|y1y2|)4x(1)x(2)1+2|y1y2|,(B.8)
    where the second line follows from the triangle inequality and the fact that |ϑi|1, and the third line follows from Hölder’s inequality and the fact that ϑ1. This implies that for all x(1),x(2)[0,1]d1 and y1,y2[0,1], we have
    |x,ysgi(x(1),y1;ϑ)x,ysgi(x(2),y2;ϑ)|4(x(1),y1)(x(2),y2)1.(B.9)

    Thus, L2=4 in this example, and each gi(·;ϑ) belongs to a (2,4)-Hölder class.

  3. Strong convexity: It is straightforward to see that the (unregularized) squared loss function θ(yθTx)2 is convex for all x, y (as it is quadratic). This implies that θfsq(x,y;θ) is μ-strongly convex for all x, y (Bubeck 2015, section 3.4).

B.2. Example: Regularized Logistic Regression

In this subsection, we consider 2-regularized logistic regression with soft labels. As before, let the parameter space be [1,1]d1, so that any parameter vector θ[1,1]d1 and each labeled training data sample (x,y)[0,1]d1×[0,1] be such that the label y is a “soft” belief. (Note that the use of soft labels instead of canonical binary classification labels has been studied in the literature (cf. Nguyen et al. 2011).) Consider the 2-regularized cross-entropy loss function:

fent(x,y;θ)y log(1+eθTx)+(1y)log(1+eθTx)+μ2θ22,(B.10)
where μ>0 is a hyperparameter that determines the level of regularization as before, and we again assume for convenience that the bias parameter in logistic regression is zero. Given a training data set (x(1),y1),,(x(n),yn)[0,1]d1×[0,1], the associated empirical risk is
F(θ)=1ni=1nfent(x(i),yi;θ)=1ni=1n(yi log(1+eθTx(i))+(1yi)log(1+eθTx(i)))+μ2θ22,(B.11)
which indeed corresponds to the well-known setting of 2-regularized logistic regression (Hastie et al. 2009, section 4.4). We next verify that fent in (B.10) satisfies the smoothness and strong convexity assumptions in Section 2.2:
  1. Smoothness in parameter: Observe that θfent(x,y;θ)=((1y)1+eθTxy1+eθTx)x+μθ, which means that for all x, y and all θ(1),θ(2)Rd1,

    θfent(x,y;θ(1))θfent(x,y;θ(2))2x2|(1y)1+eθ(1)Txy1+eθ(1)Tx(1y)1+eθ(2)Tx+y1+eθ(2)Tx|+μθ(1)θ(2)2x2(|11+eθ(1)Tx11+eθ(2)Tx|+|11+eθ(2)Tx11+eθ(1)Tx|)+μθ(1)θ(2)2x22|(θ(1)θ(2))Tx|+μθ(1)θ(2)2x222θ(1)θ(2)2+μθ(1)θ(2)2(d12+μ)θ(1)θ(2)2,(B.12)
    where the first inequality follows from the triangle inequality, the second inequality holds because max{|y|,|1y|}1, the third inequality holds due to the Lipschitz continuity of the sigmoid function and the fact that maxtR|ddt(1+et)1|=14, the fourth inequality follows from the Cauchy-Schwarz inequality, and the final inequality follows from the bound x22d1. Hence, the Lipschitz constant L1=d12+μ. (As explained earlier, such dependence of L1 on d does not affect our analysis in this work.)

  2. Smoothness in data: Observe that for any fixed ϑ[1,1]d1 and any i[d1], the ith partial derivative gi(x,y;ϑ)=((1y)1+eϑTxy1+eϑTx)xi+μϑi. Once again, let us arbitrarily choose η = 2 for the purposes of illustration. Then, for any s=(0,,0,1,0,,0)Z+d with a value of one at the jth position and |s|=η1=1, we have for all x[0,1]d1 and y[0,1],

    x,ysgi(x,y;ϑ)={2fentθixi(x,y;ϑ)=ϑixieϑTx(1+eϑTx)2+11+eϑTxy,j=i,2fentθixj(x,y;ϑ)=ϑjxieϑTx(1+eϑTx)2,j[d1],ji,2fentθiy(x,y;ϑ)=xi,j=d.(B.13)

    Hence, for all x(1),x(2)[0,1]d1 and y1,y2[0,1], we obtain

    |x,ysgi(x(1),y1;ϑ)x,ysgi(x(2),y2;ϑ)|{34x(1)x(2)1+|y1y2|,j=i,12x(1)x(2)1,j[d1],ji,|xi(1)xi(2)|,j=d,(B.14)
    where the third case follows from direct calculation, the first case follows from a similar argument to the second case, and the second case holds because
    |x,ysgi(x(1),y1;ϑ)x,ysgi(x(2),y2;ϑ)|=|ϑjxi(1)eϑTx(1)(1+eϑTx(1))2ϑjxi(2)eϑTx(2)(1+eϑTx(2))2||xi(1)eϑTx(1)(1+eϑTx(1))2xi(2)eϑTx(2)(1+eϑTx(2))2||eϑTx(1)(1+eϑTx(1))2eϑTx(2)(1+eϑTx(2))2|+eϑTx(2)(1+eϑTx(2))2|xi(1)xi(2)||eϑTx(1)(1+eϑTx(1))2eϑTx(2)(1+eϑTx(2))2|+14|xi(1)xi(2)|14|ϑT(x(1)x(2))|+14|xi(1)xi(2)|14x(1)x(2)1+14|xi(1)xi(2)|12x(1)x(2)1,(B.15)
    where the second line follows from the fact that |ϑj|1, the third line follows from the triangle inequality and the fact that xi(1)[0,1], the fourth line holds because et(1+et)214 for all tR, the fifth line follows from Lipschitz continuity and the fact that maxtR|ddtet(1+et)2|14, and the sixth line follows from Hölder’s inequality and the fact that ϑ1. This implies that for all x(1),x(2)[0,1]d1 and y1,y2[0,1], we have
    |x,ysgi(x(1),y1;ϑ)x,ysgi(x(2),y2;ϑ)|(x(1),y1)(x(2),y2)1.(B.16)

    Thus, L2=1 in this example, and each gi(·;ϑ) belongs to a (2,1)-Hölder class.

  3. Strong convexity: It is well-known that the (unregularized) cross-entropy loss function θy log(1+eθTx)+(1y)log(1+eθTx) is convex for all x, y (Hastie et al. 2009, section 4.4). This can be directly checked by computing derivatives. This implies that θfent(x,y;θ) is μ-strongly convex for all x, y (Bubeck 2015, section 3.4).

Appendix C. Conjecture on Scaling of Minimum Eigenvalue

Let B be the D × D symmetric matrix defined in Lemma 1. We conjecture that the scaling of its minimum eigenvalue in (52) can be significantly improved as follows.

Conjecture C.1

(Scaling of Minimum Eigenvalue of B). If l=Θ(poly(d)), then we have

1λmin(B)=Θ(eΘ(poly(d))).

Indeed, using the Courant-Fischer-Weyl min-max theorem (Horn and Johnson 2013, theorem 4.2.6) (also see the Schur-Horn theorem (Horn and Johnson 2013, theorem 4.3.45)), the minimum eigenvalue of B is upper bounded by the minimum diagonal entry of B as follows:

λmin(B)minsZ+d:|s|l2d(s!)2j=1d12sj+12d(2l+1)(l!)22d1e2lπ(2l+1)l2l+12d2e2lπl2l+2,(C.1)
where the second inequality follows from setting s1=l and s2==sd=0 (which is reasonable because it asymptotically achieves the minimum), and the third inequality follows from the Stirling’s formula bound (cf. Feller 1968, chapter II, section 9, equation (9.15)). Thus, λmin(B)1 has the potential to scale as
1λmin(B)πl2l+22d2e2l=Ω(eΘ(poly(d))),(C.2)
when l=Θ(poly(d)). As shown in Conjecture C.1, we believe that this scaling is likely to be the actual scaling of λmin(B)1. Because we exactly evaluate the trace and determinant in the proof of Lemma 1, the looseness of our minimum eigenvalue lower bound in Lemma 1 stems from the AM–GM based result in (49). Hence, a fruitful future direction would be to obtain tighter bounds on the minimum eigenvalue in terms of trace and determinant that are still sufficiently tractable so as to permit analytical evaluation.

Endnotes

1 This compactness assumption affords us some analytical tractability and circumvents boundary effects in the sequel.

2 Note that one of the most general frameworks for inexact gradient descent with strongly convex objective functions can be found in Devolder et al. (2013). However, our algorithm does not require many of the intricacies of this general framework. So, we do not present it here for brevity.

3 We remark that as long as h=O(Λ(d,l)l/d), we can choose a kernel K in the sequel such that (25) holds.

4 Alternatively, the Legendre polynomials can be obtained by applying the Gram-Schmidt process to the monomials 1,t,t2, with respect to the standard inner product, using a three-term recurrence relation, solving Legendre’s differential equation, or expanding certain generating functions (Chihara 2011).

5 Note that the usual upper bound of Stirling’s approximation is j!ej(j/e)j for all jN. In order to include the case j = 0, we loosen this bound in our analysis.

6 This is a simple consequence of the arithmetic mean–geometric mean (AM–GM) inequality.

7 This bound can also be proved using the AM–GM inequality.

8 Indeed, it is straightforward to verify that given two positive sequences {an}nN and {bn}nN such that limnan=limnbn=+, if lim supn log(an)/log(bn)<1, then limnan/bn=0.

References

  • Agarwal A, Shah D, Shen D, Song D (2021) On robustness of principal component regression. J. Amer. Statist. Assoc. 116(536):1731–1745.Google Scholar
  • Agarwal A, Wainwright MJ, Bartlett PL, Ravikumar PK (2009) Information-theoretic lower bounds on the oracle complexity of convex optimization. Proc. Adv. Neural Inform. Processing Systems 22 NeurIPS (Curran Associates Inc., Red Hook, NY), 1–9.Google Scholar
  • Apostol TM (1974) Mathematical Analysis, 2nd ed. (Addison-Wesley, Reading, MA).Google Scholar
  • Ausset G, Clémençon S, Portier F (2021) Nearest neighbour based estimates of gradients: Sharp nonasymptotic bounds and applications. Proc. 24th Internat. Conf. Artificial Intelligence Statist. AISTATS (PMLR, New York), 532–540.Google Scholar
  • Belkin M, Niyogi P (2001) Laplacian eigenmaps and spectral techniques for embedding and clustering. Proc. Adv. Neural Inform. Processing Systems 14 NeurIPS (MIT Press, Cambridge, MA), 585–591.Google Scholar
  • Bishop CM (2006) Pattern Recognition and Machine Learning, Information Science and Statistics (Springer, New York).Google Scholar
  • Bottou L, Curtis FE, Nocedal J (2018) Optimization methods for large-scale machine learning. SIAM Rev. 60(2):223–311.Google Scholar
  • Bubeck S (2015) Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning, vol. 8 (Now Publishers Inc., Hanover, MA).Google Scholar
  • Carmon Y, Duchi JC, Hinder O, Sidford A (2021) Lower bounds for finding stationary points II: First-order methods. Math. Program. Ser. A 185:315–355.Google Scholar
  • Chatterjee S (2015) Matrix estimation by universal singular value thresholding. Ann. Statist. 43(1):177–214.Google Scholar
  • Chen GH, Shah D (2018) Explaining the Success of Nearest Neighbor Methods in Prediction, Foundations and Trends in Machine Learning, vol. 10 (Now Publishers Inc., Hanover, MA).Google Scholar
  • Chihara TS (2011) An Introduction to Orthogonal Polynomials (Dover, New York).Google Scholar
  • Choi E, Xiao C, Stewart WF, Sun J (2018) MiME: Multilevel medical embedding of electronic health records for predictive healthcare. Proc. Adv. Neural Inform. Processing Systems 32 NeurIPS (Curran Associates Inc., Red Hook, NY), 4547–4557.Google Scholar
  • Cleveland WS, Loader C (1996) Smoothing by local regression: Principles and methods. Härdle W, Schimek MG, eds. Statistical Theory and Computational Aspects of Smoothing: Proceedings of the COMPSTAT ‘94 Satellite Meeting Held in Semmering, Austria August 27-28, 1994, Contributions to Statistics (Physica-Verlag, Heidelberg, Germany), 10–49.Google Scholar
  • Coifman RR, Lafon S (2006) Diffusion maps. Appl. Comput. Harmonic Anal. 21(1):5–30.Google Scholar
  • Cover TM, Hart PE (1967) Nearest neighbor pattern classification. IEEE Trans. Inform. Theory IT-13(1):21–27.Google Scholar
  • Cui C, Zhang K, Daulbaev T, Gusak J, Oseledets I, Zhang Z (2020) Active subspace of neural networks: Structural analysis and universal attacks. SIAM J. Math. Data Sci. 2(4):1096–1122.Google Scholar
  • Dasgupta S, Gupta A (2003) An elementary proof of a theorem of Johnson and Lindenstrauss. Random Structures Algorithms 22(1):60–65.Google Scholar
  • De Brabanter K, De Brabanter J, De Moor B, Gijbels I (2013) Derivative estimation with local polynomial fitting. J. Machine Learn. Res. 14(1):281–301.Google Scholar
  • Dekel O, Gilad-Bachrach R, Shamir O, Xiao L (2012) Optimal distributed online prediction using mini-batches. J. Machine Learn. Res. 13(6):165–202.Google Scholar
  • Delecroix M, Rosa AC (1996) Nonparametric estimation of a regression function and its derivatives under an ergodic hypothesis. J. Nonparametric Statist. 6(4):367–382.Google Scholar
  • Devolder O, Glineur F, Nesterov Y (2013) First-order methods with inexact oracle: The strongly convex case. CORE Discussion Papers 2013(16), Université catholique de Louvain, Louvain-la-Neuve, Belgium.Google Scholar
  • Dutton RD, Brigham RC (1986) Computationally efficient bounds for the Catalan numbers. Eur. J. Combin. 7(3):211–213.Google Scholar
  • Fan J, Gijbels I (1996) Local Polynomial Modeling and Its Applications, Monographs on Statistics and Applied Probability, vol. 66 (Chapman and Hall, London).Google Scholar
  • Fang C, He H, Long Q, Su WJ (2021) Exploring deep neural networks via layer-peeled model: Minority collapse in imbalanced training. Proc. Natl. Acad. Sci. USA 118(43):1–12.Google Scholar
  • Feller W (1968) An Introduction to Probability Theory and Its Applications, vol. 1, 3rd ed. (John Wiley & Sons, Inc., New York).Google Scholar
  • Fix E, Hodges JL (1951) Discriminatory analysis. Nonparametric discrimination: Consistency properties. Technical report, University of California, Berkeley, CA, and USAF School of Aviation Medicine, Randolph Field, San Antonio, TX.Google Scholar
  • Forbes C, Evans M, Hastings N, Peacock B (2011) Statistical Distributions, 4th ed. (John Wiley & Sons, Inc., Hoboken, NJ).Google Scholar
  • Friedlander MP, Schmidt M (2012) Hybrid deterministic-stochastic methods for data fitting. SIAM J. Sci. Comput. 34(3):A1380–A1405.Google Scholar
  • Galanti T, Siegel ZS, Gupte A, Poggio T (2023) SGD and weight decay provably induce a low-rank bias in neural networks. Preprint, submitted January 27, https://arxiv.org/abs/2206.05794v3.Google Scholar
  • Gooneratne M, Sim KC, Zadrazil P, Kabel A, Beaufays F, Motta G (2020) Low-rank gradient approximation for memory-efficient on-device training of deep neural network. Proc. IEEE Internat. Conf. Acoustics Speech Signal Processing ICASSP (IEEE, Piscataway, NJ), 3017–3021.Google Scholar
  • Gur-Ari G, Roberts DA, Dyer E (2018) Gradient descent happens in a tiny subspace. Preprint, submitted December 12, https://arxiv.org/abs/1812.04754.Google Scholar
  • Hastie T, Tibshirani R, Friedman J (2009) The Elements of Statistical Learning: Data Mining, Inference, and Prediction, 2nd ed., Springer Series in Statistics (Springer, New York).Google Scholar
  • Horn RA, Johnson CR (2013) Matrix Analysis, 2nd ed. (Cambridge University Press, New York).Google Scholar
  • Hotelling H (1933) Analysis of a complex of statistical variables into principal components. J. Ed. Psych. 24(6):417–441, 498–520.Google Scholar
  • Hotelling H (1936) Relations between two sets of variates. Biometrika 28(3/4):321–377.Google Scholar
  • Huang SL, Makur A, Wornell GW, Zheng L (2024) Universal Features for High-Dimensional Learning and Inference, Foundations and Trends in Communications and Information Theory, vol. 21 (Now Publishers Inc., Hanover, MA).Google Scholar
  • Hui L, Belkin M, Nakkiran P (2022) Limitations of neural collapse for understanding generalization in deep learning. Preprint, submitted February 17, https://arxiv.org/abs/2202.08384.Google Scholar
  • Jadbabaie A, Makur A, Shah D (2023) Federated optimization of smooth loss functions. IEEE Trans. Inform. Theory 69(12):7836–7866.Google Scholar
  • Jain P, Kar P (2017) Non-Convex Optimization for Machine Learning, Foundations and Trends in Machine Learning, vol. 10 (Now Publishers Inc., Hanover, MA).Google Scholar
  • Johnson R, Zhang T (2013) Accelerating stochastic gradient descent using predictive variance reduction. Proc. Adv. Neural Inform. Processing Systems 26 NeurIPS (Curran Associates Inc., Red Hook, NY), 315–323.Google Scholar
  • Jolliffe IT (1982) A note on the use of principal components in regression. J. Roy. Statist. Soc. Ser. C. Appl. Statist. 31(3):300–303.Google Scholar
  • Körner TW (1988) Fourier Analysis (Cambridge University Press, Cambridge, UK).Google Scholar
  • Krantz SG, Parks HR (2002) A Primer of Real Analytic Functions, 2nd ed., Birkhäuser Advanced Texts: Basler Lehrbücher (Birkhäuser, Boston).Google Scholar
  • Le T, Jegelka S (2022) Training invariances and the low-rank phenomenon: Beyond linear networks. Proc. Tenth Internat. Conf. Learn. Representations (ICLR, Appleton, WI), 1–26.Google Scholar
  • Le Roux N, Schmidt M, Bach F (2012) A stochastic gradient method with an exponential convergence rate for finite training sets. Proc. Adv. Neural Inform. Processing Systems 25 NeurIPS (Curran Associates Inc., Red Hook, NY), 1–9.Google Scholar
  • Makur A (2019) Information contraction and decomposition. ScD thesis in Electrical Engineering and Computer Science, Massachusetts Institute of Technology, Cambridge, MA.Google Scholar
  • Marshall AW, Olkin I, Arnold BC (2011) Inequalities: Theory of Majorization and Its Applications, 2nd ed., Springer Series in Statistics (Springer, New York).Google Scholar
  • Masry E (1996) Multivariate regression estimation local polynomial fitting for time series. Stochastic Processes Appl. 65(1):81–101.Google Scholar
  • Masry E (1999) Multivariate regression estimation of continuous-time processes from sampled data: Local polynomial fitting approach. IEEE Trans. Inform. Theory 45(6):1939–1953.Google Scholar
  • Merikoski JK, Virtanen A (1997) Bounds for eigenvalues using the trace and determinant. Linear Algebra Appl. 264:101–108.Google Scholar
  • Mukherjee S, Wu Q (2006) Estimation of gradients and coordinate covariation in classification. J. Machine Learn. Res. 7(88):2481–2514.Google Scholar
  • Mukherjee S, Zhou DX (2006) Learning coordinate covariances via gradients. J. Machine Learn. Res. 7(18):519–549.Google Scholar
  • Nadaraya EA (1964) On estimating regression. Theory Probab. Appl. 9(1):141–142.Google Scholar
  • Nechyba MC, Xu Y (1994) Neural network approach to control system identification with variable activation functions. Proc. 9th IEEE Internat. Sympos. Intelligent Control (IEEE, Piscataway, NJ), 358–363.Google Scholar
  • Nedić A (2015) Distributed optimization. Baillieul J, Samad T, eds. Encyclopedia of Systems and Control (Springer, London), 308–317.Google Scholar
  • Nelles O (2001) Nonlinear System Identification: From Classical Approaches to Neural Networks and Fuzzy Models (Springer, Berlin, Heidelberg).Google Scholar
  • Nemirovski A, Juditsky A, Lan G, Shapiro A (2009) Robust stochastic approximation approach to stochastic programming. SIAM J. Optim. 19(4):1574–1609.Google Scholar
  • Nemirovskiĭ AS, Yudin DB (1983) Problem Complexity and Method Efficiency in Optimization, Wiley-Interscience Series in Discrete Mathematics and Optimization (John Wiley & Sons Inc., New York).Google Scholar
  • Nesterov YE (1983) A method of solving a convex programming problem with convergence rate O(1k2). Dokl. Akad. Nauk SSSR 269(3):543–547.Google Scholar
  • Nesterov Y (2004) Introductory Lectures on Convex Optimization: A Basic Course, Applied Optimization, vol. 87 (Springer, New York).Google Scholar
  • Netrapalli P (2019) Stochastic gradient descent and its variants in machine learning. J. Indian Inst. Sci. 99(2):201–213.Google Scholar
  • Nguyen Q, Valizadegan H, Hauskrecht M (2011) Learning classification with auxiliary probabilistic information. Proc. IEEE Internat. Conf. Data Mining ICDM (IEEE, Piscataway, NJ), 477–486.Google Scholar
  • Papyan V (2019) The full spectrum of deepnet Hessians at scale: Dynamics with SGD training and sample size, Preprint, submitted June 3, https://arxiv.org/abs/1811.07062.Google Scholar
  • Papyan V, Han XY, Donoho DL (2020) Prevalence of neural collapse during the terminal phase of deep learning training. Proc. Natl. Acad. Sci. USA 117(40):24652–24663.Google Scholar
  • Pearson K (1901) On lines and planes of closest fit to systems of points in space. Philos. Mag. 2(11):559–572.Google Scholar
  • Poggio T, Banburski A, Liao Q (2020) Theoretical issues in deep networks. Proc. Natl. Acad. Sci. USA 117(48):30039–30045.Google Scholar
  • Polyak BT (1964) Some methods of speeding up the convergence of iteration methods. USSR Comput. Math. Math. Phys. 4(5):1–17.Google Scholar
  • Rudin W (1976) Principles of Mathematical Analysis, 3rd ed., International Series in Pure and Applied Mathematics (McGraw Hill, New York).Google Scholar
  • Sagun L, Evci U, Güney VU, Dauphin Y, Bottou L (2018) Empirical analysis of the Hessian of over-parametrized neural networks. Proc. Sixth Internat. Conf. Learn. Representations ICLR Workshop (ICLR, Appleton, WI), 1–14.Google Scholar
  • Schmidt M, Le Roux N, Bach F (2017) Minimizing finite sums with the stochastic average gradient. Math. Program. Ser. A 162:83–112.Google Scholar
  • Shalev-Shwartz S, Zhang T (2013) Stochastic dual coordinate ascent methods for regularized loss minimization. J. Mach. Learn. Res. 14:567–599.Google Scholar
  • Singh SP, Bachmann G, Hofmann T (2021) Analytic insights into structure and rank of neural network Hessian maps. Proc. Adv. Neural Inform. Processing Systems 34 NeurIPS, 23914–23927.Google Scholar
  • So AMC, Zhou Z (2017) Non-asymptotic convergence analysis of inexact gradient methods for machine learning without strong convexity. Optim. Methods Softw. 32(4):963–992.Google Scholar
  • Stein EM, Shakarchi R (2003a) Complex Analysis, Princeton Lectures in Analysis, vol. 2 (Princeton University Press, Princeton, NJ).Google Scholar
  • Stein EM, Shakarchi R (2003b) Fourier Analysis: An Introduction, Princeton Lectures in Analysis, vol. 1 (Princeton University Press, Princeton, NJ).Google Scholar
  • Stone CJ (1977) Consistent nonparametric regression. Ann. Statist. 5(4):595–620.Google Scholar
  • Tsybakov AB (2009) Introduction to Nonparametric Estimation, Springer Series in Statistics (Springer, New York).Google Scholar
  • Wang Y, Du SS, Balakrishnan S, Singh A (2018) Stochastic zeroth-order optimization in high dimensions. Proc. 21st Internat. Conf. Artificial Intelligence Statist. AISTATS (PMLR, New York), 1356–1365.Google Scholar
  • Wasserman L (2019) Statistical methods for machine learning. Lecture Notes 36–708, Department of Statistics and Data Science, Carnegie Mellon University, Pittsburgh.Google Scholar
  • Watson GS (1964) Smooth regression analysis. Sankhya Ser. A 26(4):359–372.Google Scholar
  • Wu Y, Zhu X, Wu C, Wang A, Ge R (2021) Dissecting Hessian: Understanding common structure of Hessian in neural networks. Preprint, submitted June 16, https://arxiv.org/abs/2010.04261v5.Google Scholar
  • Xu J (2018) Rates of convergence of spectral methods for graphon estimation. Proc. 35th Internat. Conf. Machine Learn. ICML (PMLR, New York), 5433–5442.Google Scholar
  • Zhang ZG, Chan SC (2011) On kernel selection of multivariate local polynomial modelling and its application to image smoothing and reconstruction. J. Signal Processing Systems 64(3):361–374.Google Scholar
  • Zhou S, Wolfe DA (2000) On derivative estimation in spline regression. Statist. Sinica 10(1):93–108.Google Scholar