Many machine learning problems can be formulated as empirical risk minimization, which is in the form of finite-sum optimization as follows:

where each can be a convex or nonconvex function. In this paper, we are particularly interested in nonconvex finite-sum optimization, where each is nonconvex. This is often the case for deep learning (LeCun et al., 2015). In principle, it is hard to find the global minimum of (1.1) because of the NP-hardness of the problem (Hillar and Lim, 2013), thus it is reasonable to resort to finding local minima (a.k.a., second-order stationary points). It has been shown that local minima can be the global minima in certain machine learning problems, such as low-rank matrix factorization (Ge et al., 2016; Bhojanapalli et al., 2016; Zhang et al., 2018b) and training deep linear neural networks (Kawaguchi, 2016; Hardt and Ma, 2016). Therefore, developing algorithms to find local minima is important both in theory and in practice. More specifically, we define an ()-approximate local minimum x of F(x) as follows

where 0 are predefined precision parameters. The most classic algorithm to find the approximate local minimum is cubic-regularized (CR) Newton method, which was originally proposed in the seminal paper by Nesterov and Polyak (2006). Generally speaking, in the k-th iteration, cubic regularization method solves a subproblem, which minimizes a cubic-regularized second-order Taylor expansion at the current iterate . The update rule can be written as follows:

where M > 0 is a penalty parameter. Nesterov and Polyak (2006) proved that to find an ()-approximate local minimum of a nonconvex function F, cubic regularization requires at most iterations. However, when applying cubic regularization to nonconvex finite-sum optimization in (1.1), a major bottleneck of cubic regularization is that it needs to compute n individual gradients ) and Hessian matrices ) at each iteration, which leads to a total complexity (i.e., number of queries to the stochastic gradient oracle ) Hessian complexity (i.e., number of queries to the stochastic Hessian oracle some i and x). Such computational overhead will be extremely expensive when n is large as is in many large-scale machine learning applications.

To overcome the aforementioned computational burden of cubic regularization, Kohler and Lucchi (2017); Xu et al. (2017) used subsampled gradient and subsampled Hessian, which achieve ) gradient complexity and ) Hessian complexity. Zhou et al. (2018d) proposed a stochastic variance reduced cubic regularization method (SVRC), which uses novel semi-stochastic gradient and semi-stochastic Hessian estimators inspired by variance reduction for first-order finite-sum optimization (Johnson and Zhang, 2013; Reddi et al., 2016; Allen-Zhu and Hazan, 2016), which attains ) Second-order Oracle (SO) complexity. Zhou et al. (2018b); Wang et al. (2018b); Zhang et al. (2018a) used a simpler semi-stochastic gradient compared with (Zhou et al., 2018d), and semi-stochastic Hessian, which achieves a better Hessian complexity, i.e., ). However, it is unclear whether the gradient and Hessian complexities of the aforementioned SVRC algorithms can be further improved. Furthermore, all these algorithms need to use the semi-stochastic Hessian estimator, which is not compatible with Hessian-vector product-based cubic subproblem solvers (Agarwal et al., 2017; Carmon and Duchi, 2016, 2018). Therefore, the cubic subproblem (1.4) in each iteration of existing SVRC algorithms has to be solved by computing the inverse of the Hessian matrix, whose computational complexity is at least This makes existing SVRC algorithms not very practical for high-dimensional problems.

In this paper, we first show that the gradient and Hessian complexities of SVRC-type algorithms can be further improved. The core idea is to use novel recursively updated semi-stochastic gradient and Hessian estimators, which are inspired by the stochastic path-integrated differential estimator (SPIDER) (Fang et al., 2018) and the StochAstic Recursive grAdient algoritHm (SARAH) (Nguyen et al., 2017) for first-order optimization. We show that such kind of estimators can be extended to second-order optimization to reduce the Hessian complexity. Nevertheless, our analysis is very different from that in Fang et al. (2018); Nguyen et al. (2017), because we study a fundamentally different optimization problem (i.e., finding local minima against finding first-order stationary points) and a completely different optimization algorithm (i.e., cubic regularization versus gradient method). In addition, in order to reduce the runtime complexity of existing SVRC algorithms, we further propose a Hessian-free SVRC method that can not only use the novel semi-stochastic gradient estimator, but also leverage the Hessian-vector product-based fast cubic subproblem solvers. Experiments on benchmark nonconvex finite-sum optimization problems illustrate the superiority of our newly proposed SVRC algorithms over the state-of-the-art (Due to space limit, we include the experiments in Appendix 6).

In detail, our contributions are summarized as follows:

1. We propose a new SVRC algorithm, namely SRVRC, which can find an ()-approximate local minimum with ) gradient complexity and + ) Hessian complexity. Compared with previous work in cubic regularization, the gradient and Hessian complexity of SRVRC is strictly better than the algorithms in Zhou et al. (2018b); Wang et al. (2018b); Zhang et al. (2018a), and better than that in Zhou et al. (2018d); Shen et al. (2019) in a wide regime.

2. We further propose a new algorithm SRVRC, which requires ) stochastic gra-dient and Hessian-vector product computations to find an ()-approximate local minimum. SRVRCis strictly better than the algorithms in (Agarwal et al., 2017; Carmon and Duchi, 2016; Tripuraneni et al., 2018) when 1. The runtime complexity of SRVRCis also better than that of SRVRC when the problem dimension d is large.

In an independent and concurrent work (Shen et al., 2019), two stochastic trust region methods namely STR1 and STR2 were proposed, which are based on the same idea of variance reduction using SPIDER, and are related to our first algorithm SRVRC. Our SRVRC is better than STR1 because it enjoys the same Hessian complexity but a better gradient complexity than STR1. Compared with STR2, our SRVRC has a consistently lower Hessian complexity and lower gradient complexity in a wide regime (i.e., ). Since Hessian complexity is the dominating term in cubic regularization method (Zhou et al., 2018b; Wang et al., 2018b), our SRVRC is arguably better than STR2, as verified by our experiments.

