Convergence Rate Analysis of a Stochastic Trust Region Method via Submartingales

Jose Blanchet, Coralia Cartis, Matt Menickelly, Katya Scheinberg

Introduction

In this paper we aim to solve a stochastic unconstrained, possibly nonconvex, optimization problem

Stochastic optimization methods, in particular stochastic gradient descent (SGD), have recently become the focus of much research in optimization, especially in applications to machine learning domains. This is because in machine learning the objective function of the optimization problem is typically a sum of a (possibly) very large number of terms, each term being the loss function evaluated using one data example. This objective function can also be viewed as an expected loss, in which case it cannot be accurately computed, but can only be evaluated approximately, given a subset of data samples. During the last decade significant theoretical and algorithmic advances were developed for convex optimization problems, such as logistic regression and support vector machines. However, with the recent practical success of deep neural networks and other nonlinear, nonconvex ML models, the focus has shifted to the analysis and development of methods for nonconvex optimization problems. While SGD remains the method of choice in the nonconvex setting for ML applications, theoretical results are weaker than those in the convex case. In particular, little has been achieved in terms of convergence rates. A notable paper is the first to provide convergence rates guarantee of a sort for a randomized stochastic gradient method in nonconvex setting. This method, however, utilizes a carefully chosen step size and a randomized stopping scheme, which are quite different from what is used in practice.

As an alternative to the basic SGD several variance reducing stochastic methods have been proposed recently, such as SAGA , SVRG and SARAH . They enjoy much stronger convergence rates than SGD, and have been extended to nonconvex problems . However, these methods specifically exploit the structure of the ML problems, where the objective function is a sum over a deterministic (if large) set of data. SVRG requires the full gradient of the objective function to be computed on some (but not all) of the iterations. Hence, SVRG, essentially is a hybrid between SGD and the full gradient method applied to a finite sum. The overall convergence rate per number of data accesses for SVRG seems better than those for the other two methods. From the practical perspective, however, SGD has low per-iteration complexity and high number of iterations and thus is not effective in a distributed setting, while each iteration of a full gradient method can be efficiently distributed, reducing the overall wall-clock time. SVRG (as well as SARAH) being a hybrid does not easily fit with either setting because it alternates between cheap stochastic gradient computation, which have to be sequential and expensive full gradient computations, which can be distributed. In other words, the superior theoretical computational complexity of the SVRG does not necessarily reflect its practical performance. Moreover, the assumption that the data set is fixed (deterministic) contradicts the ultimate goal of learning, which is to obtain a solution with good generalization performance. The method we describe in this paper is applicable in the purely stochastic setting (without assuming that there is a fixed finite set of data) and as our theory shows, it relies on variance reduction that can simply be achieved by choosing adaptive sample sizes that tend to grow as the algorithm progresses to optimality. Such adaptive schemes have been proposed in the literature primarily for gradient descent methods and in a convex setting .

With the rise of interest in nonconvex optimization, the ML community started to consider a classical alternative to gradient descent/line search methods - trust region methods . Their usefulness is largely dictated by their ability to utilize negative curvature in Hessian approximations and hence, potentially, escape the neighborhoods of saddle points , which can significantly slow down or even trap a line search method. It is argued, that while saddle points are undesirable, the local minima are typically sufficient for the purposes of training nonconvex ML models, such as deep neural networks. There has been a number of recent works that propose trust-region methods that use stochastic gradient and Hessian estimates , but they all assume that the objective function is deterministic. One of the original stochastic trust region methods for this stochastic optimization setting has been proposed in and a more sophisticated adaptive method has been recently introduced in . For both methods the convergence is achieved by repeatedly sampling the function values (and gradients, when applicable) so that eventually the estimates become asymptotically error-free with probability 1. No convergence rates have been derived for these algorithms, because the progress happens in the assymptotics. Fully stochastic versions of trust region methods with adaptive sampling, such as may be used in ML context, have not yet been explored to our knowledge. In addition, our analysis applies to the setting where the objective functions and gradient estimates may be biased.

STORM uses adaptive trust region radii and is close to what is known to be efficient in practice, hence here we focus on the theoretical analysis of this method in the first and second order settings. We recover the convergence rates whose dependence on ϵ\epsilon is the same as of those for deterministic trust region method. Since the method is stochastic our convergence rates are derived in the form of the bound on the expected number of iterations the algorithms takes until achieving the ϵ\epsilon-accuracy. In contrast, convergence rate result for SGD in , for example, only bounds the expected sum of the norms of all the gradients up to iteration TT, as a function of TT. Other weaker types of convergence rates are established in and . In a trust region and a cubic regularization methods based on sampled Hessian are considered. The number of samples is selected in such a way that the error in the Hessian approximation is smaller than ϵ\epsilon with large probability pp. Then the deterministic convergence rate can be established under the assumption that the condition on the Hessian approximation holds at each iteration until ϵ\epsilon-accuracy is reached. Hence, the established bound on the number of iterations TT holds with probability pTp^{T} and with probability 1−pT1-p^{T} no bound is known. The same type of complexity result is derived in for a cubic regularization method, where gradients and Hessians are also sampled at a rate dictated by ϵ\epsilon and the resulting bound holds only with some probability.

Algorithms in have some similarities with algorithms analyzed in and . In and , the global rates of convergence of a trust region, a line search and an adaptive cubic regularization methods are analyzed under the assumption that first and second order information is inexact, but sufficiently accurate with some probability, however, the analysis in all of these papers relies heavily on the assumption that function values are computed accurately, in particular that no increasing steps are allowed. This implies that the results in and cannot be applied in a stochastic setting. , on the other hand, does not explicitly use function values, because it does not utilize adaptive step sizes. This paper can be seen as an extension of and to the case of stochastic functions.

The goal of our paper is twofold: First, we introduce a novel framework for bounding expected complexity of a stochastic optimization method. This framework is based on defining a renewal-reward process associated with the algorithm as well as its stopping time, which is the time when the algorithm reaches desired accuracy. Then, under certain assumptions, we derive a bound on the expected stopping time. This framework, in principal, can be used for analysis of convergence rates of a variety of algorithms - for instance it applies to all algorithms in and . In recent work it has been applied to analyze a stochastic line-search method. In this paper, specifically, we use this general framework to derive a bound on the convergence rate of the STORM algorithm defined in , by proving that these assumptions are satisfied by this algorithm. In particular, we show that the expected number of iterations required to achieve ∥∇f(x)∥≤ϵ\|\nabla f(x)\|\leq\epsilon is bounded by O(ϵ−2/(2p−1))O(\epsilon^{-2}/(2p-1)), which is an improvement on the result in and a similar one to those in , in terms of dependence on ϵ\epsilon, but such that, in principal, it never requires computation of the true gradient. The result is a natural extension of the standard, best-known worst-case complexity of any first order method for nonconvex optimization . In this paper we also make a significant improvement upon the results in by relaxing a very restrictive condition on the size of the steps taken by the algorithm. By applying the general analytic framework again, we also provide a second order complexity analysis. We show that a second order STORM variant takes an expected number of iterations that is at most O(ϵ−3/(2p−1))O(\epsilon^{-3}/(2p-1)) to ensure max⁡{∥∇f(x)∥,−λmin⁡(∇2f(x))}≤ϵ\max\{\|\nabla f(x)\|,-\lambda_{\min}(\nabla^{2}f(x))\}\leq\epsilon; this result requires slightly stronger assumptions on the function estimates but provides generalization of results in to the stochastic case.

Our main complexity results does not yet provide a termination criterion that would guarantee that ∥f(xˉ)∥≤ϵ\|f(\bar{x})\|\leq\epsilon, where xˉ\bar{x} is the last iterate. However, the analysis provides a foundation for establishing such a criterion. In particular, while in this paper we simply bound the expected complexity, bounding the tail of the complexity distribution will follow from the analysis here.

In the next section we present and analyze our generic framework, while the STORM algorithm is analyzed in Section 3.

The rest of the paper is organized as follows: we begin by introducing the stochastic framework and deriving the bound on its expected stopping time in Section 2. In Section 3 we provide the first order complexity analysis of the STORM algorithm by showing that it fits into the framework introduced in Section 2. The second order complexity analysis follows in Section 4.

We will also use I(A)I\left(A\right) to denote the indicator of a random event AA occurring.

A Renewal-Reward Martingale Process

In this section we consider a general random process and a stopping time TT, which posses certain properties. We analyze the behavior of this random process and derive a bound on the expected stopping time. These results will be used later in the paper in the specific setting of convergence of a stochastic trust region method to first order stationary points. We argue that the framework presented in this section can be used for convergence analysis of a variety of stochastic algorithms. We start by defining a stopping time of a discrete time stochastic process.

Given a stochastic process {Xk}={Xk:k≥0}\{X_{k}\}=\{X_{k}:k\geq 0\}, we say that TT is a stopping time with respect to {Xk}\{X_{k}\} if for each m≥0m\geq 0 the occurrance of the event {T=m}\{T=m\} is determined by observing X1,…,XmX_{1},\dots,X_{m}. That is, {T=m}∈σ(X0,...,Xm)\{T=m\}\in\sigma\left(X_{0},...,X_{m}\right), the σ\sigma-field generated by X1,...,XmX_{1},...,X_{m}, for each m≥0m\geq 0.

Now, let {(Φk,Δk)}\{\left(\Phi_{k},\Delta_{k}\right)\} be a random process such that Φk∈[0,∞)\Phi_{k}\in[0,\infty) and Δk∈[0,∞)\Delta_{k}\in[0,\infty) for k≥0k\geq 0. Let Vk+1=Φk+1−ΦkV_{k+1}=\Phi_{k+1}-\Phi_{k} for k≥0k\geq 0. We also assume the existence of a sequence {Wk}k=1∞\left\{W_{k}\right\}_{k=1}^{\infty}, defined on the same probability space as {(Φk,Δk)}\{\left(\Phi_{k},\Delta_{k}\right)\}, we introduce W0=1W_{0}=1 and let Fk\mathcal{F}_{k} denote the σ\sigma-algebra generated by {(Φ0,Δ0,W0),⋯ ,(Φk,Δk,Wk)}\{\left(\Phi_{0},\Delta_{0},W_{0}\right),\cdots,\left(\Phi_{k},\Delta_{k},W_{k}\right)\}. We assume that {Wk}k=1∞\left\{W_{k}\right\}_{k=1}^{\infty} satisfies

