Stochastic Model-Based Minimization of Weakly Convex Functions 1: Stochastic One-Sided Model Methods Drive the Moreau-Envelope Gradient E‖∇φ_{1/ρ̄}(x_{t*})‖² to Zero at Rate O(1/√T)Research Paper
Motivation
Many problems in statistics, signal processing and machine learning minimize an expected loss that is neither smooth nor convex: robust phase retrieval, blind deconvolution, covariance estimation with the loss, training of networks with ReLU activations, and composite problems with convex and smooth . A common structural feature of these objectives is weak convexity: adding a quadratic makes the function convex. For such problems the classical stochastic algorithms — the stochastic proximal point method, the stochastic prox-linear method, and the stochastic proximal subgradient method — were used in practice long before any nonasymptotic complexity guarantee was available for them.
Damek Davis and Dmitriy Drusvyatskiy (arXiv:1803.06523, SIAM J. Optim. 29(1), 2019) gave such guarantees for all of these methods at once. They abstract each method as minimization of a stochastic one-sided model of the objective, and prove that the expected squared gradient of the Moreau envelope at a randomly selected iterate decays at rate , i.e. a sample complexity of . This mission formalizes that main theorem (Theorem 4.3) and the lemmas its proof uses.
Timeline. Nurminskii (1973, 1974) showed that when the sample losses are -weakly convex and there is no regularizer, the stochastic subgradient method converges subsequentially to stationary points almost surely; no rate was given. Ghadimi and Lan (2013) proved complexity for smooth nonconvex stochastic problems. Davis and Grimmer (arXiv:1707.03505, 2017) gave a rate for a proximally guided variant of the stochastic subgradient method on weakly convex problems. Davis and Drusvyatskiy (2018) obtained the complexity for the basic proximal stochastic subgradient method and, through the one-sided model framework, for the stochastic proximal point and prox-linear methods.
Setting
Work on with the Euclidean norm. An extended-valued function has domain and is closed when its epigraph is closed. A function is -weakly convex when is convex. For the Moreau envelope and proximal map are
The problem is with closed and locally Lipschitz, where is accessible only through a random model , . Assumption B asks for constants , an open convex and a measurable such that: and on (one-sided accuracy); is -weakly convex for almost every ; and .
Algorithm 4.1 fixes and parameters , draws i.i.d. samples , and sets
It returns , with drawn with . With it is the stochastic proximal point method; with a linearization it is the proximal stochastic subgradient method; with it is the stochastic prox-linear method.
Formalization targets
Goal: Theorem 4.3, (4.15) and (4.16)
and for with ,
Milestones, in the order the proof uses them
- Lemma 4.1: is -weakly convex and is -Lipschitz on .
- Lemma 4.2, (4.7): , with .
- Lemma 2.2: for -weakly convex and , is with .
- Theorem 4.3, (4.14): the one-step expected descent of .
Significance
The result. By Lemma 2.2, , and the proximal point satisfies and . A small envelope gradient therefore certifies that is close to a nearly stationary point. Theorem 4.3 gives this certificate, with explicit constants, for every algorithm fitting Assumption B, in particular the first complexity bounds for the stochastic proximal point and stochastic prox-linear methods on weakly convex problems. Companion results in the same paper (Theorems 4.1 and 4.2, separate missions) specialize to convex and strongly convex models.
Formalizing it. The theorem is proved on paper; no machine-checked version exists, on this platform or in Mathlib, of Theorem 4.3, of Lemma 2.2 for extended-valued weakly convex functions, or of the one-sided-model framework. A formalization yields a reusable Moreau-envelope library for weakly convex functions with constraints (via the domain of ), and a template for iterate-level stochastic analysis on product probability spaces.
Difficulty
The algebra of each step is short; what is hard is making it rigorous. First, Lemma 2.2 must be proved for extended-valued closed functions: existence and uniqueness of the proximal point need strong convexity of the subproblem together with lower semicontinuity and a quadratic minorant, and differentiability of the envelope is not a consequence of a generic envelope theorem. Second, the conditional expectations in Lemma 4.2 must be realized on the product space of samples, with integrability of , and established along the iterates rather than assumed; the paper takes all of these for granted. Third, Assumption B gives the Lipschitz bound (4.2) only in one direction and only almost surely, and the argmin of Algorithm 4.1 exists only almost surely, so every pointwise step must be carried through null sets that depend on the current iterate.
Formalization scope
- Carrier and encoding. is
EuclideanSpace ℝ (Fin d). The extended-valued (and ) is a pair ; closedness, weak convexity, the envelope (an infimum over ) and proximal points are defined in this encoding. The envelope parameter is writtenlam. - Randomness. The samples are a point of under the product measure, and is the Bochner integral. In Lemma 4.2, is the integral over one fresh sample with the current iterate fixed.
- The algorithm. The step is any jointly measurable selection that lies in and minimizes the subproblem for almost every sample; measurability is an added hypothesis.
- The output. is the weighted average of with weights , its exact value for drawn independently of the samples.
- is any real lower bound of on ; the bound is monotone in it.
- Added hypotheses, implicit in the paper: (the subproblem is -strongly convex; Assumption B allows any real , so this does not follow from ), in (4.16) (its derivation from (4.15) needs it), , , and, in Lemma 2.2, that is closed and proper.
- Typos. The statement of Theorem 4.3 twice says "Algorithm 3.1"; it means Algorithm 4.1. The constant in (4.16) is the printed one, which is valid (slightly weaker than what (4.15) yields). The proof's "Setting " is a slip; (4.16) keeps .
- No trivialization. Every bounded expectation comes with an integrability conclusion (a Bochner integral of a non-integrable function is ), the update is constrained by the argmin property of Algorithm 4.1, and is a genuine lower bound; the hypotheses are satisfiable (e.g. , , zero model).
- Contributions welcome: a general Moreau-envelope library for weakly convex extended-valued functions (Lemma 2.2), measurability of argmin selections, and the one-step analysis lemmas.
Selected references
- D. Davis, D. Drusvyatskiy, Stochastic Model-Based Minimization of Weakly Convex Functions, SIAM J. Optim. 29(1):207–239, 2019. arXiv:1803.06523
- D. Davis, B. Grimmer, Proximally Guided Stochastic Subgradient Method for Nonsmooth, Nonconvex Problems, SIAM J. Optim. 29(3), 2019. arXiv:1707.03505
- S. Ghadimi, G. Lan, Stochastic First- and Zeroth-Order Methods for Nonconvex Stochastic Programming, SIAM J. Optim. 23(4):2341–2368, 2013. doi:10.1137/120880811
- R. T. Rockafellar, R. J.-B. Wets, Variational Analysis, Springer, 1998. doi:10.1007/978-3-642-02431-3