Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Łojasiewicz Condition V: Under the Proximal-PL Inequality, Proximal-Gradient Converges at Rate (1 − μ/L)^kResearch Paper
Motivation
Many problems in machine learning and signal processing minimize a smooth loss plus a simple non-smooth regularizer: -regularized least squares, group-sparse regression, and constrained problems written with an indicator function. The standard algorithm for this composite form is the proximal-gradient method, a gradient step on the smooth part followed by a proximal step on the regularizer. Its classical linear convergence guarantee needs the smooth part to be strongly convex, a hypothesis that fails for -regularized least squares with fewer observations than variables and for most problems of interest without explicit regularization.
For smooth problems the Polyak–Łojasiewicz (PL) inequality gives a linear rate for gradient descent without convexity (Polyak 1963). Karimi, Nutini and Schmidt (arXiv:1608.04636, ECML PKDD 2016) propose a generalization to the composite setting, the proximal-PL inequality, under which the proximal-gradient method converges linearly with a one-paragraph proof. They show that it contains the PL case, the strongly convex case, and convex objectives with quadratic growth, and that it is equivalent to the Kurdyka–Łojasiewicz inequality (Bolte, Nguyen, Peypouquet, Suter 2017) and to the proximal error bound used earlier by Luo, Tseng and Drusvyatskiy–Lewis.
Timeline:
- 1963: Polyak proves linear convergence of gradient descent under the PL inequality.
- 1993: Luo and Tseng prove linear convergence of feasible descent methods under an error bound.
- 2015–2016: Bolte et al. and Li–Pong obtain linear rates of proximal gradient under the KL inequality; Drusvyatskiy and Lewis relate error bounds and quadratic growth (arXiv:1602.06661).
- 2016: Karimi, Nutini and Schmidt introduce the proximal-PL inequality and Theorem 5.
Setting
Let and consider problem (11),
The function is differentiable with an -Lipschitz gradient, with . The function is convex. has a minimizer , and .
For a base point and a parameter , the proximal model is
and the paper's measure of non-stationarity is
satisfies the proximal-PL inequality with modulus if
When is constant, and (12) is the PL inequality.
The proximal-gradient method with step size starts at an arbitrary and sets
The proximal residual function of Appendix E is .
Formalization targets
Goal: Theorem 5 (p. 8)
Under the hypotheses above,
Milestones (proof of Theorem 5)
- The quadratic upper bound (2), p. 2: an -Lipschitz gradient gives .
- Descent by , p. 9: if minimizes , then .
- The one-step contraction (15), p. 9: .
Companion results
- Lemma 2 (p. 17): for convex , is non-increasing on .
- Lemma 1 = Corollary 1 (p. 17): for .
- Appendix F, item 1 (p. 18): for constant , , and proximal-PL is PL.
- Appendix F, item 2 (p. 18): for -strongly convex , .
- Appendix F, item 4 (p. 19): for convex and with quadratic growth of modulus , .
- Two well-definedness statements: for convex and a positive parameter, the minima in (13) and (27) are attained.
Significance
Theorem 5 gives a global linear rate for the proximal-gradient method under a hypothesis that does not require strong convexity, or convexity of . Through the classes of Section 4.1 it covers -regularized least squares, the composite strongly convex case, and convex composite problems with quadratic growth. The quantity also drives the analysis of proximal coordinate descent (Theorem 6, mission VI of this series), and Lemma 1 is what lets the sufficient conditions of Appendix F, which hold with for , be used in (12).
The results are proved in the paper; none of them has a machine-checked proof that this mission knows of. The work here is to formalize the known proofs, including the properties of and that the paper uses without stating: the minimum in (13) exists, and the value is finite.
Difficulty
The proof of Theorem 5 is short; the difficulty sits in the objects. is defined through a minimum over of a function involving a non-smooth convex , so every step that replaces the minimum by its value at a particular point needs the minimum to be finite and attained. The paper never states these facts, and they fail for an arbitrary : for and the model is unbounded below and the minimum in (13) does not exist. The monotonicity of in its second argument (Lemma 1) rests on the convexity inequality of , and Appendix F item 4 relies on the projection onto the solution set, whose existence depends on that set being closed, convex and non-empty.
Formalization scope
- is
EuclideanSpace ℝ (Fin d), is Mathlib'sgradient f, and the Lipschitz hypothesis is written literally,‖gradient f x - gradient f y‖ ≤ L * ‖x - y‖, with0 < L. - The non-empty solution set is a point
xstarwithF xstar ≤ F yfor ally; isf xstar + g xstar, never a free real. - is real-valued and convex (
ConvexOn ℝ Set.univ g). Indicator functions of convex sets, which the paper lists among its examples, take the value and are outside this formalization; the formulas (13)–(15) subtract values of , which is undefined at . - The minimum in (13) and (27) is a real infimum. A real infimum of a set that is unbounded below has the junk value in Lean, so is used only with and convex , where the infimum is a minimum; this rules out the trivializing reading in which vanishes and (12) becomes .
- A proximal-gradient run is any sequence whose every step is a minimizer of the model; the minimizer is unique for convex , but uniqueness is not assumed.
- No relation between and is assumed, as in the paper.
- The PL predicate in Appendix F item 1 is the published
ShiOptRates.PolyakLojasiewicz(the paper's (3) multiplied by 2), and (2) is the publishedShiOptRates.QuadUpper.
Quadratic growth is stated with the distance to the solution set, , which equals for the projection the paper uses. Appendix F item 3 (Hoffman's bound with an unspecified constant ) and the equivalences of Appendix G are not included.
Contributions of every kind are welcome: proofs of the milestones, of the companion lemmas, and reusable facts about (finiteness, attainment, continuity in ) that other proximal-method missions can import.
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
- B. T. Polyak, Gradient methods for minimizing functionals, USSR Computational Mathematics and Mathematical Physics 3(4), 1963. https://doi.org/10.1016/0041-5553(63)90382-3
- Z.-Q. Luo, P. Tseng, Error bounds and convergence analysis of feasible descent methods: a general approach, Annals of Operations Research 46, 1993. https://doi.org/10.1007/BF02096261
- J. Bolte, T. P. Nguyen, J. Peypouquet, B. W. Suter, From error bounds to the complexity of first-order descent methods for convex functions, Mathematical Programming 165, 2017. https://doi.org/10.1007/s10107-016-1091-6
- D. Drusvyatskiy, A. S. Lewis, Error bounds, quadratic growth, and linear convergence of proximal methods, Mathematics of Operations Research 43(3), 2018. https://arxiv.org/abs/1602.06661
- G. Li, T. K. Pong, Calculus of the exponent of Kurdyka–Łojasiewicz inequality and its applications to linear convergence of first-order methods, Foundations of Computational Mathematics 18, 2018. https://arxiv.org/abs/1602.02915