Convex Optimization: Algorithms and Complexity V: Projected Gradient Descent on a Convex β-Smooth Function Has Rate (3β‖x₁ − x*‖² + f(x₁) − f(x*))/tTextbook
Motivation
Many optimization problems in statistics, machine learning and operations research ask to minimize a smooth convex function over a simple convex set: a box, a ball, a simplex, a cone of positive semidefinite matrices. Projected gradient descent is the most basic first-order method for such problems. It takes a gradient step and then returns to the feasible set by Euclidean projection. Its cost per iteration is one gradient evaluation and one projection, independent of the dimension apart from the cost of manipulating vectors, which is why it remains a workhorse in very high dimension.
This mission is the fifth in a series formalizing S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning 8(3–4), 2015; arXiv:1405.4980). Chapter 3 of the monograph treats "dimension-free" convex optimization. Its §3.2 first proves that gradient descent on a convex -smooth function on has rate (Theorem 3.3, the subject of mission IV), and then turns to the constrained case, which is the subject of this mission. The analysis follows Nesterov's Introductory Lectures on Convex Optimization (2004).
Setting
Write for the Euclidean inner product on and for the Euclidean norm. The constraint set is compact and convex; this is the standing assumption of Chapter 3. The Euclidean projection of onto is
which exists and is unique for a nonempty closed convex set.
A differentiable function is -smooth on if its gradient is -Lipschitz there:
It is -strongly convex on if for . A minimizer of over is assumed to exist, as everywhere in the monograph.
Projected gradient descent with step size starts at and iterates
For a point with projected step , the gradient mapping is . Without constraints it equals .
Formalization targets
Goal: Theorem 3.7 (p. 270)
For convex and -smooth on , projected gradient descent with satisfies, for every ,
Milestones
- Lemma 3.1 (p. 263): for and , , and hence . The first claim is an already proved platform theorem.
- (3.7) (p. 270): for .
- Lemma 3.6 (p. 270): for ,
- Descent and gap (pp. 270–271): and .
- Distances decrease (p. 271): .
- Recursion and induction (p. 271): with , , and an induction on this recursion gives the bound of Theorem 3.7.
Companion: Theorem 3.10 (p. 278)
If is in addition -strongly convex on , with condition number , then the strengthened Lemma 3.6, display (3.14), gives for every
Significance
Theorem 3.7 shows that a constraint set with a cheap projection costs nothing in the order of convergence: the rate of unconstrained gradient descent survives, with a constant that depends on . Lemma 3.6 is the reusable part. It isolates the gradient mapping as the quantity that measures progress under constraints, and the same inequality, with a strong-convexity term added, gives the linear rate of Theorem 3.10. It is also the starting point of the analysis of proximal gradient methods for composite objectives.
All of these results are classical and proved in the textbook. As far as a search of the platform shows, none of them is formalized there. A related platform theorem on constrained gradient descent for well-conditioned functions (from Hazan's Introduction to Online Convex Optimization) bounds function values with a different exponent. It is not Theorem 3.10, which bounds distances with rate . This mission adds machine-checked statements of the constrained analysis in Bubeck's formulation, with the book's constants. The definitions of -smoothness on a set, the gradient mapping and the run predicate can be reused by later missions on accelerated and proximal methods.
Difficulty
The obvious attempt is to copy the unconstrained proof, which rests on the descent inequality . That inequality fails under constraints: the projection can cut the step short, so the decrease in is not controlled by . At a boundary minimizer, for instance, while no step makes progress. The proof has to replace the gradient by the gradient mapping throughout. It also has to show separately that the iterates do not move away from , a fact that came for free without constraints from co-coercivity of the gradient (display (3.6)).
The closing induction is not routine either. The step from to works only from on, and the step from to needs a direct computation that uses the term in the numerator.
Formalization scope
- Space. is
EuclideanSpace ℝ (Fin n), and is⟪x, y⟫_ℝ. - Constraint set and minimizer. Every statement assumes compact (
IsCompact X) and convex (Convex ℝ X), the standing assumption of Chapter 3. Every statement about a run assumes a minimizer with for all , the monograph's standing assumption. - Gradient. The gradient is an explicit map
gwithHasGradientAt f (g x) xat every point of , so is defined at boundary points of . -smoothness on (IsBetaSmoothOn X f g β) asks this together with and the Lipschitz bound on only. It is not the quadratic upper bound (3.4), which is a consequence. - Strong convexity. It is the published predicate
OnlineConvexOpt.ConvexBasics.StronglyConvexOn X f g α, which is display (3.13) with the same gradient map. Display (3.14) and Theorem 3.10 use . - Projection. The projection is the published relation
IsMetricProjection X y p: and for every . No choice function is used. - Run. A run is
IsProjGDRun X g β x: , , and for every the iterate is a projection of . Indices start at as in the book, andx 0is unused. - Gradient mapping.
gradMap β x xplustakes the projected point of the run as an argument. - Positivity. is assumed, as the step size requires, and for Theorem 3.10, as requires.
- The recursion. The recursion of milestone 6 is stated multiplied by , so the case needs no division.
- No trivial readings. A run predicate with a free step size, or -smoothness replaced by the quadratic bound (3.4), would change the content. Both are excluded: the step is fixed to , and smoothness is the Lipschitz condition on the gradient.
Contributions are welcome on any milestone. Lemma 3.1 and (3.7) need only the variational characterization of the projection. Lemma 3.6 needs the quadratic upper bound for a function whose gradient is Lipschitz on a convex set, together with the first-order characterization of convexity relative to a set. Both are of independent use. The induction milestone is a self-contained statement about real sequences.
Selected references
- S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. arXiv:1405.4980
- Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer Academic Publishers, 2004. doi:10.1007/978-1-4419-8853-9
- E. Hazan, Introduction to Online Convex Optimization, 2nd ed., MIT Press, 2022. arXiv:1909.05207