Improved Iteration Complexity Bounds of Cyclic Block Coordinate Descent for Convex Problems
Ruoyu Sun, Mingyi Hong
Introduction
Consider the following convex optimization problem
Algorithm 1: The Cyclic Block Coordinate Descent (BCD) At each iteration , update the variable blocks by: (2)
where we have used the following short-handed notations:
The convergence analysis of the BCD has been extensively studied in the literature, see . For example it is known that for smooth problems (i.e. is continuous differentiable but possibly nonconvex, ), if each subproblem has a unique solution and is non-decreasing in the interval between the current iterate and the minimizer of the subproblem (one special case is per-block strict convexity), then every limit point of is a stationary point [5, Proposition 2.7.1]. The authors of have derived relaxed conditions on the convergence of BCD. In particular, when problem (1) is convex and the level sets are compact, the convergence of the BCD is guaranteed without requiring the subproblems to have unique solutions . Recently Razaviyayn et al have shown that the BCD converges if each subproblem (2) is solved inexactly, by way of optimizing certain surrogate functions.
Luo and Tseng in have shown that when problem (1) satisfies certain additional assumptions such as having a smooth composite objective and a polyhedral feasible set, then BCD converges linearly without requiring the objective to be strongly convex. There are many recent works on showing iteration complexity for randomized BCGD (block coordinate gradient descent), see and the references therein. However the results on the classical cyclic BCD is rather scant. Saha and Tewari show that the cyclic BCD achieves sublinear convergence for a family of special LASSO problems. Nutini et al show that when the problem is strongly convex, unconstrained and smooth, BCGD with certain Gauss-Southwell block selection rule could be faster than the randomized rule. Recently Beck and Tetruashvili show that cyclic BCGD converges sublinearly if the objective is smooth. Subsequently Hong et al in show that such sublinear rate not only can be extended to problems with nonsmooth objective, but is true for a large family of BCD-type algorithm (with or without per-block exact minimization, which includes BCGD as a special case). When each block is minimized exactly and when there is no per-block strong convexity, Beck proves the sublinear convergence for certain 2-block convex problem (with only one block having Lipschitzian gradient). It is worth mentioning that all the above results on cyclic BCD directly apply to randomly permuted BCD in which the blocks are randomly sampled without replacement in each cycle, since the proof techniques do not require the same order to be used in each cycle. However, we suspect that special proof techniques tailored for randomly permutation are necessary for establishing tight bounds of randomly permuted BCD; in fact, a recent work of Sun et al on randomly permuted ADMM provides some theoretical evidence that the cyclic rule is worse than the random permutation rule.
To illustrate the rates developed for the cyclic BCD algorithm, let us define to be the optimal solution set for problem (1), and define the constant
Also assume that has Lipschitz continuous gradient with respect to each , i.e.,
Let and . It is known that the cyclic BCPG has the following iteration complexity Note that the assumptions made in and are slightly different, but the rates derived in both cases have similar dependency on the problem dimension .
where is some constant independent of problem dimension. Similar bounds are provided for cyclic BCD in [7, Theorem 6.1]. In contrast, it is well known that when applying the classical gradient descent (GD) method to problem (1) with the constant stepsize , we have the following rate estimate [11, Corollary 2.1.2]
Note that unlike (6), here the constant in front of the term is independent of the problem dimension. In fact, the ratio of the bound given in (6) and (7) is
which is at least in the order of . For big data related problems with over millions of variables, a multiplicative constant in the order of can be a serious issue. In a recent work by Saha and Tewari , the authors show that for a LASSO problem with special data matrix, the rate of cyclic BCD (with special initialization) is indeed -independent. Unfortunately, such a result has not yet been extended to any other convex problems. An open question posed by a few authors are: is such a factor gap intrinsic to the cyclic BCD or merely an artifact of the existing analysis?
In this paper, we provide improved iteration complexity bounds of cyclic block coordinate descent methods for convex composite function minimization. Our analyses do not depend on the update order of block variables inside each cycle, thus our results also apply to BCD methods with random permutation (random sampling without replacement, another popular variant). Recall that is the number of blocks, is the global Lipshitz constant of , and is the Lipshitz constant of with respect to the -th block.
For minimizing the sum of a convex quadratic function (not necessarily strongly convex) and sepearable non-smooth function (including LASSO as a special case), we prove that the iteration complexity bound of cyclic block coordinate gradient descent (C-BCGD) with a small stepsize matches that of GD (gradient descent) up to an -factor. When the stepsize for the -th block update is , we establish an iteration complexity bound that is -times better than the existing bound, but still times worse than GD. We also improve the bound of the exact BCD by an order of (note that if each block has size , exact CD is equivalent to CGD with stepsize ; otherwise, exact BCD is different from BCGD).
For general smooth convex optimization (i.e. ), we prove a meta iteration complexity bound of cyclic coordinate gradient descent (C-CGD) that is proportional to the spectral norm of a “moving-iterate Hessian”. By an preliminary estimate of this spectral norm, this meta bound implies an iteration complexity bound that, in certain scenarios, matches the bound of GD and -times better than the known bounds. For the quadratic case, the moving-iterate Hessian becomes a constant matrix and the meta bound reduces to the bound mentioned in the first bullet which matches GD up to an -factor for small stepsize .
For illustration, we summarize the comparison of our main results with other existing bounds in several simple settings in Table 1. For simplicity, we only list the results of BCGD for the smooth case; note that the results listed in the last row for QP also apply to LASSO (or more general, sum of a quadratic function plus separable non-smooth function), and for cyclic exact BCD the results are similar to the last row (see Theorem 3.1). We assume and consider two extreme values of : and . The first case represents a separable objective function with a block diagonal Hessian matrix, and the second case represents a highly non-separable objection function with a full Hessian matrix, assuming all ’s have the same size.
Improved Bounds of Cyclic BCPG for Nonsmooth Quadratic Problem
In this section, we consider the following nonsmooth quadratic problem
We consider the following cyclic BCPG algorithm.
Algorithm 2: The Cyclic Block Coordinate Proximal Gradient (BCPG) At each iteration , update the variable blocks by: (9)
Here is the inverse of the stepsize for , which satisfies
Define and . Note that for the least square problem (smooth quadratic minimization, i.e. ), BCPG reduces to the widely used BCGD method.
The optimality condition for the th subproblem is given by
In what follows we show that the cyclic BCPG for problem (8) achieves a complexity bound that only dependents on , and apart from such log factor it is at least times better than those known in the literature. Our analysis consists of the following three main steps:
Estimate the descent of the objective after each BCPG iteration;
Estimate the cost yet to be minimized (cost-to-go) after each BCPG iteration;
Combine the above two estimates to obtain the final bound.
First we show that the BCPG achieves the sufficient descent.
We have the following estimate of the descent when using the BCPG:
Proof. We have the following series of inequalities
where the second inequality uses the optimality condition (11). Q.E.D.
To proceed, let us introduce two matrices and given below, which have dimension and , respectively
By utilizing the definition of in (10) we have the following inequalities (the second inequality comes from [12, Lemma 1])
where is the identity matrix and the notation “” denotes the Kronecker product.
We have the following estimate of the optimality gap when using the BCPG:
Proof. First note that being convex quadratic implies that its second order Taylor expansion is tight
Using this fact we can estimate by the following series of inequalities
where in we have used the optimality condition of the subproblem (9) (i.e., (11)); in the last equality we have defined a lower triangular matrix
where “” denotes the Hadamard product; is a lower triangular matrix similarly as defined in (31), but of dimension . Combining this identity and (25), we have
where uses the Cauchy-Schwartz inequality and the fact that {\mbox{\widetilde{P}}}\otimes I_{N}\succeq{\mbox{\widetilde{A}}}^{T}{\mbox{\widetilde{A}}}; is true for all . Inequality is true due to a result on the spectral norm of the triangular truncation operator; see [1, Theorem 1]. In particular, Define
Our third step combines the previous two steps and characterizes the iteration complexity. This is the main result of this section.
The iteration complexity of using BCPG to solve (8) is given below.
When the stepsizes are chosen conservatively as , we have
When the stepsizes are chosen as . Then we have
Proof. For notational simplicity, let us define
Taking a square of the cost-to-go estimate (2.2) and the sufficient descent estimate (12), we obtain
Utilizing a result from [2, Lemma 3.5], the above inequality implies that
When for all , the bound reduces to
When the problem is smooth and unconstrained, we have
The bounds derived in Theorem 2.1 is again at least times better than existing bounds of cyclic BCPG. For example, when the problem is smooth and unconstrained, the ratio between our bound (35) and the bound (6) is given by
where in the last inequality we have used the fact that .
For unconstrained smooth problems, let us compare the bound derived in the second part of Theorem 2.1 (stepsize ) with that of the GD (7). If for all (problem badly conditioned), our bound is about times worse than that of the GD. This indicates a counter-intuitive phenomenon: by choosing conservative stepsize the iteration complexity of BCGD is times better compared with choosing a more aggressive stepzise . It also indicates that the factor may hide an additional factor of .
Improved Bounds of Cyclic BCD for Quadratic Problems
In the previous section, we analyzed an inexact cyclic BCD algorithm, the BCPG algorithm. In this section we analyze the performance of the cyclic BCD algorithm (with exact minimization, cf. (2)), when applied to the quadratic problem (8).
We will divide our analysis into three cases ( denotes the minimum eigenvalue):
Each has full column rank with , where is some known constant;
Each has full row rank with , where is some known constant;
Each has neither full column rank nor full row rank.
Note that in the last two cases the subproblems are not strongly convex hence may not have unique solutions. Without uniqueness, the only known iteration complexity bound for the cyclic BCD is developed in , but such bound is at least proportional to .
Our analysis again follows the three-step approach used in the previous section. To estimate the descent, we have the following lemma.
We have the following estimate of the descent when using the BCD
Proof. We have the following series of inequalities
where in we have used the fact that the second order Taylor expansion of a quadratic problem is exact; in we have used the optimality of ; cf. (11). Q.E.D.
The cost-to-go estimate is given by the following lemma.
Let and . We have the following estimate of the optimality gap when using the cyclic BCD to solve (8).
When each has full column rank, we have
When each has full row rank, we have
When each has neither full column rank nor full row rank, we have
Proof. We again bound . First we have
where in we have used the optimality condition of (cf. (11)); in is the following lower triangular matrix
Below we bound for three different cases.
where is the lower triangular matrix; in the last inequality we have again used the property of lower triangular truncation operator; see the proof of Lemma 2.2.
where in {\mbox{\widetilde{A}}}^{\dagger} is the Moore-Penrose pseudoinverse of , and we have utilized the fact that when is invertible, its Moore-Penrose pseudoinverse is given by ; in we again have used the property of lower triangular truncation operator; see the proof of Lemma 2.2.
The third step again combines the previous two steps to derive the final bounds.
The iteration complexity of using cyclic BCD to solve (8) is given below.
When each has full column rank, we have
When each has full row rank, we have
When each has neither full column rank nor full row rank, we have
Proof. First, consider the case where ’s all have full column rank. In this case the descent estimate (39) can be further expressed as
Using this relation and squaring both sides of (40), we obtain
Second, when ’s all have full row rank, we can simply combine (39) and (41) and obtain
Third, when ’s all neither full row rank nor full column rank, we can combine (39) and (42) and obtain
Again by utilizing result from [2, Lemma 3.5] to the recursions (54)–(56) we obtained the desired results. Q.E.D.
Except for the log factor, such bound is again no longer explicitly dependent on , and is about times better than the existing bound on the cyclic BCD .
Viewing and as constants, the bound derived in (35) for BCPG is . An immediate question is: is it possible to further reduce these bounds by an additional order of , i.e. ? The answer is negative at least for , as shown below. Note that the analysis below does not exclude the possibility that BCD can achieve an iteration complexity or , where is a constant depending on . Finding the iteration complexity lower bound of exact BCD for smooth quadratic minimization remains an interesting open question.
Note that , but it is not necessarily full rank. Clearly is one of the optimal solutions for problem (58), while is also the the optimal objective value. Further, we know that the maximum eigenvalue for the matrix is:
which means that . Therefore the Lipschitz constant of is bounded by , regardless of the dimension . For the classical gradient decent method, we must have (cf. (7))
where is the th column of . For this block structure, we have , , and .
Based upon the above block structure, it is easy to verify that the cyclic BCD with exact minimization generates the same sequence as the cyclic BCPG with stepsizes . Therefore our result below holds for both algorithms. Consider problem (67) with given by (65). Let
We can show through explicit computation that after running cyclic BCD/BCPG for one pass of all blocks, the optimality gap is bounded below by
The derivation is relegated to the Appendix. This result implies that it is not possible to derive a global complexity bound for BCD and BCPG with stepsizes
Iteration Complexity for General Convex Problems
In this section, we consider improved iteration complexity bounds of BCD for general unconstrained smooth convex problems. We prove a general iteration complexity result, which includes a result of Beck et al. as a special case. Our analysis for the general case also applies to smooth quadratic problems, but is very different from the analysis in previous sections for quadratic problems. For simplicity, we only consider the case (scalar blocks); the generalization to the case is left as future work.
Let us assume that the smooth objective has second order derivatives . When each block is just a coordinate, we assume Then and . For unconstrained smooth convex problems with scalar block variables, the BCPG iteration reduces to the following coordinate gradient descent (CGD) iteration:
where and means that is a linear combination of and ( is the -th block unit vector).
The general framework follows the standard three-step approach that combines sufficient descent and cost-to-go estimate; nevertheless, the analysis of the sufficient descent is very different from the methods used in the previous sections.
Proof. Since and only differ by the -th block, and is Lipschitz continuous with Lipschitz constant , we have A stronger bound is , where but since , the improvement ratio of using this stronger bound is no more than a factor of .
where the last inequality is due to .
by the mean-value theorem, there must exist such that
where is the second order derivative of . Then
Let and let be defined as in (72), then , which implies
Plugging into (79), we obtain the desired result. Q.E.D.
The above lemma provides an estimate of the sufficient descent. The intuition is that CGD can be viewed as an inexact gradient descent method, thus the amount of descent can be bounded in terms of the norm of the full gradient. It would be difficult to further tighten this bound if the goal is to obtain a sufficient descent based on the norm of the full gradient. To further improve this bound, we suspect that either the third order derivatives of or the relation between and should be considered (in the derivation of Lemma 4.1 we treat and independently).
Having established the sufficient descent in terms of the full gradient , we can easily prove the iteration complexity result, following the standard analysis of GD (see, e.g. [11, Theorem 2.1.13]).
Proof. Denote , then (71) becomes
This relation also implies , thus by the definition of in (3) we have . By the convexity of and the Cauchy-Schwartz inequality, we have
Let , we obtain
This completes the proof of the claim. Q.E.D.
In the following corollary, we provide an initial estimate of and obtain a more concrete iteration complexity bound.
For CGD with , we have , thus
Proof. First, from the fact that is a scalar bounded above by , thus
We provide the second bound of below. Let denote the -th row of , then . Therefore, we have
Plugging this bound and (84) into (81), we obtain the desired result. Q.E.D.
Let us compare this bound with the bound derived in [4, Theorem 3.1] (replacing the denominator by ), which is
We have demonstrated that our bound can match GD in some cases, but can possibly be times worse than . An interesting question is: for general convex problems can we obtain an bound for cyclic BCGD, matching the bound of GD? Removing the -factor in (85) will lead to an bound for conservative stepsize no matter how large and are. Unfortunately, our approach which is based on estimating the norm of probably cannot lead to removal of this -factor as it seems unlikely to prove a bound . Our bound is tight in some cases, but not always tight. A better bound of (and thus a better iteration complexity bound) may be obtained by answering the following question (below represents the Hessian matrix corresponding to , , and denotes the ’th row of a matrix ):
The answer to this question seems to be rather complicated and is left as future work.
We conjecture that an bound for cyclic BCGD cannot be achieved for general convex problems. That being said, we point out that the iteration complexity of cyclic BCGD may depend on other intrinsic parameters of the problem such as and, possibly, third order derivatives of . Thus the question of finding the best iteration complexity bound of the form , where is a function of , may not be the right question to ask for BCD type algorithms.
Our complexity results apply directly to a popular variant of the BCD – the permuted BCD, in which the blocks are randomly sampled without replacement. This is because our analysis only depends on the behavior of the algorithm during one iteration in which all blocks are updated once. The order of the block update across different iterations is irrelevant in the analysis. However, we mention that the key research question for permuted BCD is not whether it converges, but rather whether performing the random permutation delivers improved complexity bounds. To our knowledge this is still an open question that worth further investigation.
Conclusion
In this paper, we provide new analysis and improved complexity bounds for cyclic BCD-type methods. Our results also apply to BCD methods with random permutation (random sampling without replacement). For minimizing the sum of a convex quadratic function and separable non-smooth functions, we show that cyclic BCGD with small stepsize has an iteration complexity bound , which is independent of and matches the bound of GD (except for a mild factor). For general smooth convex problems, we consider cyclic CGD and prove a meta iteration complexity bound that is proportional to the spectral norm of a “moving-iterate Hessian”. This meta bound leads to an iteration complexity bound that is sometimes -times better than existing bound of , and matches the bound of GD for the quadratic case when using small stepsize . The derivation of our meta iteration complexity bound is rather tight, but still this bound seems to be worse than the bound of GD in certain scenarios. It remains an interesting open question whether this gap in the general convex case is artificial.
Claim. Consider problem (67) with given by (65). Let
Then after running BCD/BCPG for one iteration, the optimality gap is bounded below by
Proof. Assume that we start with a vector , let us generate the iterate by using cyclic BCD (assuming that coordinates are picked in the order of ). The first variable is updated by
Here the second equality is true by the property of the matrix . Clearly we have
The second variable is updated by
Similarly, the third variable is updated by
In general, for all , we have
Also we have that is updated by
This means that after the first iteration, the objective becomes
Combining the above result, we see that when applying the BCD/BCPG to the problem (58), the complexity bound is at least
This completes the proof of the claim. Q.E.D.