Relatively-Smooth Convex Optimization by First-Order Methods, and Applications
Haihao Lu, Robert M. Freund, Yurii Nesterov
Introduction, Definition of “Relative-Smoothness,” and Basic Properties
There are by now very many first-order methods for tackling the optimization problem (1), see for example , , ; virtually all such methods are designed to solve (1) when the gradient of satisfies a uniform Lipschitz condition on , namely there exists a constant for which:
One can prove for the standard gradient descent scheme that after iterations it holds for any that:
which is an sublinear rate of convergence , . Furthermore, if is also uniformly -strongly convex for some , namely:
then one can prove linear convergence for the gradient descent scheme, see , , i.e., for any we have that:
More general versions of first-order methods are not restricted to the Euclidean () norm, and use a differentiable “prox function” , which is a -strongly convex function on , to define a Bregman distance:
The standard Primal Gradient Scheme (with Bregman distance), see , has the following update formula:
Notice in (8) by construction that the update requires the capability to solve instances of a subproblem of the general form:
for suitable iteration-specific values of ; indeed, (8) is an instance of the subproblem (9) with at iteration . It is especially important to note that the Primal Gradient Scheme is somewhat meaningless whenever we do not have the capability to efficiently solve (9), a point which we will return to later on. In a typical design and implementation of a first-order method for solving (1), one attempts to specify the norm and the strongly convex prox function in consideration of the shape of the feasible domain while also ensuring that the subproblem (9) is efficiently solvable.
Regarding computational guarantees, one can prove for the Primal Gradient Scheme that after iterations it holds for any that:
which is an exact generalization of (4), see , .
Notice that unlike quadratic functions, the second-order terms of the functions in the above examples vary dramatically on – and especially as (or as goes to infinity in ). It therefore becomes unreasonable to use a uniform bound of the form to upper-bound second-order information.
Motivated by the above drawbacks in standard first-order methods, we develop a notion of “relative smoothness” and relative strong convexity, relative to a given “reference function” and which does not require the specification of any particular norm – and indeed need not be either strictly or strongly convex. Armed with relative smoothness and relative strong convexity, we demonstrate the capability to solve a more general class of differentiable convex optimization problems (without uniform Lipschitz continuous gradients), and we also demonstrate linear convergence results for both a Primal Gradient Scheme and a Dual Averaging Scheme when the function is both relatively smooth and relatively strongly convex.
There is a certain overlap of ideas and results herein with the paper by Bolte, Bauschke, and Teboulle. For starters, the relative smoothness condition definition in the present paper in Definition 1.1 is equivalent to the (LC) condition in except that also requires the reference function to be essentially smooth and strictly convex, which we do not need in this paper. The main developments in are based on generalizing a key descent lemma and applying this generalization to tackle (additive) composite optimization problems using the primal gradient scheme (called the NoLips Algorithm in ) with associated complexity analysis involving a symmetry measure of the Bregman distance . These results are then illustrated in the application of composite optimization to Poisson inverse problems. While the NoLips Algorithm in is structurally the same as Algorithm 1 herein, they are both instantiations of the standard primal gradient scheme; however, as will be seen in Section 3 here, we do not need any symmetry measure in constructing step-sizes or in the complexity analysis. The paper by Zhou, Liang, and Shen also tackles composite optimization using the standard primal gradient scheme which therein is called PGA-, with a focus on demonstrating equivalence of proximal gradient and proximal point methods more broadly. Here we develop measures of relative smoothness and also relative strong convexity, which can improve the computational guarantees of the primal gradient scheme, see Theorem 3.1. We further present computational guarantees for the dual averaging scheme in Theorem 3.2. In Section 2 we show that many differentiable convex functions are relatively smooth with respect to a correspondingly fairly-simple reference function that is easy to construct and for which algorithmic computations can be effeciently be performed. In Section 4 we apply our approach to develop a new first-order method for the -optimal design problem, with associated computational complexity analysis. Throughout the current paper, we compare and clarify similarities and differences between our work and in the context of the specific contributions as they arise.
2 Relative Smoothness and Relative Strong Convexity
Let be any given differentiable convex function (it need not be strongly nor even strictly convex) defined on . We will henceforth refer to as the “reference function.” We define “relative smoothness” and “relative strong convexity” of relative to using the Bregman distance (7) associated with as follows.
The following proposition presents equivalent definitions of relative smoothness and relative strong convexity. In the case when both and are twice differentiable, parts (a-iii) and (b-iii) of the proposition demonstrate that the above definitions are equivalent to
which is an intuitively simple condition on the Hessian matrices of the two functions.
is -smooth relative to ,
is a convex function on ,
is -strongly convex relative to ,
is a convex function on ,
The first part of Proposition 1.1 is almost equivalent to Proposition 1 of .
Proof: For define . Using (11) and (7) it follows that (a-i) holds if and only if for all , which is equivalent to the convexity of from Theorem 2.1.2 of , thus showing that (a-i) (a-ii). It follows from Theorem 2.1.3 of applied to that is convex if and only if for all , which shows that (a-ii) (a-iv). If and are twice differentiable, then it follows from Theorem 2.1.4 of that (a-ii) (a-iii).
Similar proofs can be applied for part (b).∎
For notational convenience, let us denote by that is a convex function, whereby this also means is -smooth with respect to from Proposition 1.1. Similarly means is a convex function and so is -strongly convex with respect to . (In the case when both and are twice differentiable, the relation “” on two functions is consistent with the Löwner partial order on the Hessians of these two functions from Propositon 1.1.) Then the condition that is -smooth with respect to is equivalent to ; similarly the condition that is -strongly convex with respect to is equivalent to . In addition, relative-smoothness and relative strong convexity are each transitive, so that and implies that .
We can also work with sums and linear transformations of relatively smooth and/or relatively strongly convex functions, as the next proposition states.
If and , then for all it holds that .
If and , then for all it holds that .
If , and is a linear transformation of appropriate dimension, then .
If , and is a linear transformation of appropriate dimension, then .
Proof: The proofs of the first two arguments follow directly from the definitions of relative smoothness and relative strong convexity in Definitions 1.1 and 1.2. The proofs of the last two arguments follow from the equivalent definition (a-iv) and (b-iv) in Proposition 1.1.∎
3 Constructive Algorithmic Set-up
Let us now discuss criteria for choosing the reference function in the context of computational schemes for solving the optimization problem (1). To be concrete, consider a simple Primal Gradient Scheme as shown in Algorithm 1. Note that this scheme is essentially as described in the update formula (8), except that the uniform smoothness constant is replaced by the relative smoothness parameter of with respect to the reference function as defined in Definition 1.1, and the only formal requirement for is that the pair must satisfy the conditions of Definition 1.1.
In order to efficiently execute the update step in Algorithm 1 we also require of that the subproblem (9) is efficiently solvable for any given . In summary, to solve the optimization problem (1) using Algorithm 1, we need to specify a reference function that has the following two properties:
is -smooth relative to on , and
the subproblem (9) always has a solution, and the solution is efficiently computable.
In Section 2 we will see how this can be done for several useful classes of problems that are not otherwise solvable by traditional first-order methods that require uniform Lipschitz continuity of the gradient. In Section 3 we analyze the computational guarantees associated with the Primal Gradient Scheme (Algorithm 1) as well as a Dual Averaging Scheme. In Section 4, we apply the computational guarantees of Section 3 to the -optimal design problem.
Examples of Relatively Smooth Optimization Problems
Here we show several classes of optimization problems (1) for which one can easily construct a reference function with the two properties mentioned above, namely (i) is -smooth relative to for an easily determined value , and (ii) the subproblem (9) is efficiently solvable.
Then the following proposition states that is -smooth relative to for an easily computable value . This implies that no matter how fast the Hessian of grows as , can still be smooth relative to the simple reference function , even though need not exhibit uniform Lipschitz continuity.
Suppose is twice differentiable and satisfies where is an -degree polynomial of . Let be such that for . Then is -smooth relative to .
Proof: It follows from elementary rules of differentiation that
and so is -smooth relative to by part (iii) of Proposition 1.1. ∎
Suppose . In Proposition 2.1, one simple way to set is to use . Then
whereby for .
Solving the subproblem (9). Let us see how we can solve the subproblem (9) for this class of optimization problems. The subproblem (9) can be written as
and the first-order optimality conditions are simply:
whereby for some , and it remains to simply determine the value of the nonnegative scalar . If , then satisfies the optimality conditions. For , notice from above that must satisfy:
which is a univariate polynomial in with a unique positive root. For , this root can be computed in closed form. Otherwise, the root can be computed (up to machine precision) using any scalar root-finding method.
We can incorporate in problem (15) a simple set constraint provided that we can easily compute the Euclidean projection on . In the case when is a convex function of , the subproblem (9) can be converted to a -dimensional convex optimization problem, see Appendix A.1 for details.
which is -degree polynomial in with coefficients , , and . Therefore following Remark 2.1 it suffices to set
which is -degree polynomial in with coefficients , , and . Therefore following Remark 2.1 it suffices to set
(where the last matrix inequality follows since ), and thus is -strongly convex relative to .
In place of the simple reference function in (13) one can instead consider a “re-centered” version of the form:
where the “center” value is suitably chosen to align with and possibly attain better values of and . Note that introducing the given center value does not increase the difficulty of solving the subproblem (9). We illustrate this idea with a simple univariate example. Suppose that our objective function is . From the results in Section 2.1 we know we can use the reference function . We can also translate by the center point and use the reference function . Straightforward calculation yields values of for and for , whereby yields a better value of than for this example.
2 D𝐷D-Optimal Design Problem
where the first and the second matrix inequality above each follows from the fact that and the Hadamard product of two symmetric positive semidefinite matrices is also a symmetric positive semidefinite matrix. The result then follows using property (a-iii) of Proposition 1.1. ∎
Solving the subproblem (9). Let us see how we can solve the subproblem (9) for and given above. The subproblem (9) can be written as
and the first-order optimality conditions are simply:
for some scalar multiplier . Given , it then follows that for , and it remains to simply determine the value of the scalar . Now notice that must satisfy:
for some in the interval . Notice that is strictly decreasing on , and as and as , whereby (18) has a unique solution in . Furthermore, as suggested by results in Ye or , one can use Newton’s method (or any other suitable scalar solution-finding method) to efficiently compute the solution of (18) (up to machine precision) on the interval .
3 Generalized Volumetric Function Optimization
For a given integer parameter , let us also study optimization on the simplex of the following generalization of the volumetric barrier function:
Proof: By elementary calculus, the gradient of is
where the first inequality follows from the fact that the Hadamard product of two symmetric positive semidefinite matrices is also a symmetric positive semidefinite matrix and is a positive semidefinite matrix, and the first equation follows since is itself a diagonal matrix. The result then follows by property (iii) of Proposition 1.1. ∎
Solving the subproblem (9). Using , the subproblem (9) here is identical to that for the -optimal design problem, since the reference function and the feasible domain are the same. Therefore the methodology discussed in Section 2.2 applies here as well.
Then the following proposition states that is -smooth relative to for an easily computable value . This implies that no matter how fast grows as approaches the open boundary of the region , is smooth relative to the simple reference function , even though need not exhibit uniform Lipschitz continuity on .
Suppose is twice differentiable on and satisfies where is an -degree polynomial in . Let be such that for all . Then is -smooth relative to .
where the second matrix inequality uses and the third matrix inequality is due to . Therefore is -smooth relative to by part (iii) of Proposition 1.1.∎
Suppose . In Proposition 2.4, one simple way to set is to use . This implies for that
Solving the subproblem (9). Let us see how we can solve the subproblem (9) for this class of optimization problems. After rescaling by , the subproblem (9) can be equivalently written as
Let , then the optimality conditions for (23) can be written as:
for . For a given , define using the above rule (24), and it remains to simply determine the value of the positive scalar in the interval that satisfies
Notice that is strictly increasing on , and (since for any ) and as . Therefore (25) has a unique solution in , which can be solved with high accuracy using any suitable root-finding method, for example binary search combined with -dimensional Newton’s method.
In a sense, there are basically two ways that a twice-differentiable convex function can fail to have a uniformly Lipschitz gradient: (i) when the Hessian grows without limit as , and/or (ii) when the Hessian grows without limit as . Section 2.1 has provided a mechanism for constructing a reference function for case (i) when the growth is polynomial, and Section 2.4 has provided such a mechanism for case (ii) when the growth is polynomial. By utilizing the additivity and linear transformation properties of relative smoothness in Proposition 1.2, it should be possible to construct suitable reference functions for many convex functions of interest.
Computational Analysis for the Primal Gradient Scheme and the Dual Averaging Scheme
In this section we present computational guarantees for two algorithms: the Primal Gradient Scheme (Algorithm 1) as well as a Dual Averaging Scheme (Algorithm 2).
Our main result for the Primal Gradient Scheme is the following sublinear and linear convergence bounds.
Consider the Primal Gradient Scheme (Algorithm 1). If is -smooth and -strongly convex relative to for some and , then for all and , sequence is monotonically decreasing, and the following inequality holds:
where, in the case when , the middle expression is defined in the limit as . ∎
The first inequality in (26) shows linear convergence when ; indeed, in this case it holds that
(This inequality holds trivially for , and induction on establishes the result for .) Furthermore, when is large the term in the denominator of the left-hand side can be ignored which yields the asymptotic bound . The second inequality in (26) shows an sublinear convergence rate. In particular, the convergence rate in (26) is when .
Note that Algorithm 1 herein and the NoLips algorithm in as well as algorithm PGA- in are structurally identical (they are all instantiations of the primal gradient methodology). However, the step-size rule in as well as the complexity analysis in depends on a symmetry measure of , namely , whereas there is no such dependence here. The instantiation of Algorithm 1 in uses a smaller “step-size” of as opposed to in the update computation in Algorithm 1 (since it must always hold that ), and proves a computational guarantee of . The bound in Theorem 3.1 is better than this symmetry-based bound, but only by a multiplicative constant factor when ; it is of course far better (linear convergence rather than sublinear convergence) when .
The proof of the bound in Theorem 3.1 relies on the following standard Three-Point Property:
(Three-Point Property of Tseng ) Let be a convex function, and let be the Bregman distance for . For a given vector , let
Proof of Theorem 3.1: Define a parameter sequence
where the second equality “” follows from elementary geometric series’ analysis, and holds only when . In particular, if . For any and we have:
where the first inequality follows from the definition of -smoothness relative to , the second inequality is due to the Three-Point Property with and , , and the last inequality uses the -strong convexity of relative to , which implies . Substituting in (28) shows in particular that which proves monotonicity of the sequence .
It then follows using induction and (28) that
Using the monotonicity of and the nonnegativity of , this implies that
The proof of the second inequality in (26) follows by noting that . ∎
2 Dual Averaging Scheme and Analysis
Another algorithm for solving our optimization problem (1) is the Dual Averaging Scheme , which we present here in Algorithm 2. Somewhat akin to the Primal Gradient Scheme, the update step in the Dual Averaging Scheme also requires the solution of a subproblem exactly of the form (9). Notice that we need the coefficient of strong convexity in order to implement Algorithm 2, in contrast to the Primal Gradient Scheme (Algorithm 1). One can always conservatively set in Algorithm 2 if no reasonable lower bound on best value of is known.
We have the following result regarding computational guarantees for the Dual Averaging Scheme.
Consider the Dual Averaging Scheme (Algorithm 2). If is -smooth and -strongly convex relative to with , then for all and , the following inequality holds:
where in the case , the middle expression is defined as the limits as .
Similar to the result in Theorem 3.1, the first inequality in (32) shows linear convergence when , since
this follows using identical logic as in (27).
Proof of Theorem 3.2: Define for and , whereby and . It follows from the definition of relative strongly convexity (Definition 1.2) that for any :
for all , and where the second equality “” above follows from elementary geometric series’ analysis and holds only when ; note that when .
The function is a sum of a linear function and the reference function multiplied by the coefficient . Therefore and define the same Bregman distance, whereby for any it holds that:
where the last inequality utilizes as well as the first order optimality condition of . Therefore:
where the last inequality uses (35) with . Taking into account that , and using the relative smoothness of (Definition 1.1), we obtain:
where the second inequality is from (34). The proof is completed by rearranging (36) and taking the minimum over . ∎
3 On Optimization Problems with a Composite Function
Sometimes we are interested in solving the composite optimization problem :
under the same assumptions on and as in (1), but now the objective function includes another function that is assumed to be convex but not necessarily differentiable, and for which the following subproblem is efficiently solvable:
for any given . Under this assumption it is straightforward to show that Algorithm 1 naturally extends to cover the case of the composite optimization problem (37) (see and ) and that the computational guarantee in Theorem 3.1 extends to composite optimization as well. (Indeed, when this extension is implied in principle from .) It turns out that one can actually view composite optimization as working with the objective function that is -smooth relative to the reference function . However, the definition of the reference function has been premised on being differentiable on , which might not hold for as just defined. This can all be taken care of by a suitable modification of the theory, see Appendix A.2 for details.
4 Questions: Accelerated Methods, Conjugate Duality, Choosing the Reference Function
We have shown here in Section 3 that the computational guarantees of two standard first-order methods for smooth optimization – the Primal Gradient Scheme and the Dual Averaging Scheme – extend in precise ways to the case when is -smooth relative to the reference function . The proof techniques used here suggest that very many other first-order algorithms for smooth optimization should extend similarly with analogous computational guarantees. However, we have not been able to extend any accelerated methods, i.e., methods that attain an convergence guarantee such as , , , to the relatively smooth case. One avenue for further research is to answer the question whether one can develop computational guarantees for an accelerated method in the case when is -smooth relative to the reference function ?
Another question that arises concerns conjugate (duality) theory for the setting of relatively smooth convex functions. One simple result in conjugate duality theory is that when is -smooth (relative to ) the conjugate function is -strongly convex (relative to ), see . Is there a way to develop a more general conjugate duality theory that yields an analogous result when is -smooth relative to a general convex function ?
A third question is how can we choose the reference function in order to lower the value of the bounds in Theorems 3.1 and 3.2? Several ways to think about this question are discussed in Appendix A.3.
D𝐷D-Optimal Design Revisited: Computational Guarantees using the Primal Gradient or Dual Averaging Scheme
Let us now apply the computational guarantees for the Primal Gradient Scheme (Theorem 3.1) and the Dual Averaging Scheme (Theorem 3.2) to the -optimal design optimization problem (16) discussed in Section 2.2. Recall from the exposition in Section 2.2 that and is -smooth relative to the logarithmic barrier function
and that the subproblem (9) is efficiently solvable. The following theorem presents a computational guarantee for using the Primal Gradient Scheme to approximately solve the -optimal design optimization problem (16).
Consider using the Primal Gradient Scheme (Algorithm 1) with the reference function (39) to solve the -optimal design problem (16) using the initial point , and suppose that . If
Proof: Let . Then since . Let . It follows from the convexity of that
where the second equality uses which then implies , and the inequality follows since . Therefore, for satisfying the inequality in the statement of the theorem, we have:
where the first inequality follows from Theorem 3.1 using , as well as (40), the second inequality is from (41) and the definition of , and the third inequality follows since . ∎
For the Dual Averaging Scheme (Algorithm 2), one obtains the identical bound as in Theorem 4.1. This is proved by following virtually the same logic as above, except we use Theorem 3.2 which bounds the smallest optimality gap using instead of . However, it follows from (41) that these two quantities are the same in this case. Also, in the case of the Dual Averaging Scheme, the relevant final quantity of interest is instead of .
It is instructive to compare the computational guarantees in Theorem 4.1/Remark 4.1 to those of the Frank-Wolfe method applied to -optimal design (first analyzed by Khachiyan and re-evaluated in based in part on work by Yildirim ). Table 1 shows such a comparison, where absolute constants have been suppressed in order to highlight the dependencies on particular quantities of interest. The second column of Table 1 compares the iteration bound of the methods using the starting point , where we emphasize that is the target optimality gap for the -optimal design problem. While it follows from observations in that for , we do not show this in Table 1, as we wish to highlight where the dependence on the initial iterate arises. Examining the first column of Table 1, note that the number of iterations of the Primal Gradient Scheme (or Dual Averaging Scheme) can be less than that of the Frank-Wolfe method, especially when is not too small and when . However, as the second column of Table 1 shows, the Frank-Wolfe method requires only operations per iteration in the worst – i.e., dense matrix – case, as it does a rank- update of a matrix inverse in the computation of ), whereas the Primal Gradient Scheme (or Dual Averaging Scheme) requires operations per iteration in the dense case (it must re-compute a matrix inverse in order to work with ). Therefore the total bound on operations of the Frank-Wolfe method (shown in the last column of Table 1) is superior.
The bound for the Frank-Wolfe method applied to the -optimal design problem is based on analysis that is uniquely designed for evaluating the -optimal design problem, and is not part of the general theory for the Frank-Wolfe method (that we are aware of). Even though the Primal Gradient Scheme and the Dual Averaging Scheme have inferior computational guarantees to the Frank-Wolfe method applied to the -optimal design problem, they are the first (that we are aware of) first-order methods for which one has a general theory (Theorems 3.1 and 3.2) that can be meaningfully applied to yield computational guarantees for the -optimal design problem. We hope that this analysis will spur further interest in developing improved algorithms for -optimal design and its dual problem – the minimum volume enclosing ellipsoid problem.
Acknowledgement
The authors are grateful to the three referees for their comprehensive efforts and their suggestions on ways to improve the readability of the paper.
Appendix A Appendix
whose domain we denote by . Since is a convex function, we know from conjugacy theory that . Therefore (43) becomes
where the second equality above holds whenever the min and the sup operators can be exchanged (which is akin to strong duality). Notice that is a Euclidean projection problem. Therefore the subproblem (9) becomes a -dimensional concave maximization problem if the Euclidean projection problem can be easily solved and one can conveniently form and work with the univariate convex conjugate function .
A.2 Extension to Composite Optimization
Here we discuss some details of the extension of the ideas and results of this paper to composite optimization as described in Section 3.3, using the definitions , and as defined in Section 3.3. Note that and are not necessarily differentiable on since they include the function . However, we can use the equivalent condition from (a-ii) of Proposition 1.1 to define relative smoothness. Let us now show how convergence results for the Primal Gradient Scheme still hold in this more general setting using an extension of the proof of Theorem 3.1.
Let be a specific subgradient of at , and we will use the same subgradient of at when constructing a subgradient of and/or , namely and . Then Algorithm 1 has the following update:
where in the third equality above the term involving arising in cancels out the corresponding term involving arising in as part of the expansion of . There is therefore no actual need to compute in the update. Indeed, this update (44) corresponds exactly to the update in the NoLips algorithm (up to the step-size) and the PGA- algorithm in (up to the step-size) for composite optimization.
The proof of the computational guarantee in Theorem 3.1 can be generalized directly to the composite optimization setting as follows. Let us denote
Notice that ; therefore from the first-order optimality conditions there is a subgradient for which for all . From the additivity property of subgradients, we can write for some , and let us assign , which then is used to define the subgradient , , and the Bregman distance in the proof. Recall that the Primal Gradient Scheme does not rely on the choice of subgradient of , thus the choice of is only used in the proof and it is well-defined.
Utilizing the above method for specifying the subgradients of at each of the iterates of the Primal Gradient Scheme, we can prove the following more specialized form of the Three Point Property which we can use in the proof of Theorem 3.1 for the setting composite optimization.
For any , we have for any ,
Proof: Notice that and so is a linear function of , whereby it holds that
where the inequality follows from the choice of . Rearranging the above and recalling the definition of then completes the proof.∎
The proof of Theorem 3.1 in the setting of composite optimization follows directly by replacing , , and by , , and , respectively, and utilizing (45) to deduce the second inequality in (28).
A.3 Criteria for choosing the reference function h(⋅)ℎ⋅h(\cdot)
One natural question is how can we choose in order to lower the value of the bound in Theorem 3.1? Let us consider the simple case when is twice differentiable and is not strongly convex, namely , and attains its optimum at some point . Then the convergence bound (26) can be re-written as:
There is a trade-off between how small the Hessian is and how hard it will be to solve the subproblem (9). If we choose , the Hessian of the gap function is , but solving the subproblem (9) is as hard as solving the original problem (1). On the other hand, in standard gradient descent we use in which case the subproblem (9) can be easily solved, while the Hessian of the gap function can be huge – thus implying a poorer convergence bound. There are a number of ways to try to manage this trade-off. For example, in gradient descent with preconditioning we can use , where is a computationally-friendly positive definite matrix – typically a diagonal matrix. The criteria for designing usually involves (i) ensuring that solving equations with is easy (so that the subproblem (9) can be easily solved), and (ii) is “close to” the Hessian of (so that the Hessian of the gap function is small).