Accelerated Gradient Methods for Nonconvex Nonlinear and Stochastic Programming 2: One Aggressive Stepsize Policy Drives the AG Gradient Mapping to Zero on Nonconvex Composite ProblemsResearch Paper
Motivation
Many problems in statistics, signal processing and machine learning have the form "smooth loss plus simple non-smooth term": a differentiable data-fit term together with a constraint set or a sparsity-inducing penalty such as the norm. When the loss is convex, Nesterov's accelerated gradient (AG) method and its proximal variants are known to be optimal among first-order methods: after iterations the optimality gap is , against for plain proximal gradient (Nesterov 1983, reference [24] of the paper; Beck & Teboulle 2009). When the loss is nonconvex, the classical accelerated schemes come with no convergence guarantee at all, and practitioners must choose between a method that is fast on convex problems and one that is guaranteed to make progress on nonconvex ones.
Ghadimi and Lan (arXiv:1310.3787v1, 2013; Math. Program. 156, 2016) proposed a single AG scheme that does both. This mission formalizes their result for composite problems (§2.2, Theorem 2), where the smooth part is split into a possibly nonconvex piece with Lipschitz constant and a convex piece, and a non-smooth convex term is handled through a proximal step. The theorem shows that one aggressive step-size policy, the one that is optimal in the convex case, still drives a stationarity measure to zero when the smooth part is nonconvex.
Setting
Work in with the Euclidean norm. The problem (1.3) is
where:
- is differentiable with -Lipschitz gradient and may be nonconvex;
- is convex and differentiable with -Lipschitz gradient, so has -Lipschitz gradient with ;
- is a convex, possibly non-smooth function with bounded domain , for instance the indicator of a convex compact set, or that indicator plus .
The prox map (2.37) is
and the gradient mapping (2.38) is . With it reduces to when , and a small certifies approximate stationarity of (1.3). Assumption 2 asks for a constant with for all and ; it holds whenever is proper, closed, convex with bounded domain (Lemma 2).
Algorithm 2 takes step sizes with , for , , , starts from , and for sets
The weights are , (2.6).
Formalization targets
Goal: Theorem 2 (p. 11)
Fix and assume, for ,
Let be an optimal solution of (1.3). Then
and, if ,
The goal is stated for general step sizes, so it does not hard-code the constants of a particular policy; Corollary 2 (the policy , , ) is a specialization left for follow-up work.
Milestones
In the order of the proof: Lemma 1 (unrolling a -weighted recursion) and (2.16); the identity (2.42) linking to ; Lemma 5, the two-sided bound ; the descent bound (2.46) (already proved on the platform); the one-step inequalities (2.47)–(2.51); the summed inequality after (2.51); and the bounds (2.52)–(2.53).
Significance
The result. Theorem 2 gives a uniform treatment of convex and nonconvex composite problems. The same step sizes yield the optimal optimality gap when is affine, and a bound on the gradient mapping when is nonconvex. Under Corollary 2's policy that bound gives an iteration count of for , which depends on the Lipschitz constant of the convex part more mildly than projected gradient does. The proof template (a weighted Lyapunov argument in which the nonconvexity appears only through ) recurs in later stochastic and variance-reduced accelerated methods, including the paper's own Theorem 4.
Formalizing it. The result has been proved and refereed since 2016. As far as is known it has not been formalized in any proof assistant, and the platform holds no accelerated method with a nonconvex guarantee. A complete development would give a machine-checked proximal three-point inequality for convex functions on a convex domain, a checked composite AG analysis, and a checked version of Lemma 5. These are reusable pieces for any accelerated proximal method.
Difficulty
The obvious route is to copy the convex analysis of accelerated methods. It breaks at the convexity step: the bound needs convexity of , which is now available only up to -quadratic errors. Those errors involve and , which do not telescope and are not controlled by the usual potential; without some a priori bound on the iterates they can grow with . A formal proof also has to track the extended-valued carefully, since its convexity is used only at points of its domain, and the first iteration, where is not a prox output.
Formalization scope
- Space. is
EuclideanSpace ℝ (Fin n). Smoothness of and is the platform'sConvexOptAlg.SmoothGD.IsBetaSmooth(differentiable with an explicit gradient map, Lipschitz gradient, nonnegative constant). and are written as and . - The convex term. is represented by its domain
Kand its finite valuesX : E n → ℝonK, withConvexOn ℝ K X. Points outsideKnever enter: is compared only onK, is optimal overK, and (2.48)–(2.51) quantify over . A real-valued on all of is ruled out, because Assumption 2 is then unsatisfiable ( as ) and the goal would hold vacuously. - The prox map is a function
P x y cwith the hypothesisIsProxMap K X P: for every ,P x y c ∈ Kand it minimizes the prox objective overK. Without this minimizer hypothesis the three-point inequalities and the goal are false, so it cannot be dropped. - Indexing is 1-based as in the paper. , so . The minimum in (2.44) is
Finset.inf'over . Conditions (2.9)–(2.10) are imposed for , and is written . - One added hypothesis, . The paper uses it silently in (2.52) at ; it holds whenever .
- A typo is corrected. (2.53) prints after substituting ; the Lean states .
Contributions welcome: proofs of any milestone. The three-point inequality (2.48)–(2.49) and Lemma 5 are self-contained and reusable. Also welcome: a proof of Lemma 2 (bounded domain implies Assumption 2), and Corollary 2 as a follow-up statement.
Selected references
- S. Ghadimi, G. Lan, Accelerated Gradient Methods for Nonconvex Nonlinear and Stochastic Programming, arXiv:1310.3787v1, 2013; Math. Program. 156 (2016) 59–99. https://arxiv.org/abs/1310.3787v1 · https://doi.org/10.1007/s10107-015-0871-8
- A. Beck, M. Teboulle, A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems, SIAM J. Imaging Sci. 2 (2009) 183–202. https://doi.org/10.1137/080716542
- S. Ghadimi, G. Lan, H. Zhang, Mini-batch stochastic approximation methods for nonconvex stochastic composite optimization, Math. Program. 155 (2016) 267–305 (reference [13] of the paper: the projected-gradient bound compared against). https://doi.org/10.1007/s10107-014-0846-1
- S. Ghadimi, G. Lan, Optimal stochastic approximation algorithms for strongly convex stochastic composite optimization, I: a generic algorithmic framework, SIAM J. Optim. 22 (2012) 1469–1492 (reference [11] of the paper; its Lemma 2 is the three-point inequality (2.48)–(2.49)). https://doi.org/10.1137/110848864