Distributed Optimization and Statistical Learning via the Alternating Direction Method of Multipliers II: Soft Thresholding Is the Proximity Operator of the ℓ1 NormTextbook
Why closed-form ADMM updates matter
The alternating direction method of multipliers (ADMM) solves problems of the form subject to by alternating a minimization in , a minimization in and a dual ascent step. Its practical appeal, as surveyed by Boyd, Parikh, Chu, Peleato and Eckstein in Distributed Optimization and Statistical Learning via the Alternating Direction Method of Multipliers (Foundations and Trends in Machine Learning, 2011, doi:10.1561/2200000016), rests on the fact that each of these minimizations is often trivial to carry out: a projection, a linear solve, or a componentwise formula. Chapter 4 of the book, "General Patterns", catalogues the cases that recur in every later application (lasso, basis pursuit, sparse inverse covariance selection, consensus and sharing problems). This mission formalizes the statements of that chapter, culminating in the identification of the update for an penalty with soft thresholding.
Setting
Vectors live in with the Euclidean norm . An extended-real-valued function is given by its effective domain and its finite values there. Fix a penalty parameter , a matrix and a vector . The -update of ADMM (p. 25) is
where "argmin" denotes a minimizer over . When the right-hand side, as a function of , is the proximity operator (§4.1).
The chapter evaluates in four situations:
- is the indicator function of a closed nonempty convex set ( on , off it); then is the Euclidean projection , the nearest point of to ;
- is a convex quadratic, with symmetric positive semidefinite, possibly restricted to an affine set ;
- is componentwise separable and is diagonal;
- with and .
The soft thresholding operator (§4.4.3, p. 32) is, for ,
applied componentwise to vectors.
Formalization targets
Goal: soft thresholding is the proximity operator of the ℓ1 norm (§4.4.3, p. 32)
For , and , a vector minimizes over if and only if
The statement gives both that the soft-thresholded vector is a minimizer and that it is the only one.
Milestones
- Projection (§4.1, p. 26): for closed, nonempty and convex, exists and is unique, and for every the -update of the indicator of is exactly .
- Nonnegative orthant (§4.1, p. 26): for the indicator of , , uniquely.
- Quadratic update (4.1) (p. 26): if is invertible it is positive definite, and is the unique minimizer.
- Matrix inversion lemma (§4.2.4, p. 28): when the inverses exist. This is the Woodbury identity, already proved on the platform, and is referenced rather than re-posed.
- Equality-constrained quadratic (§4.2.5, p. 29): minimizes the update over iff some solves the KKT system; when the affine set is nonempty, is an affine function of .
- Component separability (§4.4.2, p. 31): with diagonal, the -update splits into independent scalar minimizations.
- Positive-part form (p. 32): .
- Shrinkage form (4.2) (p. 32): for .
- Scalar update (p. 32): is the unique minimizer of .
Significance
The results of this chapter are what make ADMM an implementable algorithm rather than a conceptual one. Soft thresholding is the -update of the lasso (§6.4), of distributed -regularized regression (§8.2) and of sparse inverse covariance selection (§6.5); projections are the updates of every constrained problem in Chapter 5; the linear-system form (4.1) and the KKT system are the updates of least squares and quadratic programming. Every later chapter of the book cites these identities without proof, so a formal library of them is the substrate on which formal statements about specific ADMM instances can be built.
None of these statements is open as a mathematical question: all are classical facts of convex analysis. The book states them without proof (the goal is justified by a reference to subdifferential calculus in Rockafellar's Convex Analysis). This mission poses their formal statements as Lean proof goals. The matrix inversion lemma already has a machine-checked proof on the platform (the Woodbury identity) and is reused. A related but distinct statement, that singular value shrinkage is the proximity operator of the nuclear norm (Cai, Candès and Shen, 2010), is posed separately on the platform and is not this mission's target.
Difficulty
The goal objective is not differentiable, so setting a gradient to zero does not apply; the minimizer sits exactly on the kink of whenever . A proof has to handle the three regimes of and must establish global optimality, not only stationarity, together with uniqueness. Passing from the vector problem to the scalar ones requires the separability statement, which is a statement about minimizers of sums over a product domain, not about any particular function. For the quadratic milestones, the difficulty is linear-algebraic: showing that invertibility of forces positive definiteness, and, for the equality-constrained case, that a minimizer exists and depends affinely on even when is rank deficient and the KKT matrix is singular.
Formalization scope
Vectors are EuclideanSpace ℝ (Fin n), so ‖·‖ is ; matrices are Matrix (Fin p) (Fin n) ℝ acting through Matrix.toEuclideanLin. The conventions committed to are:
- Functions as (domain, values). An extended-real-valued is a pair: a domain set and a real function on it. An indicator function is (the set, ).
- Argmin as a predicate.
IsXUpdate,IsProxandIsProjectionsay " is a minimizer"; no choice function is used. Where the book says "the solution", the theorems state an equivalence " is a minimizer equals the formula", which gives existence and uniqueness at once. - Explicit hypotheses. and (written
lam, sinceλis a Lean keyword); for the soft thresholding identities (implicit in the book, where , and false for ); in (4.2), as on the page; symmetric positive semidefinite (Matrix.PosSemidef); invertibility of in (4.1); closed, convex and nonempty for the projection; nonemptiness of for the existence part of §4.2.5; of arbitrary rank. - ℓ1 norm. is defined as , separately from the Euclidean norm.
- Separability is stated for a general with diagonal, with the scalar problems written out explicitly (the book does not write them).
The trivializing formalization to avoid is stating the goal as the subgradient condition , or as agreement of the three formulas for : the goal is global minimality and uniqueness of the soft-thresholded vector for the objective. Not formalized: the flop counts and caching remarks of §§4.2.1–4.2.4 and the iterative-solver discussion of §4.3, which make no mathematical claim.
Contributions welcome: proofs of any milestone; the general block-separability statement of §4.4.1; a lemma connecting IsProx with Mathlib's orthogonal projection onto complete convex sets.
Selected references
- S. Boyd, N. Parikh, E. Chu, B. Peleato, J. Eckstein, Distributed Optimization and Statistical Learning via the Alternating Direction Method of Multipliers, Foundations and Trends in Machine Learning 3(1), 2011, pp. 1–122. https://doi.org/10.1561/2200000016
- R. T. Rockafellar, Convex Analysis, Princeton University Press, 1970 (§23, subdifferentials). https://doi.org/10.1515/9781400873173
- J.-J. Moreau, Proximité et dualité dans un espace hilbertien, Bulletin de la Société Mathématique de France 93, 1965, pp. 273–299. https://doi.org/10.24033/bsmf.1625
- J.-F. Cai, E. J. Candès, Z. Shen, A Singular Value Thresholding Algorithm for Matrix Completion, SIAM Journal on Optimization 20(4), 2010, pp. 1956–1982. https://doi.org/10.1137/080738970