For the ease of comparison, we summarize the comparison of methods which need to compute the Hessian explicitly in Table 1, the Hessian-free or Hessian-vector product based methods in Table 2.

Table 1: Comparisons of different methods to find an ()-local minimum on gradient and Hessian complexity.

In this section, we review additional related work that is not discussed in the introduction section. Cubic Regularization and Trust-Region Methods Since cubic regularization was first proposed by Nesterov and Polyak (2006), there has been a line of followup research. It was extended to adaptive regularized cubic methods (ARC) by Cartis et al. (2011a,b), which enjoy the same iteration complexity as standard cubic regularization while having better empirical performance. The first attempt to make cubic regularization a Hessian-free method was done by Carmon and Duchi (2016), which solves the cubic sub-problem by gradient descent, requiring in total ) stochastic gradient and Hessian-vector product computations. Agarwal et al. (2017) solved cubic sub-problem by fast matrix inversion based on accelerated gradient descent, which requires + ) stochastic gradient and Hessian-vector product computations. In the pure stochastic optimization setting, Tripuraneni et al. (2018) proposed stochastic cubic regularization method, which uses subsampled gradient and Hessian-vector product-based cubic subproblem solver, and requires ) stochastic gradient and Hessian-vector product computations. A closely related second-order method to cubic regularization methods are trust-region methods (Conn et al., 2000; Cartis et al., 2009, 2012, 2013). Recent studies (Blanchet et al., 2016; Curtis et al., 2017; Mart´ınez and Raydan, 2017) proved that the trust-region method can achieve the same iteration complexity as the cubic regularization method. Xu et al. (2017) also extended trust-region method to subsampled trust-region method for nonconvex finite-sum optimization.

Local Minima Finding Besides cubic regularization and trust-region type methods, there is

Table 2: Comparisons of different methods to find an ()-local minimum both on stochastic gradient and Hessian-vector product computations.

another line of research for finding approximate local minima, which is based on first-order optimization. Ge et al. (2015); Jin et al. (2017a) proved that (stochastic) gradient methods with additive noise are able to escape from nondegenerate saddle points and find approximate local minima. Carmon et al. (2018); Royer and Wright (2017); Allen-Zhu (2017); Xu et al. (2018); Allen-Zhu and Li (2018); Jin et al. (2017b); Yu et al. (2017, 2018); Zhou et al. (2018a); Fang et al. (2018) showed that by alternating first-order optimization and Hessian-vector product based negative curvature descent, one can find approximate local minima even more efficiently. Very recently, Fang et al. (2019); Jin et al. (2019) showed that stochastic gradient descent itself can escape from saddle points. Variance Reduction Variance reduction techniques play an important role in our proposed algorithms. Variance reduction techniques were first proposed for convex finite-sum optimization, which use semi-stochastic gradient to reduce the variance of the stochastic gradient and improve the gradient complexity. Representative algorithms include Stochastic Average Gradient (SAG) (Roux et al., 2012), Stochastic Variance Reduced Gradient (SVRG) (Johnson and Zhang, 2013; Xiao and Zhang, 2014), SAGA (Defazio et al., 2014) and SARAH (Nguyen et al., 2017), to mention a few. For nonconvex finite-sum optimization problems, Garber and Hazan (2015); Shalev-Shwartz (2016) studied the case where each individual function is nonconvex, but their sum is still (strongly) convex. Reddi et al. (2016); Allen-Zhu and Hazan (2016) extended SVRG to noncovnex finite-sum optimization, which is able to converge to first-order stationary point with better gradient complexity than vanilla gradient descent. Fang et al. (2018); Zhou et al. (2018c); Wang et al. (2018a); Nguyen et al. (2019) further improved the gradient complexity for nonconvex finite-sum optimization to be (near) optimal.

and the minimal function value is bounded.

Assumption 3.1. For any function F(x) and an initial point , there exists a constant 0

We also need the following L-gradient Lipschitz and -Hessian Lipschitz assumption.

Assumption 3.2. For each i, we assume that -gradient Lipschitz continuous and Lipschitz continuous, where we have

Note that L-gradient Lipschitz is not required in the original cubic regularization algorithm (Nesterov and Polyak, 2006) and the SVRC algorithm (Zhou et al., 2018d). However, for most other SVRC algorithms (Zhou et al., 2018b; Wang et al., 2018b; Zhang et al., 2018a), they need the L-gradient Lipschitz assumption.

In addition, we need the difference between the stochastic gradient and the full gradient to be bounded.

Assumption 3.3. We assume that F has M-bounded stochastic gradient, where we have

It is worth noting that Assumption 3.3 is weaker than the assumption that each is Lipschitz continuous, which has been made in Kohler and Lucchi (2017); Zhou et al. (2018b); Wang et al. (2018b); Zhang et al. (2018a). We would also like to point out that we can make additional assumptions on the variances of the stochastic gradient and Hessian, such as the ones made in Tripuraneni et al. (2018). Nevertheless, making these additional assumptions does not improve the dependency of the gradient and Hessian complexities or the stochastic gradient and Hessian-vector product computations on . Therefore we chose not making these additional assumptions on the variances.

In this section, we present SRVRC, a novel algorithm which utilizes new semi-stochastic gradient and Hessian estimators compared with previous SVRC algorithms. We also provide a convergence analysis of the proposed algorithm.

4.1 Algorithm Description