Note that under the assumption (2) the WkW_{k}’s are independent and also independent of the sequence {(Φk,Δk)}\left\{\left(\Phi_{k},\Delta_{k}\right)\right\}.

Let {Tϵ}ϵ>0\left\{T_{\epsilon}\right\}_{\epsilon>0} be a family of stopping times with respect to {Fk}k≥0\left\{\mathcal{F}_{k}\right\}_{k\geq 0}, parametrized by some quantity ϵ>0\epsilon>0. The following assumptions will be imposed on {(Φk,Δk)}\{\left(\Phi_{k},\Delta_{k}\right)\} and TϵT_{\epsilon}.

where Wk+1W_{k+1} satisfies (2) with p>12p>\frac{1}{2}.

There exists a nondecreasing function h(⋅):[0,∞)→(0,∞)h(\cdot):[0,\infty)\rightarrow(0,\infty) and a constant Θ>0\Theta>0 such that

In order to define this renewal process we first introduce an auxiliary process. Define {Zk}k=0∞\left\{Z_{k}\right\}_{k=0}^{\infty} as follows. First, let Z0=jϵZ_{0}=j_{\epsilon} and set

Note that the process {Zk}k=0∞\left\{Z_{k}\right\}_{k=0}^{\infty} is a birth-death process on the set {k:k≤jϵ}\left\{k:k\leq j_{\epsilon}\right\}. Then, define the renewal process, A0=0A_{0}=0 and An=inf⁡{m>An−1:Zm=jϵ}A_{n}=\inf\{m>A_{n-1}:Z_{m}=j_{\epsilon}\}. By Assumption (3) and using a simple inductive argument for the second inequality below we have that

In other words, on Tϵ>kT_{\epsilon}>k, the process AnA_{n} only counts the iterations for which Δk\Delta_{k} has value at least Δϵ\Delta_{\epsilon}. The interarrival times of this renewal process are defined for all k≥1k\geq 1 by

As a final piece of notation, we define the counting process

which is the number of renewals that occur before time kk.

Let τn\tau_{n} be defined as above. Then, for all nn

Define the process Zˉk+1=Zˉk+Wk+1\bar{Z}_{k+1}=\bar{Z}_{k}+W_{k+1}, which is a simple random walk. Suppose that Zˉ0=−1\bar{Z}_{0}=-1 and define τˉ=inf⁡{n≥0:Zˉn=0}\bar{\tau}=\inf\{n\geq 0:\bar{Z}_{n}=0\} it is well known (in fact, this follows by Wald’s identity) that

On the other hand, by first step analysis (i.e. conditioning on W1W_{1}) we have that

The above identity follows because the distribution of τ1\tau_{1} conditioned on Z1=jϵ−1Z_{1}=j_{\epsilon}-1 is the same as the distribution of τˉ\bar{\tau}. So, we conclude that

the last equality follows by simplifying the expression above. ∎

We now bound the number of renewals that can occur before the time TϵT_{\epsilon}.

For ease of notation, let k∧Tϵ=min⁡{k,Tϵ}k\wedge T_{\epsilon}=\min\{k,T_{\epsilon}\}. Consider the stochastic process defined via R0=Φ0R_{0}=\Phi_{0} and

for k≥1k\geq 1, where Θ\Theta is defined in (5). Observe that RkR_{k} is a non-negative supermartingale with respect to {Fk}\left\{\mathcal{F}_{k}\right\}, to see this we first write

where the last equality follows because TϵT_{\epsilon} is a stopping time and therefore the random variable the expectation is Fk\mathcal{F}_{k}-measurable.

On the other hand, since {Tϵ≥k+1}={Tϵ>k}={Tϵ≤k}c∈Fk\left\{T_{\epsilon}\geq k+1\right\}=\left\{T_{\epsilon}>k\right\}=\left\{T_{\epsilon}\leq k\right\}^{c}\in\mathcal{F}_{k} we conclude, using (5), that

as claimed. We then conclude, since Φk≥0\Phi_{k}\geq 0 for each k≥0k\geq 0, that

Now, since h(⋅)≥0h(\cdot)\geq 0, observe that

as k→∞k\rightarrow\infty, note that this conclusion holds also on the event {Tϵ=∞}\{T_{\epsilon}=\infty\}. Therefore, by the Monotone Convergence Theorem

Now, by the definition of the counting process N(⋅)N(\cdot), since the renewal times AnA_{n} when ΔAn≥Δϵ\Delta_{A_{n}}\geq\Delta_{\epsilon}, are a subset of the iterations 0,1,…,Tϵ0,1,\dots,T_{\epsilon}, and since h(⋅)h(\cdot) is nondecreasing, we have

the term +1 being added to N(Tϵ−1)N(T_{\epsilon}-1) comes from the fact that A0=0A_{0}=0. Inserting this in (8),

We now state and prove a well known theorem on expected stopping time, known as the Wald’s Identity (non-negative increments case) (e.g., see Theorem 2.2.4 in ). The reason we provide a proof here, is that in the literature this result is typically shown under the assumption that the stopping time is finite a.s. Dropping this condition is particularly important in our framework, as this condition is equivalent to a convergence result for the optimization algorithm which generates the stochastic process. It is convenient and useful not to have to prove the convergence result before establishing the convergence rates bounds, since the convergence immediately follows from these bounds.

Wald’s Identity (non-negative increments). Suppose that {Yi}i=1n\left\{Y_{i}\right\}_{i=1}^{n} is a sequence of independent random variables such that Yi∈[0,∞]Y_{i}\in[0,\infty] with probability one. Define E(Yi)=μi∈[0,∞]E\left(Y_{i}\right)=\mu_{i}\in[0,\infty] and let N∈[0,∞]N\in[0,\infty] be a stopping time with respect to the filtration generated by the YnY_{n}’s. Define Sn=Y1+...+YnS_{n}=Y_{1}+...+Y_{n}, S0=0S_{0}=0, sn=μ1+...+μns_{n}=\mu_{1}+...+\mu_{n} and s0=0s_{0}=0. Then

Let m>0m>0 be an arbitrary integer and define Yi(m)=min⁡(Yi,m)Y_{i}\left(m\right)=\min\left(Y_{i},m\right), Nm=min⁡(N,m)N_{m}=\min\left(N,m\right), μi(m)=E(Yi(m))\mu_{i}\left(m\right)=E\left(Y_{i}\left(m\right)\right), Sn(m)=Y1(m)+...+Yn(m)S_{n}\left(m\right)=Y_{1}\left(m\right)+...+Y_{n}\left(m\right) and sn(m)=μ1(m)+...+μn(m)s_{n}\left(m\right)=\mu_{1}\left(m\right)+...+\mu_{n}\left(m\right). Note that all of these quantities are non-negative and non-decreasing in mm. By the optional sampling theorem applied to the martingale Mn=Sn(m)−sn(m)M_{n}=S_{n}\left(m\right)-s_{n}\left(m\right), we have that

as m→∞m\rightarrow\infty. For the case N=∞N=\infty, we interpret SN=sup⁡n≥0sup⁡mSn(m)S_{N}=\sup_{n\geq 0}\sup_{m}S_{n}\left(m\right). Similarly,

as m→∞m\rightarrow\infty. By the monotone convergence theorem we then conclude that

If μi=μ\mu_{i}=\mu, then E(SN)=μ⋅E(N)E\left(S_{N}\right)=\mu\cdot E\left(N\right). If μ=0\mu=0, then Xi=0X_{i}=0 almost surely and SN=0S_{N}=0. Therefore, if μ=0\mu=0, we interpret μ⋅E(N)=0\mu\cdot E\left(N\right)=0, even if E(N)=∞E\left(N\right)=\infty. This interpretation is consistent with the case in which N=0N=0 almost surely as well, in this case 0=μ⋅E(N)=E(SN)0=\mu\cdot E\left(N\right)=E\left(S_{N}\right), even μ=∞\mu=\infty.

We now apply this theorem to Sn=An=∑i=0nτiS_{n}=A_{n}=\sum_{i=0}^{n}\tau_{i} and obtain the main result of this section, which will be used in the following sections to establish the main complexity result.

Define Gn=FAn\mathcal{G}_{n}=\mathcal{F}_{A_{n}}, that is,

Note that AnA_{n} is a stopping time with respect to {Fn}n≥0\left\{\mathcal{F}_{n}\right\}_{n\geq 0}, so Gn\mathcal{G}_{n} is well defined. We claim that N(Tϵ−1)+1N\left(T_{\epsilon}-1\right)+1 is a stopping time with respect to {Gn}n≥0\left\{\mathcal{G}_{n}\right\}_{n\geq 0}. To see this, note, since N(k)≤kN\left(k\right)\leq k

where the last inclusion follows because N(k)+1N\left(k\right)+1 is a stopping time with respect to {FAn}n≥0\left\{\mathcal{F}_{A_{n}}\right\}_{n\geq 0} and because An≥nA_{n}\geq n, so Fn⊆FAn\mathcal{F}_{n}\subseteq\mathcal{F}_{A_{n}}, which implies that TϵT_{\epsilon} is also stopping time with respect to {Gn}n≥0\left\{\mathcal{G}_{n}\right\}_{n\geq 0}.

Now, because of the independence assumption implied by (2) we have that

Recalling that AN(Tϵ−1)+1=∑k=1N(Tϵ−1)+1τkA_{N(T_{\epsilon}-1)+1}=\sum_{k=1}^{N(T_{\epsilon}-1)+1}\tau_{k}, we can invoke Wald’s identity to conclude that

Since AN(Tϵ−1)+1≥Tϵ−1A_{N(T_{\epsilon}-1)+1}\geq T_{\epsilon}-1, we have by Lemmas 2.1 and 2.2

The statement of the theorem follows from the last inequality. ∎

The first order STORM algorithm

We now state and analyze a stochastic trust region (TR) algorithm (Algorithm 1) which is essentially very similar to its deterministic counterpart . This method uses the inexact (noisy) information about ff and its derivatives, just as the deterministic method uses the exact information. This algorithm, as stated, and the assumptions on its steps that we will impose below aim at convergence to a first order stationary point. In this section we will analyze the global rate of convergence of this algorithm to such a point (while in Section 4, we extend Algorithm 1 to calculate second order critical points).

For every kk, the step sks_{k} is computed so that the well-known Cauchy decrease condition is satisfied,

