Dissolving Constraints for Riemannian Optimization: Near the Feasible Manifold, the Constraint Dissolving Function and the Constrained Problem Have the Same Local MinimizersResearch Paper
Motivation
Many problems in statistics, signal processing and machine learning minimize a smooth function over a set defined by smooth equations, subject to : orthogonality constraints (the Stiefel manifold), unit-norm constraints (spheres and oblique manifolds), fixed-rank and symplectic constraints. Riemannian optimization treats the feasible set as a manifold and moves along it with retractions and vector transports, which have to be designed and implemented for each manifold (Absil, Mahony & Sepulchre 2008). Exact penalty methods avoid that geometry, but the classical smooth exact penalty of Fletcher (1970) needs second derivatives of just to evaluate its gradient.
Xiao, Liu & Toh (arXiv:2203.10319v3) propose the constraint dissolving function (CDF), an unconstrained smooth function whose gradient needs only . They prove that, near the feasible set and above an explicit penalty threshold, the CDF and the constrained problem have the same stationary points and the same local minimizers. Any unconstrained solver can then be applied to the CDF directly. The paper later appeared in Mathematics of Operations Research (2024); this mission follows the arXiv v3 numbering throughout.
Setting
Let and , and let be the feasible region of (OCP). The transposed Jacobian has the gradients as columns. Assumption 1.1 asks that and be locally Lipschitz and that LICQ hold on : has full column rank for .
A constraint dissolving operator is a map with locally Lipschitz Jacobian such that and for every (Assumption 1.2). For the unit sphere one can take . The CDF with penalty parameter is
Around a feasible point the paper introduces local constants (§2.3, p.7). These are:
- ;
- a radius on whose closed ball one has ;
- the suprema , , of , , over ;
- the Lipschitz constants , , , of , , and on .
From these it builds a radius , two closed balls around , and the threshold
The Lean development uses the same names: transJac, cdf, rhoX, MXA, LXb, epsX, omegaBarRadius, betaX, Ainf.
Formalization targets
Goal: Theorem 3.18
For every , every and every :
The condition in the second implication corrects the printed statement: can be , and with the implication fails.
Milestones, in proof order
- The operator (Lemmas 3.1–3.4). An error bound on . Then and . Finally, the iterates converge to a feasible point for .
- The linear algebra of (Lemmas 3.5–3.7). On , is idempotent with null space the normal space , and fixes the tangent space .
- Calculus of the CDF (Propositions 3.8, 3.9). The formula and the Hessians. Stationary points of OCP and of CDF coincide on .
- Exactness estimates (Theorem 3.10, Proposition 3.13). The bound holds on . Applying decreases : and on .
Significance
Theorem 3.18 makes the CDF an exact penalty for local minimization. With a penalty parameter above , unconstrained local minimization of near finds exactly the constrained local minimizers. Together with Theorem 3.10, this lets convergence and complexity results for unconstrained methods transfer to Riemannian optimization without retractions (§3.3 of the paper). The bounds of Lemmas 3.1–3.4 also justify the post-processing step used to restore feasibility.
The results are proved in the paper, and to our knowledge none of them has a machine-checked proof. Formalizing them yields a checked exact-penalty theorem with explicit constants. It also yields reusable infrastructure: an error bound for LICQ-regular equality constraints, nearest-point arguments for the zero set of a smooth map, and fixed-point iteration estimates near a manifold.
Difficulty
The obvious argument for the first implication of the goal compares with at a nearby feasible point. No such point is directly available: Lean has no projection onto , and the paper's projection argument needs the first-order optimality condition at a nearest point. That condition requires a Lagrange multiplier theorem under LICQ. The paper instead uses , whose existence (Lemma 3.4) is a quantitative contraction argument: one must show that the iterates never leave . The printed proof of the goal also applies Proposition 3.13 at the minimizer with the constants of , although the hypothesis concerns . A correct proof reruns the estimates with the constants of on a small ball around inside .
The step in the proof of Proposition 3.13 bounds along a segment where does not apply directly, and needs a separate argument. Every constant in the statements has to be tracked exactly, since the threshold is explicit.
Formalization scope
- Spaces and maps. is
EuclideanSpace ℝ (Fin n). Transposed Jacobians are adjoints of Fréchet derivatives, with the operator norm, and is the infimum of over unit . Theorems that use assume . - Constants. The constants of §2.3 are
sSups over the closed ball . They are finite under the assumptions, and no boundedness hypothesis is added. is the largest satisfying the two linear constraints. This is the printed minimum with , so an affine constraint (where ) does not collapse to a point; a proved sanity check confirms on such an instance. - Local minimizers. A local minimizer of OCP is a feasible point minimizing on near it. A local minimizer of CDF minimizes on a full neighbourhood in . Restricting the CDF neighbourhood to or to would trivialize the second implication.
- Penalty threshold. The threshold is the printed . Replacing it by " large enough" is a weaker theorem and not acceptable.
- Proposition 3.8. The Hessian terms and are stated as and . The literal directional-derivative reading of (3.9) is false.
Welcome contributions are a Lagrange multiplier rule for nearest points of under LICQ, Lipschitz-on-compact lemmas for locally Lipschitz maps, and proofs of the milestones in order.
Selected references
- N. Xiao, X. Liu, K.-C. Toh, Dissolving Constraints for Riemannian Optimization, arXiv:2203.10319v3, 2022; Mathematics of Operations Research, 2024. https://arxiv.org/abs/2203.10319
- P.-A. Absil, R. Mahony, R. Sepulchre, Optimization Algorithms on Matrix Manifolds, Princeton University Press, 2008. https://doi.org/10.1515/9781400830244
- R. Fletcher, A class of methods for nonlinear programming with termination and convergence properties, in Integer and Nonlinear Programming, North-Holland, 1970.