In order to reduce the computational complexity for calculating full gradient and full Hessian in (1.3), several ideas such as subsampled/stochastic gradient and Hessian (Kohler and Lucchi, 2017; Xu et al., 2017; Tripuraneni et al., 2018) and variance-reduced semi-stochastic gradient and Hessian (Zhou et al., 2018d; Wang et al., 2018b; Zhang et al., 2018a) have been used in previous work. SRVRC follows this line of work. The key idea is to use a new construction of semi-stochastic gradient and Hessian estimators, which are recursively updated in each iteration, and reset periodically after certain number of iterations (i.e., an epoch). This is inspired by the first-order variance reduction algorithms SPIDER (Fang et al., 2018) and SARAH (Nguyen et al., 2017). SRVRC constructs semi-stochastic gradient and Hessian as in (3.1) and (3.2) respectively. To be more specific, in the t-th iteration when mod() = 0 or mod() = 0, where are the epoch lengths of gradient and Hessian, SRVRC will set the semi-stochastic gradient and Hessian to be a subsampled gradient ) and Hessian , respectively. In the t-th iteration when mod(= 0 or mod(= 0, SRVRC constructs semi-stochastic gradient and Hessian and based on previous estimators and recursively. With semi-stochastic gradient , semi-stochastic Hessian and t-th Cubic penalty parameter , SRVRC constructs the t-th Cubic subproblem and solves for the solution to -th update direction as (3.3). If less than a given threshold which we set it as, SRVRC returns as its output. Otherwise, SRVRC updates and continues the loop.

The main difference between SRVRC and previous stochastic cubic regularization algorithms (Kohler and Lucchi, 2017; Xu et al., 2017; Zhou et al., 2018d,b; Wang et al., 2018b; Zhang et al., 2018a) is that SRVRC adapts new semi-stochastic gradient and semi-stochastic Hessian estimators, which are defined recursively and have smaller asymptotic variance. The use of such semi-stochastic gradient has been proved to help reduce the gradient complexity in first-order nonconvex finite-sum optimization for finding stationary points (Fang et al., 2018; Wang et al., 2018a; Nguyen et al., 2019). Our work takes one step further to apply it to Hessian, and we will later show that it helps reduce the gradient and Hessian complexities in second-order nonconvex finite-sum optimization for finding local minima.

4.2 Convergence Analysis

In this subsection, we present our theoretical results about SRVRC. While the idea of using variance reduction technique for cubic regularization is hardly new, the new semi-stochastic gradient and Hessian estimators in (3.1) and (3.2) bring new technical challenges in the convergence analysis.

To describe whether a point x is a local minimum, we follow the original cubic regularization work (Nesterov and Polyak, 2006) to use the following criterion

It is easy to note that if and only if x is an ()-approximate local minimum. Thus, in order to find an ()-approximate local minimum, it suffices to find a point x which satisfies

The following theorem provides the convergence guarantee of SRVRC for finding an ()-approximate local minimum.

Theorem 4.2. Under Assumptions 3.1, 3.2 and 3.3, set the cubic penalty parameter = 4for any t and the total iteration number 40∆. For t such that mod(= 0 or mod(= 0, set the gradient sample size and Hessian sample size

For ) = 0, set the gradient sample size and Hessian sample size

Then with probability at least 1 , SRVRC outputs satisfying 600, i.e., an ()-approximate local minimum.

Next corollary spells out the exact gradient complexity and Hessian complexity of SRVRC to find an ()-approximate local minimum.

Corollary 4.3. Under the same conditions as Theorem 4.2, if we set as = as their lower bounds in (4.1)- (4.4), then with probability at least 1 , SRVRC will output an ()-approximate local minimum within

stochastic Hessian evaluations and

stochastic gradient evaluations.

Remark 4.4. For SRVRC, if we assume to be constants, then its gradient complexity is ), and its Hessian complexity is ). Regarding Hessian complexity, suppose that 1, then the Hessian complexity of SRVRC can be simplified as ). Compared with existing SVRC algorithms (Zhou et al., 2018b; Zhang et al., 2018a; Wang et al., 2018b), SRVRC outperforms the best-known Hessian sample complexity by a factor of . In terms of gradient complexity, SRVRC outperforms STR2 (Shen et al., 2019) by a factor of

Remark 4.5. Note that both Theorem 4.2 and Corollary 4.3 still hold when Assumption 3.3 does not hold. In that case, and SRVRC’s Hessian complexity remains the same, while its gradient complexity can be potentially worse, i.e., ), which degenerates to that of STR1 (Shen et al., 2019).

While SRVRC adapts novel semi-stochastic gradient and Hessian estimators to reduce both the gradient and Hessian complexities, it has three limitations for high-dimensional problems with (1) it needs to compute and store the Hessian matrix, which needs ) computational time and storage space; (2) it needs to solve cubic subproblem exactly, which requires ) computational time because it needs to compute the inverse of a Hessian matrix (Nesterov and Polyak, 2006); and (3) it cannot leverage the Hessian-vector product-based cubic subproblem solvers (Agarwal et al., 2017; Carmon and Duchi, 2016, 2018) because of the use of the semi-stochastic Hessian estimator. It is interesting to ask whether we can modify SRVRC to overcome these shortcomings.

5.1 Algorithm Description

We present a Hessian-free algorithm SRVRCto address above limitations of SRVRC for high-dimensional problems, which only requires stochastic gradient and Hessian-vector product computations. SRVRCuses the same semi-stochastic gradient as SRVRC. As opposed to SRVRC which has to construct semi-stochastic Hessian explicitly, SRVRConly accesses to stochastic

Hessian-vector product. In detail, at each iteration t, SRVRCsubsamples an index set and defines a stochastic Hessian-vector product function as follows:

Note that although the subproblem depends on never explicitly computes this matrix. Instead, it only provides the subproblem solver access to through stochastic Hessian-vector product function ]. The subproblem solver performs gradient-based optimization to solve the subproblem ) as ) depends on only via ]. In detail, following Tripuraneni et al. (2018), SRVRCuses Cubic-Subsolver (See Algorithms 3 and 4 in Appendix G) and Cubic-Finalsolver from (Carmon and Duchi, 2016), to find an approximate solution to the cubic subproblem in (3.3). Both Cubic-Subsolver and Cubic-Finalsolver only need to access gradient and Hessian-vector product function ] along with other problem-dependent parameters. With the output from Cubic-Subsolver, SRVRCdecides either to update as or to exit the loop. For the later case, SRVRCwill call Cubic-Finalsolver to output , and takes as its final output.