for some constant κfcd∈(0,1].\kappa_{fcd}\in(0,1]. This condition is standard for the TR methods, easy to enforce in practice and is discussed in detail in the literature . Iterations on which xk+1=xk+skx_{k+1}=x_{k}+s_{k} occurs are called successful.

Algorithm 1 generates a random process. The source of randomness are the random models and random estimates constructed on each iteration, based on some random information obtained from the stochastic function f(x,ε)f(x,\varepsilon). MkM_{k} will denote a random model in the kk-th iteration, while we will use the notation mk=Mk(ω)m_{k}=M_{k}(\omega) for its realizations. As a consequence of using random models, the iterates XkX_{k}, the trust-region radii Δk\Delta_{k} and the steps SkS_{k} are also random quantities, and so xk=Xk(ω)x_{k}=X_{k}(\omega), δk=Δk(ω){\delta}_{k}=\Delta_{k}(\omega), sk=Sk(ω)s_{k}=S_{k}(\omega) will denote their respective realizations. Similarly, let random quantities {Fk0,Fks}\{F_{k}^{0},F_{k}^{s}\} denote the estimates of f(Xk)f(X_{k}) and f(Xk+Sk)f(X_{k}+S_{k}), with their realizations denoted by fk0=Fk0(ω)f_{k}^{0}=F_{k}^{0}(\omega) and fks=Fks(ω)f_{k}^{s}=F_{k}^{s}(\omega). In other words, Algorithm 1 results in a stochastic process {Mk,Xk,Sk,Δk,Fk0,Fks}\{M_{k},X_{k},S_{k},\Delta_{k},F_{k}^{0},F_{k}^{s}\}. Our goal is to show that under certain conditions on the sequences {Mk}\{M_{k}\} and {Fk}={(Fk0,Fks)}\{F_{k}\}=\{(F_{k}^{0},F_{k}^{s})\} the resulting stochastic process has desirable convergence rate. In particular, we will assume that models MkM_{k} and estimates Fk0,FksF_{k}^{0},F_{k}^{s} are sufficiently accurate with sufficiently high probability, conditioned on the past.

The key to the analysis lies in the assumption that the accuracy improves in coordination with the perceived progress of the algorithm. The main challenge of the analysis lies in the fact that, while in the deterministic case the function f(x)f(x) never increases from one iteration to another, this can easily happen in the stochastic case. The analysis is based on properties of supermartingales where the increments of a supermartingale depend on the function change between iterates (which as we will show, tend to decrease). To make the analysis simpler we need a technical assumption that these increments are bounded from above. Hence, overall we make the following assumptions on ff:

We assume that all iterates xkx_{k} generated by Algorithm 1 the gradient ∇f\nabla f is LL-Lipschitz continuous and

The assumptions of Lipschitz continuity of ∇f\nabla f and boundedness of ff from below are standard. Here for simplicity and w.l.o.g. we assume that the lower bound on ff is nonnegative.

Let Fk−1M⋅F\mathcal{F}_{k-1}^{M\cdot F} denote the σ\sigma-algebra generated by M0,⋯ ,Mk−1M_{0},\cdots,M_{k-1} and F0,⋯ ,Fk−1F_{0},\cdots,F_{k-1} and let Fk−1/2M⋅F\mathcal{F}_{k-{1}/{2}}^{M\cdot F} denote the σ\sigma-algebra generated by M0,⋯ ,MkM_{0},\cdots,M_{k} and F0,⋯ ,Fk−1F_{0},\cdots,F_{k-1}.

1) A function mkm_{k} is a κ\kappa-fully linear model of ff on B(xk,δk)B(x_{k},{\delta}_{k}) provided, for κ=(κef,κeg)\kappa=(\kappa_{ef},\kappa_{eg}) and ∀y∈B(xk,δk)\forall y\in B(x_{k},\delta_{k}),

2) The estimates fk0f_{k}^{0} and fksf_{k}^{s} are said to be ϵF\epsilon_{F}-accurate estimates of f(xk)f(x_{k}) and f(xk+sk)f(x_{k}+s_{k}), respectively, for a given δk\delta_{k} if

A sequence of random models {Mk}\{M_{k}\} is said to be α\alpha-probabilistically κ\kappa-fully linear with respect to the corresponding sequence {B(Xk,Δk)}\{B(X_{k},\Delta_{k})\} if the events

A sequence of random estimates {Fk}\{F_{k}\} is said to be β\beta-probabilistically ϵF\epsilon_{F}-accurate with respect to the corresponding sequence {Xk,Δk,Sk}\{X_{k},\Delta_{k},S_{k}\} if the events

where ϵF\epsilon_{F} is a fixed constant.

Next is the key assumption on the nature of the stochastic (and deterministic) information used by our algorithm.

The following hold for the quantities used in Algorithm 1

The model Hessians satisfy ∥Hk∥2≤κbhm\|H_{k}\|_{2}\leq\kappa_{bhm} for some κbhm≥1\kappa_{bhm}\geq 1, for all kk, deterministically.

The sequence of random models MkM_{k}, generated by Algorithm 1, is α\alpha-probabilistically κ\kappa-fully linear, for some κ=(κef,κeg)\kappa=(\kappa_{ef},\kappa_{eg}) and for a sufficiently large α∈(0,1)\alpha\in(0,1).

The sequence of random estimates {Fk}\{F_{k}\} generated by Algorithm 1 is β\beta-probabilistically ϵF\epsilon_{F}-accurate for ϵF≤κef\epsilon_{F}\leq\kappa_{ef} and ϵF<14η1η2κfcdmin⁡{η2κbhm,1}\epsilon_{F}<\frac{1}{4}\eta_{1}\eta_{2}\kappa_{fcd}\min\left\{\frac{\eta_{2}}{\kappa_{bhm}},1\right\}, and for a sufficiently large β∈(0,1)\beta\in(0,1).

In the analysis of the algorithm requires an additional assumption that η2≥κef\eta_{2}\geq\kappa_{ef} and for simplicity it is further assumed that η2≥κbhm\eta_{2}\geq\kappa_{bhm}. This assumption is undesirable since it restricts the size of the steps that can be taken by the trust region algorithm. In this paper we manage to improve the analysis and drop this assumption, hence allowing η2\eta_{2} to be set to a small value. Note that small values of η2\eta_{2} imply small ϵF\epsilon_{F} because of Assumption 3.2(c), so there is a potential trade-off in choosing η2\eta_{2}. On the other hand, this relationship indicates that when ϵF=0\epsilon_{F}=0, that is when there is no error in the function estimates, then η2\eta_{2} can be arbitrarily small.

Under Assumption 3.2, P{IkJk=1∣Fk−1M⋅F}≥αβP\{I_{k}J_{k}=1|\mathcal{F}_{k-1}^{M\cdot F}\}\geq\alpha\beta and P{Ik+Jk=0∣Fk−1M⋅F}≤(1−α)(1−β)P\{I_{k}+J_{k}=0|\mathcal{F}_{k-1}^{M\cdot F}\}\leq(1-\alpha)(1-\beta). At iteration kk, if IkJk=1I_{k}J_{k}=1 then the behavior of the algorithm reduces to that of an (inexact) deterministic algorithm; while if Ik+Jk=0I_{k}+J_{k}=0, then not only may the algorithm produce a bad step (that is a step which increases the objective function), but it also may accept this bad step by mistaking it for an improving step (that is a step that decreases the function value). In the cases when only one of Ik=0I_{k}=0 and Jk=0J_{k}=0 holds, then either the model is good but the estimates are faulty, or the estimates are good and the model is faulty. In this case an improving step is still possible, but a bad step is not. In the worst case, no step is taken and the trust region radius is reduced. The main idea of our framework is to choose probabilities of IkJk=1I_{k}J_{k}=1 and Ik+Jk=0I_{k}+J_{k}=0 occurring according to the possible corresponding decrease and increase in f(x)f(x), so that in expectation, f(x)f(x) is sufficiently reduced.

2 Useful existing results

Algorithm 1 is analyzed in and the following almost-sure stationarity result is shown: there exists a selection of α\alpha and β\beta such that under Assumption 3.2 with additional requirement that η2≥κef\eta_{2}\geq\kappa_{ef}, the sequence of random iterates generated by Algorithm 1, {Xk}\{X_{k}\}, almost surely satisfies lim⁡k→∞∥∇f(Xk)∥=0\underset{k\to\infty}{\lim}\|\nabla f(X_{k})\|=0. The important observation is that α\alpha and β\beta do not have to increase as the algorithm progresses. Hence with the same, constant but small enough, probabilities our models and estimates can be arbitrarily erroneous.

Our primary goal in this paper is to bound the expected number of steps that the algorithm takes until ∥∇f(Xk)∥≤ϵ\|\nabla f(X_{k})\|\leq\epsilon occurs and the secondary goal is to relax the assumption η2≥κef\eta_{2}\geq\kappa_{ef}. We will modify the analysis that led to the above stationarity result in . First, we state (without proof) several auxiliary lemmas from .

[Good model ⇒\Rightarrow function reduction in ∥gk∥\|g_{k}\|] Suppose that a model mkm_{k} is a (κef,κeg)(\kappa_{ef},\kappa_{eg})-fully linear model of ff on B(xk,δk)B(x_{k},{\delta}_{k}). If

then the trial step sks_{k} leads to an improvement in f(xk+sk)f(x_{k}+s_{k}) such that

[Good model ⇒\Rightarrow function reduction in ∥∇f(xk)∥\|\nabla f(x_{k})\|] Under Assumption 3.2(a), suppose that a model is (κef,κeg)(\kappa_{ef},\kappa_{eg})-fully linear on B(xk,δk)B(x_{k},{\delta}_{k}). If

then the trial step sks_{k} leads to an improvement in f(xk+sk)f(x_{k}+s_{k}) such that

for any C1≤κfcd4⋅max⁡{κbhmκbhm+κeg,8κef8κef+κfcdκeg}.C_{1}\leq\frac{\kappa_{fcd}}{4}\cdot\max\left\{\frac{\kappa_{bhm}}{\kappa_{bhm}+\kappa_{eg}},\frac{8\kappa_{ef}}{8\kappa_{ef}+\kappa_{fcd}\kappa_{eg}}\right\}.

