A Randomized Nonmonotone Block Proximal Gradient Method for a Class of Structured Nonlinear Programming
Zhaosong Lu, Lin Xiao
Introduction
Nowadays first-order (namely, gradient-type) methods are the prevalent tools for solving large-scale problems arising in science and engineering. As the size of problems becomes huge, it is, however, greatly challenging to these methods because gradient evaluation can be prohibitively expensive. Due to this reason, block coordinate descent (BCD) methods and their variants have been studied for solving various large-scale problems (see, for example, ). Recently, Nesterov proposed a randomized BCD (RBCD) method, which is promising for solving a class of huge-scale convex optimization problems, provided the involved partial gradients can be efficiently updated. The iteration complexity for finding an approximate optimal solution is analyzed in . More recently, Richtárik and Takáč extended Nesterov’s RBCD method to solve a more general class of convex optimization problems in the form of
where is convex differentiable in and is a block separable convex function. More specifically,
where each denotes a subvector of with cardinality , form a partition of the components of , and each is a closed convex function.
Given a current iterate , the RBCD method picks uniformly, solves a block-wise proximal subproblem in the form of
and sets and for all , where is the partial gradient of with respect to and is the Lipschitz constant of with respect to the norm (see Assumption 1 for details). The iteration complexity of finding an approximate optimal solution with high probability is established in and has recently been improved by Lu and Xiao . Very recently, Patrascu and Necoara extended this method to solve problem (1) in which is nonconvex, and they studied convergence of the method under the assumption that the block is chosen uniformly at each iteration.
One can observe that for , the RBCD method becomes a classical proximal (full) gradient method with a constant stepsize . It is known that the latter method tends to be practically much slower than the same type of methods but with variable stepsizes, for example, spectral-type stepsize ) that utilizes partial local curvature information of the smooth component . The variable stepsize strategy shall also be applicable to the RBCD method and improve its practical performance dramatically. In addition, the RBCD method is a monotone method, that is, the objective values generated by the method are monotonically decreasing. As mentioned in the literature (see, for example, ), nonmonotone methods often produce solutions of better quality than the monotone counterparts for nonconvex optimization problems. These motivate us to propose a randomized nonmonotone block proximal gradient method with variable stepsizes for solving a class of (possibly nonconvex) structured nonlinear programming problems in the form of (1) satisfying Assumption 1 below.
Throughout this paper we assume that the set of optimal solutions of problem (1), denoted by , is nonempty and the optimal value of (1) is denoted by . For simplicity of presentation, we associate with the standard Euclidean norm, denoted by . We also make the following assumption.
is differentiable (but possibly nonconvex) in . Each is a (possibly nonconvex nonsmooth) function from to for . The gradient of function is coordinate-wise Lipschitz continuous with constants in , that is,
This paper is organized as follows. In Section 2 we propose a RNBPG method for solving structured nonlinear programming problem (1) and analyze its convergence. In Section 3 we analyze the convergence of RNBPG for solving structured convex problem. In Section 4 we conduct numerical experiments to compare RNBPG method with the RBCD method with fixed or variable stepsizes.
Before ending this section we introduce some notations that are used throughout this paper and also state some known facts. The domain of the function is denoted by . stands for for any real number . Given a closed set and a point , denotes the distance between and . For symmetric matrices and , means that is positive semidefinite. Given a positive definite matrix and a vector , . In addition, denotes the Euclidean norm. Finally, it immediately follows from Assumption 1 that
By Lemma 2 of Nesterov and Assumption 1, we also know that is Lipschitz continuous with constant , that is,
Randomized nonmonotone block proximal gradient method
In this section we propose a RNBPG method for solving structured nonlinear programming problem (1) and analyze its convergence.
We start by presenting a RNBPG method as follows. At each iteration, this method randomly picks a block according to any prescribed (not necessarily uniform) probability distribution and solves typically several associated proximal subproblems in the form of (2) with replaced by some until a certain progress on objective value is achieved.
Randomized nonmonotone block proximal gradient (RNBPG) method Choose , , , , integer , and for such that . Set .
Set . Pick with probability . Choose .
Let . Compute
Set , and go to step 1).
The above method becomes a monotone method if .
Before studying convergence of RNBPG, we introduce some notations and state some facts that will be used subsequently.
Let denote the vector obtained in Step (2) of RNBPG if is chosen to be . Define
One can observe that for and there exist and the smallest nonnegative integer such that and
Let denote the block diagonal matrix , where is the identity matrix. By the definition of and (9), we observe that
After iterations, RNBPG generates a random output , which depends on the observed realization of random vector
We define . Also, define
The following lemma establishes some relations between the expectations of and .
Let be generated by RNBPG and defined in (7). There hold
Proof. By (13) and the definitions of and , we can observe that
The conclusion of this lemma follows by taking expectation with respect to on both sides of the above inequalities.
We next show that the inner loops of the above RNBPG method must terminate finitely. As a byproduct, we provide a uniform upper bound on .
Let be the sequence generated by RNBPG, defined above, and defined in (14). There hold
.
.
Proof. (i) It is clear that . We now show by dividing the proof into two cases.
Case (i) . Since , it follows that and the conclusion holds.
Case (ii) for some integer . Suppose for contradiction that . By (13) and (14), we then have
Let such that for and
On the other hand, using (3), (10), (17), (18) and the definition of , we have
which is a contradiction to (19). Hence, and the conclusion holds.
(ii) Let be defined above. It follows from statement (i) that , which together with the definition of implies that statement (ii) holds.
The next result provides some bound on the norm of a proximal gradient, which will be used in the subsequent analysis on convergence rate of RNBPG.
Let be generated by RNBPG, and defined in (11) and (14), respectively, and
Assume that is convex. There holds
We note that by the definition in (14), we have , which implies
Therefore, the expression under the square root in (21) is always positive.
In this subsection we show that the sequence of expected objective values generated by the method converge to the expected limit of the objective values obtained by a random single run of the method.
The following lemma studies uniform continuity of the expectation of with respect to random sequences.
Suppose that is uniform continuous in some . Let and be two random vectors in generated from . Assume that there exists such that for all , and moreover,
Proof. Since is uniformly continuous in , it follows that given any , there exists such that for all satisfying . Using these relations, the Markov inequality, and the assumption that for all and , where , we obtain that for sufficiently large ,
Due to the arbitrarily of , we see that the first statement of this lemma holds. The second statement immediately follows from the first statement and the well-known inequality
We are ready to establish the first main result, that is, the expected objective values generated by the RNBPG method converge to the expected limit of the objective values obtained by a random single run of the method.
Let and be the sequences generated by the RNBPG method. Assume that is uniform continuous in , where is defined in (12). Then the following statements hold:
and for some , where .
and
We first show by induction that the following relations hold for all :
It follows from this relation and (28) that
One can also observe that and hence . Using this fact, (24), (30), Lemma 2.5, and uniform continuity of over , we obtain that
Using (24), (31), (32), the induction hypothesis, and a similar argument as above, we can obtain that
These relations, together with Lemma 2.5, uniform continuity of over and the induction hypothesis, yield
Hence, (25) and (26) hold for , and the proof of (25) and (26) is completed.
These, together with (25), (26), (34), Lemma 2.5 and uniform continuity of over , imply that
It follows from (35) that . Using this, (23) and (24), one can see that . Hence, statement (i) holds. Notice that . Combining this relation with (36), we have
Using (37) and (38), we conclude that .
Finally, we claim that . Indeed, we know that . Hence, , where . It follows that
Using this relation and dominated convergence theorem (see, for example, [2, Theorem 5.4]), we have
which, together with , implies that . Hence, statement (ii) holds.
2 Convergence to stationary points
In this subsection we show that when is sufficiently large, is an approximate stationary point of (1) with high probability.
Let be generated by RNBPG, and and defined in (7). Assume that is uniformly continuous and is locally Lipschitz continuous in , where is defined in (12). Then there hold
where denotes the Clarke subdifferential of .
Any accumulation point of is a stationary point of problem (1) almost surely.
Suppose further that is uniformly continuous in
Then . Moreover, for any and , there exists such that for all ,
Proof. (i) We know from Theorem 2.6 (ii) that , which together with (16) implies . Notice that is an optimal solution of problem (11). By the first-order optimality condition (see, for example, Proposition 2.3.2 of ) of (11) and , one can have
Using this relation along with Lemma 2.3 (ii) and (41), we obtain that
which together with the first relation of (39) implies that the second relation of (39) also holds.
(ii) Let be an accumulation point of . There exists a subsequence such that . Since , it follows that almost surely. This together with the second relation of (39) and outer semi-continuity of yields
almost surely. Hence, is a stationary point of problem (1) almost surely.
(iii) Recall that . It follows from (4) that
Using this relation and Lemma 2.3 (ii), we have
Using this relation and the fact that and , one can obtain that
In addition, since and , it follows from (8) that . Hence, one has
This inequality together with (43) yields
and hence is bounded. Also, this inequality together with and the definition of implies that , for all . In addition, by statement (i), we know . In view of these facts and invoking Lemma 2.5, one has
Using these inequalities, (44) and statement (i), we see that
The rest of statement (iii) follows from this relation and the Markov inequality.
3 Convergence rate analysis
In this subsection we establish a sublinear rate of convergence of RNBPG in terms of the minimal expected squared norm of certain proximal gradients over the iterations.
Let , , and be defined in (13), (20) and (14), respectively, and the optimal value of (1). The following statements hold
Assume further that is convex. Then
Proof. (i) Using , Lemma 2.3 (ii), and (15), one can observe that
Let and for all . One can see from (29) that
Summing up the above inequality over , we have
which together with implies that
Given any , let . Observe that
which together with (45) implies that statement (i) holds.
Using this relation and a similar argument as above, one has
Statement (ii) immediately follows from this inequality and (21).
Convergence analysis for structured convex problems
In this section we study convergence of RNBPG for solving structured convex problem (1). To this end, we assume throughout this section that and are both convex functions.
The following result shows that can be arbitrarily close to the optimal value of (1) with high probability for sufficiently large .
Let be generated by the RNBPG method, and let and the optimal value and the set of optimal solutions of (1), respectively. Suppose that and are convex functions and is uniformly continuous in , where is defined in (40). Assume that there exists a subsequence such that is bounded. Then there hold:
For any and , there exists such that for all ,
Proof. (i) Let be defined in(7). Using the assumption that is uniformly continuous in and Theorem 2.7, one has
which, together with (47) and the assumption that is bounded, implies that
Using this relation, (48) and (49), we obtain that
(ii) Statement (ii) immediately follows from statement (i), the Markov inequality, and the fact .
In the rest of this section we study the rate of convergence of a monotone version of RNBPG, i.e., , or equivalently, (6) is replaced by
The following lemma will be subsequently used to establish a sublinear rate of convergence of RNBPG with .
Suppose that a nonnegative sequence satisfies
Proof. We divide the proof into two cases.
Case (i): Suppose for all . Let . It follows from (51) that
which together with implies that
where . By the definition of , one can see that (53) holds for . Suppose it holds for some . We now need to show (53) also holds for . Indeed, since , we have
Using this inequality, (52) and the induction hypothesis , we obtain that
namely, (53) holds for . Hence, the induction is completed and (53) holds for all . The conclusion of this lemma follows from (53) and the definitions of and .
We next establish a sublinear rate of convergence on the expected objective values for the RNBPG method with when applied to problem (1), where and are assumed to be convex. Before proceeding, we define the following quantities
where denotes the set of optimal solutions of (1) and is defined in (12).
Let be defined in (14), (54), (55), respectively. Assume that and are finite. Suppose that is -Lipschitz continuous in , namely,
for some . Let be generated by RNBPG with . Then
Proof. Let be defined in (7). For each , let such that . Due to and (54), we know that . By the definition of and (11), one can observe that
Using this inequality, (55), and (56), we have
where the first inequality follows from convexity of and (56), the second inequality is due to (55), the third inequality follows from (58), and the last inequality is due to . The preceding inequality, (16) and the fact yield
In addition, using and (50), one has
Let . Combining the preceding two inequalities, we obtain that
where is defined in (57). Notice that . Using this relation, the definition of , and Lemma 3.2, one can see that the conclusion of this theorem holds.
The next result shows that under an error bound assumption the RNBPG method with is globally linearly convergent in terms of the expected objective values.
Let be generated by RNBPG. Suppose that there exists such that
where is given in (20) and denotes the set of optimal solutions of (1). Then there holds
Proof. For each , let such that . Let be defined in (7), and
Using this inequality, (11) and Lemma 2.3 (ii), we have that
where . Using this relation and (59), one can obtain that
It follows from this inequality and (21) that
In addition, by (3) and the definition of , we have
Using these two inequalities, we can obtain that
where the first inequality follows from (61) and the second inequality is due to (62). Taking expectation with respect to on both sides of the above inequality gives
Using this inequality and (60), we obtain that
where is defined above. In addition, it follows from (50) that
Combining these two inequalities, we obtain that
and the conclusion of this theorem immediately follows.
The error bound condition (59) holds for a class of problems, especially when is strongly convex. More discussion about this condition can be found, for example, in .
Numerical experiments
where , , and is a regularization parameter. Clearly, this problem is a special case of the general model (1) with and and thus our proposed RNBPG method can be suitably applied to solve it.
We generated a random instance with and following the procedure described in [19, Section 6]. The advantage of this procedure is that an optimal solution is generated together with and , and hence the optimal value is known. We generated an instance where the optimal solution has only nonzero entries, so this can be considered as a sparse recovery problem. We compare RNBPG with the following methods:
RBCD: The RBCD method with constant step sizes determined by the Lipschitz constants . Here, where is the th column block corresponding to the block partitions of and is the matrix spectral norm.
RBCD-LS: A variant of RBCD method with variable stepsizes that are determined by a block-coordinate-wise backtracking line search scheme. This method can also be regarded as a variant of RNBPG with , but which has the property of monotone descent.
As discussed in , the structure of the least-squares function allows efficient computation of coordinate gradients, with cost of operations for block as opposed to for computing the full gradient. We note that the same structure also allows efficient computation of the function value, which costs the same order of operations as computing coordinate gradients. Therefore the backtracking line search used in RBCD-LS as well as the nonmonotone line search used in RNBPG (both relies on computing function values), have the same order computational cost as evaluating coordinate gradients at each iteration. Therefore we can focus on comparing their required number of iterations to obtain the same accuracy in reducing the objective value.
We run each algorithm with four different block coordinate sizes for all . For each blocksize, we pick the block coordinates uniformly at random at each iteration. Note that gives the full gradient versions of the methods considered, which are deterministic algorithms. We choose the same initial point for all three methods.
For the RNBPG method, we used the parameters , , , and . In addition, we used the Barzilai-Borwein spectral method to compute the initial estimate . That is, we choose
Figure 1 shows the behavior of different algorithms with the four different block coordinate sizes. For in Figure 1, RBCD has slightly better convergence speed than RBCD-LS and RNBPG. The reason is that in this case, along each block becomes an one-dimensional quadratic function, and the value gives the accurate second partial derivative of along each dimension. Therefore in this case the RBCD method essentially uses the best step size, which is generally better than the ones used in RBCD-LS and RNBPG.
When the blocksize is larger than one, the value is the magnitude of second derivative along the most curved direction. Line search based methods may take advantage of the possibly much smaller local curvature along the search direction by taking larger step sizes. Figure 1 , and show that RBCD-LS converges much faster that RBCD while RNBPG (with ) converges substantially faster than RBCD-LS.
Figure 2 shows more comprehensive study of the performance of the three methods: RBCD, RBCD-LS, and RNBPG. Figure 2 shows the number of iterations of different methods required to reach the precision , when using 10 different block sizes ranging from 1 to 2000 with equal logarithmic spacing. Figure 2 shows the number of epochs required to reach the same precision, where each epoch corresponds to one equivalent pass over the dataset , that is, equivalent to iterations. For each method and each block size, we record the results of 10 runs with different random sequences to pick the block coordinates, and plot the mean with the standard deviation as error bars. As we can see, the number of iterations in general decreases when we increase the block size, because each iteration involves more coordinates and more computation. On the other hand, the number of epochs required increases with the block size, meaning that larger block size updates are less efficient than small block size updates.
The above observations suggest that using larger block sizes is less efficient in terms of the overall computation work (e.g., measured in total flops). However, this does not mean longer computation time. In particular, using larger block sizes may better take advantage of modern multi-core computers for parallel computing, thus may take less computation time. Figure 2 shows the computation time required to reach the same precision on a 12 core Intel Xeon computer. We used the Intel Math Kernel Library (MKL) to carry out parallel dense matrix and vector operations. The results suggest that using appropriate large block size may take the least amount of computation time. We note that such timing results heavily depend on the specific architecture of the computer, in particular its cache size for fast access, the relative size of the data matrix , and other implementation details. For example, Figure 2 shows the timing results on the same computer for a different problem instance with and . Here the best block size is smaller than one shown in Figure 2, because the size of each column of is doubled and the operations involved in each coordinate (corresponding to a column of the matrix) has increased. However, for any fixed block size, the relative performance of the three algorithms are consistent; in particular, RNBPG substantially outperforms the other two methods in most cases.
We also conducted experiments on using randomized block coordinate methods to solve a dual SVM problem in machine learning (specifically, the dual of a smoothed SVM problem described in [27, Section 6.2]). We used two real datasets from the LIBSVM web site , whose characteristics are summarized in Table 1. In the dual SVM problem, the dimension of the dual variables are the same as the number of samples , and we partition the dual variables into blocks to apply the three randomized block coordinate gradient methods. Figure 3 shows the reduction of the objective value with the three methods on the two datasets, each illustrated with two block sizes: and . We observe that the RNBPG method converges faster than the other two methods, especially with relatively larger block sizes.
To conclude, our experiments on both synthetic and real datasets clearly demonstrate the advantage of the nonmonotone line search strategy (with spectral initialization) for randomized block coordinate gradient methods.