Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Łojasiewicz Condition VI: Randomized Proximal Coordinate Descent Has Expected Rate (1 − μ/(dL))^kResearch Paper
Motivation
Many problems in machine learning and statistics have the composite form , where is a smooth loss and is a non-smooth regularizer such as . When is separable, , the method of choice at scale is often randomized proximal coordinate descent: at each step one coordinate is drawn at random and a one-dimensional proximal problem is solved in that coordinate only. Each step is cheap, and the method underlies standard solvers for -regularized regression and for the dual of support vector machines.
Classical linear-convergence guarantees for this method assume strong convexity of or of (Richtárik and Takáč, 2014), or rely on the Kurdyka–Łojasiewicz property with constants that are hard to make explicit. Karimi, Nutini and Schmidt (arXiv:1608.04636, ECML PKDD 2016) introduced the proximal-PL inequality, a generalization of the Polyak–Łojasiewicz condition to composite problems, and showed that it alone yields linear rates for proximal-gradient methods. Theorem 6 of that paper, the subject of this mission, extends the analysis to randomized proximal coordinate descent.
Timeline. Polyak (1963) and Łojasiewicz (1963) introduced the gradient-domination inequality for smooth problems. Nesterov (2012) analysed randomized coordinate descent for smooth convex problems; Richtárik and Takáč (2014) extended it to composite separable problems under strong convexity. Karimi, Nutini and Schmidt (2016) obtained the rate for the proximal coordinate method under proximal-PL alone (Theorem 6, p. 10, proof in Appendix H, p. 23).
Setting
Write for the Euclidean space with inner product and norm ; is the -th coordinate of , the -th basis vector, and the -th coordinate of the gradient. The data are:
- a differentiable whose gradient is -Lipschitz, , with ;
- convex univariate functions and the separable ;
- , which has a minimizer ; write .
For the quantity
measures the best decrease of a quadratic upper model of at . satisfies the proximal-PL inequality with constant if
The coordinate-wise proximal-gradient method starts at . At step it draws uniformly from , independently of the past, and sets , where minimizes
Formalization targets
Goal: Theorem 6 (p. 10)
For every ,
The constant is the paper's; no further constant is fixed.
Milestones (Appendix H, p. 23)
- (62), the coordinate upper model: .
- The expected-improvement identity: .
- (63), the expected descent: .
A companion statement (not a milestone, not in the paper) asserts that the one-dimensional step (16) exists for convex and .
Significance
The result. Theorem 6 gives a linear rate for proximal coordinate descent on every composite problem satisfying proximal-PL, a class that contains -regularized least squares and other problems that are not strongly convex. The rate matches that of plain randomized coordinate descent under PL (Theorem 3 of the paper) with the composite term added, and the factor is the usual price of updating one coordinate per step.
Formalizing it. The result is proved on paper; no machine-checked proof is known. A formal proof requires the coordinate descent lemma, the exchange of a separable minimum with a sum, and an exact accounting of the expectation over random index sequences. The definitions of , proximal-PL and the random-coordinate run built here are reusable for other proximal and coordinate methods.
Difficulty
The one-step argument is short once the expected-improvement identity is available; the difficulty is in that identity. Its left side averages over coordinates the minima of one-dimensional models, while is a minimum over the whole space. Equating them requires that the full-space minimum split across coordinates, which is where separability of is used, and that each minimum be attained, which requires convexity of the and . A second point is the expectation: the iterate after steps is a function of the whole index word , and the one-step expected descent must be propagated through the tower of conditional expectations to the -step bound.
Formalization scope
- The space is
EuclideanSpace ℝ (Fin d)with (0 < d); the gradient is Mathlib'sgradient f, and isgradient f x i. - The Lipschitz gradient is the literal hypothesis with and
Differentiable ℝ f, as in the setup of Theorem 5. Appendix H uses only coordinate-wise Lipschitz continuity, which follows. - is a point minimizing ; is , never a free real.
- The are real-valued convex functions. Extended-valued (box indicators, used for the SVM dual in §4.4) are outside this formalization.
- The minimum in is a real infimum; it is attained, hence equal to the paper's minimum, for convex and , and every statement using it carries these hypotheses.
- The update is any step selection with a minimizer of (16); the selection is a hypothesis-constrained parameter, never an arbitrary map. Equation (16) as printed assigns the minimizing step to ; the iterate of Appendix H is what is encoded.
- is the exact average over all index words, each of probability , which is the law of i.i.d. uniform coordinates.
- A formalization that leaves the update unconstrained, takes as a free constant, or computes as a junk infimum would make the goal false or vacuous; these are ruled out by the conventions above.
Contributions welcome: proofs of the three milestones and the goal, the existence of the coordinate step, and a version with extended-valued .
Selected references
- H. Karimi, J. Nutini, M. Schmidt, Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Łojasiewicz Condition, ECML PKDD 2016; arXiv:1608.04636v4, 2020. https://arxiv.org/abs/1608.04636
- P. Richtárik, M. Takáč, Iteration complexity of randomized block-coordinate descent methods for minimizing a composite function, Mathematical Programming 144, 2014. https://doi.org/10.1007/s10107-012-0614-z
- Yu. Nesterov, Efficiency of coordinate descent methods on huge-scale optimization problems, SIAM J. Optim. 22(2), 2012. https://doi.org/10.1137/100802001
- B. T. Polyak, Gradient methods for minimizing functionals, USSR Comput. Math. Math. Phys. 3(4), 1963. https://doi.org/10.1016/0041-5553(63)90382-3