[Good model ++ good estimates ⇒\Rightarrow successful step] Under Assumption 3.2(a), suppose that mkm_{k} is (κef,κeg)(\kappa_{ef},\kappa_{eg})-fully linear on B(xk,δk)B(x_{k},{\delta}_{k}) and the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-accurate with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}. If

[ Good estimates ++ successful step ⇒\Rightarrow function reduction in δk2\delta_{k}^{2}] Under Assumption 3.2(a), suppose that the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-accurate with ϵF<14η1η2κfcdmin⁡{η2κbhm,1}\epsilon_{F}<\frac{1}{4}\eta_{1}\eta_{2}\kappa_{fcd}\min\left\{\frac{\eta_{2}}{\kappa_{bhm}},1\right\}. If a trial step sks_{k} is accepted (a successful iteration occurs), then the improvement in ff is bounded below as follows

We now explain briefly the role of the constants η2,ϵF,α,\eta_{2},\epsilon_{F},\alpha, and β\beta and their expected magnitude. First, we note that constants κef,κeg\kappa_{ef},\kappa_{eg}, and κbhm\kappa_{bhm} can be chosen arbitrarily large, but ideally should be chosen as small as possible while guaranteeing Assumption 3.2. Let us assume that κef,κeg\kappa_{ef},\kappa_{eg} and κbhm\kappa_{bhm} can be all chosen as Θ(L)\Theta(L)Note that it is possible to have κef\kappa_{ef} and κeg\kappa_{eg} of different magnitudes, namely when κeg\kappa_{eg} is small as we have sufficiently accurate gradients, but κef\kappa_{ef} remains as Θ(L)\Theta(L). Our analysis and results apply then as well., where LL is the Lipschitz constant of ∇f(x)\nabla f(x) in X\cal{X}, even though it may not be explicitly known (see for construction of fully-linear models in the case of unavailable derivative estimates). Once these constants are chosen, ϵF\epsilon_{F} is chosen so that it satisfies the conditions in Assumption 3.2(c). Note that if η2\eta_{2} is chosen to equal LL, this means that Algorithm 1 only takes steps when δk≤∥gk∥L\delta_{k}\leq\frac{\|g_{k}\|}{L}, which is similar to constraining step size by 1L\frac{1}{L} in gradient descent. In this case, ϵF\epsilon_{F} can be chosen relatively large and thus the estimates need to be slightly more accurate than the models, but the order of required accuracy is similar. On the other hand, since in trust region methods, step sizes are meant to be chosen adaptively, it is desirable to allow larger steps, which can be done by setting η2\eta_{2} to be small. This, in turn, implies that ϵF\epsilon_{F} has to be chosen small and hence the estimates will have to be a lot more accurate than the models. Another trade-off when choosing a small value for η2\eta_{2} will become apparent in our main complexity results, as we will see that the expected improvement per iteration may depend on η2\eta_{2}. But reasonable values for η2\eta_{2} allow the removal of this dependency.

To simplify expressions for various constants we will assume that η1=0.1\eta_{1}=0.1, γ=2\gamma=2 and κfcd=0.5\kappa_{fcd}=0.5 which are typical values for these constants. We also assume, w.l.o.g., that κbhm≤12κef\kappa_{bhm}\leq 12\kappa_{ef} and η2≤κeg\eta_{2}\leq\kappa_{eg}. To simplify expressions further we will consider κef=κeg\kappa_{ef}=\kappa_{eg}. It is clear that if κef\kappa_{ef} or κeg\kappa_{eg} happen to be smaller, somewhat better bounds than the ones we derive here will result, because the models give tighter approximations of the true function. We are interested in deriving bounds for the case when κef\kappa_{ef} or κeg\kappa_{eg} may be large. The analysis can be performed for any other values of the above constants, hence the choice here is done merely for convenience and simplicity.

The conditions on α\alpha and β\beta under the above choice of constants will be shown in our results below.

We consider a random process {Φk,Δk}\{\Phi_{k},\Delta_{k}\} derived from the process generated by Algorithm 1, with Δk\Delta_{k} - the trust region radius and

where ν∈(0,1)\nu\in(0,1) is a deterministic, large enough constant, which will be defined later. Clearly Φk≥0\Phi_{k}\geq 0. Recall the notation ϕk\phi_{k} for realizations of Φk\Phi_{k}. Here we defined Fk\mathcal{F}_{k} as FkM⋅F\mathcal{F}_{k}^{M\cdot F}.

It is easy to see that TϵT_{\epsilon} is a stopping time for the stochastic process defined by Algorithm 1 and hence for {Φk,Δk}\{\Phi_{k},\Delta_{k}\}.

Let us show that Assumptions 2.1(i)-(ii) hold with the following Δϵ\Delta_{\epsilon}

Note that with our choice of algorithmic parameters the above is satisfied by ζ=20κeg\zeta=20\kappa_{eg}.

For simplicity of the presentation and without loss of generality, we assume that Δϵ=γiδ0\Delta_{\epsilon}=\gamma^{i}\delta_{0}, for some integer i≤0i\leq 0. If not, we can always choose ζ\zeta within a factor of γ{\gamma} of its lower bound in (22). It follows that for any kk, Δk=γikΔϵ\Delta_{k}=\gamma^{i_{k}}\Delta_{\epsilon}, for some integer iki_{k}. Choosing λ\lambda in Assumption 2.1(i)-(ii) so that eλ=γe^{\lambda}=\gamma Assumption 2.1(i) follows immediately from the definition of {Φk,Δk}\{\Phi_{k},\Delta_{k}\}, and the choice of δmax\delta_{max} imposed by Algorithm 1 and for Assumption 2.1(ii) we now only need to show that the dynamics (3) hold for Δk\Delta_{k}.

Let Assumptions 3.1 and 3.2 hold. Let α\alpha and β\beta be such that αβ≥1/2\alpha\beta\geq 1/2, then Assumption 2.1(ii) is satisfied for Wk=2(IkJk−12)W_{k}=2(I_{k}J_{k}-\frac{1}{2}), λ=log⁡(γ)\lambda=\log(\gamma) and p=αβp=\alpha\beta.

Clearly inequality (3) holds when I(Tϵ>k)=0I(T_{\epsilon}>k)=0. We will show that conditioned on Tϵ>kT_{\epsilon}>k (i.e. I(Tϵ>k)=1I(T_{\epsilon}>k)=1) we have

First we note that for each realization when δk>Δϵ\delta_{k}>\Delta_{\epsilon}, we have δk≥γΔϵ\delta_{k}\geq\gamma\Delta_{\epsilon} and hence δk+1≥Δϵ\delta_{k+1}\geq\Delta_{\epsilon}. Now, assume that δk≤Δϵ\delta_{k}\leq\Delta_{\epsilon}, then, because Tϵ>kT_{\epsilon}>k, we have ∥∇f(xk)∥>ϵ\|\nabla f(x_{k})\|>\epsilon and hence, from the definition of ζ\zeta, we know that

Assume that Ik=1I_{k}=1 and Jk=1J_{k}=1, i.e., both the model and the estimates are good on iteration kk. Since the model mkm_{k} is κ\kappa-fully linear and

and the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-accurate, with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}, condition (17) in Lemma 3.3 holds. Hence, iteration kk is successful, i.e. xk+1=xk+skx_{k+1}=x_{k}+s_{k} and δk+1=max⁡{δmax,γδk}{\delta}_{k+1}=\max\{\delta_{max},\gamma{\delta}_{k}\}. If IkJk=0I_{k}J_{k}=0, then δk+1≥γ−1δk{\delta}_{k+1}\geq\gamma^{-1}{\delta}_{k} simply by the dynamics of Algorithm 1.

Finally, observing that P{IkJk=1∣Fk−1M⋅F}≥p=αβP\{I_{k}J_{k}=1|\mathcal{F}_{k-1}^{M\cdot F}\}\geq p=\alpha\beta we conclude that (23) implies Assumption 2.1(ii). ∎

We now show that Assumption 2.1(iii) holds, which is the key theorem in this section and is similar to Theorem 4.11 in , while dropping the restrictive conditions on η2\eta_{2} and simplifying the proof. We will omit the parts of the proof that are identical to those of Theorem 4.11 in .

There exist probabilities α\alpha and β\beta such that under Assumptions 3.1 and 3.2 there exists a constant Θ>0\Theta>0 such that, conditioned on Tϵ>kT_{\epsilon}>k

Moreover, under the particular choice of constants described in the last section, let α\alpha and β\beta satisfy

ThenNote that β>12\beta>\frac{1}{2} and so Θ=11800κeg−1\Theta=\frac{1}{1800}\kappa_{eg}^{-1}, independently of η2\eta_{2}, provided η2≥2κeg−1\eta_{2}\geq 2\kappa_{eg}^{-1}; the latter implies that small values are allowed for η2\eta_{2} as κeg\kappa_{eg} values of interest are large., Θ=11800min⁡{η2β,κeg−1}\Theta=\frac{1}{1800}\min\left\{\eta_{2}\beta,\kappa_{eg}^{-1}\right\}.

Since (24) holds trivially if Tϵ≤kT_{\epsilon}\leq k, we assume henceforth in this proof that ∇f(Xk)>ϵ\nabla f(X_{k})>\epsilon. We will consider two possible cases: ∥∇f(xk)∥≥ζδk\|\nabla f(x_{k})\|\geq\zeta\delta_{k} and ∥∇f(xk)∥<ζδk\|\nabla f(x_{k})\|<\zeta\delta_{k}. We will show that (24) holds in both cases and hence it holds on every iteration. Let ν∈(0,1)\nu\in(0,1) be such that

Let xkx_{k}, δk\delta_{k}, sks_{k}, gkg_{k}, and ϕk\phi_{k} denote realizations of random quantities XkX_{k}, Δk\Delta_{k}, SkS_{k}, GkG_{k}, and Φk\Phi_{k}, respectively. Let us consider some realization of Algorithm 1. Note that on all successful iterations, xk+1=xk+skx_{k+1}=x_{k}+s_{k} and δk+1=min⁡{γδk,δmax}\delta_{k+1}=\min\{\gamma\delta_{k},\delta_{max}\} with γ>1\gamma>1, hence

On all unsuccessful iterations, xk+1=xkx_{k+1}=x_{k} and δk+1=1γδk\delta_{k+1}=\frac{1}{\gamma}\delta_{k}, i.e.

with C1C_{1} defined in Lemma 3.2 and C3=1+3L2ζC_{3}=1+\frac{3L}{2\zeta}.

