Douglas-Rachford Splitting and ADMM for Nonconvex Optimization: Tight Convergence Results III: ADMM Is Douglas-Rachford Splitting on the Image-Function Problem, for Every RelaxationResearch Paper
Motivation
The alternating direction method of multipliers (ADMM) solves linearly constrained problems
by alternating a minimization in , a minimization in and a multiplier update. It is one of the standard methods for distributed optimization, statistical learning and signal processing, and in practice it is often applied to nonconvex or (see Boyd et al., Distributed optimization and statistical learning via the alternating direction method of multipliers, 2011, doi:10.1561/2200000016).
For convex problems, ADMM with is classically Douglas–Rachford splitting (DRS) applied to the Fenchel dual of (1.2) (Gabay 1983). The dual reading needs convexity. Themelis and Patrinos (arXiv:1709.05747v4, §5; SIAM J. Optim. 30(1), 2020, doi:10.1137/18M1163993) prove a primal equivalence instead: for arbitrary and for every relaxation parameter , one ADMM update is one DRS step on a reformulation of (1.2) built from image functions. A primal equivalence for was noted earlier by Bauschke and Koch (2015, Rem. 3.14) for , and Yan and Yin (2016, Thm. 1) showed that this case loses no generality. This mission formalizes the general statement and the convergence theorem for nonconvex ADMM that the paper derives from it.
Setting
All spaces are Euclidean, , , , . A function is proper if it never takes the value and is finite somewhere. The proximal mapping of with stepsize is the set
For and a linear map , the image function is
The -augmented Lagrangian of (1.2) is . One ADMM update with penalty and relaxation maps to :
One DRS step with stepsize and relaxation on maps to , where and . The reformulation of (1.2) is
that is, and . The regular subdifferential is , the limiting one .
Formalization targets
Goal: primal equivalence (Theorem 5.5)
Let be proper, large enough that every ADMM subproblem has a solution, and . With , , and the same expressions at , and ,
together with , , , and .
Milestones
Proposition 5.2(i)–(iii) (properness of , , ), Proposition 5.3 (), the inclusions (5.3a) and (5.3b), and the identity .
Companions
Theorem 5.5(vi): if is surjective, is -smooth and , then . Theorem 5.6: under Assumption II and in the ranges of Theorem 4.1 with , decreases by per step, , cluster points are KKT points with common cost, and the iterates are bounded when is level bounded.
Significance
Theorem 5.5 reduces nonconvex ADMM to nonconvex DRS: any descent, convergence or rate statement for DRS on becomes one for ADMM, with the Douglas–Rachford envelope becoming the augmented Lagrangian (5.5(vi)). This is how the paper obtains Theorem 5.6, with explicit penalty bounds for every permitted by Theorem 4.1, under assumptions weaker than earlier nonconvex ADMM analyses (Li and Pong, 2015, doi:10.1137/140998135, require and coercivity).
The results are proved in the paper. No machine-checked proof of a primal ADMM–DRS equivalence is known; the platform holds formalizations of proximal ADMM variants and of convex DRS for monotone operators, but not of image functions. The mission produces a reusable library of image functions and their proximal and subdifferential calculus, and a precise statement of relaxed ADMM in the nonconvex, extended-real setting.
Difficulty
The individual steps are short, but the objects are extended-real and set-valued. An image function may equal , and its infimum need not be attained; that is proper and that the infimum is attained at must be derived from the solvability of the ADMM subproblems, not assumed. The proximal mappings are multivalued, so the equivalence is an inclusion between argmin sets, and the algebra relating , and has to be matched to the relaxation exactly. The subgradient claims require regular subdifferentials of extended-real functions, and (v) requires the optimality condition of the -subproblem, whose quadratic term couples with the residual. Theorem 5.6 additionally rests on the whole DRS convergence analysis of §4.
Formalization scope
Points are EuclideanSpace ℝ (Fin _), matrices are continuous linear maps, is ContinuousLinearMap.adjoint, and is the operator norm. Extended-real functions are EReal-valued; the image function imageFn is the EReal infimum over , so it is off the range and may be . Proximal mappings and are sets, and an ADMM update (IsADMMStep) is any selection of minimizers in the order . The two subproblems minimize the terms of that depend on the minimized variable, because the dropped constant may be at a starting point and would make every a minimizer. "Large enough penalty" is SubproblemsSolvable. The regular and limiting subdifferentials are the published IsRegularSubgrad and LimitingSubdiff.
Added hypotheses: and proper in Theorem 5.5 and 5.5(vi), and proper in Proposition 5.2(i). The page omits them, and they are necessary: for every is the whole space while is not proper. Claim (v) is stated as the existence of with , which is stronger than the distance bound and avoids the distance to an empty set. Theorem 5.6 uses the constants of Theorem 4.1 as derived in its proof ((4.8), (4.10)); the printed (4.3) has the sign of one entry of a max reversed and the printed (4.5) an extra factor , and both are then false (counterexamples in the companion mission on DRS convergence). Theorem 5.6 adds ; its run is indexed so that the paper's is the term , with the starting triple at index .
The goal does not assume the DRS relations or (5.3a)–(5.3b), and are defined from the ADMM iterates by (5.2), not chosen independently; a formalization that quantifies over an arbitrary prox selection, or that hard-codes , is a different statement.
Needed infrastructure: image functions and their proximal calculus, regular subgradients of extended-real functions under linear maps, and, for Theorem 5.6, the DRS convergence theory of §4 (formalized in the companion mission I of this series). Proofs of individual milestones are welcome independently.
Selected references
- A. Themelis, P. Patrinos, Douglas–Rachford splitting and ADMM for nonconvex optimization: tight convergence results, SIAM J. Optim. 30(1):149–181, 2020. arXiv:1709.05747v4, doi:10.1137/18M1163993
- H. H. Bauschke, V. R. Koch, Projection methods: Swiss army knives for solving feasibility and best approximation problems with halfspaces, in Infinite Products of Operators and Their Applications, Contemp. Math. 636, pp. 1–40, AMS, 2015.
- M. Yan, W. Yin, Self equivalence of the alternating direction method of multipliers, in Splitting Methods in Communication, Imaging, Science, and Engineering, pp. 165–194, Springer, 2016.
- G. Li, T. K. Pong, Global convergence of splitting methods for nonconvex composite optimization, SIAM J. Optim. 25(4):2434–2460, 2015. doi:10.1137/140998135
- S. Boyd, N. Parikh, E. Chu, B. Peleato, J. Eckstein, Distributed optimization and statistical learning via the alternating direction method of multipliers, Found. Trends Mach. Learn. 3(1), 2011. doi:10.1561/2200000016
- D. Gabay, Applications of the method of multipliers to variational inequalities, in Augmented Lagrangian Methods, Studies in Mathematics and Its Applications 15, pp. 299–331, Elsevier, 1983.