The main differences between SRVRC and SRVRCare two-fold. First, SRVRConly needs to compute stochastic gradient and Hessian-vector product. Since both of these two computations only take O(d) time in many applications in machine learning, SRVRCis suitable for high-dimensional problems. In the sequel, following Agarwal et al. (2017); Carmon et al. (2018); Tripuraneni et al. (2018), we do not distinguish stochastic gradient and Hessian-vector product computations and consider them to have the same runtime complexity. Second, instead of solving cubic subproblem exactly, SRVRCadopts approximate subproblem solver Cubic-Subsolver and Cubic-Finalsolver, both of which only need to access gradient and Hessian-vector product function, and again only take O(d) time. Thus, SRVRCis computational more efficient than SRVRC when

5.2 Convergence Analysis

We now provide the convergence guarantee of SRVRC, which ensures that SRVRCwill output an ()-approximate local minimum.

Theorem 5.1. Under Assumptions 3.1, 3.2, 3.3, suppose ). Set the cubic penalty parameter = 4for any t and the total iteration number 25∆. Set the Hessianvector product sample size

For t such that mod(= 0, set the gradient sample size

For t such that mod() = 0, set the gradient sample size

Then with probability at least 1 satisfying ()-approximate local minimum.

The following corollary calculates the total amount of stochastic gradient and Hessian-vector product computations of SRVRCto find an ()-approximate local minimum.

Corollary 5.2. Under the same conditions as Theorem 5.1, if set = and set as their lower bounds in (5.1)-(5.3), then with probability at least 1 , SRVRCwill output an ()-approximate local minimum within

stochastic gradient and Hessian-vector product computations.

, if we assume are constants, then (5.4) is For stochastic algorithms, the regime is of most interest. In this regime, (5.4) becomes ). Compared with other local minimum finding algorithms based on stochastic gradient and Hessian-vector product, SRVRCoutperforms the results achieved by Tripuraneni et al. (2018) and Allen-Zhu (2018) by a factor of also matches the best-known result achieved by a recent first-order algorithm proposed in (Fang et al., 2018). Note that the algorithm proposed by Fang et al. (2018) needs to alternate the first-order finite-sum optimization algorithm SPIDER and negative curvature descent. In sharp contrast, SRVRCis a pure cubic regularization type algorithm and does not need to calculate the negative curvature direction.

Remark 5.4. It is worth noting that both Theorem 5.1 and Corollary 5.2 still hold when Assumption 3.3 does not hold, and SRVRC’s runtime complexity remains the same. The only difference is: without Assumption 3.3, we need to use full gradient (i.e., ) instead of subsampled gradient at each iteration t.

5.3 Discussions on runtime complexity

We would like to further compare the runtime complexity between SRVRC and SRVRC. In specific, SRVRC needs O(d) time to construct semi-stochastic gradient and ) time to construct semi-stochastic Hessian. SRVRC also needs ) time to solve cubic subproblem for each iteration. Thus, with the fact that the total number of iterations is ) by Corollary 4.3, SRVRC needs

runtime to find an ()-approximate local minimum if we regard as constants. As we mentioned before, for many machine learning problems, both stochastic gradient and Hessian-vector product computations only need O(d) time, therefore the runtime of SRVRCWe conclude that SRVRCoutperforms SRVRC when d is large, which is in accordance with the fact that Hessian-free methods are superior for high dimension machine learning tasks. On the other hand, a careful calculation can show that the runtime of SRVRC can be less than that of SRVRCis moderately small. This is also reflected in our experiments in Section 6.

In this section, we present numerical experiments on different nonconvex empirical risk minimization (ERM) problems and on different datasets to validate the advantage of our proposed SRVRC and SRVRCalgorithms for finding approximate local minima.

Baselines: We compare our algorithms with the following algorithms: SPIDER+ (Fang et al., 2018), which is the local minimum finding version of SPIDER, stochastic trust region (STR1, STR2) (Shen et al., 2019), subsampled cubic regularization (SCR) (Kohler and Lucchi, 2017), stochastic cubic regularization (STC) (Tripuraneni et al., 2018), stochastic variance-reduced cubic regularization (SVRC) (Zhou et al., 2018d), sample efficient SVRC (Lite-SVRC) (Zhou et al., 2018b; Wang et al., 2018b; Zhang et al., 2018a).

Parameter Settings and Subproblem Solver For each algorithm, we set the cubic penalty parameter adaptively based on how well the model approximates the real objective as suggested in (Cartis et al., 2011a,b; Kohler and Lucchi, 2017). For SRVRC, we set = = S for the

Figure 1: Plots of logarithmic function value gap with respect to CPU time (in seconds) for nonconvex regularized binary logistic regression on (a) a9a (b) ovtype (c) ijcnn1 and for nonconvex regularized multiclass logistic regression on (d) mnist (e) cifar10 (f) SVHN. Best viewed in color.

simplicity and set gradient and Hessian batch sizes as follows:

For SRVRC, we set gradient batch sizes the same as SRVRC and Hessian batch sizes over the grid over the grid {n, n/10, n/20, n/100}, and over the grid {50, 100, 500, 1000} for the best performance. For SCR, SVRC, Lite-SVRC, and SRVRC, we solve the cubic subproblem using the cubic subproblem solver discussed in (Nesterov and Polyak, 2006). For STR1 and STR2, we solve the trust-region subproblem using the exact trust-region subproblem solver discussed in (Conn et al., 2000). For STC and SRVRC, we use Cubic-Subsolver (Algorithm 3 in Appendix G) to approximately solve the cubic subproblem. All algorithms are carefully tuned for a fair comparison.

