On Geometric Connections of Embedded and Quotient Geometries in Riemannian Fixed-rank Matrix Optimization: Gradient and Hessian Connections for Fixed-rank PSD Matrices via YYᵀResearch Paper
Motivation
Many estimation problems in statistics, signal processing and machine learning ask for a low-rank positive semidefinite matrix: phase retrieval, matrix sensing, PCA-type problems and the Burer–Monteiro approach to semidefinite programs. A standard formulation is
and Riemannian optimization solves it by treating the feasible set as a smooth manifold. Two geometries are in common use. In the embedded geometry the set of rank- PSD matrices is a submanifold of with the Euclidean metric. In the quotient geometry one writes with a full-column-rank factor and optimizes over equivalence classes . Algorithms built on the two geometries behave similarly in experiments, and landscape results (absence of spurious local minima, strict saddles) have been proved separately for each. Luo, Li and Zhang (arXiv:2110.12121, Mathematics of Operations Research 2024) explain this by an explicit dictionary between the Riemannian gradients and Hessians of the two geometries. This mission formalizes their first main result, Theorem 3, for the full-rank factorization of PSD matrices.
Setting
Matrices carry the Frobenius inner product . The objective is twice continuously differentiable and symmetric, (the paper's standing assumptions); and are its Euclidean gradient and Hessian.
Embedded geometry. , viewed as . Its tangent space is the span of the tangent cone. The Riemannian gradient is the orthogonal projection of onto , and the Riemannian Hessian quadratic form is the derivative at along any curve in with , .
Quotient geometry. The total space is , the matrices of full column rank, with the objective (18a). A weight , smooth and symmetric positive definite, defines the metric . The vertical space is tangent to the orbit , and the horizontal space is its -orthogonal complement. Tangent vectors of the quotient are represented by horizontal vectors: is the horizontal part of the total-space gradient , and for horizontal , where is the Levi-Civita connection of (Koszul formula).
Comparison data. ; spans the top eigenspace of , , . For a quadratic form on a -dimensional space with inner product , is its -th largest eigenvalue relative to (). are the extreme singular values of an matrix.
Formalization targets
Goal: Theorem 3 (p. 21)
If in addition for all orthogonal , then for every and ,
and if is a Riemannian first-order stationary point of (18a),
both Hessians have eigenvalues, and lies between and .
Milestones
- Lemma 6: iff ; iff , ; .
- Lemma 3: is a smooth embedded submanifold of dimension , with tangent space in the block form (14).
- Lemma 4 (i): explicit , and their dimensions.
- Lemma 1 on : the total-space gradient is already horizontal.
- Proposition 1, embedded bullet: the formulas (19) for and .
- Proposition 1, quotient bullet: the formulas (20) for the lifted gradient and Hessian.
- Proposition 2: is a bijection .
- Proposition 2, (23): .
- Theorem 2: the abstract sandwich inequality for spectra of two quadratic forms related by a bijection with bounded metric distortion.
Significance
Theorem 3 says that at stationary points the two geometries see the same second-order landscape: the Hessians are congruent through , so their eigenvalues have the same signs and comparable magnitudes, with explicit factors depending on and the spectrum of . The paper derives from it (Corollary 1) that first-order stationary points, second-order stationary points and strict saddles correspond between the embedded and quotient formulations, so a landscape result proved in one geometry transfers to the other. The choice of weight (flat , right-invariant , or ) changes the comparison constants, which is how the paper relates the conditioning of the two approaches.
The result is proved in the paper; it is not formalized anywhere. A formal development adds machine-checked Riemannian gradients and Hessians on a concrete embedded submanifold and on a concrete quotient, a checked horizontal-space computation, and a reusable abstract sandwich theorem for max–min eigenvalues. Corollary 1 and the parallel results for the polar factorization and for general rank- matrices (Theorems 4–7) are natural follow-up missions on the same definitions.
Difficulty
The identities (19), (20) and (34) are matrix calculus once the geometric objects are pinned down; the difficulty is that the definitions are intrinsic. The embedded Hessian requires differentiating the tangent projection along the manifold, which produces the curvature term ; the obvious computation that ignores the moving tangent space gives the wrong Hessian. The quotient Hessian involves the derivative of the weight and needs the horizontal-space description of Lemma 4 to see that the projection in (5) can be dropped. Identifying of the tangent cone with the block form (14) needs an explicit chart of the rank- PSD set. The final sandwich requires a max–min (Courant–Fischer) argument relative to two different inner products, and the order of the bounds flips for negative eigenvalues.
Formalization scope
- is
EuclideanSpace ℝ (Fin a × Fin b)(Frobenius inner product) with matrix viewstoMat/ofMat. The weight takes values in in the same form so that is defined. - The embedded tangent space and Riemannian gradient are the published platform definitions
ProjLikeRetr.Retractor.tangentSpace(span of the tangent cone) andZORiem.RGD.rgrad(projected Euclidean gradient); "smooth embedded submanifold of dimension " is the published coordinate-slice predicateProjLikeRetr.Retractor.IsSubmanifold k d, for every . - The embedded Hessian is a curve-defined predicate on a quadratic form; the quotient Hessian is the Koszul-formula form . Neither is defined by the paper's formulas (19), (20) or (34): those are theorems, and a formalization that defined the Hessians or gradients by them would make the mission trivial.
- is the max–min value over -dimensional subspaces (the paper's Lemma 13); "sandwiched between and " means ; of the symmetric positive definite are its extreme Rayleigh quotients.
- Conventions: ; is and symmetric; is positive definite on and is smooth there (the meaning of "Riemannian metric"); " spans the top eigenspace of " is , ; ; is defined as rather than by the block formula (22).
- In Theorem 2 the map is assumed linear, as in every application in the paper, and the printed domain in (12)–(13) is read as .
- Not encoded: the half of Lemma 3 about general rank- matrices, Lemma 4 (ii) (no quotient-manifold structure in Mathlib), the polar factorization geometry and Theorems 4–7, Corollaries 1–2.
Useful contributions include Mathlib-level lemmas on the tangent cone of the rank- PSD set, Courant–Fischer for a form relative to a non-standard inner product, and matrix-calculus lemmas for .
Selected references
- Y. Luo, X. Li, A. R. Zhang, On Geometric Connections of Embedded and Quotient Geometries in Riemannian Fixed-rank Matrix Optimization, Mathematics of Operations Research, 2024; arXiv:2110.12121v2. https://arxiv.org/abs/2110.12121
- P.-A. Absil, R. Mahony, R. Sepulchre, Optimization Algorithms on Matrix Manifolds, Princeton University Press, 2008. https://doi.org/10.1515/9781400830244
- M. Journée, F. Bach, P.-A. Absil, R. Sepulchre, Low-rank optimization on the cone of positive semidefinite matrices, SIAM Journal on Optimization 20(5), 2010. https://doi.org/10.1137/080731359
- B. Vandereycken, S. Vandewalle, A Riemannian optimization approach for computing low-rank solutions of Lyapunov equations, SIAM J. Matrix Anal. Appl. 31(5), 2010. https://doi.org/10.1137/090764566
- R. Bhatia, Matrix Analysis, Springer GTM 169, 1997. https://doi.org/10.1007/978-1-4612-0653-8