Ik=1I_{k}=1 and Jk=1J_{k}=1, i.e., both the model and the estimates are good on iteration kk. The proof is almost identical to that in Theorem 4.11 , but with small modification due to the different definition of ζ\zeta because we no longer assume that η2≥κbhm\eta_{2}\geq\kappa_{bhm}.

By observing that Lemma 3.2 and Lemma 3.3 hold we can derive

Ik=1I_{k}=1 and Jk=0J_{k}=0, i.e., we have a good model and bad estimates on iteration kk. The proof is identical to that in Theorem 4.11 where it is shown that (27) holds.

Ik=0I_{k}=0 and Jk=1J_{k}=1, i.e., we have a bad model and good estimates on iteration kk. Again (27) holds, as is shown in Theorem 4.11 .

Ik=0I_{k}=0 and Jk=0J_{k}=0, i.e., both the model and the estimates are bad on iteration kk. The proof of Theorem 4.11 applies, where it is shown that

Next, following the proof of Case 1 of Theorem 4.11 in we combine the four outcomes to obtain that under condition (28), we have

where last inequality is due to ∥∇f(Xk)∥≥ζΔk\|\nabla f(X_{k})\|\geq\zeta\Delta_{k}.

We now derive the bounds on the expectation of Φk+1−Φk\Phi_{k+1}-\Phi_{k} in the remaining case. The proof of this case is different than that of Case 2 of Theorem 4.11 , because of the dropped bound on η2\eta_{2}.

First we note that if ∥gk∥<η2δk\|g_{k}\|<\eta_{2}\delta_{k}, then we have an unsuccessful step and (27) holds. Hence, we now assume that ∥gk∥≥η2δk\|g_{k}\|\geq\eta_{2}\delta_{k}. Here we consider only two outcomes, in particular, we will show that when the estimates are good, (27) holds. Otherwise, because ∥∇f(xk)∥<ζδk\|\nabla f(x_{k})\|<\zeta\delta_{k}, the increase in ϕk\phi_{k} can be bounded from above by a multiple of δk2\delta_{k}^{2}. Hence by selecting appropriate value for probability β\beta we will be able to establish the bound on expected decrease in Φk\Phi_{k} as in Case 1.

Jk=1J_{k}=1, i.e., the estimates are good on iteration kk, while the model might be good or bad.

The iteration may or may not be successful. On successful iterations, the good estimates ensure reduction in ff, while on unsuccessful iterations, δk\delta_{k} is reduces. Applying the same argument as in the Case 1(c) we establish that (27) always holds.

Jk=0J_{k}=0, i.e., the estimates are bad on iteration kk, while the model might be good or bad.

Here, as in Case 1, we bound the maximum possible increase in ϕk\phi_{k}. Using the Taylor expansion, the Lipschitz continuity of ∇f(x)\nabla f(x) and taking into account the bound ∥∇f(xk)∥<ζδk\|\nabla f(x_{k})\|<\zeta\delta_{k} we have

We are now ready to bound the expectation of ϕk+1−ϕk\phi_{k+1}-\phi_{k} as we did in Case 1 except that now we simply combine (31), which holds with probability at most (1−β)(1-\beta), and (27), which holds otherwise:

If we choose probability 0<β≤10<\beta\leq 1 so that the following holds,

then the first term in (3.3), which is negative, is at least twice as large in absolute value as the second term, which is positive. We thus have

To complete the proof of the lemma it remains to substitute the appropriate constants in the above expressions. In particular, because of our assumptions that κbhm≤12κef\kappa_{bhm}\leq 12\kappa_{ef} and κef=κeg\kappa_{ef}=\kappa_{eg}, we can choose C1=110C_{1}=\frac{1}{10}, and, recalling the choice of ζ=20κeg\zeta=20\kappa_{eg}, γ=2\gamma=2, η1=0.1\eta_{1}=0.1, κfcd=0.5\kappa_{fcd}=0.5 and η2≤κeg\eta_{2}\leq\kappa_{eg}, (25) reduces to

We can assume that ν>12\nu>\frac{1}{2} without loss of generality.

Case 1: For the probabilities α\alpha and β\beta to satisfy (28) with C3=1+3L2ζC_{3}=1+\frac{3L}{2\zeta}, it is sufficient that

Then, using ν>12\nu>\frac{1}{2} (3.3) implies

Case 2: Recalling the expression for C3C_{3} and the values for constant ν\nu, ζ\zeta and γ=2\gamma=2, and choosing ν\nu so that (35) is satisfied with equality, we see that (33) is satisfied if

Then, observing that ν\nu is chosen so that 1−ν=η2320+η21-\nu=\frac{\eta_{2}}{320+\eta_{2}}, from (34) and η2<320\eta_{2}<320 (as ν>12\nu>\frac{1}{2}),

for Θ=11800min⁡{η2β,κeg−1}\Theta=\frac{1}{1800}\min\left\{\eta_{2}\beta,\kappa_{eg}^{-1}\right\}, which completes the proof.

The almost-sure stationarity result follows immediately from Theorem 3.1 with the same proof as in , but this time without the assumption η2≥κef\eta_{2}\geq\kappa_{ef}.

Let Assumptions 3.1 and 3.2 hold, and let α\alpha and β\beta satisfy conditions of Theorem 3.1, then the sequence of random iterates generated by Algorithm 1, {Xk}\{X_{k}\}, almost surely satisfies

The validity of the Assumption 2.1(iii) follows from Theorem 3.1. We state the result below for completeness and convenience of reference.

Let the assumptions of Theorem 3.1 hold. Then Assumption 2.1(iii) is satisfied, with Θ=11800min⁡{η2β,κeg−1}\Theta=\frac{1}{1800}\min\left\{\eta_{2}\beta,\kappa_{eg}^{-1}\right\} for the process {Φk,Δk}\{\Phi_{k},\Delta_{k}\}, where Φk\Phi_{k} is defined as in (20) with ν\nu satisfying (25) and h(δ)=δ2h(\delta)=\delta^{2}.

4 Complexity result for first order STORM algorithm

Consider Algorithm 1 and the corresponding stochastic process. Let TϵT_{\epsilon} be defined as in (21). Then, under the assumptions of Theorem 3.1,

where Θ=11800min⁡{η2β,κeg−1}\Theta=\frac{1}{1800}\min\left\{\eta_{2}\beta,\kappa_{eg}^{-1}\right\}, Φ0\Phi_{0} defined as in (20) with k=0k=0, with ν\nu satisfying (25).

5 Example of models and estimates satisfying Assumption 3.2

While the assumption 3.2, which allows us to develop the general complexity analysis, is fairly general it is easy to satisfy in practice in the classical stochastic optimization setting by taking a sufficient number of samples of the function, gradient and Hessian estimates. A number of recent papers rely on this technique, for producing sufficiently accurate gradient and Hessian approximations. For example Lemma 4 in uses matrix concentration results from to show that given the bound on the variance of the gradient

We want to note that all sample sizes are determined by quantities that are generally either chosen or known by the algorithm or can be correctly estimated.

In the case of simulation optimization, when ∇f(x,ξ)\nabla f(x,\xi) is not available, κ\kappa-fully-linear models mkm_{k} can be constructed via polynomial interpolation , and α\alpha-probabilistically κ\kappa-fully-linear models are similarly obtained by combining interpolation and sufficiently accurate function value estimates (see, e.g. ).

The second order STORM algorithm

We now introduce a variant of Algorithm 1 that attempts to achieve second order criticality in the stochastic setting; we use the same notation as in Algorithm 1. Firstly, the model minimization may need to provide more than just the Cauchy decrease (9), namely, we require that on each iteration kk and for all model realizations mkm_{k} (as defined in Step 2) of MkM_{k}, we are able to compute a step sks_{k}, so that the following level of second order improvement is achieved,

for some constant κscd∈(0,1]\kappa_{scd}\in(0,1]. A step satisfying this (typical second order) assumption is given, for instance, by computing both the Cauchy step and, in the presence of negative curvature in the model, the eigenstep, and by choosing the one that provides the largest reduction in the modelThe eigenstep is the minimizer of the quadratic model in the trust region along an eigenvector corresponding to the smallest (negative) eigenvalue of HkH_{k}. .

In our analysis (not in the algorithm), we will use — instead of just the true gradient of ff — the following measure of proximity to a second order stationary point for the objective ff,

The corresponding optimality measure for the model mkm_{k} is defined slightly differently than above, following ,

The additional term in (39) compared to (38) is needed because there is no longer a bound on the model Hessians on all iterations, as in the first order case. We will only use (39) at the iterate xkx_{k}, in which case, it becomes

We are now ready to present our second order STORM algorithm, by modifying the first order STORM algorithm.

The analysis for the second order STORM variant will again use the framework proposed in Section 2, thus serving as another illustration of the applicability of our generic set up. Before applying this framework we need to describe our assumptions required for a second order analysis.

In terms of problem assumptions, we will need one more order of smoothness compared to first order ones (Assumption 3.1).

Assume that ff satisfies Assumption 3.1 and that it is twice continuously differentiable on X{\cal X}, and also that the Hessian ∇2f\nabla^{2}f is LHL_{H}-Lipschitz continuous.

Let us now introduce a measure of second order quality or accuracy of the models mkm_{k} (see for more details).

1) A function mkm_{k} is a κ\kappa-fully quadratic model of ff on B(xk,δk)B(x_{k},{\delta}_{k}) provided, for κ=(κef,κeg,κeh)\kappa=(\kappa_{ef},\kappa_{eg},\kappa_{eh}) and ∀y∈B(xk,δk)\forall y\in B(x_{k},\delta_{k}),

2) The estimates fk0f_{k}^{0} and fksf_{k}^{s} are said to be ϵF\epsilon_{F}-s.o.-accurate (s.o. for ”second order”) estimates of f(xk)f(x_{k}) and f(xk+sk)f(x_{k}+s_{k}), respectively, for a given δk\delta_{k} if

A sequence of random models {Mk}\{M_{k}\} is said to be α\alpha-probabilistically κ\kappa-fully quadratic with respect to the corresponding sequence {B(Xk,Δk)}\{B(X_{k},\Delta_{k})\} if the events

A sequence of random estimates {Fk0,Fks}\{F_{k}^{0},F_{k}^{s}\} is said to be β\beta-probabilistically ϵF\epsilon_{F}-s.o.-accurate with respect to the corresponding sequence {Xk,Δk,Sk}\{X_{k},\Delta_{k},S_{k}\} if the events

