Global Convergence of Splitting Methods for Nonconvex Composite Optimization III: For Semi-Algebraic Problems the ADMM Sequence Converges and Has Finite LengthResearch Paper
Motivation
The alternating direction method of multipliers (ADMM) is a standard method for problems of the form , in which a smooth loss is composed with a structured, possibly nonsmooth regularizer through a linear map . Its convergence theory was developed for convex problems, yet it is routinely run on nonconvex ones: sparse recovery with the constraint, low-rank matrix problems, and total-variation-type models with nonconvex penalties. For such problems a practitioner wants a guarantee about the iterates actually produced, not only about the existence of good subsequences.
Li and Pong (arXiv:1407.0753v6, SIAM J. Optim. 25(4), 2015) gave the first such guarantee for the classical ADMM on nonconvex composite problems with a surjective . Their Theorem 1 shows that cluster points of the (proximal) ADMM are stationary; their Theorem 3, the subject of this mission, shows that for semi-algebraic data the whole sequence converges. The argument adapts the Kurdyka–Łojasiewicz (KL) framework of Attouch, Bolte and Svaiter (Math. Program. 137, 2013) to a setting where the ADMM only decreases its merit function in the -block.
Setting
Fix linear, twice continuously differentiable with bounded Hessian, and proper (finite somewhere) and closed (lower semicontinuous). For the augmented Lagrangian is
The ADMM produces from arbitrary by
Assumption 1 with asks for , and symmetric maps with , and for all , , and .
The limiting subdifferential of consists of limits of regular subgradients at points with . A point is stationary if .
A set in is semi-algebraic if it is a finite union of sets cut out by finitely many polynomial equations and strict inequalities ; a function is semi-algebraic if its graph is. A proper has the KL property at if there are , a neighbourhood of and a continuous concave with , , , such that whenever and . A KL function is proper, closed, and KL at every point of .
In the Lean development is augLag h P M β x y z, and also augLagX h P M β as a single function on the triple space with the Euclidean inner product.
Formalization targets
Goal: Theorem 3 (p. 13)
Under the standing assumptions and Assumption 1 with , if and are semi-algebraic and the ADMM sequence has a cluster point , then
No constants are fixed: every parameter is quantified exactly as in the paper.
Milestones
- (35): some has for .
- (36): for .
- (39): .
- Finite termination when reaches its limit value.
- (41): the one-step KL estimate.
- Remark 4(1): the goal with " is a KL function" in place of semi-algebraicity.
- is semi-algebraic when and are.
- Proper closed semi-algebraic functions are KL functions, with .
Milestones 6, 7 and 8 together imply the goal.
Significance
The result. Theorem 3 upgrades subsequential convergence to convergence of the whole iterate sequence, with finite length of the -trajectory, for a nonconvex ADMM without any convexity of or . Semi-algebraicity covers the paper's applications: polynomial losses, the constraint, and indicators of polyhedral or algebraic sets. Remark 4(1) isolates the only property actually used, the KL property of , so the result extends to any class of functions for which that property is known (for instance, globally subanalytic or o-minimal definable data).
Formalizing it. The result is proved in the paper and, as far as is known, formalized nowhere. The mission produces a machine-checked version of the paper's convergence argument, a Lean definition of the KL property with the correct convention for empty subdifferentials, and a semi-algebraic set predicate over MvPolynomial. Milestone 8 is a published theorem of real algebraic geometry and nonsmooth analysis (Bolte–Daniilidis–Lewis 2007) that the paper quotes without proof; it is part of what a complete development of the goal requires.
Difficulty
The obvious route is to invoke the abstract convergence theorem of Attouch–Bolte–Svaiter for descent methods. It does not apply: its sufficient-decrease hypothesis requires to drop by a multiple of , while the ADMM only guarantees a drop proportional to (Remark 4(2)). The relative-error bound (35) is likewise in terms of the -step alone, and relating the - and -blocks back to the -block uses the surjectivity of and the specific structure of the multiplier update. The neighbourhood on which the KL inequality holds is a neighbourhood of the full triple, whereas the trajectory is controlled only in .
The semi-algebraic part has a separate difficulty: showing that is semi-algebraic needs closure of semi-algebraic sets under projection (the Tarski–Seidenberg theorem), and the KL property of semi-algebraic functions needs the Łojasiewicz inequality for subanalytic or semi-algebraic functions. Mathlib has neither.
Formalization scope
Spaces are EuclideanSpace ℝ (Fin n); is a continuous linear map and its adjoint. and take values in EReal, never passed through toReal except where the value is provably finite. is Mathlib's Loewner order on self-maps; the Hessian is fderiv ℝ (gradient h). The ADMM is the proximal-ADMM relation with ; argmin steps are "value at most the value anywhere", with no uniqueness; is unconstrained. The triple space is the nested product, so its inner product is the sum of the block inner products.
The KL inequality is stated for every , which encodes . Writing it with Metric.infDist 0 (∂f x) would give and make the KL property fail at every point with empty subdifferential, so that "semi-algebraic implies KL" becomes false and the goal becomes a statement about a different notion. The KL property is required only at points of , only on a neighbourhood and only for values in ; is differentiable only on the open interval. Semi-algebraicity of an extended-valued function is that of its graph over its real values. is a positive real, which is equivalent to the paper's .
The hypotheses and are part of the theorem, not a simplification: the analogous statement for the proximal ADMM is open (Remark 4(3)). The cluster point is assumed, not derived.
Needed infrastructure: calculus of the limiting subdifferential (a smooth-plus-separable sum rule), the Tarski–Seidenberg theorem, and the Łojasiewicz/KL inequality for semi-algebraic functions. The last two are reusable far beyond this mission; contributions toward them, and toward the analytic core (milestones 1–6), are welcome.
Selected references
- G. Li, T. K. Pong, Global convergence of splitting methods for nonconvex composite optimization, SIAM J. Optim. 25(4), 2015. https://arxiv.org/abs/1407.0753 (v6 is the cited version)
- H. Attouch, J. Bolte, P. Redont, A. Soubeyran, Proximal alternating minimization and projection methods for nonconvex problems: an approach based on the Kurdyka–Łojasiewicz inequality, Math. Oper. Res. 35(2), 2010. https://doi.org/10.1287/moor.1100.0449
- H. Attouch, J. Bolte, B. F. Svaiter, Convergence of descent methods for semi-algebraic and tame problems, Math. Program. 137, 2013. https://doi.org/10.1007/s10107-011-0484-9
- J. Bolte, A. Daniilidis, A. Lewis, The Łojasiewicz inequality for nonsmooth subanalytic functions with applications to subgradient dynamical systems, SIAM J. Optim. 17(4), 2007. https://doi.org/10.1137/050644641
- R. T. Rockafellar, R. J.-B. Wets, Variational Analysis, Springer, 1998. https://doi.org/10.1007/978-3-642-02431-3