Datasets and Optimization Problems We use 6 datasets a9a, covtype, ijcnn1 , mnist, cifar10 and SVHN from Chang and Lin (2011) . For binary logistic regression problem with a nonconvex regularizer on a9a, covtype, and ijcnn1, we are given training data , where and are feature vector and output label corresponding to the i-th training example. The

nonconvex penalized binary logistic regression is formulated as follows

where ) is the sigmoid function and . For multiclass logistic regression problem with a nonconvex regularizer on mnist, cifar10 and SVHN, we are given training data , where are feature vectors and multilabels corresponding to the i-th data points. The nonconvex penalized multiclass logistic regression is formulated as follows

where softmax() is the softmax function and We plot the logarithmic function value gap with respect to CPU time in Figure 1. From Figure

1(a) to 1(f), we can see that for the low dimension optimization task on a9a, covtype and ijcnn1, our SRVRC outperforms all the other algorithms with respect to CPU time. We can also observe that the stochastic trust region method STR1 is better than STR2, which is well-aligned with our discussion before. The SPIDER+ does not perform as well as other second-order methods, even though its stochastic gradient and Hessian complexity is comparable to second-order methods in theory. Meanwhile, we also notice that SRVRCalways outperforms STC, which suggests that the variance reduction technique is useful. For high dimension optimization task mnist, cifar10 and SVHN, only SPIDER+, STC and SRVRCare able to make notable progress and SRVRCoutperforms the other two. This is again consistent with our theory and discussions in Section 5. Overall, our experiments clearly validate the advantage of SRVRC and SRVRC, and corroborate the theory of both algorithms.

In this work we present two faster SVRC algorithms namely SRVRC and SRVRCto find approximate local minima for nonconvex finite-sum optimization problems. SRVRC outperforms existing SVRC algorithms in terms of gradient and Hessian complexities, while SRVRCoutperforms the best-known runtime complexity for existing CR based algorithms. Whether our algorithms have achieved the optimal complexity under the current assumptions is still an open problem, and we leave it as a future work.

We define the filtration ) as the -algebra of to . Recall that and are the semi-stochastic gradient and Hessian respectively, is the update parameter, and is the cubic penalty parameter appeared in Algorithm 1 and Algorithm 2. We denote ) := ). In this section, we define the simplicity.

A.1 Proof of Theorem 4.2

To prove Theorem 4.2, we need the following lemma adapted from Zhou et al. (2018d), which characterizes that ) can be bounded by and the norm of difference between semi-stochastic gradient and Hessian.

Lemma A.1. Suppose that ) := 2 + 6 and = argmin). If 2, then for any

Next lemma gives bounds on the inner products

We also need the following two lemmas, which show that semi-stochastic gradient and Hessian estimators are good approximations to true gradient and Hessian.

Lemma A.3. Suppose that , then conditioned on probability at least 1 ), we have that for all

Lemma A.4. Suppose that , then conditioned on probability at least 1 ), we have that for all

Given all the above lemmas, we are ready to prove Theorem 4.2.

Proof of Theorem 4.2. Suppose that SRVRC terminates at iteration

all 0 1. We have

where the second inequality holds due to the fact that ) = 0, = 4and Lemma A.2. By Lemmas A.3 and A.4, with probability at least 1 , for all 0 1, we have that

for all 0 1. Substituting (A.4) into (A.3), we have

Telescoping (A.5) from 1, we have

Recall that we have from the condition of Theorem 4.2, then by (A.6), we have . Thus, we have 1, then we have

where the first inequality holds due to Lemma A.1 with ) = 0 and . This completes our proof.

A.2 Proof of Corollary 4.3

Proof of Corollary 4.3. Suppose that SRVRC terminates at 1 iteration. Telescoping (A.5) from 1, we have

where the last inequality holds since T is set to be 40∆as the conditions of Corollary 4.3 suggests. (A.7) implies that . Thus, we have

where the first inequality holds due to H¨older’s inequality inequality, and the second inequality is due to . We first consider the total gradient sample complexity which can be bounded as

where ), the second inequality holds due to (A.8), and the last equality holds due to the choice of . We then consider the total Hessian sample complexity

, which can be bounded as

where ), the second inequality holds due to (A.8), and the last equality holds due to the choice of

In this section, we denote ) for simplicity.

B.1 Proof of Theorem 5.1

We need the following two lemmas, which bound the variance of semi-stochastic gradient and Hessian estimators.

Lemma B.1. Suppose that satisfies (5.2) and (5.3), then conditioned on , with probability at least 1 ), we have that for all

Proof of Lemma B.1. The proof is very similar to that of Lemma A.3, hence we omit it.

Lemma B.2. Suppose that , then conditioned on , with probability at least 1 , we have that

Proof of Lemma B.2. The proof is very similar to that of Lemma A.4, hence we omit it.

We have the following lemma to guarantee that by Algorithm 3 Cubic-Subsolver, the output satisfies that sufficient decrease of function value will be made and the total number of iterations is bounded by

iterations, where 0 is a constant.

We have the following lemma which provides the guarantee for the dynamic of gradient steps in Cubic-Finalsolver.

Lemma B.4. (Carmon and Duchi, 2016) For , suppose that . We denote that

Then for Cubic-Finalsolver, suppose that , then each iterate ∆ in Cubic-Finalsolver satisfies that

With these lemmas, we begin our proof of Theorem 5.1.

Proof of Theorem 5.1. Suppose that SRVRCterminates at iteration first claim that . Otherwise, suppose , then we have that for all 0

where the second inequality holds due to = 4and Lemma A.2. By Lemma B.3 and union bound, we know that with probability at least 1