where ϵF\epsilon_{F} is a fixed constant.

We will no longer assume that the Hessian HkH_{k} of the models is bounded in norm, since we cannot simply disregard large Hessian model values without possibly affecting the chances of the model being fully quadratic. However, a simple analysis can show that ∥Hk∥\|H_{k}\| is uniformly bounded from above for any fully quadratic model mkm_{k} (although we may not know what this bound is and hence may not be able to use it in an algorithm).

Let Assumption 4.1 hold. Given constants κeh\kappa_{eh}, κeg\kappa_{eg}, κef\kappa_{ef}, and δmax⁡\delta_{\max}, there exists a constant κbhm≥1\kappa_{bhm}\geq 1 such that for every kk and every realization mkm_{k} of MkM_{k} which is a (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic model of ff on B(xk,δk)B(x_{k},\delta_{k}) with xk∈Xx_{k}\in{\cal X} and δk≤δmax⁡\delta_{k}\leq\delta_{\max} we have

The proof follows trivially from the definition of fully quadratic models and the assumption that ∥∇2f∥≤L\|\nabla^{2}f\|\leq L is bounded above on X{\cal X}, which follows from the gradient of ff being Lipschitz continuous with constant LL. Then we can let κbhm:=δmax⁡κeh+L\kappa_{bhm}:=\delta_{\max}\kappa_{eh}+L.

For our convergence analysis we again need to impose conditions on the nature of the stochastic (and deterministic) information used by our algorithm.

The following hold for the quantities used in Algorithm 2

The sequence of random models MkM_{k}, generated by Algorithm 2, is α\alpha-probabilistically κ\kappa-fully quadratic, for some κ=(κef,κeg,κeh)\kappa=(\kappa_{ef},\kappa_{eg},\kappa_{eh}) and for a sufficiently large α∈(0,1)\alpha\in(0,1).

The sequence of random estimates {Fk0,Fks}\{F_{k}^{0},F_{k}^{s}\} generated by Algorithm 2 is β\beta-probabilistically ϵF\epsilon_{F}-s.o.accurate for ϵF≤κef\epsilon_{F}\leq\kappa_{ef} and ϵF<14η1η2κscdmin⁡{η2,1}\epsilon_{F}<\frac{1}{4}\eta_{1}\eta_{2}\kappa_{scd}\min\{\eta_{2},1\}, and for a sufficiently large β∈(0,1)\beta\in(0,1).

Note that as in the first order case, we are able to allow for unrestricted values of η2\eta_{2} in Algorithm 2, with a potential trade-off of increased accuracy on the function estimates.

2 Useful preliminary results for second order STORM analysis

The analysis of Algorithm 2 is similar to that of the first order STORM described in Section 3. However, there are more cases to consider and the convergence rate to the second order stationary point is different, as it is in the deterministic case. There will also be another significant difference, such as a requirement for an additional assumption on function estimates, to be detailed in the next section. First, we state and prove the analogues of Lemmas 3.1–3.4 for the function decrease in terms of first and second order optimality. The first three lemmas are almost identical to Lemmas 3.1–3.3, except that the models are assumed to be fully quadratic, instead of fully linear; the model decrease condition (37) is now used and condition ∥Hk∥≤κbhm\|H_{k}\|\leq\kappa_{bhm} is only valid when the model mkm_{k} is fully-quadratic according to Lemma 4.1. For completeness, we have delegated the proofs of Lemmas 4.2–Lemmas 4.4 to the Appendix.

[Good quadratic model ⇒\Rightarrow function reduction in ∥gk∥\|g_{k}\|] Let Assumption 4.1 hold. Suppose that a model mkm_{k} is a (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic model of ff on B(xk,δk)B(x_{k},{\delta}_{k}). If δk≤1\delta_{k}\leq 1 and

then the trial step sks_{k} leads to an improvement in f(xk+sk)f(x_{k}+s_{k}) such that

[Good quadratic model ⇒\Rightarrow function reduction in ∥∇f(xk)∥\|\nabla f(x_{k})\|] Let Assumption 4.1 hold. Suppose that a model is (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic on B(xk,δk)B(x_{k},{\delta}_{k}). If δk≤1\delta_{k}\leq 1 and

then the trial step sks_{k} leads to an improvement in f(xk+sk)f(x_{k}+s_{k}) such that

for any C1≤κscd4⋅max⁡{κbhmκbhm+κeg,8κef8κef+κscdκeg}.C_{1}\leq\frac{\kappa_{scd}}{4}\cdot\max\left\{\frac{\kappa_{bhm}}{\kappa_{bhm}+\kappa_{eg}},\frac{8\kappa_{ef}}{8\kappa_{ef}+\kappa_{scd}\kappa_{eg}}\right\}.

[Good quadratic model + good s.o. estimates ⇒\Rightarrow successful step] Let Assumption 4.1 hold. Suppose that mkm_{k} is (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic on B(xk,δk)B(x_{k},{\delta}_{k}) and the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-s.o. accurate with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}. If δk≤1\delta_{k}\leq 1 and

The remaining lemmas address the case of negative curvature in the model and that of second order accurate estimates.

[Good quadratic model ⇒\Rightarrow function reduction in λmin⁡(Hk)\lambda_{\min}(H_{k})] Let Assumption 4.1 hold. Suppose that a model mkm_{k} is a (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic model of ff on B(xk,δk)B(x_{k},{\delta}_{k}). If

then the trial step sks_{k} leads to an improvement in f(xk+sk)f(x_{k}+s_{k}) such that

Whenever λmin⁡(Hk)<0\lambda_{\min}(H_{k})<0, the optimal decrease condition (37) ensures that

Since the model is κ\kappa-fully quadratic, the improvement in ff achieved by sks_{k} is

where the last inequality is implied by (47). ∎

[Good quadratic model ⇒\Rightarrow function reduction in λmin⁡(∇2f(xk))\lambda_{\min}(\nabla^{2}f(x_{k}))] Let Assumption 4.1 hold. Suppose that a model is (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic on B(xk,δk)B(x_{k},{\delta}_{k}). If

then the trial step sks_{k} leads to an improvement in f(xk+sk)f(x_{k}+s_{k}) such that

for any C4≤κscd4⋅8κef8κef+κscdκeh.C_{4}\leq\frac{\kappa_{scd}}{4}\cdot\frac{8\kappa_{ef}}{8\kappa_{ef}+\kappa_{scd}\kappa_{eh}}.

The definition of a κ\kappa-fully-quadratic model, by Corollary 8.5.6 from yield that

Since condition (49) implies that −λmin(∇2f(xk))≥(8κefκscd+κeh)δk-\lambda_{min}(\nabla^{2}f(x_{k}))\geq(\frac{8\kappa_{ef}}{\kappa_{scd}}+\kappa_{eh}){\delta}_{k}, we have

Hence, the conditions of Lemma 4.5 hold and we have

[Good quadratic model + good s.o. estimates ⇒\Rightarrow successful step] Let Assumption 4.1 hold. Suppose that mkm_{k} is (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic on B(xk,δk)B(x_{k},{\delta}_{k}) and the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-s.o.accurate with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}. If

The model mkm_{k} being (κef,κeg)(\kappa_{ef},\kappa_{eg})-fully quadratic implies that

Since the estimates are ϵF\epsilon_{F}-s.o.accurate with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}, we obtain

where we have used the assumptions δk≤κscd(1−η1)8κef(−λmin(Hk)){\delta}_{k}\leq\frac{\kappa_{scd}(1-\eta_{1})}{8\kappa_{ef}}(-\lambda_{min}(H_{k})) to deduce the last inequality. Hence, ρk≥η1\rho_{k}\geq\eta_{1}. Moreover, the first term in (54) and (40) imply τkm≥(−λmin(Hk))≥η2δk\tau_{k}^{m}\geq(-\lambda_{min}(H_{k}))\geq\eta_{2}\delta_{k}. Thus the kk-th iteration is successful. ∎

[Good s.o. estimates + successful step ⇒\Rightarrow function decrease in δk3\delta_{k}^{3}] Assume the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-s.o.accurate with ϵF<14η1η2min⁡{1,η2}κscd\epsilon_{F}<\frac{1}{4}\eta_{1}\eta_{2}\min\{1,\eta_{2}\}\kappa_{scd}. If δk≤1\delta_{k}\leq 1 and a trial step sks_{k} is accepted (a successful iteration occurs), then the improvement in ff is bounded below as follows

An iteration being successful indicates that ρ≥η1\rho\geq\eta_{1} and either min⁡{∥gk∥,∥gk∥∥Hk∥}≥η2δk\min\left\{\|g_{k}\|,\frac{\|g_{k}\|}{\|H_{k}\|}\right\}\geq\eta_{2}{\delta}_{k} or −λmin(Hk)≥η2δk-\lambda_{min}(H_{k})\geq\eta_{2}{\delta}_{k}. First let us assume that min⁡{∥gk∥,∥gk∥∥Hk∥}≥η2δk\min\left\{\|g_{k}\|,\frac{\|g_{k}\|}{\|H_{k}\|}\right\}\geq\eta_{2}{\delta}_{k}; then

Let us now assume that −λmin(Hk)≥η2δk-\lambda_{min}(H_{k})\geq\eta_{2}{\delta}_{k}, thus,

Thus in both cases, using the fact that the estimates are ϵF\epsilon_{F}-s.o.accurate, we have

To simplify our calculations, just like for the first order case, we particularize our choices of constants, but we will clearly state when we use these choices. We let κscd=0.5\kappa_{scd}=0.5, η1=0.1\eta_{1}=0.1, γ=2\gamma=2, δmax⁡=1\delta_{\max}=1 and κef=κeg=κeh=Θ(L‾)\kappa_{ef}=\kappa_{eg}=\kappa_{eh}=\Theta(\overline{L}), where L‾=max⁡{L,LH}\overline{L}=\max\{L,L_{H}\}. To satisfy Assumption 4.2, we let ϵF=1160η2min⁡{1,η2}≤κeh\epsilon_{F}=\frac{1}{160}\eta_{2}\min\{1,\eta_{2}\}\leq\kappa_{eh} and η2≤18\eta_{2}\leq 18. Note that we cannot impose upper bounds on κbhm\kappa_{bhm} as the latter cannot be chosen freely, namely, from Lemma 4.1, we have κbhm=κeh+L≤2max⁡{κeh,L‾}\kappa_{bhm}=\kappa_{eh}+L\leq 2\max\{\kappa_{eh},\overline{L}\}.

As the order of the function decrease that can be guaranteed on good iterations of Algorithm 2 changes from the first order δk2\delta_{k}^{2} to δk3\delta_{k}^{3} due to second order terms, we must modify the process Φk\Phi_{k} accordingly. Namely, we let {Φk,Δk}\{\Phi_{k},\Delta_{k}\} be derived from the process generated by Algorithm 2, with Δk\Delta_{k} - the trust region radius and

where ν∈(0,1)\nu\in(0,1) is a deterministic, large enough constant, which we will define later, and Φk≥0\Phi_{k}\geq 0. We also define the random time

Very similarly to the first order case, we can show that Assumption 2.1(i)–(ii) holds with λ=log⁡γ\lambda=\log\gamma, and with the following new settings

with ϵ∈(0,1]\epsilon\in(0,1], and the (old) assumption that Δϵ=γiδ0\Delta_{\epsilon}=\gamma^{i}\delta_{0} for some i≤0i\leq 0. Note that (63), ϵ∈(0,1]\epsilon\in(0,1] and κbhm≥1\kappa_{bhm}\geq 1 imply that Δϵ≤1\Delta_{\epsilon}\leq 1.

Let Assumptions 4.1 and 4.2 hold. Let α\alpha and β\beta be such that αβ≥1/2\alpha\beta\geq 1/2, then Assumption 2.1(ii) is satisfied for Algorithm 2 with Wk=2(IkJk−12)W_{k}=2(I_{k}J_{k}-\frac{1}{2}), λ=log⁡γ\lambda=\log\gamma and p=αβp=\alpha\beta.

The proof follows similarly to that of Lemma 3.5 and we show that, conditioned on Tϵ>kT_{\epsilon}>k (i.e. I(Tϵ>k)=1I(T_{\epsilon}>k)=1), where TϵT_{\epsilon} is now defined in (62), (23) holds with Δϵ\Delta_{\epsilon} defined in (63). The only case that differs (from the first order proof) and needs addressing is when Δk≤Δϵ\Delta_{k}\leq\Delta_{\epsilon}. Then, conditioned on Tϵ>kT_{\epsilon}>k, we have that either ∥∇f(Xk)∥≥ϵ\|\nabla f(X_{k})\|\geq\epsilon or λmin⁡(∇2f(Xk))≤−ϵ\lambda_{\min}(\nabla^{2}f(X_{k}))\leq-\epsilon and hence, from the definition of ζ\zeta in (63), we know that

where we also used that κbhm≥1\kappa_{bhm}\geq 1. Assume that Ik=1I_{k}=1 and Jk=1J_{k}=1, i.e., both the model and the estimates are good on iteration kk. Since the model mkm_{k} is κ\kappa-fully quadratic and δk≤Δϵ≤1\delta_{k}\leq\Delta_{\epsilon}\leq 1, then if (64) holds, we have

As the estimates {fk0,fks}\{f_{k}^{0},f_{k}^{s}\} are ϵF\epsilon_{F}-s.o. accurate, with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}, (66) and (67) imply that condition (46) in Lemma 4.4 and (54) in Lemma 4.7 hold, respectively. Thus in both cases, iteration kk is successful, i.e. xk+1=xk+skx_{k+1}=x_{k}+s_{k} and δk+1=max⁡{δmax,γδk}{\delta}_{k+1}=\max\{\delta_{max},\gamma{\delta}_{k}\}. If IkJk=0I_{k}J_{k}=0, then δk+1≥γ−1δk{\delta}_{k+1}\geq\gamma^{-1}{\delta}_{k} simply by the dynamics of Algorithm 2. Finally, observing that P{IkJk}≥p=αβP\{I_{k}J_{k}\}\geq p=\alpha\beta we conclude that (23) implies Assumption 2.1(ii). ∎

To show that Assumption 2.1(iii) holds, we need an additional assumption on the accuracy of the function estimates. We also require, for simplicity, an upper bound on the trust-region radius in Algorithm 2This restriction can be avoided if one allows a more involved discussion on dominating terms in the proofs of Lemmas 4.2–4.4 and 4.8, and in the proof of the main result..

There exists a constant κF\kappa_{F} such that at any iteration kk,

The upper bound δmax⁡\delta_{\max} in Algorithm 2 is chosen so that δmax⁡≤1\delta_{\max}\leq 1.

Note that the bound on the expectation of ∣Fk0−f(xk0)∣|F_{k}^{0}-f(x_{k}^{0})| and ∣Fks−f(xk+sk)∣|F_{k}^{s}-f(x_{k}+s_{k})|, in principle, implies that the estimates are β\beta-probabilistically ϵF\epsilon_{F}-s.o. accurate. However, for ϵF\epsilon_{F} to satisfy the conditions in Assumption 4.2 (b) conditions would have to be imposed on κF\kappa_{F}. Thus, for our purposes here, we choose to have any finite κF>0\kappa_{F}>0 and to impose the bound only on ϵF\epsilon_{F}.

Assumption 4.3(a) is needed for the case when we have a bad model and bad estimates, and when the (true) objective may increase after a successful step. Without this assumption, it is possible that the increase in the objective is at most of order δk2\delta_{k}^{2} (due to first order terms), while the decrease (on other successful steps) may be smaller, of order δk3\delta_{k}^{3} (due to second order terms). Such a situation would make it impossible to balance out the increase and decrease in the objective over the course of the algorithm in such a way to ensure that the stochastic process Φk\Phi_{k} decreases on average.

We now prove that Assumption 2.1(iii) holds for Algorithm 2.

Let Assumptions 4.1, 4.2 and 4.3 hold. Then, there exist probabilities α\alpha and β\beta and a constant Θ>0\Theta>0 such that, conditioned on Tϵ>kT_{\epsilon}>k, for each iteration kk of Algorithm 2, we have

where TϵT_{\epsilon} is defined in (62), and Φk\Phi_{k} in (61).

Moreover, under the particular choice of constants described on page 4.2, let α\alpha and β\beta satisfy

Then, ζ=20κbhm=20(κeh+L)\zeta=20\kappa_{bhm}=20(\kappa_{eh}+L) and Θ≥6⋅10−4η2min⁡{1,η2}\Theta\geq 6\cdot 10^{-4}\eta_{2}\min\{1,\eta_{2}\}.

Since (68) easily holds if Tϵ≤kT_{\epsilon}\leq k, we assume in what follows that Tϵ>kT_{\epsilon}>k and so τ(xk)>ϵ\tau(x_{k})>\epsilon, where τ(x)\tau(x) is defined in (38). We will consider two possible cases: τ(xk)≥ζδk\tau(x_{k})\geq\zeta\delta_{k} and τ(xk)<ζδk\tau(x_{k})<\zeta\delta_{k}, where ζ\zeta is defined in (63). We show that (68) holds in both cases and so for all k<Tϵk<T_{\epsilon}. Let ν∈(0,1)\nu\in(0,1) be such that

with C1C_{1} defined as in Lemma 4.3, C4C_{4} in Lemma 4.6, and C2C_{2} in Lemma 4.8. Note that on all successful iterations, xk+1=xk+skx_{k+1}=x_{k}+s_{k} and δk+1=min⁡{γδk,δmax}\delta_{k+1}=\min\{\gamma\delta_{k},\delta_{max}\} with γ>1\gamma>1, hence

On all unsuccessful iterations, xk+1=xkx_{k+1}=x_{k} and δk+1=1γδk\delta_{k+1}=\frac{1}{\gamma}\delta_{k}, i.e.

Case 1: τ(xk)=max⁡{∥∇f(xk)∥,−λmin⁡(∇2f(xk))}≥ζδk\tau(x_{k})=\max\{\|\nabla f(x_{k})\|,-\lambda_{\min}(\nabla^{2}f(x_{k}))\}\geq\zeta\delta_{k}, where ζ\zeta is defined in (63).

Ik=1I_{k}=1 and Jk=1J_{k}=1, i.e., both the model and the estimates are good on iteration kk. From the definition of ζ\zeta and Case 1, we know that either (64) or (65) hold. Since Ik=1I_{k}=1 and δmax⁡≤1\delta_{\max}\leq 1 (Assumption 4.3(b)), (64) and (65) imply that either condition (44) in Lemma 4.3 or condition (49) in Lemma 4.6 hold. Therefore, the trial step sks_{k} leads to a decrease in ff as in (45) or as in (50). Again from Ik=1I_{k}=1 and δmax⁡≤1\delta_{\max}\leq 1, (64) or (65) imply that (66) or (67) hold, respectively. As Jk=1J_{k}=1, and ϵF≤κef\epsilon_{F}\leq\kappa_{ef}, (66) and (67) imply that condition (46) in Lemma 4.4 or (54) in Lemma 4.7 hold, respectively. Thus in both cases, iteration kk is successful, i.e. xk+1=xk+skx_{k+1}=x_{k}+s_{k} and δk+1=max⁡{δmax,γδk}{\delta}_{k+1}=\max\{\delta_{max},\gamma{\delta}_{k}\}.

with C1C_{1} defined in Lemma 4.3. Since ∥∇f(xk)∥≥ζδk\|\nabla f(x_{k})\|\geq\zeta\delta_{k} we have

with b1b_{1} defined in (73), for ν∈(0,1)\nu\in(0,1) satisfying (71).

with C4C_{4} defined in Lemma 4.6. Again, since −λmin(∇2f(Xk))≥ζδk-\lambda_{min}(\nabla^{2}f(X^{k}))\geq\zeta\delta_{k} we have

with b1b_{1} defined in (73), for ν∈(0,1)\nu\in(0,1) satisfying (71).

Ik=1I_{k}=1 and Jk=0J_{k}=0, i.e., we have a good model and bad estimates on iteration kk. Then the analysis of case (a) applies, and either by Lemma 4.3 or 4.6, sks_{k} yields a sufficient decrease in ff. However, the step can be erroneously rejected, because of inaccurate probabilistic estimates, in which case we have an unsuccessful iteration and (73) holds. Since (71) holds, (73) applies whether the iteration is successful or not.

Ik=0I_{k}=0 and Jk=1J_{k}=1, i.e., we have a bad model and good estimates on iteration kk. In this case, again, iteration kk can be either successful or unsuccessful; in the latter case, (73) holds. In the former, since the estimates are ϵF\epsilon_{F}-accurate and (41) holds, then by Lemma 4.8 and Assumption 4.3(b), (59) holds with some C2>0C_{2}>0, and so,

due to the choice of ν\nu satisfying (71).

Ik=0I_{k}=0 and Jk=0J_{k}=0, i.e., both the model and the estimates are bad on iteration kk. Inaccurate estimates can cause the algorithm to accept a bad step, which may lead to an increase both in ff and in δk\delta_{k}. Hence in this case ϕk+1−ϕk\phi_{k+1}-\phi_{k} may be positive. We can derive a bound on the increase of f(xk)f(x_{k}) on successful steps in terms of the error of the estimates

On unsuccessful steps (73) still applies, which means that the right-hand side of (79) dominates and (79) holds whether the iteration is successful or not. Note that here, in (d), we have not used the definition of Case 1.

Now we are ready to take the expectation of Φk+1−Φk\Phi_{k+1}-\Phi_{k} in Case 1. Case (d) occurs with probability at most (1−α)(1−β)(1-\alpha)(1-\beta) and in that case ϕk+1−ϕk\phi_{k+1}-\phi_{k} is bounded from above as in (79). Cases (a), (b) and (c) occur otherwise and in those cases ϕk+1−ϕk\phi_{k+1}-\phi_{k} is bounded from above by b1<0b_{1}<0, with b1b_{1} defined in (73). Hence, we obtain

Choosing 0<α≤10<\alpha\leq 1 and 0<β≤10<\beta\leq 1 such that

Case 2: τ(xk)=max⁡{∥∇f(xk)∥,−λmin⁡(∇2f(xk))}<ζδk\tau(x_{k})=\max\{\|\nabla f(x_{k})\|,-\lambda_{\min}(\nabla^{2}f(x_{k}))\}<\zeta\delta_{k}, where ζ\zeta is defined in (63).

Jk=1J_{k}=1, namely, we have good estimates but the model may be bad. This case follows similarly to Case 1(c), and (78) holds on successful steps. Thus decrease b1b_{1} can again be guaranteed for Φk\Phi_{k} whether the iteration is successful or not.

Jk=0J_{k}=0, namely, we have bad estimates and the model may be bad too. In this case, the situation of Case 1(d) may occur and both ff and δk\delta_{k} may increase. Then we can upper bound the potential increase in Φk\Phi_{k} by (79) on both successful and unsuccessful stepsNote that under additional assumptions on κef\kappa_{ef} and η2\eta_{2}, one can further refine the analysis here and take into account the decrease in Φk\Phi_{k} that could then be achieved when Ik=1I_{k}=1.

Now we are ready to take the expectation of Φk+1−Φk\Phi_{k+1}-\Phi_{k} in Case 2. Case 2(i) occurs with probability at least β\beta, and then ϕk+1−ϕk\phi_{k+1}-\phi_{k} is bounded above by b1<0b_{1}<0, with b1b_{1} defined in (73). Case 2(ii) happens with probability at most (1−β)(1-\beta), and possible increase in Φk\Phi_{k} as in (79). We obtain

Thus, in conclusion, for ν\nu satisfying (71) and α\alpha and β\beta satisfying (80) and (83), the expected decrease in Φk\Phi_{k} in (68) holds, with

Now, let us particularize the constants as on page 4.2. Firstly, using η2≤18\eta_{2}\leq 18 and κbhm=κeh+L\kappa_{bhm}=\kappa_{eh}+L, we deduce that ζ:=20κbhm=20(κeh+L)\zeta:=20\kappa_{bhm}=20(\kappa_{eh}+L) satisfies (63). Furthermore, letting C1=C4=110C_{1}=C_{4}=\frac{1}{10} satisfies the conditions in Lemmas 4.3 and 4.6, and from Lemma 4.8 and particular choice of ϵF\epsilon_{F} we conclude, C2=180η2min⁡{1,η2}C_{2}=\frac{1}{80}\eta_{2}\min\{1,\eta_{2}\}. Thus, from (71) and ϵF≤κeh≤κbhm\epsilon_{F}\leq\kappa_{eh}\leq\kappa_{bhm}, ν\nu must satisfy

We let ν=320320+η2min⁡{1,η2}∈(0,1)\nu=\frac{320}{320+\eta_{2}\min\{1,\eta_{2}\}}\in(0,1). Then (80) is equivalent to

which is implied by (69). The bound (83) becomes

which is implied by (70). The value of Θ\Theta follows also using η2min⁡{1,η2}≤18\eta_{2}\min\{1,\eta_{2}\}\leq 18. ∎

Note the difference to first order results (Theorem 3.1): the effect of the stronger assumption on the estimates (Assumption 4.3 (a)) can be clearly seen in our results, in the presence of κF\kappa_{F} in the simplified bounds. Note that, due to the choice of constants and requirements on the accuracy of the estimates, the η2\eta_{2} terms are assumed to be smaller than terms involving Lipschitz constants L‾\overline{L} and κeh\kappa_{eh}, and hence, they remain present in the bounds. Given the definition of η2\eta_{2} in the algorithm, we can regard it as a means to control/ensure model quality.

The main complexity result for second order STORM follows.

Consider Algorithm 2 and the corresponding stochastic process. Let TϵT_{\epsilon} be defined as in (62) with ϵ∈(0,1]\epsilon\in(0,1]. Then, under the assumptions of Theorem 4.1, for sufficiently large α∈(0,1]\alpha\in(0,1] and β∈(0,1]\beta\in(0,1] with αβ>1/2\alpha\beta>1/2, we have

where Φ0\Phi_{0} is defined in (61) with k=0k=0, ν\nu in (71) and ζ\zeta in (63). Moreover, under the particular choice of constants described on page 4.2, and in Theorem 4.1, (85) becomes

where Θ≥6⋅10−4η2min⁡{1,η2}\Theta\geq 6\cdot 10^{-4}\eta_{2}\min\{1,\eta_{2}\}.

The validity of the Assumption 2.1(iii) follows from Theorem 4.1, with h(δ)=δ3h(\delta)=\delta^{3} and Δϵ\Delta_{\epsilon} defined in (63). Then Lemma 4.9 and the discussion preceding it imply that Theorem 2.2 applies, which provides (85). ∎

The lim⁡inf⁡\lim\inf-type probability one convergence result trivially follows.

Under conditions of Theorem 4.2 Algorithm 2 generates a subsequence convergent to a second order stationary point, almost surely.

As in Section 3.5, similar set-ups that sub-sample function, gradient and Hessian estimates can be provided that satisfy Assumptions 4.2 and 4.3 for second order STORM .

Conclusion

We have proposed a general framework based on a stochastic process that can be used to bound expected complexity of optimization algorithms. This framework can be applied beyond the algorithms discussed in this paper and has already been used in a new work on stochastic line search . We then applied this framework to establish that a stochastic trust region method, with dynamic stochastic estimates of the gradient, has essentially the same complexity as any other first order method in non convex setting. Similarly, given dynamic stochastic estimates of the gradient and Hessian the second order stochastic trust region method converges to second order stationary point and its expected complexity matches the deterministic case. While our algorithm requires the stochastic estimates to be progressively more accurate, it never requires to compute the full gradient, hence it applies in purely stochastic settings.

References

Appendix

This Appendix contains proofs of several lemmas that are novel, but whose proofs are similar to existing results. We include them here for completeness.

Proof of Lemma 4.2. Using the optimal decrease condition (37), the upper bound on model Hessian from Lemma 4.1, and the fact that ∥gk∥≥κbhmδk\|g_{k}\|\geq\kappa_{bhm}{\delta}_{k}, we have

Since the model is κ\kappa-fully quadratic, the improvement in ff achieved by sks_{k} is

where the last inequality is implied by δk2≤δk≤κscd8κef∥gk∥{\delta}_{k}^{2}\leq{\delta}_{k}\leq\frac{\kappa_{scd}}{8\kappa_{ef}}\|g_{k}\|. □\Box

Proof of Lemma 4.3. The definition of a κ\kappa-fully-quadratic model yields that

Since condition (44) implies that ∥∇f(xk)∥≥max⁡{κbhm+κeg,8κefκscd+κeg}δk\|\nabla f(x_{k})\|\geq\max\left\{\kappa_{bhm}+\kappa_{eg},\frac{8\kappa_{ef}}{\kappa_{scd}}+\kappa_{eg}\right\}{\delta}_{k}, using δk≤1{\delta}_{k}\leq 1, we have

Hence, the conditions of Lemma 4.2 hold and we have

Since ∥gk∥≥∥∇f(x)∥−κegδk\|g_{k}\|\geq\|\nabla f(x)\|-\kappa_{eg}\delta_{k} in which δk{\delta}_{k} satisfies (44), we also have

Combining (86) and (87) yields (45). □\Box

Proof of Lemma 4.4. Since δk≤∥gk∥κbhm{\delta}_{k}\leq\frac{\|g_{k}\|}{\kappa_{bhm}}, the model decrease condition (37) and the uniform bound on HkH_{k} under Lemma 4.1 immediately yield that

The model mkm_{k} being (κef,κeg,κeh)(\kappa_{ef},\kappa_{eg},\kappa_{eh})-fully quadratic implies that

Since the estimates are ϵF\epsilon_{F}-s.o.accurate with ϵF≤κef\epsilon_{F}\leq\kappa_{ef}, we obtain

where we have used the assumptions δk2≤δk≤κscd(1−η1)8κef∥gk∥{\delta}_{k}^{2}\leq{\delta}_{k}\leq\frac{\kappa_{scd}(1-\eta_{1})}{8\kappa_{ef}}\|g_{k}\| to deduce the last inequality. Hence, ρk≥η1\rho_{k}\geq\eta_{1}. Moreover, since ∥gk∥≥η2κbhmδk\|g_{k}\|\geq\eta_{2}\kappa_{bhm}{\delta}_{k}, then τkm≥min⁡{∥gk∥,∥gk∥κbhm}≥η2δk\tau_{k}^{m}\geq\min\left\{\|g_{k}\|,\dfrac{\|g_{k}\|}{\kappa_{bhm}}\right\}\geq\eta_{2}\delta_{k} and the kk-th iteration is successful. □\Box