Optimizing Static Linear Feedback: Gradient Method III: Gradient Descent with the Hessian Step Size Converges Linearly on Strongly Convex FunctionsResearch Paper
Motivation
Gradient descent needs a step size, and the classical choices each ask for something the user may not have. A constant step needs the Lipschitz constant of the gradient, which is rarely known and often pessimistic. Backtracking needs repeated function evaluations. The exact line search needs a one-dimensional minimization at every iteration. Fatkhullin and Polyak (arXiv:2004.09875v2, SIAM J. Control Optim. 2021, doi:10.1137/20M1329858) proposed a step size for the static linear-quadratic regulator (their rule (4.8), §4.3, p. 11). In §6.1 they point out that the same rule applies to any smooth unconstrained problem . The rule divides the squared gradient norm by the Hessian quadratic form along the gradient. It needs one Hessian–vector product per iteration and neither nor the strong convexity constant . In the paper's LQR experiment (§5, Figure 8, p. 13) the algorithm built on this step converges much faster than gradient descent with a constant step tuned at the first iterations.
The paper proves the method converges linearly for strongly convex functions (Theorem 6.1, p. 13). The proof takes one page (Appendix D.4, p. 19). This mission formalizes that theorem. It is the third mission of a series on this paper; the other two concern the LQR gradient method and gradient flow, and this one uses none of their control-theoretic objects.
Setting
Let be twice differentiable, with gradient and Hessian . Three constants describe it.
- is -strongly convex, : for all and with .
- is Lipschitz with constant : .
- is Lipschitz with constant : in the operator norm.
Let be the global minimizer of . The Hessian step size at a point is
the minimizer of the second-order Taylor model of along . The method (6.1) runs
from a starting point . The damped method with factor runs . For a quadratic the method (6.1) is steepest descent with exact line search.
Formalization targets
Goal: Theorem 6.1 (p. 13), both parts
Under the hypotheses above:
- If and (condition (6.2)), the iterates of (6.1) satisfy
- If , the damped iterates from any satisfy
The constants are the paper's, stated exactly.
Milestones (Appendix D.4, p. 19)
- Cubic Taylor bound (first display): .
- One-step inequality (third display): with , .
- (D.1): .
- (D.2): one damped step gives .
Significance
Theorem 6.1 gives a rate for a step size computed from local second-order information alone. Part 1 says that near the minimizer the method converges at least as fast as gradient descent with step , with no step-size parameter to tune. Part 2 gives convergence from every starting point, at the price of knowing a lower bound on for the damping. The same step appears in the paper's LQR method (rule (4.8)) and in gradient projection methods (p. 13, citing [37]), so the one-step inequalities are reusable beyond this theorem.
The result is proved in the paper. As of this writing none of it has a machine-checked proof. The formal work is the full development: the cubic Taylor bound from a Lipschitz second derivative on , the two one-step inequalities, and the inductions that give the rates.
Difficulty
The obvious argument for gradient descent uses the quadratic upper bound and a step no larger than . The Hessian step can be as large as , far outside that range, so the quadratic bound in the Euclidean norm gives no decrease. Two different replacements are needed. For the undamped method, the cubic Taylor error must be controlled along the whole trajectory, and condition (6.2) is imposed only on : the proof must show the gradient stays small enough at every later iterate. For the damped method, the upper bound must be measured in the local Hessian norm (D.1), which trades the step's size for the condition number .
On the Lean side, Mathlib has Taylor's theorem in one variable. The cubic bound for a function on with a Lipschitz Fréchet second derivative must be assembled from it or from the integral form along a segment. Mathlib has no ready-made link between strong convexity and a lower bound on the Hessian either.
Formalization scope
The space is EuclideanSpace ℝ (Fin n). The gradient is Mathlib's gradient f. The Hessian quadratic form is fderiv ℝ (fderiv ℝ f) x v v. Twice differentiability is differentiability of f and of fderiv ℝ f everywhere. The Lipschitz constant of the Hessian is in the operator norm of the bilinear map, not the Frobenius norm. Strong convexity is StrongConvexOn Set.univ μ f with 0 < μ; Mathlib's modulus is . and are real constants in the Lipschitz inequalities. The minimizer is a hypothesis ( for all ), not constructed.
Deviations from the page, all recorded in the items' Formalization Notes:
- The damping positivity is added. It is implicit on the page.
- The damped claim is stated under the full hypotheses of Theorem 6.1, including the Lipschitz Hessian, although its proof does not use .
- The one-step inequality for (6.1) is stated under the strong convexity of Theorem 6.1, which keeps . The gradient's Lipschitz constant is not assumed there.
At a stationary point the step is ; Lean evaluates it to , so the method stays at the minimizer, and both rates remain true.
The iterates are those of the defined recursions (6.1) and its damped version. A statement about an arbitrary sequence satisfying a descent inequality would be a different, weaker theorem and does not discharge the goal. Condition (6.2) is imposed on only; a version assuming it at every iterate is also not the goal.
Contributions welcome: the multivariate cubic Taylor bound (reusable wherever a Lipschitz Hessian appears, e.g. in cubic regularization of Newton's method); the Hessian bounds from strong convexity and a Lipschitz gradient; and the inequality .
Selected references
- I. Fatkhullin, B. Polyak, Optimizing Static Linear Feedback: Gradient Method, arXiv:2004.09875v2, 2020; SIAM J. Control Optim. 59(5), 2021. https://arxiv.org/abs/2004.09875 · https://doi.org/10.1137/20M1329858
- Yu. Nesterov, B. T. Polyak, Cubic regularization of Newton method and its global performance, Math. Program. 108, 2006 (the cubic Taylor bound for Lipschitz Hessians). https://doi.org/10.1007/s10107-006-0706-8
- B. T. Polyak, Introduction to Optimization, Optimization Software, 1987 (gradient methods, strong convexity).