The Proximal Alternating Direction Method of Multipliers in the Nonconvex Setting: Convergence Analysis and Rates 1: Bounded Iterates Converge to a KKT Point with Finite Length Under the KL PropertyResearch Paper
Motivation
Many problems in signal processing, statistics and machine learning are of the form : a smooth data-fitting term , plus a possibly nonsmooth and nonconvex regularizer applied to a linear image of the variable. Examples are total-variation-type penalties with nonconvex potentials, -type sparsity terms, and constraints with nonconvex. The alternating direction method of multipliers (ADMM) treats and separately, which makes it the method of choice for this class. For convex problems its convergence theory is classical. For nonconvex problems it was long used as a heuristic, and the Kurdyka–Łojasiewicz (KŁ) framework of Attouch, Bolte, Redont, Soubeyran and Svaiter, and of Bolte, Sabach and Teboulle, is now the standard tool for proving that such methods converge.
Timeline. Li and Pong (SIAM J. Optim. 2015) proved global convergence of a proximal ADMM with for twice differentiable under a KŁ hypothesis. Hong, Luo and Razaviyayn (SIAM J. Optim. 2016) studied related nonconvex ADMM schemes under other assumptions. Boţ and Nguyen (arXiv:1801.01994v2, Math. Oper. Res. 45(2), 2020) extended this in three directions. Their scheme uses variable metrics , . It allows a relaxation parameter in the dual step. It covers both a proximal and a proximal-linearized -update. Under these conditions they proved that bounded iterates have finite length and converge to a KKT point whenever a regularized augmented Lagrangian has the KŁ property. This mission formalizes that theorem.
Setting
Let be proper and lower semicontinuous, let be differentiable with -Lipschitz gradient, and let be linear. The augmented Lagrangian with parameter is
Let and be symmetric positive semidefinite matrices, and write . Fix . From an arbitrary starting point , Algorithm 1 generates
- ,
- ,
- .
Algorithm 2 is the same, except that in the -update is replaced by its linearization . Assumption 1 asks for four things:
- and are bounded below;
- is surjective;
- ;
- and , where and are explicit in , , and .
With the constants and of §2.2 of the paper, the regularized augmented Lagrangian is
A KKT point of the problem is a triple with , and , where is the limiting subdifferential. A proper lower semicontinuous is a KŁ function if, near every point of , a concave desingularizing function satisfies .
Formalization targets
Goal: Theorem 14
Under Assumption 1, let be a bounded sequence generated by Algorithm 1 or Algorithm 2. If is a KŁ function, then
and for some KKT point .
Milestones
The milestones follow the order in which the paper's proof uses them:
- the descent lemma (Remark 1, already proved on the platform);
- the monotonicity inequality for (Lemma 4);
- the increment estimates (Lemma 5);
- the Lyapunov property and convergence of (Theorem 7);
- explicit subgradients of and with bounds (Lemma 8, Lemma 9, Corollary 10);
- the limit-set properties (Lemmas 11 and 12);
- the uniformized KŁ property (Lemma 13, shared with the PALM mission);
- a summability lemma for three-step recursions (Lemma 3).
Theorem 6, the boundedness of the iterates under coercivity, is included as a supporting item that discharges the boundedness hypothesis.
Significance
The theorem gives global convergence of the whole sequence, not just subsequences, for a broad family of nonconvex splitting methods. The family includes full-splitting variants, in which makes the -update explicit. The KŁ hypothesis on holds, for example, whenever and are semi-algebraic. The finite length conclusion is also the starting point for the rates of Section 3 of the paper, which are the subject of the companion mission.
The paper result is proved; this mission formalizes the known proof. As far as the platform records go, no machine-checked proof of convergence of a nonconvex ADMM exists. The Li–Pong ADMM theorems on the platform are open statements about a different algorithm. A completed development would supply the following reusable pieces:
- the Lyapunov analysis of a variable-metric ADMM with relaxed dual step;
- subdifferential calculus for an augmented Lagrangian on a product space;
- a second instance of the generic KŁ finite-length argument.
Difficulty
The obvious Lyapunov candidate, itself, does not decrease along the iterates: the dual step increases it by . That increase has to be paid for by primal progress, and this is possible only because surjectivity of lets the dual increment be controlled by -increments over two consecutive steps. This is where the specific constants , , and the coupling condition (15) come from. The KŁ argument must therefore be run on the lifted function in five variables, whose subgradient bound (Corollary 10) involves three past increments and two telescoping terms. The textbook one-step KŁ argument therefore does not apply directly, and the recursion has to be closed with the three-step summability Lemma 3. For the dual update is relaxed, and the estimates must handle in both regimes and .
Formalization scope
Spaces are EuclideanSpace ℝ (Fin n), and products such as and the 5-fold domain of are nested WithLp 2 products. Their norm is the Euclidean product norm the paper uses, not the sup norm. The function and the functions , take values in EReal. The data are bundled in a structure Data, and the algorithm is a parameter Variant (prox for Algorithm 1, linearized for Algorithm 2), so every statement covers both algorithms. A run is a relation: any minimizer may be selected at each step, and existence of minimizers is not claimed. and are replaced by parameters with the inequalities the proofs use ( with , and ). The paper's statements are the instance , . The subdifferential is the published limiting subdifferential, and "KŁ function" is the published IsKLFunction. In that definition, is read as a bound for every subgradient, so . Corrections read past: the paper indexes although exists only for , and it writes "defined in Assumption 1" for and , which are defined on p. 10.
There are three ways to formalize the statement trivially, and all are excluded:
- a run predicate that no sequence satisfies, or a junk value making so that Assumption 1 cannot hold (a sorry-free check shows that Assumption 1 and the run relation are jointly satisfiable for both algorithms);
- a KŁ hypothesis on an arbitrary function: the hypothesis is on the specific built from and ;
- the weaker conclusion " is stationary": the conclusion is the full KKT triple.
Contributions are welcome at every milestone. Lemma 3 and the subdifferential calculus behind Lemmas 8–9 are independent of the algorithm, and are good entry points.
Selected references
- R. I. Boţ, D.-K. Nguyen, The proximal alternating direction method of multipliers in the nonconvex setting: convergence analysis and rates, arXiv:1801.01994v2, 2020; Math. Oper. Res. 45(2), 2020. https://arxiv.org/abs/1801.01994, https://doi.org/10.1287/moor.2019.1008
- J. Bolte, S. Sabach, M. Teboulle, Proximal alternating linearized minimization for nonconvex and nonsmooth problems, Math. Program. 146, 2014. https://doi.org/10.1007/s10107-013-0701-9
- 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
- G. Li, T. K. Pong, Global convergence of splitting methods for nonconvex composite optimization, SIAM J. Optim. 25(4), 2015. https://doi.org/10.1137/140998135
- M. Hong, Z.-Q. Luo, M. Razaviyayn, Convergence analysis of alternating direction method of multipliers for a family of nonconvex problems, SIAM J. Optim. 26(1), 2016. https://doi.org/10.1137/140990309