Motivation
Iterative methods for optimization and equation solving on a smooth constraint set M (orthogonal matrices, fixed-rank matrices, spheres, Stiefel and Grassmann manifolds) compute an update vector u in the tangent space at the current iterate x and then have to return to M. The Riemannian exponential map does this along geodesics, but computing it means solving an ordinary differential equation. The notion of retraction, introduced by Adler, Dedieu, Margulies, Martens and Shub (IMA J. Numer. Anal., 2002) and developed in the book of Absil, Mahony and Sepulchre (Princeton, 2008), captures what such a return map needs for Newton's method to keep its local quadratic convergence and for gradient methods to converge: smoothness, R(x,0)=x, and first-order agreement with the exponential.
Absil and Malick (SIAM J. Optim., 2012; HAL hal-00651608v2) give a general recipe for building retractions on submanifolds of a Euclidean space: move tangentially from x to x+u, then come back to M along a prescribed family of admissible directions. This mission formalizes that recipe, Theorem 4.2 of the paper ("retractors give retractions"), together with the two lemmas its proof rests on.
Setting
Let E be a Euclidean space of dimension n (in the paper's examples, Rn×m with the Frobenius inner product). A set M⊆E is a Ck submanifold of dimension d if around every xˉ∈M it is a coordinate slice: there are an open neighbourhood U of xˉ and a Ck diffeomorphism ϕ of U onto an open subset of Rn with M∩U={x∈U:ϕd+1(x)=⋯=ϕn(x)=0}. Throughout, k≥2.
The tangent space TM(x) is the linear subspace of E of tangent directions of M at x, the normal space NM(x) is its orthogonal complement, and the tangent bundle is TM={(x,u):x∈M, u∈TM(x)}.
A map R from TM to M is a retraction around xˉ (Definition 2.1) if, on a neighbourhood U of (xˉ,0) in TM, it is of class Ck−1, satisfies R(x,0)=x, and u↦R(x,u) has derivative idTM(x) at u=0. It is a retraction on M if this holds around every point.
A retractor (Definition 4.1) is a Ck−1 map D, defined on a neighbourhood of the zero section of TM, with values in the Grassmann manifold Gr(n−d,E) of (n−d)-dimensional linear subspaces, such that D(x,0)∩TM(x)={0} for every x∈M. Given D, set D(x,u)=x+u+D(x,u) and let
R(x,u)={points of M∩D(x,u) nearest to x+u}.
Formalization targets
Goal: Theorem 4.2 (retractors give retractions)
∀xˉ∈M ∃r:R(x,u)={r(x,u)} for (x,u)∈TM near (xˉ,0),and r is a retraction around xˉ.
The theorem asserts both that the point-to-set map R is single-valued near the zero section and that it is a retraction there.
Milestone: Lemma 4.7 (the normal case D(x,u)=NM(x))
Near (xˉ,0) there is one and only one smallest v(x,u)∈NM(x) with x+u+v(x,u)∈M; Duv(x,0)=0; and R(x,u)=x+u+v(x,u) is a retraction around xˉ, hence on M.
Milestone: Lemma 4.8 (straightening up)
On a neighbourhood of the zero section, D(x,u)={v+A(x,u)v: v∈NM(x)} for a unique linear A(x,u):NM(x)→TM(x) depending Ck−1 on (x,u).
Further: Theorem 4.9 (second order)
If k≥3 and D(x,0)=NM(x) for all x∈M, then dt2d2R(x,tu)∣t=0∈NM(x) for all (x,u)∈TM.
Significance
Theorem 4.2 reduces the construction of a retraction to the choice of a smooth field of subspaces transverse to the tangent space at u=0. The orthographic retraction (D=NM(x)) and the projective retraction R(x,u)=PM(x+u) (D=NM(PM(x+u))) are both instances, as are the gnomonic, orthographic and stereographic projections on the sphere. Theorem 4.9 then certifies, by a check at u=0 only, that a retraction agrees with the exponential to second order, which matters for the superlinear convergence of Riemannian trust-region and Newton methods. Lemma 4.7 goes beyond an earlier result on tangential parameterizations (reference [28, Th. 3.4] of the paper) by giving Ck−1 regularity jointly in (x,u), not only in u (Remark 4.4 of the paper).
The results are proved in the paper. None of them is machine-checked: Mathlib has the implicit function theorem and smooth manifolds, but no embedded submanifolds of a Euclidean space with their tangent and normal bundles, no retractions, and no smooth Grassmannian-valued maps. The work here is to formalize the known proof and to build this layer, which every later mission of this series (projective, spectral, fixed-rank and Stiefel retractions) also needs.
Difficulty
The statement is an implicit-function argument, but the obvious one does not apply directly. The unknown v lives in the normal space NM(x), which moves with x, and (x,u) ranges over the tangent bundle, a Ck−1 submanifold of E×E rather than an open set of a vector space. The equation has to be read in charts of the bundle, which costs one derivative; the uniqueness given by the implicit function theorem holds only locally in (x,u,v), and turning it into "the smallest v", or "the nearest point of M∩D(x,u)", requires excluding far-away intersection points. For a general retractor, the subspace D(x,u) also moves, and has to be written as a graph over the normal space with a Ck−1 dependence before a second implicit-function argument applies.
Formalization scope
E is a finite-dimensional real inner product space E, n is Module.finrank ℝ E, and k,d are natural numbers with 2≤k and d≤n (the latter inside the submanifold definition). The coordinate slice uses an open partial homeomorphism E → EuclideanSpace ℝ (Fin n), Ck in both directions, with 0-based coordinates. TM(x) is the span of Mathlib's tangent cone (chart-free), NM(x) its orthogonal complement. Retractions and retractors are total maps on E×E constrained only on O∩TM with O open; "of class Ck−1 on a subset of TM" is ContDiffOn on that set. A Grassmannian-valued map is Ck−1 when its orthogonal projector PD(x,u) is, and the dimension n−d of D(x,u) is part of the definition. The set-valued R uses the published nearest-point predicate RandomGradFree.Nonsmooth.IsMetricProjection. In Lemma 4.8, A(x,u) is extended by 0 on TM(x) so that it is an operator on E. Theorem 4.9 assumes k≥3, the standing assumption of the definition of second-order retractions (display (2.3)), and reads the page's "D(xˉ,0)=NM(x)" as D(x,0)=NM(x) for all x∈M.
The goal is not satisfied by exhibiting some retraction: it requires the nearest-point set R(x,u) to equal a singleton {r(x,u)} (so it is nonempty) and that very r to be a retraction. Theorem 4.9 requires the curve t↦R(x,tu) to be twice differentiable, so a junk second derivative of 0 does not satisfy it.
A complete development needs: finite-rank facts for tangent spaces of slices (dimTM(x)=d), smoothness of x↦PNM(x), charts of TM and of the Whitney sum TM⊕NM, ContDiffOn on submanifolds, and local orthonormal frames of a smooth field of subspaces. All of these are reusable beyond this mission. Contributions to any of them, and proofs of the two lemmas, are welcome.
Selected references
- P.-A. Absil, 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-00651608)
- P.-A. Absil, R. Mahony, R. Sepulchre, Optimization Algorithms on Matrix Manifolds, Princeton University Press, 2008. https://press.princeton.edu/absil
- R. L. Adler, J.-P. Dedieu, J. Y. Margulies, M. Martens, M. Shub, Newton's method on Riemannian manifolds and a geometric model for the human spine, IMA J. Numer. Anal. 22(3):359–390, 2002. https://doi.org/10.1093/imanum/22.3.359