An optimal randomized incremental gradient method
Guanghui Lan, Yi Zhou
Introduction
The basic problem of interest in this paper is the convex programming (CP) problem given by
and is a given constant. Hence, the objective function is strongly convex whenever . For notational convenience, we also denote and . It is easy to see that for some ,
Throughout this paper, we assume subproblems of the form
are easy to solve. CP given in the form of (1.1) has recently found a wide range of applications in machine learning, statistics, and image processing, and hence becomes the subject of intensive studies during the past few years.
Stochastic (sub)gradient descent (SGD) (a.k.a. stochastic approximation (SA)) type methods have been proven useful to solve problems given in the form of (1.1). SGD was originally designed to solve stochastic optimization problems given by
Note however, that there is a significant gap on the complexity bounds between SGDs and deterministic FOMs if ’s are smooth convex functions. For the sake of simplicity, let us focus on the strongly convex case when and let be the optimal solution of (1.1). In order to find a solution s.t. , the total number of gradient evaluations for ’s performed by optimal FOMs can be bounded by
which was first achieved by the well-known Nesterov’s accelerated gradient method Nest83-1 ; Nest04 , see also relevant extensions in Nest13-1 ; BecTeb09-2 ; tseng08-1 . On the other hand, a direct application of optimal SGDs to the aforementioned stochastic optimization reformulation of (1.1) would yield an
iteration complexity bound on the number of gradient evaluations for ’s, which was first achieved by the accelerated stochastic approximation method (Lan10-3 ; GhaLan12-2a ; GhaLan13-1 ). Here denotes variance of the stochastic gradients. Clearly, the latter bound is significantly better than the one in (1.7) in terms of its dependence on , but much worse in terms of its dependence on accuracy and a few other problem parameters (e.g., and ).
It should be noted that the optimality of (1.8) for general stochastic programming (1.6) does not preclude the existence of more efficient algorithms for solving (1.1), because (1.1) is a special case of (1.6) with finite support . Last few years have seen very active and fruitful research in this field (e.g., SchRouBac13-1 ; JohnsonZhang13-1 ; DefBacLac14-1 ; ShaZhang15-1 ; Yuchen14 ). In particular, Schmidt, Roux and Bach SchRouBac13-1 presented a stochastic average gradient (SAG) method, which recursively computes an estimator of by aggregating the gradient of a randomly selected with some other previously computed gradient information. They proved that the complexity of SAG is bounded by , see also Johnson and Zhang JohnsonZhang13-1 and Defazio et al. DefBacLac14-1 for similar complexity results for solving (1.1). In a related but different line of research, Shalev-Shwartz and Zhang ShaZhang15-1 studied a special class of CP problems given in the form of (1.1) with given by , where denotes an affine mapping. Under the assumption that , they presented an accelerated stochastic dual coordinate ascent (A-SDCA) method, obtained by properly restarting a stochastic coordinate ascent method in ShaZhang13-1 applied to the dual of (1.1). Shalev-Shwartz and Zhang show that the iteration complexity of this method can be bounded by However, each iteration of A-SDCA requires, instead of the computation of , the solution of a subproblem given in the form of
where denotes the conjugate function of . Moreover, these methods were also designed for solving a more special class of problems than (1.1). More recently, Lin, Lu, and Xiao LinLuXiao14-1 proposed to apply the accelerated coordinate descent methods by Nesterov Nest10-1 , and Fercoq and Richtárik’s fr13 to obtain similar results for solving these “regularized empirical loss functions” as in ShaZhang15-1 . Zhang and Xiao Yuchen14 had also obtained similar results by using different stochastic primal-dual coordinate decomposition techniques.
In this paper, we focus on randomized incremental gradient methods that can access the first-order information of only one randomly selected smooth component at each iteration (see Bertsekas Bertsekas10-1 for an introduction to incremental gradient methods). It should be noted that while the algorithms in SchRouBac13-1 ; JohnsonZhang13-1 ; DefBacLac14-1 belong to incremental gradient methods, generally speaking, the dual coordinate algorithms in LinLuXiao14-1 ; ShaZhang15-1 ; Yuchen14 cannot be considered as incremental gradient methods because they require the solutions of a different subproblem rather than the computation of the gradient of . The previous attempts to improve the complexity of the existing incremental gradient methods, e.g., based on the extrapolation idea in Nesterov Nest83-1 , however, turned out to be tricky and unsuccessful, see Section 1.2 of Bertsekas Bertsekas10-1 and Section 5 of Agarwal and Bottou AgrBott14-1 for more discussions. Another important yet unresolved issue is that there does not exist a valid lower complexity bound for randomized incremental gradient methods in the literature. Hence, it remains unknown what would be the best possible performance that one can expect for these types of methods. Regarding this question, Agarwal and Bottou AgrBott14-1 recently suggested a lower complexity bound for solving problems given in the form of (1.1). However, as pointed out by them in a recent ISMP talk in 2015, the lower complexity bound in AgrBott14-1 is deterministic by construction, and hence cannot be used to justify the optimality or suboptimality for the randomized incremental gradient methods in SchRouBac13-1 ; JohnsonZhang13-1 ; DefBacLac14-1 or dual coordinate methods in LinLuXiao14-1 ; ShaZhang15-1 ; Yuchen14 .
Our contribution in this paper mainly lies on the following several aspects. Firstly, we present a new class of deterministic FOMs, referred to as the primal-dual gradient (PDG) methods, which can achieve the optimal black-box iteration complexity in (1.7) for solving (1.1). The novelty of these methods exists in: 1) a proper reformulation of (1.1) as a primal-dual saddle point problem and 2) the incorporation of a new non-differentiable prox-function (or Bregman distance) based on the conjugate functions of in the dual space. As a consequence, we are able to show that the PDG method covers a variant of the well-known Nesterov’s accelerated gradient method as a special case. In particular, the computation of the gradient at the extrapolation point of the accelerated gradient method is equivalent to a primal prediction step combined with a dual ascent step (employed with the aforementioned dual prox-function) in the PDG method. While it is often difficult to interpret Nesterov’s method, the development of the PDG method allows us to view this method as a natural iterative buyer-supplier game. Such a game-theoretic view of the accelerated gradient method seems to be new in the literature. In fact, the obtained complexity results for the PDG method are slightly stronger than the one in (1.7) and those in Nest83-1 ; Nest04 for Nesterov’s accelerated gradient method, because a stronger primal-dual termination criterion has been used in our analysis.
Secondly, we develop a randomized primal-dual gradient (RPDG) method, which is an incremental gradient method using only one randomly selected component at each iteration. A variant of PDG, this algorithm incorporates an additional dual prediction step before performing the primal descent step (with a properly defined primal prox-function). We prove that the number of iterations (and hence the number of gradients) required by RPDG is bounded by
both in expectation and with high probability. The complexity bounds of the RPDG method are established in terms of not only the distance from the iterate to the optimal solution, but also the primal optimality gap based on the ergodic mean of the iterates. In comparison with the accelerated stochastic dual coordinate ascent method in ShaZhang15-1 , RPDG deals with a wider class of problems and can be applied to the cases when ’s involve a more complicated composite structure (see examples in Bertsekas10-1 ) and/or a more general regularization term that is strongly convex with respect to an arbitrary norm (see open problems in Section 7 of ShaZhang15-1 ). Moreover, each iteration of RPDG only involves the computation , rather than the more complicated subproblem in (1.9), which sometimes may not have explicit solutions ShaZhang15-1 (e.g., the logistics regression problem). The RPDG method also admits an interesting game theoretic interpretation, implying that by properly incorporating randomization, the buyer and supplier can reach the equilibrium with possibly fewer price changes at the expense of more order transactions.
whenever the dimension is sufficiently large. This bound is obtained by carefully constructing a special class of separable quadratic programming problems and tightly bounding the expected distance to the optimal solution for any arbitrary distribution used to choose at each iteration. Comparing (1.10) with (1.11), we conclude that the complexity of the RPDG method is optimal if is large enough. To the best of our knowledge, this is the first time that such a lower complexity bound has been presented for randomized incremental gradient methods in the literature. As a byproduct, we also derived a lower complexity bound for randomized block coordinate descent methods by utilizing the separable structure of the aforementioned worst-case instances. These methods have been intensively studied recently, but a valid lower complexity bound is still missing in the literature.
Finally, we generalize RPDG for problems which are not necessarily strongly convex (i.e., ) and/or involve structured nonsmooth terms . We show that for all these cases, the RPDG can save times gradient computations (up to certain logarithmic factors) in comparison with the corresponding optimal deterministic FOMs. In particular, we show that when both the primal and dual of (1.1) are not strongly convex, the total number of iterations performed by the RPDG method can be bounded by (up to some logarithmic factors), which is times better, in terms of the total number of dual subproblems to be solved, than Nesterov’s smoothing technique Nest05-1 , Nemirovski’s mirror-prox method Nem05-1 , or Chambolle and Pock’s primal-dual method ChamPoc11-1 . It seems that this complexity result has not been obtained before in the literature.
It is worth mentioning a few relevant works to our development. The most two related ones are conducted independently by Dang and Lan DangLan14-1 , and Zhang and Xiao Yuchen14 . Both of these papers deal with randomized variants of the primal-dual method presented by Chambolle and Pock ChamPoc11-1 (see also extensions in CheLanOu13-1 ) for solving saddle point problems. Zhang and Xiao’s development Yuchen14 was based on a variant of the primal-dual method for solving strongly convex saddle point problems ChamPoc11-1 . They were able to show that a block-wise randomized version of the algorithm can achieve similar complexity as the A-SDCA method in ShaZhang15-1 . Since Zhang and Xiao’s algorithm targets for solving a similar class of problems and requires the solutions of a similar subproblem to ShaZhang15-1 , it appears that the aforementioned possible advantages of RPDG over A-SDCA are also applicable to the stochastic primal-dual coordinate method in Yuchen14 . Moreover, the complexity bound of Zhang and Xiao’s algorithm is only established in terms of the Euclidean distances of the iterate , to the optimal solution. They did not deal with the convergence of the ergodic mean of iterates. On the other hand, Dang and Lan’s work was motivated by the observation in ChenHeYeYuan13-1 that a direct extension of the alternating direction method of multiplier (ADMM) does not converge for multi-block problems. Their work in DangLan14-1 then focuses on the non-strongly convex case and shows that a randomized primal-dual method, which is equivalent to a randomized pre-conditioned ADMM for linear constrained problems, does converge for multi-block problems. Without incorporating the aforementioned dual prediction step, the complexity obtained in DangLan14-1 is times worse than Chambolle and Pock’s method. Nevertheless, this is the first time that randomized algorithms for saddle point optimization with an complexity has been presented in the literature. More recently, close to the end of the preparation of this paper, we notice that Lin, Mairal, and Harchaoui LinMaiHar15-1 in a concurrent work presented a catalyst scheme that can be used to accelerate the SAG method in SchRouBac13-1 and thus possibly achieve the complexity bound in (1.10) (under the Euclidean setting). While their approach is an indirect one obtained by properly restarting SAG (or other “non-accelerated” first-order methods), the proposed randomized primal-dual gradient method is a direct approach with a “built-in” acceleration. Also none of these works DangLan14-1 ; Yuchen14 ; LinMaiHar15-1 discussed the lower complexity bound for randomized methods.
This paper is organized as follows. We first study the deterministic primal-dual method in Section 2. Section 3 is devoted to the design and analysis of the randomized primal-dual method for the strongly convex case, as well as the development of the lower complexity bound in (1.11). In Section 4, we generalize the RPDG method to different classes of CP problems that are not necessarily strongly convex. Important technical results and proofs of the main theorems in Sections 2 and 3 are provided in Section 5. Some brief concluding remarks are made in Section 6.
An optimal primal-dual gradient method
Our goal in this section is to present a novel primal-dual gradient (PDG) method for solving (1.1), which will also provide a basis for the development of the randomized primal-dual gradient methods in later sections. We establish the optimal convergence of this algorithm in terms of the primal-dual optimality gap under the assumption that the gradient of is computed at each iteration. We show that PDG generalizes one variant of the well-known Nesterov’s accelerated gradient method, and allows a natural game interpretation, and hence that the latter algorithm also admits a similar interpretation.
In this subsection, we discuss both primal and dual prox-functions (proximity control functions) in the primal and dual spaces, respectively.
where is an arbitrary subgradient of at . Clearly, by the strong convexity of , we have
Note that the prox-function described above generalizes the Bregman’s distance in the sense that is not necessarily differentiable (see Breg67 ; AuTe06-1 ; BBC03-1 ; Kiw97-1 and references therein). Throughout this paper, we assume that the prox-mapping associated with , , and , given by
where denotes the normal cone of at . Once such a satisfying the above relation is identified, we will use it as a subgradient when defining in the next iteration.
It is clear that is strongly convex with modulus w.r.t. . Therefore, we can define its associated dual prox-functions and dual prox-mappings as
for any . Again, may not be uniquely defined since is not necessarily differentiable. Instead of choosing similarly to , we can explicitly specify such selections as will be discussed later in this paper.
The following simple result shows that the computation of the dual prox-mapping associated with is equivalent to the computation of .
In view of the definition of in (2.5), we have
2 Primal-dual gradient method, Nesterov’s method, and a game interpretation
By the definition of in (2.4), problem (1.1) is equivalent to:
In order to implement the above primal-dual gradient method, it is more convenient to rewrite step (2.9) in a form involving the computation of gradient rather than the dual prox-mapping . In order to do so, we shall specify explicitly the selection of the subgradient in (2.9). Denoting , we can easily see from that . Using this relation and letting in (see (2.5)), we then conclude from Lemma 1 that for any , (2.9) reduces to
With the above selection of the dual prox-function, we can specialize the primal-dual gradient method as follows.
The above PDG method is related to the well-known Nesterov’s accelerated gradient (AG) method. Let us focus on a simple variant of the AG method that has been extensively studied in the literature (e.g., Nest04 ; tseng08-1 ; Lan10-3 ; GhaLan12-2a ; GhaLan13-1 ; GhaLan13-2 ). Given , this AG algorithm updates by
for some . By (2.15) and (2.17), we have
Therefore, (2.15) is equivalent to (2.11) and (2.12) with and . Moreover, (2.16) is identical to (2.14)(and (2.10)), and (2.17) basically defines the output of the AG algorithm as an ergodic mean of the iterates . We then conclude that the above variant of Nesterov’s AG method is a special case of Algorithm 2 (and Algorithm 1). It should be noted, however, that Algorithm 1 provides more flexibility in the specification of parameters, which will be used later in the development of the RPDG method. Moreover, the presentation of the PDG method helps us to reveal a natural game interpretation out of the intertwined and somehow mysterious updating of the three search sequences in the AG method.
Algorithm 1 is also closely related to Chambolle and Pock’s primal-dual method for solving saddle point problems ChamPoc11-1 , which explains the origin of its name. Two versions of primal-dual methods were discussed in ChamPoc11-1 . One is designed for solving general saddle point problems without assuming the strong convexity of and the other one is to deal with the case when is strongly convex by incorporating an additional extrapolation step. As pointed out in Remark 3 of ChamPoc11-1 , the rate of convergence for the latter primal-dual method is only suboptimal for solving (1.1) as it uses a weaker termination criterion. On the other hand, the PDG method does not involve any additional extrapolation steps and so it shares a similar scheme to the basic version of the primal-dual method in ChamPoc11-1 . Moreover, the original primal-dual methods in ChamPoc11-1 do not employ general prox-functions, which, as shown in Lemma 1, is crucial to relate the dual step (2.9) to the computation of the gradients. It should be noted that some recent extensions of the primal-dual method in CheLanOu13-1 ; DangLan14-1 ; ChamPoc14-1 indeed consider the incorporation of prox-functions, but restricted to problems without strong convexity. Hence, none of these earlier primal-dual methods can be viewed as a generalized accelerated gradient method.
3 Convergence properties of the primal-dual gradient method
Our goal in this subsection is to show that Algorithm 1 exhibits an optimal rate of convergence for solving problem (1.1). It is worth mentioning that our analysis significantly differs from the previous studies on optimal gradient methods and those on primal-dual methods for saddle point problems.
Given a pair of feasible solutions and of (2.7), we define the primal-dual gap function by
It can be easily seen that (resp., ) is an optimal solution of (2.7) if and only if for any (resp., for any ). Therefore, one can assess the solution quality of by the primal-dual optimality gap:
It should be noted that may not be well-defined, for example, when is unbounded and is not strictly convex. In these cases, we can define a slightly modified primal-dual gap
for an arbitrary optimal solution of (1.1). Since is strongly convex, is well-defined.
The following result establishes some relationship between the primal optimality gap and the above primal-dual optimality gaps.
Let be a given pair of feasible solutions of (2.7) and denote . Also let be a pair of optimal solutions of (2.7). Then we have
It follows from the definitions of , and the gap function that
Relation (2.22) follows directly from the definitions of and .
Theorem 2.1 below describes the main convergence properties of the PDG method. More specifically, we provide in Theorem 2.1.a) a constant stepsize policy which works for the strongly convex case where , and a different parameter setting that works for the non-strongly convex case with in Theorem 2.1.b). Note that for the strongly convex case, we estimate the solution quality for the iterates , as well as that for their ergodic mean
for some , while only establishing the error bounds for for the non-strongly convex case. We put the proof of Theorem 2.1 in Section 5 since it shares many basic elements with the convergence analysis of the RPDG method.
Let be an optimal solution of (1.1), and be defined in (2.10) and (2.23), respectively.
Suppose that and that , , and are set to
Suppose that , , and are set to
Observe that when the algorithmic parameters are set to (2.24), by using an inductive argument, we can easily show that
In other words, can be written as a convex combination of and hence for any . Similarly, when the algorithmic parameters are set to (2.28), we can show by using induction that
In view of the results obtained in Theorem 2.1, the primal-dual gradient method is an optimal method for convex optimization. In fact, the rates of convergence in (2.26), (2.27), (2.29) and (2.30) associated with the ergodic mean have employed the primal-dual optimality gaps and , which are stronger than the primal optimality gap used in the previous studies for accelerated gradient methods. Moreover, whenever is bounded, the primal-dual optimality gap gives us a computable online accuracy certificates to check the quality of the solution (see lns11 ; GhaLan12-2a for some related discussions). Also observe that each iteration of the PDG method requires the computation of , and hence all the components . In the next section, we will develop a randomized PDG method that can possibly save the number of gradient evaluations for by utilizing the finite-sum structure of problem (1.1).
Randomized primal-dual gradient methods
In this section, we present a randomized primal-dual gradient (RPDG) method which needs to compute the gradient of only one randomly selected component function at each iteration. We show that RPDG can possibly achieve a better complexity than PDG in terms of the total number of gradient evaluations.
It is well-known that is an optimal solution of (3.1) if and only if for any .
Since , are strongly convex with modulus w.r.t. , we can define their associated dual prox-functions and dual prox-mappings as
for any . Accordingly, we define
Again, may not be uniquely defined since are not necessarily differentiable. However, we will discuss how to specify the particular selection of later in this subsection.
In other words, , , denote the prices that all the suppliers can possibly set up at iteration . Then we can see that
In order to implement the above RPDG method, we shall explicitly specify the selection of the subgradient in the definition of the dual prox-mapping in (3.8). Denoting , , we can easily see from that , . Using this relation and letting in the definition of in (3.8) (see (3.4)), we then conclude from Lemma 1 (with and ) and (3.8) that for any ,
Also, by the definition of and (3.9), we have
Incorporating these two ideas mentioned above, we present an efficient implementation of the RPDG method in Algorithm 4.
Clearly, the RPDG method is an incremental gradient type method since each iteration of this algorithm involves the computation of the gradient of only one component function. As shown in the following Subsection, such an randomization scheme can lead to significantly savings on the total number of gradient evaluations, at the expense of more primal prox-mappings.
2 The convergence of the RPDG algorithm
Our goal in this subsection is to describe the convergence properties of the RPDG method for the strongly convex case when . Generalization of the RPDG method for the non-strongly convex case will be discussed in Section 4.
Suppose that , , and in the RPDG method are set to
for some . Then, for any , we have
where with defined as in (2.24), and denotes the optimal solution of problem (1.1), and the expectation is taken w.r.t. .
The following corollary shows the convergence of RPDG under a non-uniform distribution for the random variables , .
Suppose that in the RPDG method are distributed over according to
Also assume that , , and are set to (3.19) with
and hence that the conditions in (3.20)-(3.22) are satisfied. Notice that by the fact that and (3.26), we have
Using the above bound in (3.23), we obtain (3.28). It follows from the facts , , and that
Using the above bound in (3.24), we obtain (3.29). Denoting , we conclude from (3.28) and the fact that for any that
Moreover, by Markov’s inequality, (3.28) and the fact that for any , we have
The proofs for the complexity bounds in terms of the primal optimality gap is similar and hence the details are skipped.
The non-uniform distribution in (3.25) requires the estimation of the Lipschitz constants , . In case such information is not available, we can use a uniform distribution for , and as a result, the complexity bounds will depend on a larger condition number given by . However, if we do have , then the results obtained by using a uniform distribution is slightly sharper than the one by using a non-uniform distribution in Corollary 1.
Suppose that in the RPDG method are uniformly distributed over according to
Also assume that , , and are set to (3.19) with
and hence that the conditions in (3.20)-(3.22) are satisfied. By the identity , we have
Using the above bound in (3.23), we obtain (3.35). Moreover, note that and we have
Using the above bound in (3.24), we obtain (3.36). The proofs for the complexity bounds are similar to those in Corollary 1 and hence the details are skipped.
Comparing the complexity bounds obtained from Corollaries 1 and 2 with those of any optimal deterministic first-order method, they differ in a factor of , whenever is dominating in (3.30). Clearly, when and are in the same order of magnitude, RPDG can save up to gradient evaluations for the component function than the deterministic first-order methods. However, it should be pointed out that can be much smaller than . In particular, when , . In the next subsection, we will construct examples in such extreme cases to obtain the lower complexity bound for general randomized incremental gradient methods.
3 Lower complexity bound for randomized methods
Our goal in this subsection is to demonstrate that the complexity bounds obtained in Theorem 3.1, and Corollaries 1 and 2 for the RPDG method are essentially not improvable. Observe that although there exist rich lower complexity bounds in the literature for deterministic first-order methods (e.g. nemyud:83 ; Nest04 ), the study on lower complexity bounds for randomized methods are still quite limited. Recently Agarwal and Bottou AgrBott14-1 suggested a lower complexity bound for minimizing the finite-sum convex optimization problem given in the form of (1.1). However, their bounds are developed for deterministic algorithms and hence not applicable to randomized incremental gradient methods.
To derive the performance limit of the incremental gradient methods, we consider a special class of unconstrained and separable strongly convex optimization problems given in the form of
where the last inequality follows from the fact that . Therefore, for any , the component functions in (3.38) are convex and their gradients are Lipschitz continuous with constant bounded by , .
We consider a general class of randomized incremental gradient methods which sequentially acquire the gradients of a randomly selected component function at iteration . More specifically, we assume that the independent random variables , , satisfy
Similar to Nest04 , we assume that these methods generate a sequence of test points such that
where denotes the linear span.
Theorem 3.2 below describes the performance limit of the above randomized incremental gradient methods for solving (3.37).
Let be the optimal solution of problem (3.37) and denote
Then the iterates generated by any randomized incremental gradient method must satisfy
As an immediate consequence of Theorem 3.2, we obtain a lower complexity bound for randomized incremental gradient methods.
if is sufficiently large, where and .
It follows from (3.43) that the number of iterations required by any randomized incremental gradient methods to find an approximate solution must satisfy
Noting that for the worst-case instance in (3.37), we have , , and hence that . Using this relation, we conclude that
The above bound holds when .
In view of Theorem 3.2, we can also derive a lower complexity bound for randomized block coordinate descent methods, which update one randomly selected block of variables at each iteration for . Here is smooth and strongly convex such that
if is sufficiently large, where denotes the condition number of .
The worst-case instances in (3.37) have a block separable structure. Therefore, any randomized incremental gradient methods are equivalent to randomized block coordinate descent methods. The result then immediately follows from (3.45).
Generalization of randomized primal-dual gradient methods
In this section, we generalize the RPDG method for solving a few different types of convex optimization problems which are not necessarily smooth and strongly convex.
Our goal in this subsection is to generalize RPDG for solving smooth problems without strong convexity (i.e., ). Different from the deterministic PDG method, it is difficult to develop a simple stepsize policy for , , and which can guarantee the convergence of this method unless a weaker termination criterion is used (see DangLan14-1 ). In order to obtain stronger convergence results, we will discuss a different approach obtained by applying the RPDG method to a slightly perturbed problem of (1.1).
In order to apply this perturbation approach, we will assume that is bounded (see Subsection 4.3 for possible extensions), i.e., given , s.t.
Now we define the perturbation problem as
for some fixed . It is well-known that an approximate solution of (4.2) will also be an approximate solution of (1.1) if is sufficiently small. More specifically, it is easy to verify that
The following result describes the complexity associated with this perturbation approach for solving smooth problems without strong convexity (i.e., ).
Let us apply the RPDG method with the parameter settings in Corollary 1 to the perturbation problem (4.2) with
iterations. Moreover, we can find a solution s.t. for any in at most
Let be the optimal solution of (4.2). Denote and
Note that problem (4.2) is given in the form of (1.1) with the strongly convex modulus , and . Hence by applying Corollary 1, we have
Observe that if we apply a deterministic optimal first-order method (e.g., Nesterov’s method or the PDG method), the total number of gradient evaluations for , , would be given by
Comparing this bound with (4.6), we can see that the number of gradient evaluations performed by the RPDG method can be times smaller than these deterministic methods when and are in the same order of magnitude.
2 Structured nonsmooth problems
In this subsection, we assume that the smooth components are nonsmooth but can be approximated closely by smooth ones. More specifically, we assume that
Nesterov in an important work Nest05-1 shows that we can approximate and , respectively, by
where is a strongly convex function with modulus such that
for any , where . Moreover, and are continuously differentiable and their gradients are Lipschitz continuous with constants given by
respectively. As a consequence, we can apply the RPDG method to solve the approximation problem
The following result provides complexity bounds of the RPDG method for solving the above structured nonsmooth problems for the case when .
Let us apply the RPDG method with the parameter settings in Corollary 1 to the approximation problem (4.13) with
iterations. Moreover, we can find a solution s.t. for any in at most
The following result holds for the RPDG method applied to the above structured nonsmooth problems when .
iterations. Moreover, we can find a solution s.t. for any in at most
Similarly to the arguments used in the proof of Proposition 2, our results follow from (4.17), and an application of Proposition 1 to problem (4.13).
3 Unconstrained smooth problems
We assume that the set of optimal solutions of this problem is nonempty.
We will still use the perturbation-based approach as described in Subsection 4.1 by solving the perturbation problem given by
We are now ready to establish the complexity of the RPDG method applied to (4.18) in terms of .
Let us apply the RPDG method with the parameter settings in Corollary 1 to the perturbation problem (4.19) with
iterations. Moreover, we can find a solution s.t. for any in at most
Let be the optimal solution of (4.19). Also let be the optimal solution of (4.18) that is closest to , i.e., . It then follows from the strong convexity of that
Moreover, using the definition of and the fact that is feasible to (4.19), we have
Now suppose that we run the RPDG method applied to (4.19) for iterations. Then by Corollary 1, we have
where the last inequality follows from (4.24) and is defined in (3.26) with . Combining the above two relations, we have
Dividing both sides of the above inequality by , we obtain
which clearly implies the bound in (4.22). The bound in (4.23) also follows from the above inequality and the Markov’s inequality.
By Proposition 4, the total number of gradient evaluations for the component functions required by the RPDG method can be times smaller than those performed by deterministic optimal first-order methods.
Complexity analysis
Our main goal in this section is to prove the main theorems in Sections 2 and 3. After introducing some basic tools and general results about PDG and RPDG methods in Subsection 5.1 and 5.2, respectively, we provide the proofs for Theorem 2.1 and Theorem 3.1, which describe the main convergence properties for the PDG and RPDG methods, in Subsection 5.3. Moreover, in Subsection 5.4, we provide the proof for the lower complexity bound in Theorem 3.2.
The following result provides a few different bounds on the diameter of the dual feasible sets and in (2.7) and (3.1).
Let be given, , , and . Assume that and in the definition of and in (3.4) and (2.5), respectively.
For any and , , we have
If is an optimal solution of (1.1) and , , then
For any and , we have
We first show part a). It follows from the definition of , (3.4), and (3.6) that
where the last inequality follows from (2.2). We now show part b). By the above relation, the convexity of and , and the optimality of , we have
The proof of part c) is similar to part a) and hence the details are skipped.
The following lemma gives an important bound for the primal optimality gap for some .
Let be a given pair of feasible solutions of (3.1), and be a pair of optimal solutions of (3.1). Then, we have
Let , and by the definition of in (3.3), we have
where the second equality follows from the fact that , are the conjugate functions of .
2 General results for both PDG and RPDG
We will establish some general convergence results in Proposition 5 which holds for both deterministic and randomized PDG methods by viewing PDG as a special case of RPDG with . Then both Theorems 2.1 and 3.1 follow as some immediate consequences of Proposition 5.
Before showing Proposition 5 we will develop a few technical results. Lemma 5 below characterizes the solutions of the prox-mapping in (2.3) and (3.5). This result generalizes some previous results (e.g., Lemma 6 of LaLuMo11-1 and Lemma 2 of GhaLan12-2a ).
for some . Also assume that the scalars and are chosen such that . If
Using these relations and (5.6), we conclude that
for any , which together with the fact that then imply that is convex. Since is an optimal solution of (5.7), we have . Combining this inequality with (5.8), we conclude that
from which the result immediately follows.
(5.9) follows immediately from the facts that and . Here denotes the conditional probability w.r.t. given . Similarly, we can show (5.10).
We now prove an important recursion about the RPDG method.
Let the gap function be defined in (3.3). Also let and be defined in (3.10) and (3.11), respectively. Then for any , we have
It follows from Lemma 5 applied to (3.10) that ,
Moreover, by Lemma 5 applied to (3.11), we have, for any and ,
Summing up these inequalities over , we have, ,
Using the definition of in (3.3), (5.12), and (5.13), we have
Also observe that by (3.8), (3.12), (5.9), and (5.10),
Taking expectation on both sides of (5.14) and using the above observations, we obtain (5.11).
We are now ready to establish a general convergence result which holds for both PDG and RPDG.
Suppose that , , and in the RPDG method satisfy
for some , . Then, for any and any given , we have
Multiplying both sides of (5.11) by and summing the resulting inequalities, we have
which, in view of the assumptions in (5.16) and (5.15), then implies that
We now provide a bound on in (5.22). Note that by (3.7), we have
where the last identity follows from the observation that by (3.8) and (3.9),
Using relation (5.24) in the definition of in (5.23), we have
Observe that by (5.20) and the fact that ,
Using the previous three relations in (5.25), we have
Regrouping the terms in the above relation, and the fact that , we obtain
where the second inequality follows from the simple relation that
and the last inequality follows from (5.17) and (5.18). Plugging the bound (5.26) into (5.22), we have
The result then immediately follows by combining the above two conclusion.
3 Proof of main convergence results
We now provide a proof for Theorem 2.1 which describes the main convergence properties of the deterministic PDG method.
We first specialize Proposition 5 for the PDG method applied to (2.7).
Suppose that , , and in the PDG method satisfy
for some , . Also let us denote , and
Then, for any and any given , we have
Notice that in the deterministic PDG method, we have , , and . It can be easily seen that the assumptions in (5.15)-(5.20) are implied by those in (5.28)-(5.32). It then follows from (5.21) that
Dividing both sides of the above inequality by and using the convexity of w.r.t. , we have
Rearranging the terms in the above relation, we obtain (5.34).
Proof of Theorem 2.1 We first show part a). It can be easily checked that (5.28)-(5.32) are satisfied with the selection of , , , and in (2.24). Using (5.34) (with and ), (5.3), and the fact that , we have
Using the parameter settings in (2.24), we conclude that
Also using (5.34) and the fact that , we have
Denoting , we conclude from (5.3) that
where the second inequality follows from the convexity of , the third inequality follows from the triangular inequality, the fourth inequality follows from and (2.25), and the last inequality follows from . Also note that by the definition of , we have
where the last inequality follows from the fact that due to (2.24). Fixing in (5.35) and using the above two relations, we obtain
The result in (2.26) then directly follows from the above relation and (2.21). If is bounded, the result in (2.27) then follows from the above relation, (2.21), and (2.22).
We now show part b). It is trivial to check that the conditions in (5.28)-(5.32) hold by using our selection of , , , and . Using (5.34) and the facts and , we have
which, in view of (2.20) and (2.21) and the fact that , clearly implies (2.29). In case is bounded, the result in (2.30) immediately follows from (2.21), (2.22), and the above inequality.
We are now ready to provide a proof for Theorem 3.1, which describes the main convergence properties of the RPDG method applied to strongly convex problems with .
Proof of Theorem 3.1. It can be easily checked that the conditions in (5.15)-(5.20) are satisfied with our requirements (3.19)-(3.22) of , , , and . Using the fact that , we then conclude from (5.21) (with and ) that, for any ,
where the first inequality follows from (3.19) and (3.20), and the second inequality follows from (3.21) and (5.1).
Let us denote , . In view of (5.4), the convexity of , and (2.2), we have
Using (5.21) (with and ), the fact that , and (5.36), we obtain
We conclude from (3.23) and the definition of that
Using the above two relations, and (5.3), we obtain
4 Proof of the lower complexity bound
This subsection is devoted to the proof of Theorem 3.2, which describes the performance limit for randomized incremental gradient methods.
The following result provides an explicit expression for the optimal solution of (3.37).
Let be defined in (3.42), is the -th element of , and define
Then is the unique optimal solution of (3.37).
It can be easily seen that is the smallest root of the equation
Note that satisfies the optimality condition of (3.37), i.e.,
Indeed, we can write the coordinate form of (5.40) as
where the first two equations follow directly from the definition of and relation (5.39), and the last equation is implied by the definitions of and in (3.39) and (5.38), respectively.
We also need a few technical results to establish the lower complexity bounds.
Let be given. If we have
We first show part a). Denote . It can be easily seen that . Moreover, for any , we have
which implies that is a strictly decreasing function for . Hence, we must have for any . Part b) follows from the following simple calculation.
Proof of Theorem 3.2 Without loss of generality, we may assume that the initial point , . Indeed, the incremental gradient methods described in Subsection 3.3 are invariant with respect to a simultaneous shift of the decision variables. In other words, the sequence of iterates , which is generated by such a method for minimizing the function starting from , is just a shift of the sequence generated for minimizing starting from the origin.
Noting that is convex w.r.t. for any and , by minimizing the RHS of the above bound w.r.t. , , subject to and , we conclude that
for any (see (3.44)) and possible selection of , satisfying (3.40), where the last inequality follows from Lemma 9.b). Noting that
we then conclude from (5.46) and Lemma 9.a) that