where we use the fact that = 4. By Lemmas B.1 and B.2, we know that with probability at least 1 , for all 0 1, we have

Substituting (B.2) and (B.3) into (B.1), we have

Telescoping (B.4) from 1, we have

where the last inequality holds since we assume = 25∆from the condition of Theorem 5.1. (B.5) leads to a contradiction, thus we have . Therefore, by union bound, with probability at least 1 , Cubic-Finalsolver is executed by SRVRCat 1 iteration. We have that

The only thing left is to check that we indeed find a second-order stationary point, , by Cubic-Finalsolver. We first need to check that the choice of = 1/(16L) satisfies that 14(

where the first inequality holds due to Lemma A.1, the second inequality holds due to the fact that from Lemma B.4, the last inequality holds due to the facts that ()from Cubic-Finalsolver and

B.2 Proof of Corollary 5.2

We have the following lemma to bound the total number of iterations of Algorithm 4 Cubic-Finalsolver.

Lemma B.5. If , then Cubic-Finalsolver will terminate within iterations, where 0 is a constant.

Proof of Corollary 5.2. We have that

where the first inequality holds due to H¨older’s inequality, the second inequality holds due to the facts that = 25∆and ∆4 by (B.5). We first consider the total stochastic gradient computations, , which can be bounded as

where = 2640 log), the second inequality holds due to (B.6), the last equality holds due to the fact = . We now consider the total amount of Hessian-vector product computations T , which includes from Cubic-Subsolver and from Cubic-Finalsolver. By Lemma B.3, we know that at k-th iteration of SRVRC, Cubic-Subsolver has iterations, which needs Hessian-vector product computations each time. Thus, we have

where ), the first inequality holds due to the fact that second inequality holds due to the fact that , the last inequality holds due to the fact that

where the first inequality holds due to the fact that ), the second inequality holds due to the fact that = by Lemma B.5. Since at each iteration we need Hessian-vector computations.

Combining (B.7), (B.8) and (B.9), we know that the total stochastic gradient and Hessian-vector product computations are bounded as

C.1 Proof of Lemma A.1

We have the following lemmas from Zhou et al. (2018d)

Lemma C.1. (Zhou et al., 2018d) If , then we have

Lemma C.2. (Zhou et al., 2018d) If , then we have

Proof of Lemma A.1. By Lemma C.1, we have

where the second inequality holds due to the fact that for any 0, we have (

2(). By Lemma C.2, we have

where the second inequality holds due to the fact that for any 0, we have (9(). Thus we have

where the inequality holds due to (C.1), (C.2) and the fact that

C.2 Proof of Lemma A.2

Proof of Lemma A.2. We have

where the first inequality holds due to CauchySchwarz inequality, the second inequality holds due to Young’s inequality. We also have

where the first inequality holds due to CauchySchwarz inequality, the second inequality holds due to Young’s inequality.

C.3 Proof of Lemma A.3

We need the following lemma:

Lemma C.3. Conditioned on , with probability at least 1

We also have

Proof of Lemma A.3. First, we have

Meanwhile, we have ] = 0. Conditioned on = 0, from Lemma C.3, we have that with probability at least 1 the following inequality holds :

where the second inequality holds due to (4.1). For mod() = 0, with probability at least 1 we have

where the second inequality holds due to (4.3). Conditioned on , by union bound, with probability at least 1 ) (C.5) or (C.6) holds for all . Then for given k, by vector Azuma-Hoeffding inequality in Lemma F.1, conditioned on, with probability at least 1

Finally, by union bound, we have that with probability at least 1 , we have (C.7) holds.

C.4 Proof of Lemma A.4

We need the following lemma:

Lemma C.4. Conditioned on , with probability at least 1, we have the following concentration inequality

We also have

Proof of Lemma A.4. First, we have

Meanwhile, we have )] = 0. Conditioned on , for mod(= 0, from Lemma C.4, we have that with probability at least 1 , the following inequality holds :

where the second inequality holds due to (4.1). For mod() = 0, with probability at least 1 we have

where the second inequality holds due to (4.3). Conditioned on , by union bound, with probability at least 1 holds for all Then for given k, by Matrix Azuma inequality Lemma F.2, conditioned on, with probability at least 1

Finally, by union bound, we have that with probability at least 1 , we have (C.12) holds.

D.1 Proof of Lemma B.3

We have the following lemma which guarantees the effectiveness of Cubic-Subsolver in Algorithm 3. Lemma D.1. (Carmon and Duchi, 2016) Let and (0, 1)(0, 1) and + 2). We denote that g(h) = 2 + and ). Then with probability at least 1

then x = Cubic-Subsolver() satisfies that

Proof of Lemma B.3. We simply set = (16= 0.5 and , then we set . With the choice of assumption that , we can check that ). We also have that (D.1) holds. Thus, by Lemma D.1, we have

By the choice of in Cubic-Subsolver, we have

D.2 Proof of Lemma B.5

We have the following lemma which provides the guarantee for the function value in Cubic-Finalsolver.

Lemma D.2. (Carmon and Duchi, 2016) We denote that g(h) = 2 + ,

