Accelerated Bregman Proximal Gradient Methods for Relatively Smooth Convex Optimization
Filip Hanzely, Peter Richtarik, Lin Xiao
convex optimization, relative smoothness, Bregman divergence, proximal gradient methods, accelerated gradient methods.
Introduction
where for all . Here, we use the gradient to construct a local quadratic approximation of around while leaving untouched. Our assumption that and are simple means that the minimization problem in (2) can be solved efficiently, especially if it admits a closed-form solution.
This smoothness assumption implies (see, e.g., [25, Lemma 1.2.3])
See, e.g., , and [5, Chapter 10]. Under the same assumption, accelerated proximal gradient methods () can achieve a faster convergence rate:
which is optimal (up to a constant factor) for this class of convex optimization problems .
While the uniform smoothness condition (3) is central in the development and analysis of first-order methods, there are many applications where the objective function does not have this property, despite being convex and differentiable. For example, in D-optimal experiment design (e.g., ) and Poisson inverse problems (e.g., ), the objective functions involve the logarithm in the form of log-determinant or relative entropy, whose gradients may blow up towards the boundary of the feasible region. In order to develop efficient first-order algorithms for solving such problems, the notion of relative smoothness was introduced by several recent works .
The function is called -smooth relative to on if there is an such that
As shown in and , this notion of relative smoothness is equivalent to the following statements:
The definition of relative smoothness in (7) gives an upper approximation of that is similar to (4). In fact, (4) is a special case of (7) with and . Therefore it is natural to consider a more general algorithm by replacing the squared Euclidean distance in (2) with a Bregman distance:
Here, our assumption that and are simple means that the minimization problem in (8) can be solved efficiently. Similar to the proximal gradient method (2), this algorithm can also be interpreted through operator splitting mechanism: it is the composition of a Bregman proximal step and a Bregman gradient step (see details in [3, Section 3.1]). Therefore, it is called the Bregman proximal gradient (BPG) method .
This is a generalization of (5). The same convergence rate for the general case (with nontrivial ) is obtained in , where the authors also discussed the effect of a symmetry measure for the Bregman distance. Similar results are also obtained in and . In addition, introduced the notion of relative strong convexity and obtained linear convergence of the BPG method when both relative smoothness and relative strong convexity hold. More recently, studied stochastic gradient descent and randomized coordinate descent methods in the relatively smooth setting, and extended this framework to minimize relatively continuous convex functions.
A natural question is whether the rate can be improved with first-order methods under the relative smoothness assumption, especially whether the accelerated rate can be achieved . Very recently, it is shown by Dragomir et al. that the rate is optimal for the class of relatively smooth functions, thus cannot be improved in general. However, we note that the class of relatively smooth functions is very broad, containing differentiable functions whose gradients has arbitrarily large Lipschitz constants. Indeed, the worst-case function constructed in to prove the lower bound is obtained by smoothing a nonsmooth function, which demonstrate pathological nonsmooth behavior. This is in sharp contrast to the situation under the uniform Lipschitz assumption, which uses a fixed quadratic function as the relatively smooth measure.
Ideally, it would be most informative to derive both upper and lower bounds on the convergence rate of first-order methods for every fixed function in the relatively smooth setting, or at least for the popular ones that are frequently encountered in application (such as the KL divergence). It is plausible that the achievable convergence rates for particular functions (more likely particular combinations of and ) can be better than in theory or at least in practice. A full spectrum investigation is beyond the scope of this paper. Instead, we study a structural property of general Bregman divergences called triangle scaling and develop adaptive first-order methods that, although without a priori guarantee, often demonstrate the convergence rate empirically in many applications. Moreover, these methods produce simple numerical certificates of the fast rates whenever they happen.
2 Contributions and outline
In Section 3, we propose a basic accelerated Bregman proximal gradient (ABPG) method that attains an convergence rate, where is the TSE of the Bregman divergence. More specifically, under the assumption (7), the basic ABPG method produces a sequence satisfying
The exact value of depends on a triangle scaling property of the Bregman distance. For , we have and , hence the result in (9) recovers that in (6). We also give an adaptive variant that can automatically search for the largest possible for which the convergence rate in (9) holds for finite even though is larger than the TSE.
In Section 5, we present an accelerated Bregman dual-averaging algorithm that has similar convergence rates as the basic ABPG method, but omit discussions of its adaptive variants.
Finally, in Section 6, we present numerical experiments with three applications: the D-optimal experiment design problem, a Poisson linear inverse problem, and relative-entropy nonnegative regression. In all experiments, the ABPG methods, especially the adaptive variants, demonstrate superior performance compared with the BPG method. Moreover, we obtain numerical certificates for the empirical rate in all our experiments.
The relative smoothness condition directly extends the upper approximation property (4) with more general Bregman distances. Nesterov took an alternative approach by extending the Lipschitz condition (3). Specifically, he considered functions with Hölder continuous gradients with a parameter :
and obtained rate with a universal gradient method and rate with accelerated schemes. These methods are called “universal” because they do not assume the knowledge of and automatically ensure the best possible rate of convergence. The accelerated rate interpolates between and with . There seems to be no simple connection or correspondence between the Hölder smoothness property and the combination of relative smoothness and the triangle scaling property studied in this paper.
Gutman and Peña studied iteration complexity of first-order methods using a general framework of perturbed Fenchel duality. Their framework provides alternative derivations of the convergence rates of Bregman proximal gradient methods under the relative smooth setting and the ones under Hölder continuity assumption.
Development and analysis of optimization methods in the relatively smooth setting require some delicate assumptions in order to cover many interesting applications without loss of rigor. Here we adopt the same assumptions made in regarding problem (1).
, i.e., problem (1) is bounded below.
Sufficient conditions for the well-posedness of (8) are given in [3, Lemma 2]. The same conditions also ensure that our proposed accelerated methods are well-posed.
Triangle scaling of Bregman distance
In this section, we define the triangle scaling property for Bregman distances and discuss two different notions of triangle scaling exponent (TSE).
We call a uniform triangle scaling exponent (TSE) of .
Figure 1 gives a geometric illustration of the points involved in the above definition.
If is jointly convex in , then the inequality (10) holds with because
The squared Euclidean distance. Let and . Obviously, here is jointly convex in its two arguments. But it is also easy to see that
Therefore the squared Euclidean distance has a uniform TSE , which is much larger than obtained by following the jointly convex argument.
Bregman distance induced by strongly convex and smooth functions. If is -strongly convex and -smooth over its domain, then the inequality (10) would hold with if the right-hand side is multiplied by an additional factor , which is the condition number of . We will prove this fact in Section 2.2.
Bregman distance based on polynomial kernels. Reference functions of the form for some recently attracted lots of attention following Nesterov’s work on tensor methods in convex optimization . In general, the global TSEs for the induced Bregman divergence can be less than for . However, the modified reference function for has , or with an additional factor on the right-hand side of (10), over a bounded domain. We will give detailed analysis for the case in Section 2.2, after introducing a relaxed version of TSE.
We observe that the largest uniform TSEs are quite different for the Bregman distances listed above. An important question is: Are these differences essential such that they lead to different convergence rates if different Bregman distances are used in an accelerated algorithm? It would be ideal to derive an intrinsic characterization that is common for most Bregman distances and essential for convergence analysis of accelerated algorithms.
Plugging the last equality into (15) and after some simple algebra, we arrive at (14). ∎
2 Bounding the triangle-scaling gain
where for some , which we denote as . Consequently, if we define
Next we consider the polynomial reference function , which does not have bounded Hessian. In this case, we have and , where is the identity matrix. Clearly . According to (16), we have
As a simple fix, we consider , whose Hessian is and it satisfies . Therefore, according to (16),
Accelerated Bregman proximal gradient method
In this section, we present an accelerated Bregman proximal gradient (ABPG) method for solving problem (1), and analyze its convergence rate under the uniform triangle-scaling property. Adaptive variants based on the intrinsic TSE are developed in Section 4.
To simplify notation, we define a lower approximation of by linearizing at a given point :
If is -smooth relative to (Definition 1), then we have both a lower and an upper approximation:
When and , Algorithm 1 reduces to the IGA (improved interior gradient algorithm) method in , which is an extension of Nesterov’s accelerated gradient method in to the Bregman proximal setting. It was shown in that the IGA method attains rate of convergence under the uniform Lipschitz condition (3). In this paper, we consider the general case under the much weaker relatively smooth condition.
We show that the ABPG method converges with a sublinear rate of . First, we state a basic property of optimization with Bregman distance [13, Lemma 3.2].
and is differentiable at , then
The following lemma establishes a relationship between the two consecutive steps of Algorithm 1. It is an extension of Proposition 1 in , which uses under the assumption (3).
First, using the upper approximation in (17) and line 1 and line 1 in Algorithm 1, we have
where in the last inequality we used the lower bound in (17). Subtracting from both sides of the inequality above, we obtain
Dividing both sides by and rearranging terms yield
Finally applying the condition (18) gives the desired result. ∎
The sequence for satisfies the condition (18).
With , we have
Recall the weighted arithmetic mean and geometric mean inequality (see, e.g., [18, Section 2.5].), i.e., for any positive real numbers , , and , it holds that
Setting , , and , we arrive at
which, together with (24) and (25), implies the inequality (18). ∎
A slightly faster converging sequence can be obtained by solving the equality in (18). Since there is no closed-form solution in general, we can find as the root of
numerically, say, using Newton’s method with as the starting point.
Let and be the solution to (27) for all . Then for all .
Let and define another sequence such that and
is monotone decreasing in . Since by Lemma 3 and by (28), we have for all .
Next we prove for all by mathematical induction. This obviously holds for since . Suppose holds for some . Then using the facts and , we obtain . Since is monotone decreasing, we conclude that . Combining with obtained above, we have . This completes the induction. ∎
Using and the initializations and , we obtain
It remains to apply the condition . ∎
2 ABPG method with exponent adaptation
The best convergence rate of the ABPG method is obtained with the largest uniform TSE for the Bregman distance. Since it is often hard to determine the largest TSE, we present in Algorithm 2 a variant of the ABPG method with automatic exponent adaptation, called the ABPG-e method.
This method starts with a large . During each iteration , it reduces by a small amount until some stopping criterion is satisfied. An obvious choice for the stopping criterion is the local triangle-scaling property
where and . According to the proof of Lemma 2, we can also use the inequality (21) as stopping criterion, which is implied by (29) and the relatively smooth assumption. For convergence analysis, we only need (21) to hold, which can be less conservative than (29). In Algorithm 2, we use the following inequality as the stopping criterion
which is equivalent to (21) (by subtracting from both sides of the inequality). In practice, this condition often leads to much faster convergence than using (29). Computationally, it is slightly more expensive since it needs to evaluate in addition to during each inner loop, while (29) does not.
By replacing inequality (18) with the one above and repeating the analysis in Section 3.1, we obtain the following result.
ABPG methods with gain adaptation
In this section, we present and analyze an adaptive ABPG method based on the concept of intrinsic TSE developed in Section 2.1. Instead of searching for the largest uniform TSE as in Algorithm 2, we can replace line 1 in Algorithm 1 by
Algorithm 3 is such a method with gain adaptation. During each iteration, the algorithm uses an inner loop to search for an value of that satisfies
which is true if the following local triangle-scaling property holds:
For any and , the following inequality holds:
The case of is obvious. Assume . The desired inequality is equivalent to
Applying the weighted arithmetic and geometric mean inequality (26), we have
where is a weighted geometric mean of the gains at each step:
We follow the same steps as in Section 3.1. In light of (31), the inequality (21) becomes
Then the same arguments in the proof of Theorem 1 lead to
Next we derive an upper bound for . For convenience, let’s define for ,
Then (32) implies . Moreover, we have
Applying Lemma 5 with and , we obtain
We can eliminate the common factor on both sides of the above inequality to obtain
Summing the above inequality from step to and using , we have
Using the weighted arithmetic and geometric mean inequality (e.g., [18, Section 2.5]) gives
Combining the last two inequalities above, we arrive at
Finally, substituting the inequality above into (38) gives the desired result. ∎
We note that the geometric mean in (34) can be much smaller than the average (arithmetic mean) of . Under the assumption of uniform Lipschitz smoothness (3), Nesterov proposed an accelerated gradient method with non-monotone line search. However, the complexity obtained there still depends on the global Lipschitz constant , more specifically, replacing in (33) with when . Our result in (33) can be tighter if the local Lipschitz constants are smaller than (equivalently with ).
In order to estimate the overhead of the gain-adaptation procedure, we follow the approach of [27, Lemma 4]. Notice that each inner loop needs to call a gradient oracle to compute , and also when we use (30) as the stopping criterion for gain adaptation. Let be the number of calls of the oracle (for ) at the th iteration, for . Then
Therefore, the total number of oracle calls is
Roughly speaking, on average each iteration need two oracle calls (unless becomes very large).
As an alternative to calculating by solving the equation (32), we can also use the following explicit update rule:
1 Towards the O(k−2)O(k^{-2}) convergence rate
where and . Suppose the sequence converges to the optimal solution and , then have . If is an interior point of the positive orthant or the simplex, meaning for all coordinates , Then we see from (40) that the bound on depends on how close is close to the boundary (assuming is bounded).
The most interesting case is when the optimal solution is on the boundary, i.e., when for some coordinates . In fact, in our numerical examples on the D-optimal design problem and Poisson linear inverse problem, most of the solutions are on the boundary. However, we emphasize that the iterates generated by the ABPG algorithm is never on the boundary, but may only converge to the boundary; see Assumption A, especially A.5. Our analysis applies to this case as well. In particular, we can show that if both sequences and converge, then implies and the convergence rate of is no faster than that of . (Here means .) More precisely, we have the following lemma.
Suppose an algorithm generates two sequences and in the strictly positive orthant, satisfying and for all . Then
If converges and for some coordinate , then there must exists an subsequence of that converges to .
Suppose both sequences and converge. If for some , then it converges at a rate that is no faster than in the following sense: For any monotone decreasing sequence that converges to and satisfies for all , we have for all . In particular, we can choose to be the monotone lower envelop of , i.e., .
By the update rule , we know that each is a convex combination of the points , which all lie in the strictly positive orthant. Therefore,
Part (a). Suppose but there is no subsequence of converging to . Then there must exist an such that for all . Since is a convex combination of , this implies
which contradicts with the assumption that . Therefore, there must exists an subsequence of that converges to .
Part (b). Suppose is monotone decreasing and converges to . If for all , then
In this sense, at a rate that is no faster than . ∎
According to the bound in (40) and Lemma 6, if both sequences and converge, then
which can be bounded by a constant, since at a rate that is no faster than . In this case, we have in Algorithm 3 bounded by a constant asymptotically, thus the convergence rate is .
However, we are not able to prove the convergence of the sequences and without additional assumptions (such as relative strong convexity). Indeed, to the best of our knowledge, convergence of these sequences have not been established even under the classical uniform Lipschitz condition. Therefore, an a priori theoretical guarantee of the rate seems to be out of reach in general, which seems to coroborate the recent result in that the rate cannot be improved in in general for the class of relatively smooth functions.
Nevertheless, we would like to reiterate the remarks at the end of Sections 1.1. In particular, the class of relatively smooth functions is very large, and the lower bound in is established with a worst-case function with pathological nonsmooth behavior. In practical applications, we always work with one particular reference function which may possess structural properties that allow fast convergence. In Algorithm 3, the sequence is readily available as part of the computation and we can easily check the magnitude of . Whenever it is small, we obtain a numerical certificate that the algorithm did converge with the rate. This is exactly what we observe in the numerical experiments in Section 6.
Accelerated Bregman dual averaging method
In this section, we present an accelerated Bregman dual averaging (ABDA) method under the relative smoothness assumption. This method extends Nesterov’s accelerated dual averaging method ( and [33, Algorithm 3]) to the relatively smooth setting. Here we focus on a simple variant in Algorithm 4 based on the uniform triangle-scaling property, although it is also possible to develop more sophisticated variants with automatic exponent or gain adaptation.
In other words, is a weighted sum of the lower approximations in (17) constructed at :
When implementing Algorithm 4, we only need to keep track of and , and there is no need to maintain the abstract form of . Here our assumption of and being simple means that the minimization problem in (43) can be solved efficiently. This requirement is equivalent to that for the BPG method (8) and all variants of the ABPG methods in this paper.
To see this, we use induction. Clearly it holds for if we choose . Suppose it holds for some , then in light of (45) and (44),
Therefore the inequality (45) holds for all .
To analyze the convergence of Algorithm 4, we need the following simple variant of Lemma 1.
Notice that for , is the minimizer of over . We use Lemma 7 to obtain
Combining the inequalities (47) and (48), we obtain
where in the last equality we used recursive definition of in (41). Dividing both sides of the above inequality by , we have
Using (44) and rearranging terms gives the desired result (46), which holds for . ∎
Suppose Assumption A holds, is -smooth relative to on , and is a uniform TSE of . The sequences generated by Algorithm 4 satisfy:
In this case, we can extend the result of Lemma 8 to hold for all . Applying the inequality (46) for iterations , we obtain
where we used and . Next using (44) and rearranging terms, we have
According to Lemma 4, we have if (44) holds, which gives (49).
where the last inequality repeats the arguments from (51) to (52). Rearranging terms leads to
and further applying Lemma 4 gives the desired result (50). ∎
where the second inequality used the upper bound in (17), and the last inequality used the lower bound in (17). Therefore, for any such that , we have
Numerical experiments
We consider three applications of relatively smooth convex optimization: D-optimal experiment design, Poisson linear inverse problem, and relative-entropy nonnegative regression. For each application, we compare the algorithms developed in this paper with the BPG method (8) and demonstrate significant performance improvement. Our implementations and experiments are shared through an open-source repository at https://github.com/linxiaolx/accbpg.
Figure 2(b) shows that for and , is mostly much smaller than . For , is much closer to but always less than . This gives a numerical certificate that the ABPG method converged with rate. For , stayed close to for the first 700 iterations and then jumped to and stayed around. The method diverges with larger value of . We didn’t plot the ABDA method (Algorithm 4) because it overlaps with ABPG for the same value of when the initial point is taken as the center of the simplex, see part (a) of Theorem 4.
We also show the comparison in terms of CPU time in Figures 3(c) and 3(d). As remarked at the end of Section 3.2, the ABPG-e method only take a constant number more iterations than ABPG, thus its their comparison is very similar to the case with number of iterations. For ABPG-g, the analysis in Section 4 on page 4 shows that the number of gradient calls and proximal computations is roughly twice of the ABPG method with the same number of iterations. This is exactly what we observe in Figures 3(c) and 3(d). Given such predictable scaling between number of iterations and CPU time, we only show comparisons in the number of iterations in the rest numerical experiments.
Figure 4 shows the comparison of different methods on another random problem instance with and . All methods converge much faster and reach very high precision. In particular, BPG and BPG-LS look to have linear convergence. This indicates that this problem instance is much better conditioned and the objective function may be strongly convex relative to Burg’s entropy. In this case, it is shown in that the BPG method attains linear convergence. The ABPG and ABPG-g methods demonstrate periodic non-monotone behavior. A well-known technique to avoid such oscillations and attain fast linear convergence is to restart the algorithm whenever the function value starts to increase . We applied restart (RS) to both ABPG and ABPG-g, which resulted in a much faster convergence as shown in Figure 4.
1.2 Experiment on real data
In our second experiment, we construct D-optimal design instances from LibSVM data . In particular, we consider several regression datasets – the goal is to find the most relevant data points where one shall run the experiment to evaluate the corresponding label.
Figure 5 shows the results on four different datasets: abalone (), bodyfat (), mpg () and housing (). The left column indicates that in each case, the best performance of ABPG is achieved with large TSE and . Furthermore, ABPG with always compared favorably over plain BPG.
Next, the second column of Figure 5 shows that both ABPG-g and ABPG (with ) always significantly outperform BPG and BPG-LS. We have chosen log-log scale of the plot to contrast the convergence rate of BPG (with line search) with the convergence rate of ABPG and ABPG-g. In the third column, we plot the local triangle-scaling gains. They serve as numerical certificates of the empirical convergence rate of ABPG and its variants. In particular, we see that for the ABPG-g algorithm is mostly flat and less than one.
2 Poisson linear inverse problem
Figure 6 shows our computational results for a randomly generated instance with and and (no regularization). The entries of and are generated following independent uniform distribution over the interval $$.
Figure 7 shows the results for a randomly generated instance with and . In this case, since , we added a regularization with . ABPG-g has the best performance. Again we observe that most of the time, which gives a numerical certificate that the ABPG methods do converge with rate.
3 Relative-entropy nonnegative regression
Acknowledgments
We thank Haihao Lu, Robert Freund and Yurii Nesterov for helpful conversations. Peter Richtárik acknowledges the support of the KAUST Baseline Research Funding Scheme.