Projection-like Retractions on Matrix Manifolds V: Within Distance σ_m(X̄) of the Stiefel Manifold, the Polar Factor Is the Unique Nearest Orthonormal FrameResearch Paper
Motivation
Optimization problems with orthonormality constraints arise throughout numerical linear algebra and its applications: computing a few eigenvectors or singular vectors, Procrustes problems in statistics and shape analysis, orthogonal factor rotation, independent component analysis, and the orthogonality constraints of electronic-structure calculations. The feasible set of such a problem is the Stiefel manifold of orthonormal -frames in . Riemannian optimization algorithms on this manifold (Riemannian gradient, Newton and trust-region methods; see Absil, Mahony and Sepulchre, Optimization Algorithms on Matrix Manifolds, 2008) take a step in a tangent direction and then need a retraction: a map that brings the updated point back onto the manifold while agreeing with the geometry to first order.
The most natural way to come back to a constraint set is to take the nearest point. P.-A. Absil and J. Malick, Projection-like retractions on matrix manifolds (SIAM J. Optim. 22(1), 2012; the mission follows the authors' version, HAL hal-00651608v2), show in their Proposition 3.2 that the metric projection onto a smooth submanifold yields a retraction, and then work out the projection explicitly for several matrix manifolds. For the Stiefel manifold, their Proposition 3.4 identifies the projection with a factor of the singular value decomposition, equivalently with the orthonormal factor of the polar decomposition. The paper notes that this result was mentioned without proof in Higham's survey of matrix nearness problems, and gives a proof along the lines of Horn and Johnson, Matrix Analysis, §7.4.
Section 4.5 of the same paper treats a second, "orthographic" retraction on the Stiefel manifold, which corrects a tangent step by a normal vector instead of projecting; for the orthogonal group (the case ) it admits a closed form through a matrix square root (Proposition 4.12).
Setting
Fix natural numbers . Matrices are real, and carries the Frobenius norm
The Stiefel manifold is
the set of matrices with orthonormal columns; is the orthogonal group .
The singular values of are , the square roots of the eigenvalues of . A singular value decomposition of is a factorization with , , and zero off its diagonal, with nonnegative nonincreasing diagonal entries. For every singular value equals ; in particular .
A projection of onto is a point with for all . A polar decomposition of is a factorization with and symmetric positive definite.
Formalization targets
Goal: Proposition 3.4 (p. 10)
Let and let satisfy . For every singular value decomposition ,
and is the of the polar decomposition : it is the orthonormal factor of every polar decomposition of , and a polar decomposition with this factor exists.
The statement asserts existence, uniqueness and the closed form of the projection on the whole open ball of radius , for whichever singular value decomposition is supplied.
Steps of the proof (milestones)
- For : .
- For every , , with equality at .
- If with , then has full rank .
- The polar factor of a full-rank matrix is unique (Horn and Johnson, Theorem 7.3.2).
Further items (§4.5)
for , skew-symmetric and symmetric with ; and Proposition 4.12,
stated as: is the unique symmetric correction of smallest Frobenius norm.
Significance
Proposition 3.4 makes the projective retraction on the Stiefel manifold computable from one singular value decomposition of an matrix, and identifies it with the polar factor, which is also what many numerical codes already compute for re-orthonormalization. Combined with Proposition 3.2 of the paper, it yields a second-order-accurate retraction usable in any Riemannian algorithm on . The same statement is the orthogonal Procrustes problem in the special case of a full-rank target: the nearest orthonormal frame to .
The result is classical and proved; the paper's argument is short but cites two external facts (Weyl's perturbation bound for singular values and the uniqueness of the polar decomposition). To our knowledge no machine-checked proof of the nearest-orthonormal-frame property, of the uniqueness of the polar factor, or of the closed form of Proposition 4.12 exists in Mathlib or on this platform. A formalization produces reusable pieces: the trace inequality over the Stiefel manifold, the uniqueness of the polar decomposition, and the full-rank property near , all of which recur in matrix analysis and in the analysis of Riemannian algorithms.
Difficulty
Existence of a nearest point is easy (the Stiefel manifold is compact), and attainment of the trace bound is a direct computation. The content is in two places. First, the bound for every requires transporting by the orthogonal factors of the singular value decomposition and bounding diagonal entries of a matrix with orthonormal columns. Second, uniqueness does not follow from the trace argument: when is rank deficient there are many maximizers. Uniqueness needs both a perturbation bound (singular values are 1-Lipschitz in the Frobenius norm, so near ) and the uniqueness of the polar factor, which in turn rests on the uniqueness of the positive-semidefinite square root of . Mathlib has singular values of linear maps and the spectral theorem but, at the pinned revision, neither Weyl's inequality for singular values nor the polar decomposition.
For Proposition 4.12, the printed proof compares only two solutions of (4.14), while (4.14) has other symmetric solutions (mixed signs of the square roots, or non-diagonal ones on repeated eigenvalues); minimality must be proved against all of them.
Formalization scope
Matrices are Matrix (Fin n) (Fin m) ℝ with the Frobenius norm brought in by open scoped Matrix.Norms.Frobenius. The Stiefel manifold is the set stiefel n m of matrices with Xᵀ * X = 1. Singular values are Mathlib's LinearMap.singularValues of Matrix.toEuclideanLin X, re-indexed to be 1-based as on the page (sv X m is ). A singular value decomposition is the predicate IsSVD X U S V of display (3.5), and is U * E * Vᵀ with E the rectangular identity (frameOfSVD U V). The projection is the set of nearest points, using the published predicate RandomGradFree.Nonsmooth.IsMetricProjection; "exists and is unique" is equality of that set with a singleton. Positive definiteness is Mathlib's Matrix.PosDef, which over includes symmetry.
Standing assumptions and added hypotheses: is the page's assumption of §3.3; is added so that is meaningful. The radius is with strict inequality, kept in that form although its value is . Two misprints of the page are corrected in the Lean and kept in the verbatim milestone texts: the display of Proposition 3.4 reads for , and the distance identity reads for . The uniqueness of the polar factor is stated for positive semidefinite factors under the rank hypothesis, which contains the positive definite case of Proposition 3.4. The matrix square root of §4.5 is defined through Mathlib's spectral theorem; Proposition 4.12 is stated for every admissible symmetric correction, and is vacuous only when no symmetric correction exists.
A trivializing formalization is ruled out: the projection is not taken as a hypothesis or chosen by definition; the goal asserts that the nearest-point set equals the singleton of the explicit matrix, for every singular value decomposition supplied, so neither existence nor uniqueness can be assumed away.
Contributions welcome beyond the stated items: Weyl's inequality in Mathlib's singular-value API, existence of a singular value decomposition in the matrix form (3.5), and the polar decomposition with its uniqueness, all reusable well beyond this mission.
Selected references
- P.-A. Absil and J. Malick, Projection-like retractions on matrix manifolds, SIAM J. Optim. 22(1):135–158, 2012. https://doi.org/10.1137/100802529 ; authors' version https://hal.science/hal-00651608v2
- P.-A. Absil, R. Mahony and R. Sepulchre, Optimization Algorithms on Matrix Manifolds, Princeton University Press, 2008. https://doi.org/10.1515/9781400830244
- R. A. Horn and C. R. Johnson, Matrix Analysis, Cambridge University Press, 1985 (cited by the paper in its 1989 printing) (Theorem 7.3.2, §7.4). https://doi.org/10.1017/CBO9780511810817
- N. J. Higham, Matrix nearness problems and applications, in Applications of Matrix Theory (M. J. C. Gover and S. Barnett, eds.), Oxford University Press, 1989, pp. 1–27 (cited by the paper as [15, §4]; no DOI).
- A. Edelman, T. A. Arias and S. T. Smith, The geometry of algorithms with orthogonality constraints, SIAM J. Matrix Anal. Appl. 20(2):303–353, 1998. https://doi.org/10.1137/S0895479895290954