From Differential Equation Solvers to Accelerated First-Order Methods for Convex Optimization 1: Gauss–Seidel Splitting of the HB and NAG Transforms Has ρ(E(α, G)) ≤ 1/√(1 + 2α) If 0 < α ≤ 2/√κ(A)Research Paper
Motivation
Accelerated first-order methods, the heavy-ball method of Polyak and Nesterov's accelerated gradient (NAG) method, reach an error on a -strongly convex, -smooth problem in gradient steps, , against for plain gradient descent. A long line of work has tried to explain where this square root comes from by reading the methods as discretizations of ordinary differential equations: Su, Boyd and Candès (arXiv:1503.01243), Wibisono, Wilson and Jordan (arXiv:1603.04245), and others.
Luo and Chen (arXiv:1909.03145, Math. Program. 2022) take the view from numerical analysis. Section 2 of their paper treats the quadratic case: the gradient flow is stiff, and an explicit scheme for it needs step size . They embed in a block matrix whose spectrum lies in the left half-plane with condition number , and they show that an explicit Gauss–Seidel splitting scheme for is stable with step size of order . This is the linear-algebra core of the paper's argument. The later sections build the nonlinear schemes (missions 2–8 of this series) on it.
Setting
Let be a real symmetric matrix. Its eigenvalues are real, and we assume
with condition number . For any real square matrix , the spectrum is the set of its eigenvalues, which are complex in general, and the spectral radius is .
The two transformations (39) of the paper are the block matrices
For the linear system , write , where is the block lower triangular part of (the two diagonal blocks and the lower-left block) and . The Gauss–Seidel splitting scheme (41) with step size is
The matrix is block lower triangular with diagonal blocks that are multiples of , so the scheme is explicit and never inverts . A one-step scheme is A-stable if .
Appendix A works with the scalar model , where and , and with its Gauss–Seidel matrix , defined in the same way from the lower triangular part of .
Formalization targets
Goal: Theorem 2.1 (p. 10)
For or , if , then the scheme is A-stable and
Milestones (Appendix A, pp. 35–36)
- (124): the closed form of , and .
- Lemma A.1: if , then .
- The heavy-ball reduction: a closed block form of , and with .
- Heavy-ball spectral radius: under (127), .
- NAG spectral radius: under (127), .
Companion statements
- Proposition 2.1 (p. 9): on and .
- The choice (p. 36): .
The goal is the weakest of these statements. Milestones 4 and 5 show that it is attained for heavy ball and strict for NAG.
Significance
The result. Theorem 2.1 gives a contraction factor per step for an explicit, matrix-free scheme. Applied to , that is the accelerated linear rate. Eliminating from shows that both transformations discretize the same second-order ODE . The theorem therefore explains acceleration as a stability property of an ODE solver, not as an algebraic trick. The paper's nonlinear schemes (predictor–corrector, gradient correction, Nesterov's method, the proximal variants) are built from this splitting. Their Lyapunov analyses generalize the spectral computation formalized here.
Formalizing it. The result is proved in the paper; to our knowledge no machine-checked proof exists. A formalization has to make the paper's "it is clear that" spectral reduction rigorous: the block diagonalisation through and the identification of the spectrum of a structured non-symmetric matrix with a union of spectra. It also has to treat spectra of real matrices over . Both are reusable well beyond this paper, since every spectral analysis of a momentum method (heavy ball, NAG, their discretizations) runs through the same reduction.
Difficulty
The obvious route bounds a norm of . It fails: is not normal, and the paper stresses that is not controlled by . The bound concerns eigenvalues only, and the eigenvalues are complex.
A second naive route applies a Gershgorin or field-of-values estimate to directly. It loses the square root, because the off-diagonal blocks have norm . The real content is that for each eigenvalue of , the Gauss–Seidel matrix has a negative discriminant exactly when the step-size condition holds. Its eigenvalues are then a conjugate pair on the circle of radius , and does not depend on . Reaching this needs three things: the reduction of the block spectrum to scalar spectra, the explicit inverse of , and a careful equivalence of the discriminant condition with (126).
Formalization scope
Matrices are Matrix (Fin d) (Fin d) ℝ, and the block matrices live on Fin d ⊕ Fin d, built with Matrix.fromBlocks.
- The splitting. It is blockwise: takes the two diagonal blocks and the lower-left block of . That is how the paper's own formula for reads it; the entrywise lower triangle would split itself.
- Spectra. The spectrum of a real matrix is
spectrum ℂof its complexification. "" isIsGreatestof the set of eigenvalue moduli. "" and "" are stated eigenvalue by eigenvalue, so no junk supremum over an empty set can make a bound vacuous. - The model matrix. and are encoded as hypotheses: both are eigenvalues of and bound every eigenvalue. is the number . Every statement assumes , the case of §2.3 where and are defined.
- Step size and inverse. The step size satisfies , which the paper leaves implicit in (124). The matrix inverse in is Mathlib's. It is genuine here, because is invertible for .
Two trivializations are ruled out explicitly. A spectral radius defined as sSup over a possibly empty spectrum would make every bound hold vacuously at . A spectrum taken over would miss the complex eigenvalues, which are all of them here.
A complete development needs:
- block-diagonalisation of the structured block matrices by and a permutation;
- the spectrum of a block-diagonal matrix of blocks;
- the eigenvalues of a real matrix with negative discriminant;
- explicit and block-triangular inverses.
Lemmas of this kind, stated for general symmetric , are welcome as separate theorems.
Selected references
- H. Luo, L. Chen, From differential equation solvers to accelerated first-order methods for convex optimization, Math. Program. 195 (2022); arXiv:1909.03145v4. https://arxiv.org/abs/1909.03145
- W. Su, S. Boyd, E. J. Candès, A differential equation for modeling Nesterov's accelerated gradient method: theory and insights, JMLR 17 (2016). https://arxiv.org/abs/1503.01243
- A. Wibisono, A. C. Wilson, M. I. Jordan, A variational perspective on accelerated methods in optimization, PNAS 113 (2016). https://arxiv.org/abs/1603.04245
- B. T. Polyak, Some methods of speeding up the convergence of iteration methods, USSR Comput. Math. Math. Phys. 4 (1964). https://doi.org/10.1016/0041-5553(64)90137-5
- E. Hairer, G. Wanner, Solving Ordinary Differential Equations II: Stiff and Differential-Algebraic Problems, Springer, 1996 (A-stability). https://doi.org/10.1007/978-3-642-05221-7