Proof of Lemma B.5. In Cubic-Finalsolver we are focusing on minimizing ). We have that and by Lemma B.3. We can check that = (16satisfies that (4(, where R is defined in Lemma B.4, when . From Lemma B.4 we also know that )-smooth, which satisfies that 1). Thus, by standard gradient descent analysis, to get a point ∆ where , Cubic-Finalsolver needs to run

iterations, where we denote by ∆the starting point of Cubic-Finalsolver. By directly computing, we have 0. By Lemma D.2, we have

Thus, (D.2) can be further bounded as

E.1 Proof of Lemma C.3

Proof of Lemma C.3. We only need to consider the case where . For each , let

then we have i.i.d., and

where the second inequality holds due to the L-smoothness of and F. Thus by vector Azuma-Hoeffding inequality in Lemma F.1, we have that with probability at least 1

For each

then we have . Thus by vector Azuma-Hoeffding inequality in Lemma F.1, we have that with probability at least 1

E.2 Proof of Lemma C.4

Proof of Lemma C.4. We only need to consider the case where

then we have i.i.d. and

where the second inequality holds due to -Hessian Lipschitz continuous of and F. Then by Matrix Azuma inequality Lemma F.2, we have that with probability at least 1

For each

then we have = 0, = , and . Then by Matrix Azuma inequality in Lemma F.2, we have that with probability at least 1

which completes the proof.

We have the following vector Azuma-Hoeffding inequality:

Lemma F.1. (Pinelis, 1994) Consider be a vector-valued martingale difference, where , then we have that with probability at least 1

We have the following Matrix Azuma inequality :

Lemma F.2. (Tropp, 2012) Consider a finite adapted sequence of self-adjoint matrices in dimension d, and a fixed sequence of self-adjoint matrices that satisfy

Due to space limit, we include the approximate solvers (Carmon and Duchi, 2016) for the cubic subproblem in this section for the purpose of self-containedness.

Agarwal, N., Allen-Zhu, Z., Bullins, B., Hazan, E. and Ma, T. (2017). Finding approximate local minima faster than gradient descent. In Proceedings of the 49th Annual ACM SIGACT Symposium on Theory of Computing.

Allen-Zhu, Z. (2017). Natasha 2: Faster non-convex optimization than sgd. arXiv preprint arXiv:1708.08694 .

Allen-Zhu, Z. (2018). Natasha 2: Faster non-convex optimization than sgd. In Advances in Neural Information Processing Systems.

Allen-Zhu, Z. and Hazan, E. (2016). Variance reduction for faster non-convex optimization. In International Conference on Machine Learning.

Allen-Zhu, Z. and Li, Y. (2018). Neon2: Finding local minima via first-order oracles. In Advances in Neural Information Processing Systems.

Bhojanapalli, S., Neyshabur, B. and Srebro, N. (2016). Global optimality of local search for low rank matrix recovery. In Advances in Neural Information Processing Systems.

Blanchet, J., Cartis, C., Menickelly, M. and Scheinberg, K. (2016). Convergence rate analysis of a stochastic trust region method for nonconvex optimization. arXiv preprint arXiv:1609.07428 .

Carmon, Y. and Duchi, J. C. (2016). Gradient descent efficiently finds the cubic-regularized non-convex newton step. arXiv preprint arXiv:1612.00547 .

Carmon, Y. and Duchi, J. C. (2018). Analysis of krylov subspace solutions of regularized non-convex quadratic problems. In Advances in Neural Information Processing Systems.

Carmon, Y., Duchi, J. C., Hinder, O. and Sidford, A. (2018). Accelerated methods for nonconvex optimization. SIAM Journal on Optimization 28 1751–1772.

Cartis, C., Gould, N. I. and Toint, P. L. (2009). Trust-region and other regularisations of linear least-squares problems. BIT Numerical Mathematics 49 21–53.

Cartis, C., Gould, N. I. and Toint, P. L. (2011a). Adaptive cubic regularisation methods for unconstrained optimization. part i: motivation, convergence and numerical results. Mathematical Programming 127 245–295.

Cartis, C., Gould, N. I. and Toint, P. L. (2012). Complexity bounds for second-order optimality in unconstrained optimization. Journal of Complexity 28 93–108.

Cartis, C., Gould, N. I. and Toint, P. L. (2013). On the evaluation complexity of cubic regularization methods for potentially rank-deficient nonlinear least-squares problems and its relevance to constrained nonlinear optimization. SIAM Journal on Optimization 23 1553–1574.

Cartis, C., Gould, N. I. M. and Toint, P. L. (2011b). Adaptive cubic regularisation methods for unconstrained optimization. Part II: worst-case function- and derivative-evaluation complexity. Springer-Verlag New York, Inc.

Chang, C.-C. and Lin, C.-J. (2011). Libsvm: a library for support vector machines. ACM transactions on intelligent systems and technology (TIST) 2 27.

Conn, A. R., Gould, N. I. and Toint, P. L. (2000). Trust region methods. SIAM.

Curtis, F. E., Robinson, D. P. and Samadi, M. (2017). A trust region algorithm with a worst-case iteration complexity of ) for nonconvex optimization. Mathematical Programming 162 1–32.

Defazio, A., Bach, F. and Lacoste-Julien, S. (2014). Saga: A fast incremental gradient method with support for non-strongly convex composite objectives. In Advances in Neural Information Processing Systems.

Fang, C., Li, C. J., Lin, Z. and Zhang, T. (2018). Spider: Near-optimal non-convex optimization via stochastic path integrated differential estimator. arXiv preprint arXiv:1807.01695 .

Fang, C., Lin, Z. and Zhang, T. (2019). Sharp analysis for nonconvex sgd escaping from saddle points. arXiv preprint arXiv:1902.00247 .

Garber, D. and Hazan, E. (2015). Fast and simple pca via convex optimization. arXiv preprint arXiv:1509.05647 .

Ge, R., Huang, F., Jin, C. and Yuan, Y. (2015). Escaping from saddle pointsonline stochastic gradient for tensor decomposition. In Conference on Learning Theory.

Ge, R., Lee, J. D. and Ma, T. (2016). Matrix completion has no spurious local minimum. In Advances in Neural Information Processing Systems.

Golub, G. H. and Van Loan, C. F. (1996). Matrix Computations (3rd Ed.). Johns Hopkins University Press, Baltimore, MD, USA.

Hardt, M. and Ma, T. (2016). Identity matters in deep learning. arXiv preprint arXiv:1611.04231 .

Hillar, C. J. and Lim, L.-H. (2013). Most tensor problems are np-hard. Journal of the ACM (JACM) 60 45.

