Iteration complexity analysis of random coordinate descent methods for $\ell_0$ regularized convex problems
Andrei Patrascu, Ion Necoara
Introduction
where function is smooth and convex and the quasinorm of is defined as:
2 Notations and preliminaries
The function has (block) coordinatewise Lipschitz continuous gradient with constants for all , i.e. the convex function satisfies the following inequality for all :
An immediate consequence of Assumption 1 is the following relation :
Characterization of local minima
In this section we present the necessary optimality conditions for problem (1) and provide a detailed description of local minimizers. First, we establish necessary optimality conditions satisfied by any local minimum. Then, we separate the set of local minima into restricted classes around the set of global minimizers. The next theorem provides conditions for obtaining local minimizers of problem (1):
For the first implication, we assume that is a local minimizer of problem (1) on the open ball , i.e. we have:
Based on Assumption 1 it follows that has also global Lipschitz continuous gradient, with constant , and thus we have:
Taking and , we obtain:
Therefore, we have , which means that:
Clearly, for any , with , we have:
Let , with . The convexity of function and the Holder inequality lead to:
We now assume that satisfies (3). For any we have , which by (4) implies that whenever . Therefore, we get:
We denote with the set of all local minima of problem (1), i.e.
and we call them basic local minimizers. It is not hard to see that when the function is strongly convex, the number of basic local minima of problem (1) is finite, otherwise we might have an infinite number of basic local minimizers.
We additionally impose the following assumptions on each function .
(iv) There exists such that and
Note that a similar set of assumptions has been considered in , where the authors derived a general framework for the block coordinate descent methods on composite convex problems. Clearly, Assumption 3 implies the upper bound (6) and in this inequality is replaced with the assumption of strong convexity of in the first argument.
We now provide several examples of approximation versions of the objective function which satisfy Assumption 3.
It satisfies Assumption 3, in particular condition holds for . This type of approximations was used by Nesterov for deriving the random coordinate gradient descent method for solving smooth convex problems and further extended to the composite convex case in [NecCli:13, 25].
2. General quadratic approximation: given , such that for all , we define the approximation version
It satisfies Assumption 3, in particular condition holds for (the smallest eigenvalue). This type of approximations was used by Luo, Yun and Tseng in deriving the greedy coordinate descent method based on the Gauss-Southwell rule for solving composite convex problems .
It satisfies Assumption 3, in particular condition holds for . This type of approximation functions was used especially in the nonconvex settings .
Based on each approximation function satisfying Assumption 3, we introduce a class of restricted local minimizers for our nonconvex optimization problem (1).
For any set of approximation functions satisfying Assumption 3, a vector is called an u-strong local minimizer for problem (1) if it satisfies:
Moreover, we denote the set of strong local minima, corresponding to the approximation functions , with .
and thus an u-strong local minimizer , has the property that each block is a fixed point of the operator defined by the minimizers of the function , i.e. we have for all :
Let the set of approximation functions satisfy Assumption 3, then any strong local minimizer is a local minimum of problem (1), i.e. the following inclusion holds:
From Definition 5 and Assumption 3 we have:
we have from the definition of that
and thus or equivalently . Since this holds for any , it follows that satisfies . Using now Theorem 2 we obtain our statement. ∎
\begin{cases}\lvert\nabla_{(j)}f(z)\rvert\leq\sqrt{2\lambda_{i}M_{i}},&\text{if}\ z_{(j)}=0\\ \lvert z_{(j)}\rvert\geq\sqrt{\frac{2\lambda_{i}}{M_{i}}},&\text{if}\ z_{(j)}\neq 0,\quad\forall i\in[N] and j\in\mathcal{S}_{i}.\end{cases}
The relations given in can be derived based on the separable structure of the approximation and of the quasinorm using similar arguments as in Lemma 3.2 from . For completeness, we present the main steps in the derivation. First, it is clear that any satisfies:
for all and . On the other hand since the optimum point in the previous optimization problems can be or different from , we have:
Let Assumption 1 hold and be two approximation functions satisfying Assumption 3. Additionally, let
Assume , i.e. it is a global minimizer of our original nonconvex problem (1). Then, we have:
and thus , i.e. we proved that . Therefore, any class of -strong local minimizers contains the global minima of problem (1).
Further, let us take . Using Definition (5) and defining
This shows that and thus . ∎
Note that if the following inequalities hold
using the Lipschitz gradient relation (2), we obtain that
Therefore, from Theorem 7 we observe that -strong local minimizers for problem (1) are included in the class of all basic local minimizers . Thus, designing an algorithm which converges to a local minimum from () will be of interest. Moreover, -strong local minimizers for problem (1) are included in the class of all -strong local minimizers. Thus, designing an algorithm which converges to a local minimum from will be of interest. To illustrate the relationships between the previously defined classes of restricted local minima and see how much they are related to global minima of (1), let us consider an example.
Random coordinate descent type methods
In order to find a local minimizer of problem (1), we introduce the family of random block coordinate descent iterative hard thresholding (RCD-IHT) methods, whose iteration is described as follows:
Choose a (block) coordinate with uniform probability
Set and .
then the iteration of (RCD-IHT) method becomes:
for all . Note that if at some iteration , then the iteration of algorithm (RCD-IHT) is identical with the iteration of the usual random block coordinate gradient descent method [NecCli:13, 22]. Further, our algorithm has, in this case, similarities with the iterative hard thresholding algorithm (IHTA) analyzed in . For completeness, we also present the algorithm (IHTA).
or equivalently for each component we have the update:
Then, it can be seen that the iteration of (RCD-IHT) in the scalar case for the exact approximation has the following form:
In general, if the function satisfies Assumption 1, computing at each iteration of (RCD-IHT) requires the minimization of an unidimensional convex smooth function, which can be efficiently performed using unidimensional search algorithms. Let us analyze the least squares settings in order to highlight the simplicity of the iteration of algorithm (RCD-IHT) in the scalar case for the approximation .
where . Under these circumstances, the iteration of (RCD-IHT) has the following closed form expression:
In the sequel we use the following notations for the entire history of index choices, the expected value of objective function w.r.t. the entire history and for the support of the sequence :
Due to the randomness of algorithm (RCD-IHT), at any iteration with , the sequence changes if one of the following situations holds for some :
In other terms, at a given moment with , we expect no change in the sequence of algorithm (RCD-IHT) if there is no index satisfying the above corresponding set of relations and . We define the notion of change of in expectation at iteration , for algorithm (RCD-IHT) as follows: let be the sequence generated by (RCD-IHT), then the sequence changes in expectation if the following situation occurs:
which implies (recall that we consider uniform probabilities for the index selection):
In the next section we show that there is a finite number of changes of in expectation generated by algorithm (RCD-IHT) and then, we prove global convergence of this algorithm, in particular we show that the limit points of the generated sequence converges to strong local minima from the class of points .
Global convergence analysis
In order to prove almost sure convergence results for our family of algorithms, we use the following supermartingale convergence lemma of Robbins and Siegmund (see e.g. ):
Let and be three sequences of nonnegative random variables satisfying the following conditions:
where denotes the collections , . Then, we have for a random variable a.s. and a.s.
Further, we analyze the convergence properties of algorithm (RCD-IHT). First, we derive a descent inequality for this algorithm.
Let be the sequence generated by (RCD-IHT) algorithm. Under Assumptions 1 and 3 the following descent inequality holds:
In conclusion, our family of algorithms belong to the class of descent methods:
Taking expectation w.r.t. we get our descent inequality. ∎
We now prove the global convergence of the sequence generated by algorithm (RCD-IHT) to local minima which belongs to the restricted set of local minimizers .
Let be the sequence generated by algorithm (RCD-IHT). Under Assumptions 1 and 3 the following statements hold:
At each change of sequence in expectation we have the following relation:
where
The sequence changes a finite number of times as almost surely. The sequence converges to some almost surely. Furthermore, any limit point of the sequence belongs to the class of strong local minimizers almost surely.
On the other hand, given , from the definition of we get:
Subtracting from both sides, leads to:
Further, if we apply the Lipschitz gradient relation given in Assumption 3 in the right hand side and use the optimality conditions for the unconstrained problem solved at each iteration, we get:
Combining with the left hand side of (15) we get:
Replacing for , it can be easily seen that, for any and , we have:
Further, assume that at some iteration a change of sequence in expectation occurs. Thus, there is an index (and block containing ) such that either or . Analyzing these cases we have:
Observing that under uniform probabilities we have:
we can conclude that at each change of sequence in expectation we get:
Further, if the sequence is constant for , then we have and for any vector satisfying . Also, for algorithm (RCD-IHT) is equivalent with the classical random coordinate descent method , and thus shares its convergence properties, in particular any limit point of the sequence is a minimizer on the coordinates for . Therefore, if the sequence is fixed, then we have for any and :
On the other hand, denoting with an accumulation point of , taking limit in (17) and using that as , we obtain the following relation:
for all and thus is the minimizer of the previous right hand side expression. Using the definition of local minimizers from the set , we conclude that any limit point of the sequence belongs to this set, which proves our statement. ∎
It is important to note that the classical results for any iterative algorithm used for solving nonconvex problems usually state global convergence to stationary points, while for our algorithms we were able to prove global convergence to local minima of our nonconvex and NP-hard problem (1). Moreover, if for all , then the optimization problem (1) becomes convex and we see that our convergence results cover also this setting.
Rate of convergence analysis
where . Using the strong convexity property for we have:
In order to derive the rate of convergence in probability for algorithm (RCD-IHT), we first define the following notion which is a generalization of relations (8) and (3) for and , respectively:
We make the following assumption on functions and consequently on :
There exist some positive constants and such that the approximation functions satisfy for all :
Note that if is strongly convex, then the set of basic local minima has a finite number of elements. Next, we show that this assumption holds for the most important approximation functions (recall that in the scalar case ).
Under Assumption 1 the following statements hold: If we consider the separable quadratic approximation , then:
For the separable quadratic approximation , using the definition of and given in (20)–(21) (see also (8)), we get:
Then, since and using the property of the norm for any two vectors and , we obtain:
For the exact approximation , using the definition of and given in (20)–(21) (see also (3)), we get:
Then, using the triangle inequality we derive the following relation:
In order to bound , it is sufficient to find upper bounds on and . For a bound on we use . Using the optimality conditions for the map and convexity of we obtain:
where in the last inequality we used the Cauchy-Schwartz inequality. On the other hand, from the global Lipschitz continuous gradient inequality we get:
In order to obtain a bound on we observe that:
where in the last inequality we used the Lipschitz gradient relation and Cauchy-Schwartz inequality. Also, from the convexity of and the Cauchy-Schwartz inequality we get:
Combining now the bounds (24) and (25) we obtain:
Therefore, from (23) and (26) we obtain a bound on :
Regarding the second quantity , we observe that:
From the upper bounds on and given in (27) and (28), respectively, we obtained our result. ∎
We further show that the second part of Assumption 13 holds for the most important approximation functions .
Under Assumption 1 the following statements hold: Considering the separable quadratic approximation , then for any fixed there exist only two values of parameter satisfying . Considering the exact approximation , then for any fixed , there exists a unique satisfying .
For the approximation we have:
Thus, we observe that is equivalent with the following relation:
which is valid for only two values of .
For the approximation we have:
where and are defined as in (20) corresponding to the exact approximation. Without loss of generality, we can assume that there exist two constants such that . In other terms, we have:
We analyze two possible cases. Firstly, if , then the above equality leads to the following relation:
which implies that , that is a contradiction. Secondly, assuming we observe from optimality of that:
On the other hand, taking into account that we have:
From (29) and (30) we get , thus implying the same contradiction. ∎
Let be the sequence generated by the family of algorithms (RCD-IHT) under Assumptions 1, 3 and 13 and the additional assumption of strong convexity of with parameter . Denote with the number of changes in expectation of as . Let be some limit point of and be some confidence level. Considering the scalar case for all , the following statements hold:
From (12) and Theorem 12 it can be easily seen that:
i.e. we have proved the first part of our theorem.
In order to establish the linear rate of convergence in probability of algorithm (RCD-IHT), we first derive a bound on the number of iterations performed between two changes in expectation of . Secondly, we also derive a bound on the number of iterations performed after the support is fixed (a similar analysis for deterministic iterative hard thresholding method was given in ). Combining these two bounds, we obtain the linear convergence of our algorithm. Recall that for any , at iteration , there is a change in expectation of , i.e.
Assume that the number of iterations performed between two changes in expectation satisfies:
We show that under relation (33), the probability (31) does not hold. First, we observe that between two changes in expectation of , i.e. , the algorithm (RCD-IHT) is equivalent with the randomized version of coordinate descent method for strongly convex problems. Therefore, the method has linear rate of convergence (18), which in our case is given by the following expression:
for all . Taking , if we apply the complexity estimate (19) and use the bound (33), we obtain:
From the Markov inequality, it can be easily seen that we have:
Let such that . From Assumption 13 and definition of parameter we see that the event implies:
The first and the last terms from the above inequality further imply:
or equivalently . In conclusion, if (33) holds, then we have:
Applying the same procedure as before for iteration we obtain:
Considering the events and to be independent (according to the definition of ), we have:
Therefore, between two changes of support the number of iterations is bounded by:
where we used the inequality for any . Denoting with the number of iterations until the last change of support, we have:
Once the support is fixed (i.e. after iterations), in order to reach some -local minimum in probability with some confidence level , the algorithm (RCD-IHT) has to perform additionally another
iterations, where we used again (19) and Markov inequality. Taking into account that the iteration is the largest possible integer at which the support of sequence could change, we can bound:
Adding up this quantity and the upper bound on , we get that the algorithm (RCD-IHT) has to perform at most
iterations in order to attain an -suboptimal point with probability at least , which proves the second statement of our theorem. ∎
Random data experiments on sparse learning
In this section we analyze the practical performances of our family of algorithms (RCD-IHT) and compare them with that of algorithm (IHTA) . We perform several numerical tests on sparse learning problems with randomly generated data. All algorithms were implemented in Matlab code and the numerical simulations are performed on a PC with Intel Xeon E5410 CPU and 8 Gb RAM memory.
Sparse learning represents a collection of learning methods which seek a tradeoff between some goodness-of-fit measure and sparsity of the result, the latter property allowing better interpretability. One of the models widely used in machine learning and statistics is the linear model (least squares setting). Thus, in the first set of tests we consider sparse linear formulation:
where denotes a parameter vector. Then, for a set of independently drawn data samples , the joint likelihood can be written as a function of . To find the maximum likelihood estimate one should maximize the likelihood function, or equivalently minimize the negative log-likelihood (the logistic loss):
where now is strongly convex with parameter . For simulation, data were uniformly random generated and we fixed the parameters and . Once an instance of random data has been generated, we ran 10 times our algorithms (RCC-IHT-) and (RCD-IHT-) and algorithm (IHTA) starting from 10 different initial points. We reported in Table 3 the best results of each algorithm obtained over all 10 trials, in terms of best function value that has been attained with associated sparsity and number of iterations. In order to report relevant information, we have measured the performance of coordinate descent methods (RCD-IHT-) and (RCD-IHT-) in terms of full iterations obtained by dividing the number of all iterations by the dimension . The column denotes the final function value attained by the algorithms, represents the sparsity of the last generated point and iter (full-iter) represents the number of iterations (the number of full iterations). Note that our algorithms (RCD-IHT-) and (RCD-IHT-) have superior performance in comparison with algorithm (IHTA) on the reported instances. We observe that algorithm (RCD-IHT-) performs very few full iterations in order to attain best function value amongst all three algorithms. Moreover, the number of full iterations performed by algorithm (RCD-IHT-) scales up very well with the dimension of the problem.