Random coordinate descent

Randomized (Block) Coordinate Descent Method is an optimization algorithm popularized by Nesterov (2010) and Richtárik and Takáč (2011). The first analysis of this method, when applied to the problem of minimizing a smooth convex function, was performed by Nesterov (2010).[1] In Nesterov's analysis the method needs to be applied to a quadratic perturbation of the original function with an unknown scaling factor. Richtárik and Takáč (2011) give iteration complexity bounds which do not require this, i.e., the method is applied to the objective function directly. Furthermore, they generalize the setting to the problem of minimizing a composite function, i.e., sum of a smooth convex and a (possibly nonsmooth) convex block-separable function:

${\displaystyle F(x)=f(x)+\Psi (x),}$

where ${\displaystyle \Psi (x)=\sum _{i=1}^{n}\Psi _{i}(x^{(i)}),}$ ${\displaystyle x\in R^{N}}$ is decomposed into ${\displaystyle n}$ blocks of variables/coordinates: ${\displaystyle x=(x^{(1)},\dots ,x^{(n)})}$ and ${\displaystyle \Psi _{1},\dots ,\Psi _{n}}$ are (simple) convex functions.

Example (block decomposition): If ${\displaystyle x=(x_{1},x_{2},\dots ,x_{5})\in R^{5}}$ and ${\displaystyle n=3}$, one may choose ${\displaystyle x^{(1)}=(x_{1},x_{3}),x^{(2)}=(x_{2},x_{5})}$ and ${\displaystyle x^{(3)}=x_{4}}$.

Example (block-separable regularizers):

1. ${\displaystyle n=N;\Psi (x)=\|x\|_{1}=\sum _{i=1}^{n}|x_{i}|}$
2. ${\displaystyle N=N_{1}+N_{2}+\dots +N_{n};\Psi (x)=\sum _{i=1}^{n}\|x^{(i)}\|_{2}}$, where ${\displaystyle x^{(i)}\in R^{N_{i}}}$ and ${\displaystyle \|\cdot \|_{2}}$ is the standard Euclidean norm.

Algorithm

Consider the optimization problem

${\displaystyle \min _{x\in R^{n}}f(x),}$

where ${\displaystyle f}$ is a convex and smooth function.

Smoothness: By smoothness we mean the following: we assume the gradient of ${\displaystyle f}$ is coordinate-wise Lipschitz continuous with constants ${\displaystyle L_{1},L_{2},\dots ,L_{n}}$. That is, we assume that

${\displaystyle |\nabla _{i}f(x+he_{i})-\nabla _{i}f(x)|\leq L_{i}|h|,}$

for all ${\displaystyle x\in R^{n}}$ and ${\displaystyle h\in R}$, where ${\displaystyle \nabla _{i}}$ denotes the partial derivative with respect to variable ${\displaystyle x^{(i)}}$.

Nesterov, and Richtarik and Takac showed that the following algorithm converges to the optimal point:

Algorithm Random Coordinate Descent Method
    Input: ${\displaystyle x_{0}\in R^{n}}$ //starting point
    Output: ${\displaystyle x}$

    set x := x_0

    for k := 1, ... do
        choose coordinate ${\displaystyle i\in \{1,2,\dots ,n\}}$, uniformly at random
        update ${\displaystyle x^{(i)}=x^{(i)}-{\frac {1}{L_{i}}}\nabla f_{i}(x)}$ 
    end for

• "←" denotes assignment. For instance, "largest ← item" means that the value of largest changes to the value of item.
• "return" terminates the algorithm and outputs the following value.

Convergence rate

Since the iterates of this algorithm are random vectors, a complexity result would give a bound on the number of iterations needed for the method to output an approximate solution with high probability. It was shown in [2] that if ${\displaystyle k\geq {\frac {2nR_{L}(x_{0})}{\epsilon }}\log \left({\frac {f(x_{0})-f^{*}}{\epsilon \rho }}\right)}$, where ${\displaystyle R_{L}(x)=\max _{y}\max _{x^{*}\in X^{*}}\{\|y-x^{*}\|_{L}:f(y)\leq f(x)\}}$, ${\displaystyle f^{*}}$ is an optimal solution (${\displaystyle f^{*}=\min _{x\in R^{n}}\{f(x)\}}$), ${\displaystyle \rho \in (0,1)}$ is a confidence level and ${\displaystyle \epsilon >0}$ is target accuracy, then ${\displaystyle Prob(f(x_{k})-f^{*}>\epsilon )\leq \rho }$.

Example on particular function

The following Figure shows how ${\displaystyle x_{k}}$ develops during iterations, in principle. The problem is

${\displaystyle f(x)={\tfrac {1}{2}}x^{T}\left({\begin{array}{cc}1&0.5\\0.5&1\end{array}}\right)x-\left({\begin{array}{cc}1.5&1.5\end{array}}\right)x,\quad x_{0}=\left({\begin{array}{cc}0&0\end{array}}\right)^{T}}$

Extension to block coordinate setting

Blocking coordinate directions into Block coordinate directions

One can naturally extend this algorithm not only just to coordinates, but to blocks of coordinates. Assume that we have space ${\displaystyle R^{5}}$. This space has 5 coordinate directions, concretely ${\displaystyle e_{1}=(1,0,0,0,0)^{T},e_{2}=(0,1,0,0,0)^{T},e_{3}=(0,0,1,0,0)^{T},e_{4}=(0,0,0,1,0)^{T},e_{5}=(0,0,0,0,1)^{T}}$ in which Random Coordinate Descent Method can move. However, one can group some coordinate directions into blocks and we can have instead of those 5 coordinate directions 3 block coordinate directions (see image).

See also

• Coordinate descent
• Gradient descent
• Mathematical optimization

References

1. Nesterov, Yurii (2010), "Efficiency of coordinate descent methods on huge-scale optimization problems", SIAM Journal on Optimization, 22 (2): 341–362, CiteSeerX 10.1.1.332.3336, doi:10.1137/100802001
2. Richtárik, Peter; Takáč, Martin (2011), "Iteration complexity of randomized block-coordinate descent methods for minimizing a composite function", Mathematical Programming, Series A, 144 (1–2): 1–38, arXiv:1107.2848, doi:10.1007/s10107-012-0614-z