Analysis of nonsmooth stochastic approximation: the differential inclusion approach
Szymon Majewski, Błażej Miasojedow, Eric Moulines
Introduction
Stochastic approximation algorithms are stochastic processes defined iteratively as
A powerful method to analyze stochastic gradient algorithm, introduced in the early works by and is the ordinary differential equation (ODE) method. The ODE method has led to an enormous literature; see for example , and the references therein. The ODE method can be informally summarized as follows: first we rewrite
The classical stochastic approximation algorithm update rule is replaced by a stochastic recursive inclusion:
where is a point-to-set map and is defined in (1). Such algorithms play an important role in game theory, as illustrated in where numerous examples of stochastic recursive inclusions are introduced.
have shown that the ”mean-limit” approach leading to the ODE method in the smooth case can be extended to the analysis of stochastic recursive inclusion. In this case, the limit ODE is replaced by a solution of the differential inclusion
In this paper, we will also consider proximal algorithms, which have become an important tool in nonsmooth optimization problems; the literature in this field is also huge, see for example ). Proximal algorithms with stochastic updates have been proposed and studied in recent years. One such algorithm is Proximal Stochastic Gradient Descent (proxSGD), that optimizes composite convex function where is a continuously differentiable function with Lipschitz-gradients and is a ”proximable” function (e.g. is lower semi-continuous and convex, but this notion can be extended to nonconvex functions). The proxSGD algorithm alternates between stochastic gradient update for and deterministic proximal step for . This above optimization problem plays a fundamental role in many machine learning problems, ranging from convex optimization such as convex regression problem with sparsity inducing penalties like LASSO to highly nonconvex problem such as optimizing the weights of deep neural networks. Numerous papers have been devoted to the case when and are both convex, gradient Lipschitz and lower semi-continuous; see for example . Atchade et al. have extended these results in the Markovian noise case. In recent years, triggered by the surge deep learning, the nonconvex case has started to attract many research efforts, at least in the smooth case ( gradient Lipschitz and ); see for example and the references therein. For the nonsmooth and nonconvex case, the results are still partial. Ghadimi et al. considered the case where is differentiable but possibly nonconvex and is non-differentiable but convex. They have analyzed the deterministic proximal gradient algorithm (where the full gradient is computed at each iteration). They have also extended their results to the stochastic case; Reddi et al. provides rates of convergence.
We consider in this paper nonconvex and nonsmooth minimization problems. We establish the convergence of stochastic inclusion equation generalizing (1) by allowing implicit steps and projections on a closed compact convex set at each iteration. Our results generalize . We also discuss the stability of the limit differential inclusion by means of locally Lipschitz continuous and regular Lyapunov functions (see Definition Definition). We in particular establish a characterization of the possible limit point of the stochastic approximation algorithm as the set of zeros of an upper-bound of the set-valued Lie derivative (see Definition Definition) of the Lyapunov function. We then apply our results to the analysis of the proximal stochastic gradient descent for the composite minimization problem , under assumptions on the noise sequence analogous to those commonly used for the SGD in the smooth nonconvex case. We also show that can play the role of a Lyapunov function. We finally analyse a projected version of stochastic subgradient algorithm.
The paper is organized as follows. In Section 2, we introduce our main assumptions and notations and introduce the proximal stochastic gradient and projected subgradient algorithms. In Section 3, we state and prove our main convergence results under the assumption that the iterates are stable. In Section 4, we extend these convergence results to the case where the updates are projected on a compact convex set. In Section 5, we consider applications of our main results to ProxSGD and projected stochastic subgradient. Finally in Section 6 we present postponed proofs.
Assumptions and Notations
In this section we introduce definitions and notations.
By convention for a convex closed set and a given set , by we denote the projection of onto , defined as .
The condition is satisfied if and only if there exists such that
[11, Definition III] deal with sequence satisfying a recursion of the form
We now show that the PAD and -PPAD formalism cover the proximal stochastic gradient descent (ProxSGD) and the stochastic (sub)gradient algorithms for nonsmooth and nonconvex minimization problems. First we introduce some additional definitions and notations.
Similarly to subgradient, the Clarke generalized gradient is a set-valued generalization of the gradient. In particular, when function is continuously differentiable at some point , then we have . Furthermore, if the function is convex and locally Lipschitz and belongs to the interior of its domain, then the Clarke generalized gradient of coincides with the subgradient.
It is shown in [16, Propositions 2.1.2] that if is Lipschitz on a neighborhood of with Lipschitz constant then is a non-empty set, compact, convex, and for any , . By [16, Proposition 2.1.5], is upper hemicontinuous at . For any compact set , showing that is locally bounded. For proofs of those results and additional properties of Clarke generalized gradient, we refer the reader to .
where is the generalized directional derivatives (see Definition Definition). \proofbox
It may happen that the usual directional derivative exists, but does not coincide with the generalized directional derivative. The classical example is . As shown in [16, Proposition 2.3.6] every convex locally Lipschitz function is regular. The same property obviously holds for any continuously differentiable function. Note finally that if the functions are regular at , then for any nonnegative weights , is also regular at , see [16, Proposition 2.3.6]. \proofbox
Now we are ready to discuss the proximal gradient descent and projected subgradient algorithms.
The characterization of the minimum by the Clarke generalized gradient (see [16, Proposition 2.3.2]) yields
We will analyse the convergence of PAD (see (3)) under the following assumptions:
Condition (A(A1)) is a rather mild regularity condition. Upper hemicontinuity replaces the continuity of the vector field which plays a key role in the classical theory of stochastic approximation . The requirement for to be convex-compact valued and locally bounded might be less obvious, but this assumption is commonly used in nonsmooth analysis. This is not a serious limitation for the minimization problems we have primarily in mind.
The assumptions (A(A2), A(A3)) are usual in stochastic approximation literature . It is worth noting, that condition (A(A3)) allows perturbations sequences which have random and deterministic components and hence our results can be used for proving almost sure convergence for ProxSGD for which the proximal operator is computed numerically and is therefore inexact (although in our framework the deterministic noise should vanish asymptotically faster than step size). The assumptions (A(A4)) allows to cover both explicit and implicit discretization of differential inclusions as illustrated in Example Example.
Convergence of Perturbed Approximate Discretisation
In this section, we state our main convergence results for PAD. First in Theorem Theorem we show that a translated and interpolated version of the PAD converges to a solution of a differential inclusion. Further in Theorem Theorem we combine these results with Lyapunov stability conditions to obtain convergence of the iterates to the set of stationary points of the differential inclusion.
Without loss of generality we can assume that (if this is not true, we can just modify and ).
Therefore for we have . Since was arbitrary positive number, we get that . T Hence, for all , there exists such that
where is defined in (8). By assumption (A(A2)) is well defined and converges to as .
Moreover, since and , by (15) we have
By assumption converges to , so also goes to as . Therefore by assumption (A(A4)) the first part of the RHS of (18) converges to . By (16), the second term in the RHS of (18) is equal to
which goes to zero by uniform convergence of to . Finally, continuity of implies that the last term of (18) also converges to . All together we have therefore established that
The limit is a solution of differential inclusion . \proofbox
Combining Theorem Theorem with stability properties of underlying differential inclusion we establish convergence of PADs. To state the result we need to define a set valued Lie derivative, introduced in .
where is the Clarke generalized gradient of , cf. Definition Definition. \proofbox
The Lie derivative plays important role in analysis of stability of solution of differential inclusions. We in particular will used an important property stated in [6, Lemma 1].
where is the set-valued Lie derivative of with respect to , cf. Definition Definition. \proofbox
and set .
The set is nonempty. \proofbox
If the by upper semicontinuity of and compactness of we would have for some . Therefore function must decrease at a rate at least , and thus . But this is a contradiction with the assumption that is bounded from below. \proofbox
Since is continuous and is compact,
and hence . By upper semicontinuity of and compactness of we get that there exists such that , and, using Lemma Lemma, we conclude that for almost every , . This means, that
Convergence of Projected Perturbed Approximate Discretisation
As in the proof of Theorem Theorem, we assume . We denote by
and we define the functions for any
we get that . Next, since
and for every , we get .
Any limit of converging subsequence is a solution of the projected differential inclusion . \proofbox
We use the following characterization of the solution of projected differential inclusion given in [5, Chapter 5, Section 6, Propositions 1 and 2]:
For all we have .
For almost every there exists such that,
Let be such, that . Since , where is defined in (15), .
we get .
and set .
The proof is along the same lines as the proof of Theorem Theorem. \proofbox
Applications
In this section we apply the result from previous sections to projected ProxSGDand projected subgradient descent algorithm.
Stochastic proximal gradient is a natural extension of Proximal Gradient algorithm to the case where the gradient cannot be computed exactly and is therefore affected by some errors. More specifically, we want to optimize a composite function of form
We consider two versions of the projected ProxSGD algorithm which are given by
Those two approaches to projection are not equivalent, and depending on and one might be easier to compute than the other. Consider the following assumptions:
To illustrate our derivations, we consider now two possible choices of sparsity inducing penalties .
where is the soft thresholding operator (here is the positive part of ). The proximal operator for is given by (see [33, Section 2.1])
The SCAD penalty () can be handled along the same lines (the expression for the proximal function can be found in [14, Section 2]). \proofbox
We denote by the set-valued map . The Clarke gradient of a locally Lipschitz function is convex-compact valued and locally bounded (see [16, Proposition 2.1.2]) and is upper hemi-continuous (see [16, Proposition 2.1.5]. Since is continuous, this implies that satisfies (A(A1)). By [16, Corollary 2.3.2], .
With this notation (31) can be written as
By [16, corollary of Proposition 2.4.3, p. 52], . Therefore there exists
The normal cone to at consists of vectors , such that for all we have . Since and using (34), (36) implies that for all ,
where is defined in (35). Setting and we get
it is enough to show that . Plugging into (37), we get that:
and using the Cauchy-Schwartz and triangle inequalities and (34), we obtain
Denoting by and by we get that
We now chack (A(A3)). The perturbation may be decomposed as where and . Since is continuous, Assumption (A(A3)) is satisfied if
Note that, since is Lipschitz, [16, Proposition 2.1.2-(a)] shows that for all , . Boundedness of on the set and assumptions (P(P2)) and (P(P3)) implies that
Because is a projection of on the set we have
showing that (A(A4)) is satisfied. \proofbox
Applying our results from Section 4, we now show that both versions of the projected proximal gradient algorithms converge.
To apply Theorem Theorem, we show that is a Lyapunov function for , . Under the stated assumptions, is locally Lipschitz regular and by [16, Corollary 2 of Proposition 2.3.3]
We now compute the Lie derivative of with respect to the field see Definition Definition). Let . Suppose that there exists . Then there exists , such that for all , . Let be such that . Note that , which implies and
Applying [5, Proposition 0.6.2], we get that . Therefore, if , then for some . Hence, for any , is either empty, or contains only non-positive elements.
For any , [16, Proposition 2.1.2] shows that is non-empty, convex and compact. Define for any as follows
By [5, Proposition 0.6.4] we get for any ,
Under (A(A1)), is upper hemicontinuous compact-convex valued. On the other hand, by [5, Chapter 5, Section 1, Theorem 1], has closed graph. Hence, the map has closed graph by [5, Proposition 1.1.2, p. 41]. By Lemma Lemma the function is upper semicontinuous. Thus we have an upper semicontinuous function , such that for all :
2 Online proximal stochastic gradient descent algorithm
In the online learning case, the gradient of the function cannot be computed but that a noisy version of the gradient is available To make the discussion simple, we assume that
We are considering the two following stochastic approximation procedures
3 Monte Carlo Proximal stochastic gradient descent algorithm
In this section, we still consider the composite minimization problem (42). We assume that is continuously differentiable and that for all satisfies
When sampling directly is doable, then an obvious choice is to use a naive Monte Carlo estimator which amounts to sample a batch independently of the past values of the parameters and of the past draws i.e. independently of the -algebra
Conditionally to , is an unbiased estimator of .
When direct sampling from is not an option, we may still construct a Markov kernel with invariant distribution . Monte Carlo Markov Chains (MCMC) provide a set of principled tools to sample from complex distributions over large dimensional spaces. In such case, conditional to the past, is a realization of a Markov chain with transition kernel and started from (the last sample draws in the previous minibatch).
We refer the reader to for the definitions and basic properties of Markov chains.
In this section, we assume that is a Monte Carlo approximation of the expectation :
for all , conditionally to the past, is a Markov chain started from and with transition kernel (we set ), where for all , is a Markov kernel with invariant distribution .
From a mathematical standpoint, the Markovian setting is trickier than the fixed batch size, because is no longer an unbiased estimator of , i.e. the bias defined by
There exists , and a measurable function such that
Sufficient conditions for the uniform-in- ergodic behavior are given e.g. in [22, Lemma 2.3], in terms of aperiodicity, irreducibility and minorization conditions on the kernels . Examples of MCMC kernels satisfying this assumption can be found in [2, Proposition 12], [38, Proposition 15].
The kernels and the stationary distributions are locally Lipschitz with respect to , i.e. for any compact set and any there exists such that
The proof follows along the same lines as [30, Proof of Lemma 27]. However for completeness we give a detailed proof in Appendix A. \proofbox
4 Projected stochastic subgradient descent algorithm
The convergence of stochastic subgradient algorithm for regular functions can be easily deduced from Section 3. Here, we show that projected stochastic subgradient descent algorithm also fits into our framework and its convergence can be established based on the results of Section 4.
Applying our results from Section 4, we now show that projected stochastic subgradient algorithms converge.
The proof follows along the sime lines as proof of Theorem Theorem. \proofbox
Note, that adaptation of results from Section 5.2 and Section 5.3 to the case of projected stochastic subgradient descent algorithm is straightforward.
Proofs
In this section we introduce some notations and preliminary facts used in the proofs of results from Section 3 and Section 4, as well as some auxiliary definitions and theorems.
which means that is upper semicontinuous. \proofbox
By [16, Proposition 2.3.6] a finite linear combination (by nonnegative scalars) of functions regular at is regular at . The proof then follows by noting that, for any , the function is regular at . \proofbox
Let be the D-subdifferential of at (see [17, chapter 3.4, subsection D-differential] for definition). Then according to [17, Proposition 4.10] , the inequality (51) is satisfied for if and only if . On the other hand, [17, Proposition 4.8, part (b)] implies that for a locally Lipschitz function we have if and only if is regular. Combined, these two facts conclude the proof. \proofbox
References
Appendix A Proof of Proposition Proposition
Since is projection of on set , by the triangle inequality we get
Note that, since is Lipschitz, [16, Proposition 2.1.2-(a)] shows that for all , . Boundedness of on the set and assumption (B(B2)) concludes the proof. \proofbox
In this proof, is a constant whose value may change upon each appearance. Observe that it is enough to show almost surely, where by construction
Geometric ergodicity (B(B1)) in turn implies the existence of a solution of the Poisson equation, and also provide bounds on the growth of this solution; see [3, Lemma 13]. For , we set . For any there exists a solution to the Poisson equation
and there exists a constant such that for any and
Therefore, since we conclude that converges almost surely.
Decompose with
Finally, from (52), (P(P2)) and we deduce that and also converges to zero almost surely and that completes the proof. \proofbox