Gradient-Based Empirical Risk Minimization Using Local Polynomial Regression
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 , 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 . 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 , 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 for any and is at the statistical limit, our method can be made to require oracle calls for any , while SGD and GD require and 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 (Nesterov 2004, theorem 2.1.15), where n is the number of training samples in the ERM objective function and is the desired approximation accuracy of the solution. In comparison, it is well-known that SGD has an oracle complexity of (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 and , 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 . Hence, the typical regime of interest is when n is large. For small p (i.e., ), 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):
To this end, in the remainder of this work, we consider the following problem formulation of ERM. For , assume that we are given n samples of d-dimensional training data , where is an arbitrarily small parameter (to be determined later; see (25)), and each training sample belongs to the hypercube .1 Note that when , the first d − 1 elements of a sample can be perceived as a -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 , fix the (parametrized) loss function , where is a p-dimensional parameter vector. Now, define the ERM objective function as follows:
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 -norm error guarantees (cf. Friedlander and Schmidt 2012, section 1.1).2 Given the objective function , a sequence of step sizes , and an initial parameter vector , an inexact gradient descent algorithm proceeds via the iterations
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 over the data space . Then, for every training data sample , we precompute the (scalar-valued) interpolation weights such that
After these precomputation stages, we run inexact gradient descent using a pertinent choice of fixed step size for an appropriately chosen number of iterations . In particular, at iteration , due to the finite sum structure of the ERM objective function in (3), we construct the following approximation of the true gradient using the previous iteration’s parameter vector :
Constructing this local polynomial interpolation-based approximation requires us to compute gradients of the loss function . Then, we compute the standard inexact gradient descent update
We note that the inexact gradients are “good” approximations of , and we can derive -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 is much smaller than the number of training samples , 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 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.
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.
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 (neglecting subdominant factors), where 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 is nonconvex.
Furthermore, we show in Propositions 3 and 4 in Section 3.2.2 that if the data dimension is sufficiently small—for example, (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, —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 and any . In particular, when we let have the smallest scaling of interest (as explained earlier) and for any , Table 1 depicts that the oracle complexities of GD, SGD, and LPI-GD are , and for any small (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.
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 1. Oracle Complexities of GD, SGD, and Our LPI-GD Algorithm for Strongly Convex Loss Functions
| Algorithm | Oracle complexity |
|---|---|
| GD | (Nesterov 2004) |
| SGD | (Bottou et al. 2018) |
| LPI-GD | (Theorem 2) |
Notes. Here, we assume that the approximation accuracy , the parameter dimension for any constant , and the data dimension , where n is the number of training samples. Moreover, 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 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, . 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 and denote the sets of natural numbers and nonnegative integers, respectively. As mentioned earlier, for any , let be the set of positive integers less than n + 1. In the sequel, we analyze a d-variate problem setting with . So, we briefly introduce the standard multi-index notation. For any d-tuple of nonnegative integers , we let , and for . Moreover, given a continuously differentiable function , we write to denote its sth partial derivative, and to denote its gradient. We will sometimes find it convenient to index certain vectors or matrices using the set . 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 with , we say that r < s if rj < sj for the first position , where ).
Furthermore, for any vector with , we let xi be the ith entry of x, where (or more generally, i belongs to an appropriate index set), and we let be the usual -norm of x with . Similarly, for any matrix with , we let be the (i, j)th element of A, where and (or, more generally, i and j belong to appropriate index sets), and we let be the Frobenius norm of A. In addition, represents the transpose of A, and when m = n, represents the inverse of A if it is invertible, represents the trace of A, represents the determinant of A, and represents the minimum eigenvalue of A. We also let In be the n × n identity matrix. For any pair of symmetric matrices , we use to denote the Löwner partial order and write to mean that A − B 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, and denote the natural exponential and logarithm functions with base e, respectively, and 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, , etc.—where the underlying limiting variable (that usually tends to infinity) will be clear from context. Moreover, we will let and 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 and a loss function , and we consider the ERM problem in (3) and (4). Let 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):
Smoothness in parameter: There exists such that for all fixed , the gradient of f with respect to θ, denoted , exists and is L1-Lipschitz continuous as a function of the parameter vector:
(10)Smoothness in data: There exist and such that for all fixed and for all , the ith partial derivative of f (with respect to θi) at ϑ, denoted , belongs to the -Hölder class as a function of the data vector x (cf. Tsybakov 2009, definition 1.2). By definition, this means that each is times differentiable, and for every such that , we have
(11)Strong convexity: There exists such that for all fixed , the map is μ-strongly convex as a function of the parameter vector:
(12)Moreover, as additional notation, we let represent the condition number of the problem.
Under these assumptions, we define an ϵ-approximate solution to the optimization problem (4) as any parameter vector that satisfies
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 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), is clearly smooth with respect to the data (assuming 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 (-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 , 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 , defined as:
Given any particular iterative algorithm, , that solves our class of ERM problems, its first-order oracle complexity (or just oracle complexity) is formally defined as the minimum number of oracle calls made by the algorithm to obtain an ϵ-approximate solution in the sense of (13) (Nesterov 2004, section 1.1). Thus, is typically a function of the notable problem variables , and the fixed constants . 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 that belongs to the -Hölder class, namely, g is times differentiable and satisfies (11) for every with . In order to present the local polynomial regression-based approximation of this function, we first state some required definitions and notation. Let be a bounded kernel function, where are some fixed constants, and we will use the extension of this kernel (to the domain ), , in the sequel with the assumption that the support of K is . For any fixed , define the uniform grid
To intuitively understand the problem in (17), let us temporarily specialize to the setting where d = 1, η = 2, and the kernel . Then, belonging to the -Hölder class means that g is continuously differentiable and is L2-Lipschitz continuous. Moreover, the uniform grid has the form and is given by . Hence, the problem in (17) becomes:
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 , which live in the neighborhood. Because the mean value theorem and L2-Lipschitz continuity of imply the first-order Taylor approximation for all y in the neighborhood of x (cf. Rudin 1976, theorem 5.10), we can intuitively interpret the (scaled) optimal regression coefficients and as estimators of the first-order Taylor approximation coefficients g(x) and , respectively. This is why 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 , construct the D × D symmetric matrix
(
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 of the weighted regression problem in (17) can be written as:
Then, defining the first coordinates of these vectors as the real-valued interpolation weights:
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 .
(
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 with the parameter given by
Let denote the “approximate solution” produced by LPI-GD at iteration , and let denote the maximum allowable supremum norm approximation error at iteration . Fix any bounded kernel with as in Section 3.1, and corresponding to each δt at iteration , consider the uniform grid , where is given by Theorem 1:
We initialize to be any arbitrary vector in at iteration t = 0. Then, at iteration , we make first-order oracle calls at all virtual data points in the uniform grid with the approximate solution at iteration t − 1:
For every , these oracle calls give us information about the partial derivatives with respect to θi. Hence, akin to (17) and (24) in Section 3.1, we can construct the local polynomial interpolator of order for the -Hölder class function as follows:
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 , 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 . 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 ). 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.
(
Input: n samples of training data
Input: first order oracle defined in (14)
Input: approximation accuracy level
Output: ϵ-approximate solution satisfying (13)
Step 1: Initialization
1: Set arbitrary initial parameter vector
2: Set number of iterations according to (32)
3: Set supremum norm approximation error ⊳ This is the same for all iterations; see (33)
4: Set grid size according to (26) ⊳ Substitute mt = m and in (26)
5: Set bandwidth ⊳ See (27)
Step 2: Calculation of Interpolation Weights
6: Construct uniform grid according to (15)
7: for j = 1 to n do ⊳ Iterate over training data
8: Compute interpolation weights via (22) and (23) using kernel
9: end for
Step 3: Inexact Gradient Descent
10: for t = 1 to T do
11: Make oracle calls to get partial derivatives
Step 3a: Local Polynomial Interpolation
12: for j = 1 to n do ⊳ Iterate over training data
13: Compute local polynomial interpolators according to (29) using the oracle queries and precomputed interpolation weights
14: end for
15: Construct estimate of gradient using interpolators as shown in (30)
Step 3b: Standard GD Update
16: Update: ⊳ See (31)
17: end for
18: return
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 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 , 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
(
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 and all other problem parameters are constant.
As in Appendix A, let be the unique global minimizer of the empirical risk in (3) so that . 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
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
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
Note that the expressions in (37) and (38) do not depend on 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 are all functions of the number of training samples n and study the asymptotics as . 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.
(
This implies the following results:
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 ). In fact, under the above assumptions, the LPI-GD algorithm beats any 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 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 are all functions of the number of training samples n and study the asymptotics as .
(
This implies the following results:
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 , 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 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 and . As noted in Section 1, 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, , 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 . However, the proof of Proposition 2 in Section 5.1 needs to bound the -norm error between (30) and the true gradient for . Translating the coordinate-wise guarantee to an -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 , 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 . 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 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 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.
(
We break this proof into several claims for clarity. For convenience in the sequel, we begin by establishing some simple facts about in Claim 1 and Claim 2. Specifically, Claim 1 shows that is positive definite.
(
Following the proof of Tsybakov (2009, lemma 1.4) mutatis mutandis, note that is positive semidefinite:
Suppose for the sake of contradiction that there exists a nonzero vector such that
Then, we must have almost everywhere (with respect to the Lebesgue measure) on . However, because the nonzero polynomial 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 for all nonzero vectors . Thus, 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 in Claim 2.
(
Because the rows and columns of are indexed by , for any with , we have
Thus, Claim 1 and Claim 2 demonstrate that is a doubly nonnegative matrix with all rational entries. To establish Lemma 1, we need to lower bound the strictly positive minimum eigenvalue of . To this end, using Claim 2, we first upper bound the trace of in Claim 3, compute the determinant of in Claim 4, and then lower bound the determinant of in Claim 5.
(
To upper bound , observe that
Note that the exponential scaling of in Claim 3 is tight, because . To lower bound , we will first compute it exactly in the next claim.
(
To prove this, we introduce the Legendre polynomials, , which are defined by Rodrigues’ formula (Chihara 2011, chapter V, equation (2.1)):
Moreover, for each , the leading coefficient (of tk) of is known to be (Chihara 2011, chapter V, equation (2.7)):
Next, for any , define the Legendre product polynomial:
Now, notice that
We next establish a lower bound on the determinant of using Claim 4.
(
Observe from Claim 4 that we can write
The hypercube maximizes the volume over all d-orthotopes with an isoperimetric constraint:6
where and the maximum is achieved by ;For any and any with , the vector majorizes the vector s, and because the map is convex, Karamata’s majorization inequality yields (cf. Marshall et al. 2011, chapter 1, section A)
which implies thatusing the monotonicity of the exponential function;and the ninth inequality follows from the simple bound:
which uses the fact that . □
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):
We make three pertinent remarks at this point. Firstly, we note that when for some constant , one can verify that
Hence, scales double exponentially in d:
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 , 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.
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 , for all , and for all with , we have
Then, applying the Courant-Fischer-Weyl min-max theorem (Horn and Johnson 2013, theorem 4.2.6), we obtain
We proceed to lower bounding . Fix any , and any with , and consider the Riemann sum
Because , it is easy to verify that the points partition the hypercube into a grid of d-cells with side lengths and volume each, where Iz is the d-cell with tag z. Moreover, because the polynomial function is Riemann integrable, if with all other problem parameters fixed, then the above Riemann sum converges to an entry of the matrix defined in Lemma 1 (cf. Apostol 1974, section 14.3):
Therefore, we have shown that when . 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 and any fixed , notice that for every ,
Hence, via repeated application of the triangle inequality, we have
Now, observe that for all with ,
Then, combining (55) and (56) produces the lower bound
Finally, because m and h satisfy , or, equivalently,
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).
(
We follow the proof of Tsybakov (2009, proposition 1.12) mutatis mutandis. Fix any and such that B(x) is positive definite, and define the vector as
Then, for every , we have
The second lemma generalizes Tsybakov (2009, lemma 1.3) to the d-variate setting and portrays an -norm bound on the sequence of interpolation weights in (23).
(
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 , we have
4.3. Proof of Theorem 1
Finally, we are in a position to prove the supremum norm interpolation guarantee in Theorem 1.
Suppose that and are such that . Then, observe that for every , we have
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 2–4 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 , 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.
(
We remark that Friedlander and Schmidt (2012, lemma 2.1) additionally assumes that the infimum in (4) can be achieved. However, this condition need not be imposed in Lemma 4, because if F is strongly convex on , its infimum is finite and achieved by some unique global minimizer . We next prove Proposition 2 using Lemma 4 and Theorem 1.
We first verify that the conditions of Lemma 4 hold. Recall the ERM objective function given in (3):
Next, we apply Lemma 4 and get
We now select appropriate values for the constants . For every , choose
Finally, to ensure that is an ϵ-approximate solution of (4), viz.
By taking logarithms of both sides, and rearranging and simplifying the resulting inequality, we obtain the equivalent inequality
5.2. Proof of Theorem 2
We next establish the oracle complexity bound in Theorem 2 using Proposition 2.
Because we choose the maximum allowable approximation errors to be constant with respect to t as in (33), a single common uniform grid 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
We remark that the proofs of Proposition 2 and Theorem 2 can be combined to determine the optimal t-dependent approximation error parameters and number of iterations such that 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.
We first establish the scaling of in (34). Because , observe that
Hence, because , we obtain
Finally, from the relations (37) and (38) and the assumptions that and , we get and . Therefore, for some constants , we obtain
5.4. Proof of Proposition 4
Lastly, we derive Proposition 4 from Theorem 2.
This proof closely follows the argument for Proposition 3. Under the assumptions of the proposition statement, for some positive constant and all sufficiently large , we have from (64) that
Next, from the existing first-order oracle complexity results in Propositions A.3 and A.4 in Appendix A, we define the upper bounds:
Using the assumptions and , we obtain that and . Therefore, as in the proof of Proposition 3, we have
Furthermore, we also obtain the stronger results
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 -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 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.
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 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 in (3) is μ-strongly convex and has L1-Lipschitz continuous gradient. Let be the unique global minimizer of the empirical risk so that and (which is the zero vector). Furthermore, assume that the parameter θ belongs to an -ball with constant radius—for example, —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 -diameter of such an -ball scales as , which means that the -distance between any two parameter vectors is . 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 follows from the facts that: (1) Each iteration of GD makes n first-order oracle queries, and (2) it takes 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).
(
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 is obtained as follows:
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 —for example, —and number of (variance reduced) SGD iterations per epoch proportional to the condition number σ—for example, (Johnson and Zhang 2013, theorem 1).
(
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 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: or . The latter regime reduces to a GD-like setting, so we only consider the 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).
(
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 , and 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 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 is L1-Lipschitz continuous, it can be shown that each iteration of GD reduces the objective value:
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 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 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 -times differentiable function with Lipschitz continuous 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 -regularized linear regression, and Section B.2 provides an example of -regularized logistic regression.
B.1. Example: Ridge Regression
In this subsection, we consider -regularized linear regression or ridge regression. Let the parameter space be so that any parameter vector , and each labeled training data sample . The compactness of the parameter space is reasonable because training usually takes place in a compact subset of the Euclidean space . Consider the -regularized squared loss function:
Smoothness in parameter: Observe that , which means that for all x, y and all ,
(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 . Hence, the Lipschitz constant .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 in (34). When the condition number ,
(B.5)which implies that grows linearly in σ. Hence, when as in this example, and, consequently, the condition number picks up an additional O(d) factor. However, this does not change the result in Proposition 3, as (see the proof in Section 5.3).Smoothness in data: Observe that for any fixed and any , the ith partial derivative . Let us arbitrarily choose η = 2 for the purposes of illustration. Then, for any with a value of one at the jth position and , we have for all and ,
(B.6)Hence, for all and , we obtain
(B.7)where the second and third cases follow from direct calculation and the fact that , and the first case holds because(B.8)where the second line follows from the triangle inequality and the fact that , and the third line follows from Hölder’s inequality and the fact that . This implies that for all and , we have(B.9)Thus, in this example, and each belongs to a (2,4)-Hölder class.
Strong convexity: It is straightforward to see that the (unregularized) squared loss function is convex for all x, y (as it is quadratic). This implies that is μ-strongly convex for all x, y (Bubeck 2015, section 3.4).
B.2. Example: Regularized Logistic Regression
In this subsection, we consider -regularized logistic regression with soft labels. As before, let the parameter space be , so that any parameter vector and each labeled training data sample 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 -regularized cross-entropy loss function:
Smoothness in parameter: Observe that , which means that for all x, y and all ,
(B.12)where the first inequality follows from the triangle inequality, the second inequality holds because , the third inequality holds due to the Lipschitz continuity of the sigmoid function and the fact that , the fourth inequality follows from the Cauchy-Schwarz inequality, and the final inequality follows from the bound . Hence, the Lipschitz constant . (As explained earlier, such dependence of L1 on d does not affect our analysis in this work.)Smoothness in data: Observe that for any fixed and any , the ith partial derivative . Once again, let us arbitrarily choose η = 2 for the purposes of illustration. Then, for any with a value of one at the jth position and , we have for all and ,
(B.13)Hence, for all and , we obtain
(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(B.15)where the second line follows from the fact that , the third line follows from the triangle inequality and the fact that , the fourth line holds because for all , the fifth line follows from Lipschitz continuity and the fact that , and the sixth line follows from Hölder’s inequality and the fact that . This implies that for all and , we have(B.16)Thus, in this example, and each belongs to a (2,1)-Hölder class.
Strong convexity: It is well-known that the (unregularized) cross-entropy loss function is convex for all x, y (Hastie et al. 2009, section 4.4). This can be directly checked by computing derivatives. This implies that is μ-strongly convex for all x, y (Bubeck 2015, section 3.4).
Appendix C. Conjecture on Scaling of Minimum Eigenvalue
Let 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.
(
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 is upper bounded by the minimum diagonal entry of as follows:
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 , 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 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 for all . 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 and such that , if , then .
References
- (2021) On robustness of principal component regression. J. Amer. Statist. Assoc. 116(536):1731–1745.Google Scholar
- (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
- (1974) Mathematical Analysis, 2nd ed. (Addison-Wesley, Reading, MA).Google Scholar
- (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
- (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
- (2006) Pattern Recognition and Machine Learning, Information Science and Statistics (Springer, New York).Google Scholar
- (2018) Optimization methods for large-scale machine learning. SIAM Rev. 60(2):223–311.Google Scholar
- (2015) Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning, vol. 8 (Now Publishers Inc., Hanover, MA).Google Scholar
- (2021) Lower bounds for finding stationary points II: First-order methods. Math. Program. Ser. A 185:315–355.Google Scholar
- (2015) Matrix estimation by universal singular value thresholding. Ann. Statist. 43(1):177–214.Google Scholar
- (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
- (2011) An Introduction to Orthogonal Polynomials (Dover, New York).Google Scholar
- (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
- (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
- (2006) Diffusion maps. Appl. Comput. Harmonic Anal. 21(1):5–30.Google Scholar
- (1967) Nearest neighbor pattern classification. IEEE Trans. Inform. Theory IT-13(1):21–27.Google Scholar
- (2020) Active subspace of neural networks: Structural analysis and universal attacks. SIAM J. Math. Data Sci. 2(4):1096–1122.Google Scholar
- (2003) An elementary proof of a theorem of Johnson and Lindenstrauss. Random Structures Algorithms 22(1):60–65.Google Scholar
- (2013) Derivative estimation with local polynomial fitting. J. Machine Learn. Res. 14(1):281–301.Google Scholar
- (2012) Optimal distributed online prediction using mini-batches. J. Machine Learn. Res. 13(6):165–202.Google Scholar
- (1996) Nonparametric estimation of a regression function and its derivatives under an ergodic hypothesis. J. Nonparametric Statist. 6(4):367–382.Google Scholar
- (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
- (1986) Computationally efficient bounds for the Catalan numbers. Eur. J. Combin. 7(3):211–213.Google Scholar
- (1996) Local Polynomial Modeling and Its Applications, Monographs on Statistics and Applied Probability, vol. 66 (Chapman and Hall, London).Google Scholar
- (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
- (1968) An Introduction to Probability Theory and Its Applications, vol. 1, 3rd ed. (John Wiley & Sons, Inc., New York).Google Scholar
- (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
- (2011) Statistical Distributions, 4th ed. (John Wiley & Sons, Inc., Hoboken, NJ).Google Scholar
- (2012) Hybrid deterministic-stochastic methods for data fitting. SIAM J. Sci. Comput. 34(3):A1380–A1405.Google Scholar
- (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
- (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
- (2018) Gradient descent happens in a tiny subspace. Preprint, submitted December 12, https://arxiv.org/abs/1812.04754.Google Scholar
- (2009) The Elements of Statistical Learning: Data Mining, Inference, and Prediction, 2nd ed., Springer Series in Statistics (Springer, New York).Google Scholar
- (2013) Matrix Analysis, 2nd ed. (Cambridge University Press, New York).Google Scholar
- (1933) Analysis of a complex of statistical variables into principal components. J. Ed. Psych. 24(6):417–441, 498–520.Google Scholar
- (1936) Relations between two sets of variates. Biometrika 28(3/4):321–377.Google Scholar
- (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
- (2022) Limitations of neural collapse for understanding generalization in deep learning. Preprint, submitted February 17, https://arxiv.org/abs/2202.08384.Google Scholar
- (2023) Federated optimization of smooth loss functions. IEEE Trans. Inform. Theory 69(12):7836–7866.Google Scholar
- (2017) Non-Convex Optimization for Machine Learning, Foundations and Trends in Machine Learning, vol. 10 (Now Publishers Inc., Hanover, MA).Google Scholar
- (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
- (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
- (1988) Fourier Analysis (Cambridge University Press, Cambridge, UK).Google Scholar
- (2002) A Primer of Real Analytic Functions, 2nd ed., Birkhäuser Advanced Texts: Basler Lehrbücher (Birkhäuser, Boston).Google Scholar
- (2022) Training invariances and the low-rank phenomenon: Beyond linear networks. Proc. Tenth Internat. Conf. Learn. Representations (ICLR, Appleton, WI), 1–26.Google Scholar
- (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
- (2019) Information contraction and decomposition. ScD thesis in Electrical Engineering and Computer Science, Massachusetts Institute of Technology, Cambridge, MA.Google Scholar
- (2011) Inequalities: Theory of Majorization and Its Applications, 2nd ed., Springer Series in Statistics (Springer, New York).Google Scholar
- (1996) Multivariate regression estimation local polynomial fitting for time series. Stochastic Processes Appl. 65(1):81–101.Google Scholar
- (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
- (1997) Bounds for eigenvalues using the trace and determinant. Linear Algebra Appl. 264:101–108.Google Scholar
- (2006) Estimation of gradients and coordinate covariation in classification. J. Machine Learn. Res. 7(88):2481–2514.Google Scholar
- (2006) Learning coordinate covariances via gradients. J. Machine Learn. Res. 7(18):519–549.Google Scholar
- (1964) On estimating regression. Theory Probab. Appl. 9(1):141–142.Google Scholar
- (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
- (2015)
Distributed optimization . Baillieul J, Samad T, eds. Encyclopedia of Systems and Control (Springer, London), 308–317.Google Scholar - (2001) Nonlinear System Identification: From Classical Approaches to Neural Networks and Fuzzy Models (Springer, Berlin, Heidelberg).Google Scholar
- (2009) Robust stochastic approximation approach to stochastic programming. SIAM J. Optim. 19(4):1574–1609.Google Scholar
- (1983) Problem Complexity and Method Efficiency in Optimization, Wiley-Interscience Series in Discrete Mathematics and Optimization (John Wiley & Sons Inc., New York).Google Scholar
- (1983) A method of solving a convex programming problem with convergence rate . Dokl. Akad. Nauk SSSR 269(3):543–547.Google Scholar
- (2004) Introductory Lectures on Convex Optimization: A Basic Course, Applied Optimization, vol. 87 (Springer, New York).Google Scholar
- (2019) Stochastic gradient descent and its variants in machine learning. J. Indian Inst. Sci. 99(2):201–213.Google Scholar
- (2011) Learning classification with auxiliary probabilistic information. Proc. IEEE Internat. Conf. Data Mining ICDM (IEEE, Piscataway, NJ), 477–486.Google Scholar
- (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
- (2020) Prevalence of neural collapse during the terminal phase of deep learning training. Proc. Natl. Acad. Sci. USA 117(40):24652–24663.Google Scholar
- (1901) On lines and planes of closest fit to systems of points in space. Philos. Mag. 2(11):559–572.Google Scholar
- (2020) Theoretical issues in deep networks. Proc. Natl. Acad. Sci. USA 117(48):30039–30045.Google Scholar
- (1964) Some methods of speeding up the convergence of iteration methods. USSR Comput. Math. Math. Phys. 4(5):1–17.Google Scholar
- (1976) Principles of Mathematical Analysis, 3rd ed., International Series in Pure and Applied Mathematics (McGraw Hill, New York).Google Scholar
- (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
- (2017) Minimizing finite sums with the stochastic average gradient. Math. Program. Ser. A 162:83–112.Google Scholar
- (2013) Stochastic dual coordinate ascent methods for regularized loss minimization. J. Mach. Learn. Res. 14:567–599.Google Scholar
- (2021) Analytic insights into structure and rank of neural network Hessian maps. Proc. Adv. Neural Inform. Processing Systems 34 NeurIPS, 23914–23927.Google Scholar
- (2017) Non-asymptotic convergence analysis of inexact gradient methods for machine learning without strong convexity. Optim. Methods Softw. 32(4):963–992.Google Scholar
- (2003a) Complex Analysis, Princeton Lectures in Analysis, vol. 2 (Princeton University Press, Princeton, NJ).Google Scholar
- (2003b) Fourier Analysis: An Introduction, Princeton Lectures in Analysis, vol. 1 (Princeton University Press, Princeton, NJ).Google Scholar
- (1977) Consistent nonparametric regression. Ann. Statist. 5(4):595–620.Google Scholar
- (2009) Introduction to Nonparametric Estimation, Springer Series in Statistics (Springer, New York).Google Scholar
- (2018) Stochastic zeroth-order optimization in high dimensions. Proc. 21st Internat. Conf. Artificial Intelligence Statist. AISTATS (PMLR, New York), 1356–1365.Google Scholar
- (2019) Statistical methods for machine learning. Lecture Notes 36–708, Department of Statistics and Data Science, Carnegie Mellon University, Pittsburgh.Google Scholar
- (1964) Smooth regression analysis. Sankhya Ser. A 26(4):359–372.Google Scholar
- (2021) Dissecting Hessian: Understanding common structure of Hessian in neural networks. Preprint, submitted June 16, https://arxiv.org/abs/2010.04261v5.Google Scholar
- (2018) Rates of convergence of spectral methods for graphon estimation. Proc. 35th Internat. Conf. Machine Learn. ICML (PMLR, New York), 5433–5442.Google Scholar
- (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
- (2000) On derivative estimation in spline regression. Statist. Sinica 10(1):93–108.Google Scholar

