A Block Successive Upper-Bound Minimization Method of Multipliers for Linearly Constrained Convex Optimization 2: Randomized BSUM-M Converges to Primal-Dual Optimal Solutions Almost SurelyResearch Paper
Motivation
Many large convex problems in signal processing, networking and statistics have a separable objective coupled by linear constraints: blocks of variables, each with its own nonsmooth regularizer and its own feasible set, tied together by . The alternating direction method of multipliers (ADMM) is the standard tool for such problems when . For the direct extension of ADMM can diverge, as shown by Chen, He, Ye and Yuan (2016), and convergence guarantees for multi-block variants typically require strong convexity or extra correction steps.
Hong, Chang, Wang, Razaviyayn, Ma and Luo propose the block successive upper-bound minimization method of multipliers (BSUM-M), which updates each primal block by minimizing a local upper bound of the augmented Lagrangian, followed by a dual gradient step, and prove that it converges for any number of blocks without strong convexity of the objective. They also analyse a randomized version, RBSUM-M, in which each iteration updates a single randomly chosen block, primal or dual. Randomized block selection matters when data or computation are distributed and not every block is available at every step. This mission formalizes the randomized half of their main theorem; the cyclic half is a separate mission.
Setting
Let with , and consider
The standing assumptions (Assumption A) are: with strictly convex and continuously differentiable; with nonnegative weights; each is a compact polyhedron; the problem is feasible and its primal and dual optimal values are attained.
For the augmented Lagrangian and augmented dual function are
and is the set of minimizers of over . The method works with approximation functions (Assumption B): majorizes in block , touches it with matching gradient at , is strongly convex with modulus and has an -Lipschitz gradient.
RBSUM-M. Fix probabilities summing to 1. At iteration draw with probability . If , take a dual step ; otherwise replace block by
leaving the other blocks and unchanged. The analysis uses the block steps (the minimizer above, computed for every ), the dual step , the gaps , , and the proximal gradient . The error bound with constant is for all and .
Formalization targets
Goal: Theorem 2.1, part 2
Under Assumptions A and B and the error bound, if the stepsizes are either constant and sufficiently small, or satisfy , , then with probability 1
where is the point of nearest to , and every limit point of is a primal and dual optimal pair. The threshold for "sufficiently small" is existential and depends only on the problem data, the , the error-bound constant and .
Milestones
In the order of the paper: Lemma 2.1 (differentiability of , , the -Lipschitz bound); Lemma 2.3(2) (expected decrease of ); Lemma 2.4(2) (proximal gradient bounded by the block and dual steps); Lemmas 2.5(2) and 2.6(2) (expected change of the dual and primal gaps); and from the proof of Theorem 2.1, (2.24)–(2.25) (expected change of ), (2.28) (constraint violation), (2.29) (the descent estimate with explicit coefficients) and (2.40) (one-step change of ).
Significance
The result shows that a randomized, single-block-per-iteration primal-dual method converges almost surely to primal-dual optimal solutions of a multi-block linearly constrained problem, under assumptions that allow a rank-deficient , nonsmooth group-sparse regularizers, and inexact (majorized) block subproblems. The potential function and the way the error bound converts it into a supermartingale estimate are reused in later analyses of multi-block and randomized ADMM-type methods.
The result is proved in the paper; to our knowledge it has no machine-checked proof. Formalizing it produces a checked convergence proof for a randomized augmented-Lagrangian method, a reusable formal treatment of the augmented dual function of a convex program with compact polyhedral constraints, and a precise record of the conventions under which the published statements hold (see the scope section).
Difficulty
The obvious route is to show that decreases. It does not: a dual step increases by , and without strong convexity there is no direct control of the distance to the solution set. The argument has to combine the primal and dual gaps into one potential, bound the residual term through an error bound that holds without strong convexity, and then pass from a conditional-expectation inequality to almost-sure convergence. Under diminishing stepsizes the supermartingale argument only yields , and upgrading this to a limit needs a separate pathwise argument on how fast can move.
Formalization scope
Blocks are EuclideanSpace ℝ (Fin (n k)) and the full vector lives in PiLp 2, so norms are Euclidean. is real-valued on all of . The augmented dual function is (the page's (1.9) prints an unconstrained minimum of plus penalty; the constrained reading is the one Lemma 2.1 and (2.19) use). The prox in the proximal gradient includes the constraint set , which is what the proof of Lemma 2.4 needs; the error bound of Lemma 2.2 is a hypothesis of the goal, in its global form on , and is not derived. Lemma 2.2 itself is not part of the mission.
Conditional expectations given are written as the explicit average over the random index, at every state; this is because the index drawn at iteration is independent of . The goal is stated on an arbitrary probability space carrying mutually independent indices with law , for every run from a deterministic start , . Iterations are indexed from as on the page.
Two printed statements are corrected and labelled in their items: Lemma 2.1 asserts (not each ) constant over , and Lemma 2.6(2) uses where the page prints . Statements are not trivialized: the setting is satisfiable (for instance , , , , , ), the stepsize threshold precedes the probability space and the run, and the solution sets entering distances are nonempty under Assumption A. A formalization in which the error bound or the gap bounds are assumed in a form that already contains the conclusion would not count.
A complete development needs: Danskin-type differentiability of and the -smoothness of the augmented dual; first-order optimality and nonexpansiveness of the prox of ; and the Robbins–Siegmund almost-supermartingale convergence theorem in Mathlib's conditional expectation framework, together with a bridge from the explicit index average to condExp. The last two are reusable well beyond this mission. Contributions to any of the milestones, or to these general lemmas, are welcome.
Selected references
- M. Hong, T.-H. Chang, X. Wang, M. Razaviyayn, S. Ma, Z.-Q. Luo, A Block Successive Upper Bound Minimization Method of Multipliers for Linearly Constrained Convex Optimization, arXiv:1401.7079v1, 2014; Mathematics of Operations Research 45(3), 2020. https://arxiv.org/abs/1401.7079, https://doi.org/10.1287/moor.2019.1010
- M. Hong, Z.-Q. Luo, On the linear convergence of the alternating direction method of multipliers, Mathematical Programming 162, 2017. https://doi.org/10.1007/s10107-016-1034-2
- C. Chen, B. He, Y. Ye, X. Yuan, The direct extension of ADMM for multi-block convex minimization problems is not necessarily convergent, Mathematical Programming 155, 2016. https://doi.org/10.1007/s10107-014-0826-5
- H. Robbins, D. Siegmund, A convergence theorem for non negative almost supermartingales and some applications, in Optimizing Methods in Statistics, Academic Press, 1971. https://doi.org/10.1016/B978-0-12-604550-5.50015-8
- M. Razaviyayn, M. Hong, Z.-Q. Luo, A unified convergence analysis of block successive minimization methods for nonsmooth optimization, SIAM Journal on Optimization 23(2), 2013. https://doi.org/10.1137/120891009
Theorem numbers, equation numbers and page numbers in this mission refer to arXiv:1401.7079v1 (printed page = PDF page − 1); the journal version renumbers them.