Jin, C., Ge, R., Netrapalli, P., Kakade, S. M. and Jordan, M. I. (2017a). How to escape saddle points efficiently. arXiv preprint arXiv:1703.00887 .

Jin, C., Netrapalli, P., Ge, R., Kakade, S. M. and Jordan, M. I. (2019). Stochastic gradient descent escapes saddle points efficiently. arXiv preprint arXiv:1902.04811 .

Jin, C., Netrapalli, P. and Jordan, M. I. (2017b). Accelerated gradient descent escapes saddle points faster than gradient descent. arXiv preprint arXiv:1711.10456 .

Johnson, R. and Zhang, T. (2013). Accelerating stochastic gradient descent using predictive variance reduction. In Advances in neural information processing systems.

Kawaguchi, K. (2016). Deep learning without poor local minima. In Advances in Neural Information Processing Systems.

Kohler, J. M. and Lucchi, A. (2017). Sub-sampled cubic regularization for non-convex optimization. arXiv preprint arXiv:1705.05933 .

LeCun, Y., Bengio, Y. and Hinton, G. (2015). Deep learning. Nature 521 436–444.

and Raydan, M. (2017). Cubic-regularization counterpart of a variable-norm trust-region method for unconstrained minimization. Journal of Global Optimization 68 367–385.

Nesterov, Y. and Polyak, B. T. (2006). Cubic regularization of newton method and its global performance. Mathematical Programming 108 177–205.

Nguyen, L. M., Liu, J., Scheinberg, K. and (2017). Sarah: A novel method for machine learning problems using stochastic recursive gradient. In International Conference on Machine Learning.

Nguyen, L. M., van Dijk, M., Phan, D. T., Nguyen, P. H., Weng, T.-W. and Kalagnanam, J. R. (2019). Optimal finite-sum smooth non-convex optimization with sarah. arXiv preprint arXiv:1901.07648 .

Pinelis, I. (1994). Optimum bounds for the distributions of martingales in banach spaces. The Annals of Probability 1679–1706.

Reddi, S. J., Hefny, A., Sra, S., Poczos, B. and Smola, A. (2016). Stochastic variance reduction for nonconvex optimization 314–323.

Roux, N. L., Schmidt, M. and Bach, F. R. (2012). A stochastic gradient method with an exponential convergence rate for finite training sets. In Advances in Neural Information Processing Systems.

Royer, C. W. and Wright, S. J. (2017). Complexity analysis of second-order line-search algorithms for smooth nonconvex optimization. arXiv preprint arXiv:1706.03131 .

Shalev-Shwartz, S. (2016). Sdca without duality, regularization, and individual convexity. In International Conference on Machine Learning.

Shen, Z., Zhou, P., Fang, C. and Ribeiro, A. (2019). A stochastic trust region method for non-convex minimization. arXiv preprint arXiv:1903.01540 .

Tripuraneni, N., Stern, M., Jin, C., Regier, J. and Jordan, M. I. (2018). Stochastic cubic regularization for fast nonconvex optimization. In Advances in Neural Information Processing Systems.

Tropp, J. A. (2012). User-friendly tail bounds for sums of random matrices. Foundations of computational mathematics 12 389–434.

Wang, Z., Ji, K., Zhou, Y., Liang, Y. and Tarokh, V. (2018a). Spiderboost: A class of faster variance-reduced algorithms for nonconvex optimization. arXiv preprint arXiv:1810.10690 .

Wang, Z., Zhou, Y., Liang, Y. and Lan, G. (2018b). Sample complexity of stochastic variance-reduced cubic regularization for nonconvex optimization. arXiv preprint arXiv:1802.07372 .

Xiao, L. and Zhang, T. (2014). A proximal stochastic gradient method with progressive variance reduction. SIAM Journal on Optimization 24 2057–2075.

Xu, P., Roosta-Khorasani, F. and Mahoney, M. W. (2017). Newton-type methods for non-convex optimization under inexact hessian information. arXiv preprint arXiv:1708.07164 .

Xu, Y., Rong, J. and Yang, T. (2018). First-order stochastic algorithms for escaping from saddle points in almost linear time. In Advances in Neural Information Processing Systems.

Yu, Y., Xu, P. and Gu, Q. (2018). Third-order smoothness helps: faster stochastic optimization algorithms for finding local minima. In Advances in Neural Information Processing Systems.

Yu, Y., Zou, D. and Gu, Q. (2017). Saving gradient and negative curvature computations: Finding local minima more efficiently. arXiv preprint arXiv:1712.03950 .

Zhang, J., Xiao, L. and Zhang, S. (2018a). Adaptive stochastic variance reduction for subsampled newton method with cubic regularization. arXiv preprint arXiv:1811.11637 .

Zhang, X., Wang, L., Yu, Y. and Gu, Q. (2018b). A primal-dual analysis of global optimality in nonconvex low-rank matrix recovery. In International conference on machine learning.

Zhou, D., Xu, P. and Gu, Q. (2018a). Finding local minima via stochastic nested variance reduction. arXiv preprint arXiv:1806.08782 .

Zhou, D., Xu, P. and Gu, Q. (2018b). Sample efficient stochastic variance-reduced cubic regularization method. arXiv preprint arXiv:1811.11989 .

Zhou, D., Xu, P. and Gu, Q. (2018c). Stochastic nested variance reduced gradient descent for nonconvex optimization. In Advances in Neural Information Processing Systems.

Zhou, D., Xu, P. and Gu, Q. (2018d). Stochastic variance-reduced cubic regularized Newton methods. In Proceedings of the 35th International Conference on Machine Learning. PMLR, 5990–5999.

Zhou, D., Xu, P. and Gu, Q. (2019). Stochastic variance-reduced cubic regularization methods. Journal of Machine Learning Research 20 1–47.