Prove2Me
Navigate
DiscoverCollectionsFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in
← Collections

Convex Optimization

Convex optimization textbooks, chapter by chapter: KKT conditions, conic duality, barrier methods, and the complexity of first-order methods from center of gravity to mirror descent.

17 completed missions

Missions

1–17 of 17
OpenCompletedAll
🏆Completed
Convex OptimizationOperations ResearchOptimization·Captain: Shuze Chen

Convex Optimization I: Prékopa's TheoremTextbook

Log-concave functions are the meeting point of convex analysis and probability: densities of Gaussian, exponential, uniform and Wishart distributions are all log-concave, and countless facts of applied probability flow from one structural theorem — integrating out variables preserves log-concavity. This mission builds the convex-analysis spine of Boyd & Vandenberghe's Convex Optimization (Chapters 2–3) — separation and supporting hyperplanes, dual cones, the first- and second-order differential characterizations of convexity, Fenchel conjugacy — and climbs to Prékopa's theorem via the Prékopa–Leindler inequality, a landmark of Brunn–Minkowski theory absent from Mathlib.

29 thms5 active usersReviewed
🏆Completed
Convex OptimizationOperations ResearchOptimization·Captain: Shuze Chen

Convex Optimization II: KKT ConditionsTextbook

The Karush–Kuhn–Tucker conditions are the central result of convex optimization: for a convex differentiable problem satisfying Slater's condition, a point is optimal exactly when primal feasibility, dual feasibility, complementary slackness and Lagrangian stationarity hold. This mission formalizes Chapters 4–5 of Boyd & Vandenberghe end to end — the first-order optimality criterion, concavity of the Lagrange dual, weak duality, Slater's strong-duality theorem with dual attainment (via the separating-hyperplane argument of §5.3.2), the saddle-point characterization, sensitivity bounds and Pareto scalarization — culminating in the full KKT characterization.

15 thms4 active usersReviewed
🏆Completed
Convex OptimizationOperations ResearchOptimization·Captain: Shuze Chen

Convex Optimization IV: Löwner–John EllipsoidsTextbook

Every full-dimensional convex body is sandwiched between an ellipsoid and its nnn-fold dilation: shrinking the minimum-volume covering (Löwner–John) ellipsoid E\mathcal{E}E about its centre x0x_0x0​ by the factor 1/n1/n1/n lands inside the body,

x0+1n (E−x0)  ⊆  C  ⊆  E,x_0 + \tfrac{1}{n}\,(\mathcal{E} - x_0) \;\subseteq\; C \;\subseteq\; \mathcal{E},x0​+n1​(E−x0​)⊆C⊆E,

and the factor nnn is tight on simplices. This rounding theorem underlies the ellipsoid method, John's theorem on the Banach–Mazur distance to the Euclidean ball, and much of modern convex geometry. The mission formalizes §8.4 of Boyd & Vandenberghe for polytopes C=conv⁡{x1,…,xm}C = \operatorname{conv}\{x_1,\dots,x_m\}C=conv{x1​,…,xm​}, exactly as the book proves it: existence and uniqueness of the extremal ellipsoid, the KKT identities at the normalized optimum (∑iλixixiT=I\sum_i \lambda_i x_i x_i^{T} = I∑i​λi​xi​xiT​=I, ∑iλixi=0\sum_i \lambda_i x_i = 0∑i​λi​xi​=0, ∑iλi=n\sum_i \lambda_i = n∑i​λi​=n), the convex-combination step that produces the 1/n1/n1/n ball, and affine invariance.

8 thms3 active usersReviewed
🏆Completed
Convex OptimizationOperations ResearchOptimization·Captain: Shuze Chen

Convex Optimization V: Newton's MethodTextbook

The classical convergence theory of smooth convex minimization. For a function that is mmm-strongly convex and MMM-smooth (mI⪯∇2f(x)⪯MImI \preceq \nabla^2 f(x) \preceq MImI⪯∇2f(x)⪯MI), gradient descent converges linearly, while Newton's method exhibits its famous two phases: a damped phase in which every backtracking step decreases the objective by a fixed amount γ\gammaγ, and a quadratically convergent phase in which the scaled gradient norm squares at each step, L2m2∥∇f(x+)∥2≤(L2m2∥∇f(x)∥2)2\tfrac{L}{2m^2}\lVert \nabla f(x^{+})\rVert_2 \le \bigl(\tfrac{L}{2m^2}\lVert \nabla f(x)\rVert_2\bigr)^22m2L​∥∇f(x+)∥2​≤(2m2L​∥∇f(x)∥2​)2. Together they give the iteration count of B&V (9.36),

#iterations  ≤  f(x(0))−p⋆γ  +  log⁡2log⁡2(ε0/ε),γ=αβη2mM2,ε0=2m3L2,\#\text{iterations} \;\le\; \frac{f(x^{(0)}) - p^{\star}}{\gamma} \;+\; \log_2\log_2(\varepsilon_0/\varepsilon), \qquad \gamma = \frac{\alpha\beta\eta^2 m}{M^2}, \quad \varepsilon_0 = \frac{2m^3}{L^2},#iterations≤γf(x(0))−p⋆​+log2​log2​(ε0​/ε),γ=M2αβη2m​,ε0​=L22m3​,

with LLL the Lipschitz constant of the Hessian and α,β\alpha,\betaα,β the backtracking parameters. This mission formalizes Chapters 9–10 of Boyd & Vandenberghe with every constant exactly as printed — a quantitative theory entirely absent from Mathlib.

12 thms4 active usersReviewed
🏆Completed
Convex OptimizationOperations ResearchOptimization·Captain: Shuze Chen

Convex Optimization III: Conic Duality and the S-procedureTextbook

Two quadratic functions can be compared losslessly. The S-procedure says that, when the constraint is strictly feasible, the implication

q1(x)≤0  ⟹  q2(x)≤0,qk(x)=xTFkx+2gkTx+hk,q_1(x) \le 0 \;\Longrightarrow\; q_2(x) \le 0, \qquad q_k(x) = x^{T}F_k x + 2g_k^{T}x + h_k,q1​(x)≤0⟹q2​(x)≤0,qk​(x)=xTFk​x+2gkT​x+hk​,

holds if and only if a single nonnegative multiplier certifies it as a matrix inequality, λ[F1g1g1Th1]⪰[F2g2g2Th2]\lambda \begin{bmatrix} F_1 & g_1 \\ g_1^{T} & h_1\end{bmatrix} \succeq \begin{bmatrix} F_2 & g_2 \\ g_2^{T} & h_2\end{bmatrix}λ[F1​g1T​​g1​h1​​]⪰[F2​g2T​​g2​h2​​] for some λ≥0\lambda \ge 0λ≥0. It is a cornerstone of control theory, trust-region methods and robust optimization, and a rare case in which a nonconvex problem has zero duality gap. The route runs through the theory this mission builds from Boyd & Vandenberghe §5.8–5.9 and Appendix B: strong alternatives for convex inequality systems, cone-program strong duality under a generalized Slater condition, semidefinite programming duality, the LMI theorems of alternatives, and the hidden convexity of the joint range of two quadratic forms.

17 thms5 active usersReviewed
🏆Completed
Convex OptimizationOperations ResearchOptimization·Captain: Shuze Chen

Convex Optimization VI: Self-Concordance and the Barrier MethodTextbook

Why do interior-point methods solve convex programs in O(mlog⁡(1/ε))O(\sqrt{m}\log(1/\varepsilon))O(m​log(1/ε)) Newton steps? Nesterov and Nemirovskii's answer is self-concordance: a convex function whose third derivative is controlled by its second, ∣φ′′′(t)∣≤2 φ′′(t)3/2|\varphi'''(t)| \le 2\,\varphi''(t)^{3/2}∣φ′′′(t)∣≤2φ′′(t)3/2 along every line, admits a Newton analysis with absolute constants and no condition number — and the logarithmic barrier is self-concordant. This mission formalizes §9.6 and Chapter 11 of Boyd & Vandenberghe: the self-concordance calculus, the Newton-decrement analysis, the duality gap m/tm/tm/t along the central path, the per-centering work bound m(μ−1−log⁡μ)/γ+cm(\mu - 1 - \log\mu)/\gamma + cm(μ−1−logμ)/γ+c, and the crown result — with the aggressive schedule μ=1+1/m\mu = 1 + 1/\sqrt{m}μ=1+1/m​ the barrier method reaches duality gap ε\varepsilonε after

⌈m log⁡2(m/(t(0)ε))⌉\Bigl\lceil \sqrt{m}\,\log_2\bigl(m/(t^{(0)}\varepsilon)\bigr)\Bigr\rceil⌈m​log2​(m/(t(0)ε))⌉

centering steps, each of uniformly bounded Newton cost.

19 thms6 active users
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity III: Projected Subgradient Descent with η = R/(L√t) Satisfies f(average) − f(x*) ≤ RL/√tTextbook

Motivation

Many convex optimization problems in machine learning and statistics have objectives that are convex but not differentiable: hinge losses, ℓ1\ell_1ℓ1​ penalties, maxima of finitely many affine functions, and the dual functions of Lagrangian relaxations. Methods that rely on gradients do not apply to them directly, while cutting-plane methods such as the ellipsoid method pay a price that grows with the dimension. The projected subgradient method replaces the gradient by an arbitrary subgradient and restores feasibility by a Euclidean projection. Its guarantee depends on the dimension only through two constants, a radius RRR and a Lipschitz constant LLL. This is the reason it, and its descendants (mirror descent, stochastic gradient descent, online gradient descent), are the standard tools for large-scale nonsmooth problems.

The rate analysed here goes back to the subgradient methods of Shor and Polyak in the 1960s and 1970s and to the lower bounds of Nemirovski and Yudin (1983). The book follows the presentation of Nesterov, Introductory Lectures on Convex Optimization (2004). The strongly convex variant with weights proportional to sss is from Lacoste-Julien, Schmidt and Bach (2012).

This mission is the third of a series formalizing S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning, 2015), and covers the preamble of Chapter 3, Section 3.1 and Section 3.4.1.

Setting

Let Rn\mathbb R^nRn carry the Euclidean inner product x⊤yx^\top yx⊤y and norm ∥⋅∥\|\cdot\|∥⋅∥. Let X⊆Rn\mathcal X\subseteq\mathbb R^nX⊆Rn be compact and convex, and let fff be a convex function on X\mathcal XX with a minimizer x∗∈Xx^*\in\mathcal Xx∗∈X.

A vector ggg is a subgradient of fff at x∈Xx\in\mathcal Xx∈X if f(x)−f(y)≤g⊤(x−y)f(x)-f(y)\le g^\top(x-y)f(x)−f(y)≤g⊤(x−y) for every y∈Xy\in\mathcal Xy∈X. The set of subgradients at xxx is written ∂f(x)\partial f(x)∂f(x). The projection ΠX(y)\Pi_{\mathcal X}(y)ΠX​(y) of a point y∈Rny\in\mathbb R^ny∈Rn is the point of X\mathcal XX nearest to yyy.

Fix step sizes ηs>0\eta_s>0ηs​>0. Projected subgradient descent starts at some x1∈Xx_1\in\mathcal Xx1​∈X and iterates, for s≥1s\ge1s≥1,

ys+1=xs−ηsgs,gs∈∂f(xs),xs+1=ΠX(ys+1).y_{s+1}=x_s-\eta_s g_s,\quad g_s\in\partial f(x_s),\qquad x_{s+1}=\Pi_{\mathcal X}(y_{s+1}).ys+1​=xs​−ηs​gs​,gs​∈∂f(xs​),xs+1​=ΠX​(ys+1​).

Any subgradient may be chosen at each step. In Section 3.1 the step is constant, ηs=η\eta_s=\etaηs​=η. The set X\mathcal XX lies in the Euclidean ball of radius RRR centred at x1x_1x1​, and the subgradients have norm at most LLL.

A function fff is α\alphaα-strongly convex on X\mathcal XX if f(x)−f(y)≤g⊤(x−y)−α2∥x−y∥2f(x)-f(y)\le g^\top(x-y)-\frac{\alpha}{2}\|x-y\|^2f(x)−f(y)≤g⊤(x−y)−2α​∥x−y∥2 for all x,y∈Xx,y\in\mathcal Xx,y∈X and g∈∂f(x)g\in\partial f(x)g∈∂f(x).

Formalization targets

Goal: Theorem 3.2

For every horizon t≥1t\ge1t≥1, projected subgradient descent with the constant step η=R/(Lt)\eta=R/(L\sqrt t)η=R/(Lt​) satisfies

f(1t∑s=1txs)−f(x∗)≤RLt.f\Big(\frac1t\sum_{s=1}^{t}x_s\Big)-f(x^*)\le\frac{RL}{\sqrt t}.f(t1​s=1∑t​xs​)−f(x∗)≤t​RL​.

Milestones

  1. Lemma 3.1. For x∈Xx\in\mathcal Xx∈X and y∈Rny\in\mathbb R^ny∈Rn: (ΠX(y)−x)⊤(ΠX(y)−y)≤0(\Pi_{\mathcal X}(y)-x)^\top(\Pi_{\mathcal X}(y)-y)\le0(ΠX​(y)−x)⊤(ΠX​(y)−y)≤0, already on the platform as a published theorem. The mission also states its consequence
∥ΠX(y)−x∥2+∥y−ΠX(y)∥2≤∥y−x∥2.\|\Pi_{\mathcal X}(y)-x\|^2+\|y-\Pi_{\mathcal X}(y)\|^2\le\|y-x\|^2 .∥ΠX​(y)−x∥2+∥y−ΠX​(y)∥2≤∥y−x∥2.
  1. The per-step inequality in the proof of Theorem 3.2:
f(xs)−f(x∗)≤12η(∥xs−x∗∥2−∥ys+1−x∗∥2)+η2∥gs∥2.f(x_s)-f(x^*)\le\frac1{2\eta}\big(\|x_s-x^*\|^2-\|y_{s+1}-x^*\|^2\big)+\frac\eta2\|g_s\|^2 .f(xs​)−f(x∗)≤2η1​(∥xs​−x∗∥2−∥ys+1​−x∗∥2)+2η​∥gs​∥2.
  1. The summed inequality for any constant step η>0\eta>0η>0:
∑s=1t(f(xs)−f(x∗))≤R22η+ηL2t2.\sum_{s=1}^{t}\big(f(x_s)-f(x^*)\big)\le\frac{R^2}{2\eta}+\frac{\eta L^2t}{2}.s=1∑t​(f(xs​)−f(x∗))≤2ηR2​+2ηL2t​.

Companion: Theorem 3.9

If fff is α\alphaα-strongly convex and its subgradients are bounded by LLL, then with ηs=2/(α(s+1))\eta_s=2/(\alpha(s+1))ηs​=2/(α(s+1)),

f(∑s=1t2st(t+1)xs)−f(x∗)≤2L2α(t+1).f\Big(\sum_{s=1}^{t}\frac{2s}{t(t+1)}x_s\Big)-f(x^*)\le\frac{2L^2}{\alpha(t+1)}.f(s=1∑t​t(t+1)2s​xs​)−f(x∗)≤α(t+1)2L2​.

Significance

Theorem 3.2 gives an oracle complexity of O(R2L2/ε2)O(R^2L^2/\varepsilon^2)O(R2L2/ε2) for reaching an ε\varepsilonε-optimal point, independent of the ambient dimension. Section 3.5 of the book shows this rate is unimprovable for black-box first-order methods once the dimension is large. Theorem 3.9 shows how strong convexity improves the rate to O(1/t)O(1/t)O(1/t), with the averaging weights changed from uniform to linear. These two bounds are the reference points against which the rest of Chapter 3 and Chapters 4 to 6 (smooth, accelerated, mirror, stochastic methods) are measured.

The results are classical and their proofs are short. Formalizing them produces a reusable, machine-checked account of the basic projected first-order step: the projection inequality, the one-step distance recursion, and the telescoping argument with Jensen's inequality for averaged iterates. To our knowledge, no machine-checked proof of the averaged-iterate bound for the projected subgradient method exists in Mathlib. Related platform items cover other algorithms or other averaging schemes.

Difficulty

The arithmetic is elementary, and the obvious argument works. The care is in the bookkeeping. The projection must be shown not to increase the distance to x∗x^*x∗, which needs convexity of X\mathcal XX and the variational characterization of the nearest point. The sum must telescope with a horizon-dependent constant step. Jensen's inequality must be applied to a finite convex combination of points of X\mathcal XX, which requires showing that the average lies in X\mathcal XX. In Theorem 3.9 the step sizes and the averaging weights are coupled, so neither can be changed independently. In Lean, the iterates are indexed from 111 with natural-number horizons, and the bounds involve t\sqrt tt​, so these casts need care.

Formalization scope

Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n). The iterates are sequences ℕ → EuclideanSpace ℝ (Fin n) with the first iterate at index 111. The projection is the published relation OnlineConvexOpt.FirstOrder.IsMetricProjection (xs+1∈Xx_{s+1}\in\mathcal Xxs+1​∈X is a nearest point to ys+1y_{s+1}ys+1​). Subgradients are taken relative to X\mathcal XX (Definition 1.2). A run is a predicate on the steps s=1,…,ts=1,\dots,ts=1,…,t, and every theorem holds for all runs, that is, for every choice of subgradients. Compactness and convexity of X\mathcal XX, convexity of fff on X\mathcal XX and the existence of the minimizer x∗x^*x∗ are the book's standing assumptions and appear as hypotheses. R>0R>0R>0, L>0L>0L>0 and α>0\alpha>0α>0 are explicit, because Lean's division by zero would otherwise turn the step size into a junk value.

The book assumes ∥g∥≤L\|g\|\le L∥g∥≤L for every subgradient at every point of X\mathcal XX. With subgradients relative to a compact X\mathcal XX, that assumption can never hold at a boundary point, since every outward normal can be added to a subgradient. Stated that way the theorems would be vacuous. The mission therefore assumes the bound only for the subgradients g1,…,gtg_1,\dots,g_tg1​,…,gt​ that the run uses. This is a weaker hypothesis and gives a stronger, non-vacuous statement. Bounding only these subgradients is a deliberate choice, not a trivialization: the bound still constrains every quantity the conclusion depends on.

Contributions welcome: proofs of the four inequalities and of the two rates, and a reusable lemma that the convex combination of finitely many points of a convex set lies in the set, together with Jensen's inequality in the form used here.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. https://arxiv.org/abs/1405.4980
  • Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer, 2004. https://doi.org/10.1007/978-1-4419-8853-9
  • A. Nemirovski and D. Yudin, Problem Complexity and Method Efficiency in Optimization, Wiley, 1983.
  • S. Lacoste-Julien, M. Schmidt and F. Bach, A simpler approach to obtaining an O(1/t) convergence rate for the projected stochastic subgradient method, 2012. https://arxiv.org/abs/1212.2002
  • N. Z. Shor, Minimization Methods for Non-Differentiable Functions, Springer, 1985. https://doi.org/10.1007/978-3-642-82118-9
7 thms4 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity IV: Gradient Descent on a Convex β-Smooth Function Has Rate 2β‖x₁ − x*‖²/(t − 1)Textbook

Motivation

Gradient descent goes back to Cauchy (1847). It is the simplest method for minimizing a differentiable function, and most of the first-order methods in large-scale optimization and machine learning are variants of it. Its appeal in high dimension is that its oracle complexity, the number of gradient evaluations needed to reach a given accuracy, can be bounded independently of the dimension. For a merely Lipschitz convex function the projected subgradient method needs on the order of 1/ε21/\varepsilon^21/ε2 steps to reach accuracy ε\varepsilonε (Theorem 3.2 of the book). Under a smoothness assumption gradient descent does much better, because the gradients shrink near the optimum and the steps adapt automatically.

This mission formalizes the basic result of that kind: Theorem 3.3 of S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning 8(3–4), 2015; arXiv:1405.4980v2). It states that gradient descent with step size 1/β1/\beta1/β on a convex β\betaβ-smooth function on Rn\mathbb R^nRn has optimality gap O(1/t)O(1/t)O(1/t) after ttt steps. The result and its proof are standard; versions appear in Nesterov's Introductory Lectures on Convex Optimization (2004, §2.1.5). It is the fourth mission in a series that formalizes the capstone results of Bubeck's monograph.

Setting

Write Rn\mathbb R^nRn for Euclidean space with inner product x⊤yx^\top yx⊤y and norm ∥⋅∥\|\cdot\|∥⋅∥. Let f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R be differentiable with gradient ∇f\nabla f∇f.

  • fff is convex if f((1−λ)x+λy)≤(1−λ)f(x)+λf(y)f((1-\lambda)x+\lambda y)\le(1-\lambda)f(x)+\lambda f(y)f((1−λ)x+λy)≤(1−λ)f(x)+λf(y) for all x,yx,yx,y and λ∈[0,1]\lambda\in[0,1]λ∈[0,1].
  • For β≥0\beta\ge0β≥0, fff is β\betaβ-smooth if its gradient is β\betaβ-Lipschitz:
∥∇f(x)−∇f(y)∥≤β∥x−y∥for all x,y∈Rn.\|\nabla f(x)-\nabla f(y)\|\le\beta\|x-y\|\qquad\text{for all }x,y\in\mathbb R^n.∥∇f(x)−∇f(y)∥≤β∥x−y∥for all x,y∈Rn.
  • A minimizer is a point x∗x^*x∗ with f(x∗)≤f(y)f(x^*)\le f(y)f(x∗)≤f(y) for every yyy. Throughout the book a minimizer is assumed to exist.
  • Gradient descent with step size η>0\eta>0η>0, started at x1∈Rnx_1\in\mathbb R^nx1​∈Rn, is the sequence
xt+1=xt−η∇f(xt),t≥1.(3.1)x_{t+1}=x_t-\eta\nabla f(x_t),\qquad t\ge1. \tag{3.1}xt+1​=xt​−η∇f(xt​),t≥1.(3.1)

The optimality gaps are δs=f(xs)−f(x∗)≥0\delta_s=f(x_s)-f(x^*)\ge0δs​=f(xs​)−f(x∗)≥0.

Formalization targets

Goal: Theorem 3.3 (p. 267)

If fff is convex and β\betaβ-smooth with β>0\beta>0β>0, x∗x^*x∗ is a minimizer, and (xt)(x_t)(xt​) is gradient descent with η=1/β\eta=1/\betaη=1/β, then for every t≥2t\ge2t≥2

f(xt)−f(x∗)≤2β∥x1−x∗∥2t−1.f(x_t)-f(x^*)\le\frac{2\beta\|x_1-x^*\|^2}{t-1}.f(xt​)−f(x∗)≤t−12β∥x1​−x∗∥2​.

Milestones

These are the statements the book's proof uses, in the book's order:

  1. Lemma 3.4 (p. 267). For any β\betaβ-smooth fff, with no convexity: ∣f(x)−f(y)−∇f(y)⊤(x−y)∣≤β2∥x−y∥2|f(x)-f(y)-\nabla f(y)^\top(x-y)|\le\frac\beta2\|x-y\|^2∣f(x)−f(y)−∇f(y)⊤(x−y)∣≤2β​∥x−y∥2.
  2. (3.4) (p. 267). For convex β\betaβ-smooth fff: 0≤f(x)−f(y)−∇f(y)⊤(x−y)≤β2∥x−y∥20\le f(x)-f(y)-\nabla f(y)^\top(x-y)\le\frac\beta2\|x-y\|^20≤f(x)−f(y)−∇f(y)⊤(x−y)≤2β​∥x−y∥2.
  3. (3.5) (p. 267). For convex fff, one step of length 1/β1/\beta1/β decreases fff by at least 12β∥∇f(x)∥2\frac1{2\beta}\|\nabla f(x)\|^22β1​∥∇f(x)∥2.
  4. Lemma 3.5 (p. 268). If (3.4) holds, then f(x)−f(y)≤∇f(x)⊤(x−y)−12β∥∇f(x)−∇f(y)∥2f(x)-f(y)\le\nabla f(x)^\top(x-y)-\frac1{2\beta}\|\nabla f(x)-\nabla f(y)\|^2f(x)−f(y)≤∇f(x)⊤(x−y)−2β1​∥∇f(x)−∇f(y)∥2.
  5. (3.6) (p. 269). Co-coercivity: (∇f(x)−∇f(y))⊤(x−y)≥1β∥∇f(x)−∇f(y)∥2(\nabla f(x)-\nabla f(y))^\top(x-y)\ge\frac1\beta\|\nabla f(x)-\nabla f(y)\|^2(∇f(x)−∇f(y))⊤(x−y)≥β1​∥∇f(x)−∇f(y)∥2.
  6. Distances decrease (proof of Theorem 3.3, p. 269). ∥xs+1−x∗∥≤∥xs−x∗∥\|x_{s+1}-x^*\|\le\|x_s-x^*\|∥xs+1​−x∗∥≤∥xs​−x∗∥ for every s≥1s\ge1s≥1.
  7. The recursion (p. 268). δs+1≤δs−12β∥x1−x∗∥2δs2\delta_{s+1}\le\delta_s-\frac{1}{2\beta\|x_1-x^*\|^2}\delta_s^2δs+1​≤δs​−2β∥x1​−x∗∥21​δs2​.
  8. From the recursion to the rate (p. 269). For ω>0\omega>0ω>0 and non-negative reals, ωδs2+δs+1≤δs\omega\delta_s^2+\delta_{s+1}\le\delta_sωδs2​+δs+1​≤δs​ for all s≥1s\ge 1s≥1 implies 1/δt≥ω(t−1)1/\delta_t\ge\omega(t-1)1/δt​≥ω(t−1).

Stronger companion: footnote 4 (p. 269)

Under the same hypotheses, f(xt)−f(x∗)≤2β∥x1−x∗∥2/(t+3)f(x_t)-f(x^*)\le 2\beta\|x_1-x^*\|^2/(t+3)f(xt​)−f(x∗)≤2β∥x1​−x∗∥2/(t+3) for every t≥1t\ge1t≥1.

Significance

Theorem 3.3 is the reference rate for first-order methods on smooth convex problems. Several later results in the book are measured against it. Nesterov's accelerated gradient descent (§3.7) improves 1/t1/t1/t to 1/t21/t^21/t2, the lower bounds of §3.5 show that 1/t21/t^21/t2 cannot be beaten by any black-box first-order method, and adding strong convexity (§3.4) upgrades 1/t1/t1/t to a linear rate. Its ingredients are reused throughout the book and the optimization literature: the descent lemma (Lemma 3.4), the one-step improvement (3.5) and co-coercivity (3.6). Co-coercivity is the finite-dimensional case of the Baillon–Haddad theorem.

The result is classical and fully proved on paper. To our knowledge Mathlib does not contain this rate for gradient descent on convex smooth functions, and no published Prove2Me theorem states it. The mission provides it, together with Lemma 3.4 and co-coercivity as reusable statements on EuclideanSpace ℝ (Fin n). These are the facts that later missions of the series on projected gradient descent, strong convexity and acceleration need. Formalizing the improved constant of footnote 4 is a welcome addition.

Difficulty

Two steps are not routine. The first is Lemma 3.4, where the book integrates the gradient along a segment. In Lean this needs a mean-value or fundamental-theorem-of-calculus argument for a function on a Euclidean space, together with a Cauchy–Schwarz estimate, and Mathlib's HasGradientAt interface has to be connected to one-variable derivatives along lines.

The second is the monotonicity of ∥xs−x∗∥\|x_s-x^*\|∥xs​−x∗∥. The natural first attempt is to telescope the one-step improvement (3.5) together with the convexity bound δs≤∥xs−x∗∥∥∇f(xs)∥\delta_s\le\|x_s-x^*\|\|\nabla f(x_s)\|δs​≤∥xs​−x∗∥∥∇f(xs​)∥. That gives a recursion involving ∥xs−x∗∥\|x_s-x^*\|∥xs​−x∗∥, which is not controlled by ∥x1−x∗∥\|x_1-x^*\|∥x1​−x∗∥ without further work. Making the recursion uniform requires co-coercivity (3.6), which comes from Lemma 3.5. That lemma's hypothesis is the two-sided inequality (3.4), not smoothness directly. The final numerical step divides by the gaps δs\delta_sδs​, so zero gaps need separate handling.

Formalization scope

  • Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n), and x⊤yx^\top yx⊤y is ⟪x, y⟫_ℝ.
  • The gradient is an explicit map g:Rn→Rng:\mathbb R^n\to\mathbb R^ng:Rn→Rn with the hypothesis ∀ x, HasGradientAt f (g x) x.
  • β\betaβ-smoothness (IsBetaSmooth f g β) is the book's definition: β≥0\beta\ge0β≥0, the gradient-existence hypothesis, and the Lipschitz bound ∥g(x)−g(y)∥≤β∥x−y∥\|g(x)-g(y)\|\le\beta\|x-y\|∥g(x)−g(y)∥≤β∥x−y∥. The book's "continuously differentiable" follows from it.
  • Convexity is Mathlib's ConvexOn ℝ Set.univ f. A minimizer is a point xstar with ∀ y, f xstar ≤ f y; its existence is the book's standing assumption, and ∇f(x∗)=0\nabla f(x^*)=0∇f(x∗)=0 is derived, not assumed.
  • A gradient-descent run (IsGDRun g η x) is a sequence x : ℕ → ℝⁿ with η>0\eta>0η>0 and xt+1=xt−ηg(xt)x_{t+1}=x_t-\eta g(x_t)xt+1​=xt​−ηg(xt​) for every t≥1t\ge1t≥1. The book's x1x_1x1​ is x 1, and index 000 is unused. Every theorem quantifies over all runs with η=1/β\eta=1/\betaη=1/β.

Disclosed side conditions:

  • β>0\beta>0β>0 wherever the page divides by β\betaβ. Convexity is retained for (3.5), as in its source context.
  • t≥2t\ge2t≥2 in the goal, where the page's bound has denominator t−1t-1t−1.
  • Statements in which the page divides by δs\delta_sδs​ or by ∥x1−x∗∥2\|x_1-x^*\|^2∥x1​−x∗∥2 are multiplied through, so that they stay true and meaningful when those quantities vanish.

A trivializing formalization is ruled out. Smoothness is the Lipschitz condition on the gradient, so Lemma 3.4 and (3.4) are not restatements of the definition, as they would be if the quadratic upper bound (3.4) were taken as the definition of smoothness. The run predicate also fixes the step size 1/β1/\beta1/β.

A complete development needs:

  • the segment integral or mean-value estimate for HasGradientAt functions;
  • first-order characterizations of convexity for differentiable functions;
  • elementary inner-product algebra.

Lemma 3.4, Lemma 3.5 and (3.6) are reusable well beyond this mission. Proofs of any milestone are welcome independently.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–357, 2015. arXiv:1405.4980v2; §3.2, pp. 266–269.
  • A. Cauchy, Méthode générale pour la résolution des systèmes d'équations simultanées, C. R. Acad. Sci. Paris 25:536–538, 1847.
  • Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer, 2004. doi:10.1007/978-1-4419-8853-9
  • J.-B. Baillon and G. Haddad, Quelques propriétés des opérateurs angle-bornés et n-cycliquement monotones, Israel J. Math. 26:137–150, 1977. doi:10.1007/BF03007664
10 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

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 β\betaβ-smooth function on Rn\mathbb R^nRn has rate O(β∥x1−x∗∥2/t)O(\beta\|x_1-x^*\|^2/t)O(β∥x1​−x∗∥2/t) (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 x⊤yx^\top yx⊤y for the Euclidean inner product on Rn\mathbb R^nRn and ∥⋅∥\|\cdot\|∥⋅∥ for the Euclidean norm. The constraint set X⊆Rn\mathcal X\subseteq\mathbb R^nX⊆Rn is compact and convex; this is the standing assumption of Chapter 3. The Euclidean projection of y∈Rny\in\mathbb R^ny∈Rn onto X\mathcal XX is

ΠX(y)=argmin⁡z∈X∥y−z∥,\Pi_{\mathcal X}(y)=\operatorname*{argmin}_{z\in\mathcal X}\|y-z\| ,ΠX​(y)=z∈Xargmin​∥y−z∥,

which exists and is unique for a nonempty closed convex set.

A differentiable function f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R is β\betaβ-smooth on X\mathcal XX if its gradient is β\betaβ-Lipschitz there:

∥∇f(x)−∇f(y)∥≤β∥x−y∥(x,y∈X).\|\nabla f(x)-\nabla f(y)\|\le\beta\|x-y\|\qquad(x,y\in\mathcal X).∥∇f(x)−∇f(y)∥≤β∥x−y∥(x,y∈X).

It is α\alphaα-strongly convex on X\mathcal XX if f(x)−f(y)≤∇f(x)⊤(x−y)−α2∥x−y∥2f(x)-f(y)\le\nabla f(x)^\top(x-y)-\frac\alpha2\|x-y\|^2f(x)−f(y)≤∇f(x)⊤(x−y)−2α​∥x−y∥2 for x,y∈Xx,y\in\mathcal Xx,y∈X. A minimizer x∗∈Xx^*\in\mathcal Xx∗∈X of fff over X\mathcal XX is assumed to exist, as everywhere in the monograph.

Projected gradient descent with step size η=1/β\eta=1/\betaη=1/β starts at x1∈Xx_1\in\mathcal Xx1​∈X and iterates

xt+1=ΠX(xt−1β∇f(xt))(t≥1).x_{t+1}=\Pi_{\mathcal X}\Bigl(x_t-\tfrac1\beta\nabla f(x_t)\Bigr)\qquad(t\ge1).xt+1​=ΠX​(xt​−β1​∇f(xt​))(t≥1).

For a point xxx with projected step x+=ΠX(x−1β∇f(x))x^+=\Pi_{\mathcal X}\bigl(x-\frac1\beta\nabla f(x)\bigr)x+=ΠX​(x−β1​∇f(x)), the gradient mapping is gX(x)=β(x−x+)g_{\mathcal X}(x)=\beta(x-x^+)gX​(x)=β(x−x+). Without constraints it equals ∇f(x)\nabla f(x)∇f(x).

Formalization targets

Goal: Theorem 3.7 (p. 270)

For fff convex and β\betaβ-smooth on X\mathcal XX, projected gradient descent with η=1/β\eta=1/\betaη=1/β satisfies, for every t≥1t\ge1t≥1,

f(xt)−f(x∗)≤3β∥x1−x∗∥2+f(x1)−f(x∗)t.f(x_t)-f(x^*)\le\frac{3\beta\|x_1-x^*\|^2+f(x_1)-f(x^*)}{t}.f(xt​)−f(x∗)≤t3β∥x1​−x∗∥2+f(x1​)−f(x∗)​.

Milestones

  1. Lemma 3.1 (p. 263): for x∈Xx\in\mathcal Xx∈X and y∈Rny\in\mathbb R^ny∈Rn, (ΠX(y)−x)⊤(ΠX(y)−y)≤0(\Pi_{\mathcal X}(y)-x)^\top(\Pi_{\mathcal X}(y)-y)\le0(ΠX​(y)−x)⊤(ΠX​(y)−y)≤0, and hence ∥ΠX(y)−x∥2+∥y−ΠX(y)∥2≤∥y−x∥2\|\Pi_{\mathcal X}(y)-x\|^2+\|y-\Pi_{\mathcal X}(y)\|^2\le\|y-x\|^2∥ΠX​(y)−x∥2+∥y−ΠX​(y)∥2≤∥y−x∥2. The first claim is an already proved platform theorem.
  2. (3.7) (p. 270): ∇f(x)⊤(x+−y)≤gX(x)⊤(x+−y)\nabla f(x)^\top(x^+-y)\le g_{\mathcal X}(x)^\top(x^+-y)∇f(x)⊤(x+−y)≤gX​(x)⊤(x+−y) for y∈Xy\in\mathcal Xy∈X.
  3. Lemma 3.6 (p. 270): for x,y∈Xx,y\in\mathcal Xx,y∈X,
f(x+)−f(y)≤gX(x)⊤(x−y)−12β∥gX(x)∥2.f(x^+)-f(y)\le g_{\mathcal X}(x)^\top(x-y)-\frac1{2\beta}\|g_{\mathcal X}(x)\|^2 .f(x+)−f(y)≤gX​(x)⊤(x−y)−2β1​∥gX​(x)∥2.
  1. Descent and gap (pp. 270–271): f(xs+1)−f(xs)≤−12β∥gX(xs)∥2f(x_{s+1})-f(x_s)\le-\frac1{2\beta}\|g_{\mathcal X}(x_s)\|^2f(xs+1​)−f(xs​)≤−2β1​∥gX​(xs​)∥2 and f(xs+1)−f(x∗)≤∥gX(xs)∥ ∥xs−x∗∥f(x_{s+1})-f(x^*)\le\|g_{\mathcal X}(x_s)\|\,\|x_s-x^*\|f(xs+1​)−f(x∗)≤∥gX​(xs​)∥∥xs​−x∗∥.
  2. Distances decrease (p. 271): ∥xs+1−x∗∥≤∥xs−x∗∥\|x_{s+1}-x^*\|\le\|x_s-x^*\|∥xs+1​−x∗∥≤∥xs​−x∗∥.
  3. Recursion and induction (p. 271): with δs=f(xs)−f(x∗)\delta_s=f(x_s)-f(x^*)δs​=f(xs​)−f(x∗), δs+1≤δs−12β∥x1−x∗∥2δs+12\delta_{s+1}\le\delta_s-\frac{1}{2\beta\|x_1-x^*\|^2}\delta_{s+1}^2δs+1​≤δs​−2β∥x1​−x∗∥21​δs+12​, and an induction on this recursion gives the bound of Theorem 3.7.

Companion: Theorem 3.10 (p. 278)

If fff is in addition α\alphaα-strongly convex on X\mathcal XX, with condition number κ=β/α\kappa=\beta/\alphaκ=β/α, then the strengthened Lemma 3.6, display (3.14), gives for every t≥0t\ge0t≥0

∥xt+1−x∗∥2≤exp⁡(−tκ)∥x1−x∗∥2.\|x_{t+1}-x^*\|^2\le\exp\Bigl(-\frac t\kappa\Bigr)\|x_1-x^*\|^2 .∥xt+1​−x∗∥2≤exp(−κt​)∥x1​−x∗∥2.

Significance

Theorem 3.7 shows that a constraint set with a cheap projection costs nothing in the order of convergence: the O(β∥x1−x∗∥2/t)O(\beta\|x_1-x^*\|^2/t)O(β∥x1​−x∗∥2/t) rate of unconstrained gradient descent survives, with a constant that depends on f(x1)−f(x∗)f(x_1)-f(x^*)f(x1​)−f(x∗). 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 exp⁡(−t/κ)\exp(-t/\kappa)exp(−t/κ). This mission adds machine-checked statements of the constrained analysis in Bubeck's formulation, with the book's constants. The definitions of β\betaβ-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 f(x−1β∇f(x))≤f(x)−12β∥∇f(x)∥2f(x-\frac1\beta\nabla f(x))\le f(x)-\frac1{2\beta}\|\nabla f(x)\|^2f(x−β1​∇f(x))≤f(x)−2β1​∥∇f(x)∥2. That inequality fails under constraints: the projection can cut the step short, so the decrease in fff is not controlled by ∥∇f(x)∥\|\nabla f(x)\|∥∇f(x)∥. At a boundary minimizer, for instance, ∇f(x∗)≠0\nabla f(x^*)\neq0∇f(x∗)=0 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 x∗x^*x∗, 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 sss to s+1s+1s+1 works only from s=2s=2s=2 on, and the step from 111 to 222 needs a direct computation that uses the term f(x1)−f(x∗)f(x_1)-f(x^*)f(x1​)−f(x∗) in the numerator.

Formalization scope

  • Space. Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n), and x⊤yx^\top yx⊤y is ⟪x, y⟫_ℝ.
  • Constraint set and minimizer. Every statement assumes X\mathcal XX compact (IsCompact X) and convex (Convex ℝ X), the standing assumption of Chapter 3. Every statement about a run assumes a minimizer x∗∈Xx^*\in\mathcal Xx∗∈X with f(x∗)≤f(y)f(x^*)\le f(y)f(x∗)≤f(y) for all y∈Xy\in\mathcal Xy∈X, the monograph's standing assumption.
  • Gradient. The gradient is an explicit map g with HasGradientAt f (g x) x at every point of Rn\mathbb R^nRn, so ∇f(x)\nabla f(x)∇f(x) is defined at boundary points of X\mathcal XX. β\betaβ-smoothness on X\mathcal XX (IsBetaSmoothOn X f g β) asks this together with β≥0\beta\ge0β≥0 and the Lipschitz bound on X\mathcal XX 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 α>0\alpha>0α>0.
  • Projection. The projection is the published relation IsMetricProjection X y p: p∈Xp\in\mathcal Xp∈X and ∥y−p∥≤∥y−z∥\|y-p\|\le\|y-z\|∥y−p∥≤∥y−z∥ for every z∈Xz\in\mathcal Xz∈X. No choice function is used.
  • Run. A run is IsProjGDRun X g β x: β>0\beta>0β>0, x1∈Xx_1\in\mathcal Xx1​∈X, and for every t≥1t\ge1t≥1 the iterate xt+1x_{t+1}xt+1​ is a projection of xt−β−1g(xt)x_t-\beta^{-1}g(x_t)xt​−β−1g(xt​). Indices start at 111 as in the book, and x 0 is unused.
  • Gradient mapping. gradMap β x xplus =β(x−x+)=\beta(x-x^+)=β(x−x+) takes the projected point of the run as an argument.
  • Positivity. β>0\beta>0β>0 is assumed, as the step size 1/β1/\beta1/β requires, and α>0\alpha>0α>0 for Theorem 3.10, as κ=β/α\kappa=\beta/\alphaκ=β/α requires.
  • The recursion. The recursion of milestone 6 is stated multiplied by 2β∥x1−x∗∥22\beta\|x_1-x^*\|^22β∥x1​−x∗∥2, so the case x1=x∗x_1=x^*x1​=x∗ needs no division.
  • No trivial readings. A run predicate with a free step size, or β\betaβ-smoothness replaced by the quadratic bound (3.4), would change the content. Both are excluded: the step is fixed to 1/β1/\beta1/β, 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
12 thms4 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity VI: Gradient Descent with η = 2/(α + β) on a β-Smooth α-Strongly Convex Function Has Rate (β/2)exp(−4t/(κ + 1))‖x₁ − x*‖²Textbook

Motivation

Gradient descent is a basic method for minimizing a differentiable function when evaluating its gradient is practical but solving the optimization problem directly is not. The rate at which its iterates approach an optimizer depends on the assumptions about the function. For a convex function with a Lipschitz gradient, the value error decreases at a sublinear rate. Adding strong convexity changes the behavior: the distance from the optimizer contracts at each step, giving an exponential bound on the value error. This section of Bubeck's monograph identifies a fixed step size that uses both the smoothness and curvature constants and gives the corresponding rate.

The result matters when a high-accuracy answer is needed. A sublinear bound makes each extra digit progressively more expensive; an exponential bound says that a fixed number of additional gradient evaluations reduces the error by a fixed factor. The theorem is a textbook result, already proved mathematically. This mission asks for its precise machine-checked statement and the source's supporting inequalities, rather than for a new optimization method.

Setting

Work in Euclidean space Rn\mathbb R^nRn with n≥1n\ge1n≥1, equipped with its usual inner product and norm. A differentiable function f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R has gradient g(x)=∇f(x)g(x)=\nabla f(x)g(x)=∇f(x). It is β\betaβ-smooth when its gradient is β\betaβ-Lipschitz: ∥g(x)−g(y)∥≤β∥x−y∥\|g(x)-g(y)\|\le\beta\|x-y\|∥g(x)−g(y)∥≤β∥x−y∥ for every x,yx,yx,y. It is α\alphaα-strongly convex when, for every x,yx,yx,y,

f(y)≥f(x)+⟨g(x),y−x⟩+α2∥y−x∥2.f(y)\ge f(x)+\langle g(x),y-x\rangle+\frac\alpha2\|y-x\|^2.f(y)≥f(x)+⟨g(x),y−x⟩+2α​∥y−x∥2.

The first condition limits how rapidly the gradient changes. The second gives a quadratic lower bound on the function around any point. Here α>0\alpha>0α>0 and β≥0\beta\ge0β≥0. In positive dimension, the two conditions together entail β≥α\beta\ge\alphaβ≥α, so the condition number κ=β/α\kappa=\beta/\alphaκ=β/α is at least one. The case α=β\alpha=\betaα=β remains part of the target.

A point x∗x^*x∗ is a global minimizer when f(x∗)≤f(y)f(x^*)\le f(y)f(x∗)≤f(y) for every yyy. The book assumes such a point exists as a standing convention. A gradient descent run is a sequence (xt)t≥1(x_t)_{t\ge1}(xt​)t≥1​ satisfying xt+1=xt−ηg(xt)x_{t+1}=x_t-\eta g(x_t)xt+1​=xt​−ηg(xt​) at each positive index. Its first iterate x1x_1x1​ is arbitrary. The step size in this mission is fixed at η=2/(α+β)\eta=2/(\alpha+\beta)η=2/(α+β), rather than chosen by line search or adapted along the run.

Formalization targets

The central target is Theorem 3.12 of Bubeck, p. 279. For every integer t≥0t\ge0t≥0, the gradient descent run satisfies

f(xt+1)−f(x∗)≤β2exp⁡ ⁣(−4tκ+1)∥x1−x∗∥2.f(x_{t+1})-f(x^*)\le \frac\beta2\exp\!\left(-\frac{4t}{\kappa+1}\right)\|x_1-x^*\|^2.f(xt+1​)−f(x∗)≤2β​exp(−κ+14t​)∥x1​−x∗∥2.

At t=0t=0t=0 this is a smoothness bound on the initial value gap. For subsequent iterations it gives a linear convergence rate with the explicit exponential factor stated in the book. No initial-radius bound or bounded domain is imposed: the actual squared distance ∥x1−x∗∥2\|x_1-x^*\|^2∥x1​−x∗∥2 appears in the conclusion.

The milestones trace the mathematical claims stated in the source. Equation (3.6) is the co-coercivity inequality for gradients of convex smooth functions. The proof of Lemma 3.11 introduces ϕ(z)=f(z)−(α/2)∥z∥2\phi(z)=f(z)-(\alpha/2)\|z\|^2ϕ(z)=f(z)−(α/2)∥z∥2, and identifies it as convex and (β−α)(\beta-\alpha)(β−α)-smooth. Lemma 3.11 combines curvature and smoothness into a sharper inequality for two gradients. The proof of Theorem 3.12 then gives a value-gap bound, a one-step distance contraction, and its iterated exponential form. These statements are separately useful: the co-coercivity and contraction bounds can be reused in analyses of related first-order methods.

Significance

The theorem states a complete guarantee for the algorithm: an explicit rule, the hypotheses on the objective, and a bound valid for every iteration count. It makes the role of κ\kappaκ visible. When κ\kappaκ is close to one, the contraction is strong; when the smoothness constant is much larger than the curvature constant, more iterations are needed for the same error reduction. The stated dependence supports comparisons with projected and accelerated gradient methods elsewhere in the same monograph.

Formalizing the result requires a common interface for actual gradients, smoothness, strong convexity, and algorithm runs. The strong-convexity predicate is an existing published definition, while the local smoothness and run definitions use Bubeck's conventions. Once these interfaces and the inequalities are proved, later missions can use the resulting declarations to compare rates without translating between informal meanings of “smooth” or changing the iterate index. The source provides a mathematical proof; these draft Lean theorems carry sorry and do not yet constitute machine-checked proofs.

Difficulty

The main issue is getting the sharp contraction factor from two assumptions that control different parts of the gradient step. A direct Lipschitz estimate on the update map does not by itself express the mixed inner-product term with the constants needed for the stated factor. Lemma 3.11 is the source's precise bridge between the gradient difference, the point displacement, and their inner product. The case α=β\alpha=\betaα=β also needs to remain valid: a proof route that divides by β−α\beta-\alphaβ−α cannot cover that boundary by the same calculation.

The last display in the proof of Theorem 3.12 joins a one-step inequality involving xtx_txt​ to an exponential inequality involving x1x_1x1​. The latter is the cumulative statement after ttt steps. Keeping these as separate milestones makes each quantified claim explicit while preserving the theorem's bound.

Formalization scope

Lean represents Rn\mathbb R^nRn as EuclideanSpace ℝ (Fin n), with n>0n>0n>0. The gradient is an explicit map ggg required to be the actual gradient of fff at every point. Smoothness is the gradient Lipschitz condition, not a quadratic upper bound used as a definition. Strong convexity uses the published OnlineConvexOpt.ConvexBasics.StronglyConvexOn predicate on the whole space; its formula is the book's (3.13). Iterates are indexed from one, and index zero imposes no condition. The norm, inner product, constants, and real exponential follow the printed formulas.

The added explicit conditions are n>0n>0n>0, α>0\alpha>0α>0, and β>0\beta>0β>0 where Equation (3.6) divides by β\betaβ. Positive dimension excludes a degenerate space where curvature imposes no restriction on smoothness. The positivity of α\alphaα makes κ\kappaκ meaningful; the source treats it as a positive strong-convexity parameter. The minimizer hypothesis is the book's standing convention. There is no assumption that g(x∗)=0g(x^*)=0g(x∗)=0: that property follows from global minimality and differentiability. A gradient map unrelated to fff would trivialize the model, so the smoothness definition includes the gradient identity.

The local development needs Euclidean inner-product identities, convexity, differentiability, Lipschitz gradient bounds, and real exponential estimates. The auxiliary function and the two co-coercivity inequalities are reusable outside this chapter. Contributions should prove the exact milestone statements and the final theorem, including t=0t=0t=0 and α=β\alpha=\betaα=β, without weakening constants or substituting another gradient descent step.

Selected references

  • Sébastien Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. arXiv:1405.4980v2; DOI:10.1561/2200000050.
9 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity VII: Conditional Gradient Descent (Frank–Wolfe) with γ_s = 2/(s + 1) Has Rate 2βR²/(t + 1) in Any NormTextbook

Motivation

Many constrained optimization problems in machine learning and statistics have a feasible set X\mathcal XX over which a linear function is cheap to minimize but a Euclidean projection is expensive: the ℓ1\ell_1ℓ1​-ball, the simplex, the nuclear-norm ball, the convex hull of a combinatorial family. Projected gradient descent needs a projection at every step. Conditional gradient descent, introduced by Frank and Wolfe in 1956 for quadratic programming, replaces the projection by a call to a linear minimization oracle over X\mathcal XX, and its iterates are convex combinations of oracle outputs, which makes them sparse when X\mathcal XX is a polytope.

This mission is the seventh of a series formalizing S. Bubeck's monograph Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning, 2015, arXiv:1405.4980v2). It covers Section 3.3, whose main result, Theorem 3.8, is the O(1/t)O(1/t)O(1/t) rate of the method in the form given by Jaggi (2013), going back to Dunn and Harshbarger (1978).

Setting

Let EEE be a finite-dimensional real vector space with an arbitrary norm ∥⋅∥\|\cdot\|∥⋅∥. For a linear form ggg on EEE, written v↦g⊤vv\mapsto g^\top vv↦g⊤v, the dual norm is ∥g∥∗=sup⁡∥v∥≤1g⊤v\|g\|_*=\sup_{\|v\|\le1}g^\top v∥g∥∗​=sup∥v∥≤1​g⊤v. Let X⊆E\mathcal X\subseteq EX⊆E be nonempty, compact and convex, with diameter R=sup⁡x,y∈X∥x−y∥R=\sup_{x,y\in\mathcal X}\|x-y\|R=supx,y∈X​∥x−y∥.

Let f:E→Rf:E\to\mathbb Rf:E→R be differentiable with gradient ∇f(x)\nabla f(x)∇f(x), a linear form on EEE. For β≥0\beta\ge0β≥0, fff is β-smooth with respect to ∥⋅∥\|\cdot\|∥⋅∥ on X\mathcal XX if

∥∇f(x)−∇f(y)∥∗≤β∥x−y∥(x,y∈X).\|\nabla f(x)-\nabla f(y)\|_*\le\beta\|x-y\|\qquad(x,y\in\mathcal X).∥∇f(x)−∇f(y)∥∗​≤β∥x−y∥(x,y∈X).

A point x∗∈Xx^*\in\mathcal Xx∗∈X with f(x∗)=min⁡x∈Xf(x)f(x^*)=\min_{x\in\mathcal X}f(x)f(x∗)=minx∈X​f(x) is fixed throughout, and δt=f(xt)−f(x∗)\delta_t=f(x_t)-f(x^*)δt​=f(xt​)−f(x∗).

Given step sizes (γs)s≥1(\gamma_s)_{s\ge1}(γs​)s≥1​, a run of conditional gradient descent is a pair of sequences with x1∈Xx_1\in\mathcal Xx1​∈X and, for every t≥1t\ge1t≥1,

yt∈argmin⁡y∈X∇f(xt)⊤y(3.8),xt+1=(1−γt)xt+γtyt(3.9).y_t\in\operatorname*{argmin}_{y\in\mathcal X}\nabla f(x_t)^\top y\quad(3.8),\qquad x_{t+1}=(1-\gamma_t)x_t+\gamma_ty_t\quad(3.9).yt​∈y∈Xargmin​∇f(xt​)⊤y(3.8),xt+1​=(1−γt​)xt​+γt​yt​(3.9).

The minimizer yty_tyt​ need not be unique; any choice is allowed.

Formalization targets

Goal: Theorem 3.8 (p. 272)

If fff is convex and β\betaβ-smooth with respect to ∥⋅∥\|\cdot\|∥⋅∥ and γs=2s+1\gamma_s=\frac{2}{s+1}γs​=s+12​ for s≥1s\ge1s≥1, then every run satisfies, for every t≥2t\ge2t≥2,

f(xt)−f(x∗)≤2βR2t+1.f(x_t)-f(x^*)\le\frac{2\beta R^2}{t+1}.f(xt​)−f(x∗)≤t+12βR2​.

Milestones

  1. Inequality (3.4) in an arbitrary norm (p. 267, used on p. 272): for x,y∈Xx,y\in\mathcal Xx,y∈X, 0≤f(x)−f(y)−∇f(y)⊤(x−y)≤β2∥x−y∥20\le f(x)-f(y)-\nabla f(y)^\top(x-y)\le\frac{\beta}{2}\|x-y\|^20≤f(x)−f(y)−∇f(y)⊤(x−y)≤2β​∥x−y∥2.
  2. The one-step recursion (pp. 272–273): for any run with γs∈[0,1]\gamma_s\in[0,1]γs​∈[0,1],
δs+1≤(1−γs)δs+β2γs2R2.\delta_{s+1}\le(1-\gamma_s)\delta_s+\frac{\beta}{2}\gamma_s^2R^2 .δs+1​≤(1−γs​)δs​+2β​γs2​R2.
  1. Initialization (p. 273): if γ1=1\gamma_1=1γ1​=1, then δ2≤β2R2\delta_2\le\frac{\beta}{2}R^2δ2​≤2β​R2.
  2. The induction (p. 273): a real sequence with δ2≤β2R2\delta_2\le\frac{\beta}{2}R^2δ2​≤2β​R2 and the recursion of milestone 2 with γs=2s+1\gamma_s=\frac{2}{s+1}γs​=s+12​ for s≥2s\ge2s≥2 satisfies δt≤2βR2t+1\delta_t\le\frac{2\beta R^2}{t+1}δt​≤t+12βR2​ for t≥2t\ge2t≥2.

Significance

Theorem 3.8 is the basic guarantee for projection-free first-order optimization. Its rate does not depend on the dimension, and it depends on the geometry only through the product βR2\beta R^2βR2, both measured in the same norm, which may be chosen to fit X\mathcal XX: for the ℓ1\ell_1ℓ1​-ball, smoothness in ∥⋅∥1\|\cdot\|_1∥⋅∥1​ with dual norm ∥⋅∥∞\|\cdot\|_\infty∥⋅∥∞​ gives much better constants than the Euclidean analysis. The book applies it this way to a LASSO-type problem (Section 3.3, pp. 274–276), and the same bound underlies the sparse-approximation corollary on the simplex (p. 273) and the many later variants of the method (away steps, stochastic and online conditional gradient).

The result is classical and proved in the book. What this mission adds is a machine-checked statement and proof in an arbitrary finite-dimensional normed space, with the dual norm as the operator norm on linear forms, and a reusable encoding of norm-smoothness and of conditional gradient runs. On Prove2Me a related result is already proved: Lan's Theorem 7.1 (First-order and Stochastic Optimization Methods), which bounds f(yk)−f∗f(y_k)-f^*f(yk​)−f∗ by 2Lk(k+1)∑i≤k∥xi−yi−1∥2\frac{2L}{k(k+1)}\sum_{i\le k}\|x_i-y_{i-1}\|^2k(k+1)2L​∑i≤k​∥xi​−yi−1​∥2 with a different indexing; after the diameter bound it yields 2βR2/t2\beta R^2/t2βR2/t at Bubeck's iterate xtx_txt​, which is weaker than Theorem 3.8 by one step.

Difficulty

The difficulty is in the bookkeeping of norms and indices, not in a deep idea. Inequality (3.4) is proved in the book only for the Euclidean norm, where ∇f(x)∈Rn\nabla f(x)\in\mathbb R^n∇f(x)∈Rn and the Cauchy–Schwarz inequality is used; in a general norm it needs the pairing between a linear form and a vector and the bound ∣g⊤v∣≤∥g∥∗∥v∥|g^\top v|\le\|g\|_*\|v\|∣g⊤v∣≤∥g∥∗​∥v∥. The rate 2βR2/(t+1)2\beta R^2/(t+1)2βR2/(t+1) is attained only by starting the induction at t=2t=2t=2, where the step γ1=1\gamma_1=1γ1​=1 erases the dependence on the starting point; at t=1t=1t=1 the bound can fail, since δ1\delta_1δ1​ is arbitrary. A first attempt that runs the induction from t=1t=1t=1 with an arbitrary δ1\delta_1δ1​ does not give the stated constant.

Formalization scope

EEE is a type with [NormedAddCommGroup E] [NormedSpace ℝ E] [FiniteDimensional ℝ E]; nothing is specialised to the Euclidean norm. The gradient is an explicit derivative map f' : E → (E →L[ℝ] ℝ) with HasFDerivAt f (f' x) x for every x; ∇f(x)⊤v\nabla f(x)^\top v∇f(x)⊤v is f' x v, and ∥⋅∥∗\|\cdot\|_*∥⋅∥∗​ is the operator norm, which equals sup⁡∥v∥≤1g⊤v\sup_{\|v\|\le1}g^\top vsup∥v∥≤1​g⊤v. RRR is Metric.diam X, which equals the supremum of ∥x−y∥\|x-y\|∥x−y∥ over X\mathcal XX because X\mathcal XX is compact. Sequences are indexed by ℕ with the first iterate at index 1.

Committed conventions: X\mathcal XX compact, convex and containing x∗x^*x∗ (hence nonempty); convexity of fff and the Lipschitz bound on the gradient are assumed on X\mathcal XX only, which is weaker than the book's global assumptions; β≥0\beta\ge0β≥0; the existence of the minimizer x∗x^*x∗ is the book's standing assumption (p. 242); the conclusion is stated for t≥2t\ge2t≥2, as in the book. Runs are predicates: yty_tyt​ is any minimizer of the linear form over X\mathcal XX and yt∈Xy_t\in\mathcal Xyt​∈X is required, so the goal quantifies over every run with γs=2/(s+1)\gamma_s=2/(s+1)γs​=2/(s+1). A formalization in which yty_tyt​ need not lie in X\mathcal XX, or in which smoothness is assumed only along the iterates, states a different theorem and is excluded.

A complete development needs the descent inequality (3.4) for Fréchet derivatives in a normed space (reusable for every smooth method in the series and beyond), the fact that the iterates stay in X\mathcal XX, and a scalar induction. Proofs of the milestones are welcome independently; milestone 4 is a statement about real sequences only.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. https://arxiv.org/abs/1405.4980
  • M. Frank and P. Wolfe, An algorithm for quadratic programming, Naval Research Logistics Quarterly 3(1–2):95–110, 1956. https://doi.org/10.1002/nav.3800030109
  • J. C. Dunn and S. Harshbarger, Conditional gradient algorithms with open loop step size rules, Journal of Mathematical Analysis and Applications 62(2):432–444, 1978. https://doi.org/10.1016/0022-247X(78)90137-3
  • M. Jaggi, Revisiting Frank–Wolfe: projection-free sparse convex optimization, Proceedings of ICML 2013, PMLR 28(1):427–435. https://proceedings.mlr.press/v28/jaggi13.html
  • G. Lan, First-order and Stochastic Optimization Methods for Machine Learning, Springer, 2020, Theorem 7.1. https://doi.org/10.1007/978-3-030-39568-1
6 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity VIII: No Black-Box Method Beats 3β‖x₁ − x*‖²/(32(t + 1)²) on β-Smooth Convex FunctionsTextbook

Why lower bounds for first-order methods

Upper bounds for an optimization method say how fast it converges; oracle complexity lower bounds say how fast any method of a given kind can possibly converge. Chapter 3 of S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning 8(3–4), 2015, arXiv:1405.4980) proves upper bounds for subgradient descent on Lipschitz functions and for gradient methods on smooth functions. Section 3.5 (Lower bounds, pp. 279–283) shows that these rates cannot be improved by more than a numerical constant, as long as the number of queries is smaller than the dimension. For smooth convex functions the matching lower bound is what identifies Nesterov's accelerated gradient descent, with its 1/t21/t^21/t2 rate, as an optimal method.

Timeline. The lower bounds first appeared in A. Nemirovski and D. Yudin, Problem Complexity and Method Efficiency in Optimization (Wiley, 1983). The presentation followed by the book, with an explicit tridiagonal quadratic as the hard instance and the "span of past gradients" restriction on the method, is that of Y. Nesterov, Introductory Lectures on Convex Optimization (Kluwer, 2004), §2.1.2. Nesterov's accelerated method (1983) attains the matching upper bound.

Setting

Work in Rn\mathbb R^nRn with the Euclidean inner product x⊤yx^\top yx⊤y, coordinates x(1),…,x(n)x(1),\dots,x(n)x(1),…,x(n), canonical basis e1,…,ene_1,\dots,e_ne1​,…,en​ and balls B2(R)={x:∥x∥≤R}\mathrm B_2(R)=\{x:\|x\|\le R\}B2​(R)={x:∥x∥≤R}. A first-order oracle for fff answers a query xxx with a subgradient g∈∂f(x)g\in\partial f(x)g∈∂f(x) (the gradient when fff is differentiable). A black-box procedure maps the history (x1,g1,…,xt,gt)(x_1,g_1,\dots,x_t,g_t)(x1​,g1​,…,xt​,gt​) to the next query xt+1x_{t+1}xt+1​. Section 3.5 restricts attention to procedures with

x1=0,xt+1∈Span(g1,…,gt)(t≥0),(3.15)x_1=0,\qquad x_{t+1}\in\mathrm{Span}(g_1,\dots,g_t)\quad(t\ge0), \tag{3.15}x1​=0,xt+1​∈Span(g1​,…,gt​)(t≥0),(3.15)

which covers gradient descent, its accelerated variants and conjugate gradient. A function is β\betaβ-smooth if its gradient is β\betaβ-Lipschitz; LLL-Lipschitz on X\mathcal XX if every subgradient at every point of X\mathcal XX has norm at most LLL; α\alphaα-strongly convex if x↦f(x)−α2∥x∥2x\mapsto f(x)-\frac\alpha2\|x\|^2x↦f(x)−2α​∥x∥2 is convex.

The hard smooth instance uses, for k≤nk\le nk≤n, the symmetric tridiagonal matrix AkA_kAk​ with entries 222 on the first kkk diagonal positions and −1-1−1 on the neighbouring off-diagonal positions of the leading k×kk\times kk×k block, zero elsewhere, and the quadratics

fk(x)=β8x⊤Akx−β4x⊤e1,fk∗=inf⁡x∈Rnfk(x).f_k(x)=\frac\beta8x^\top A_kx-\frac\beta4x^\top e_1 ,\qquad f_k^*=\inf_{x\in\mathbb R^n}f_k(x).fk​(x)=8β​x⊤Ak​x−4β​x⊤e1​,fk∗​=x∈Rninf​fk​(x).

Formalization targets

Goal: Theorem 3.14 (p. 282)

For 1≤t≤n−121\le t\le\frac{n-1}21≤t≤2n−1​ and β>0\beta>0β>0 there are a β\betaβ-smooth convex fff and a minimizer x∗x^*x∗ such that every procedure satisfying (3.15) has

min⁡1≤s≤tf(xs)−f(x∗) ≥ 3β32 ∥x1−x∗∥2(t+1)2.\min_{1\le s\le t}f(x_s)-f(x^*)\ \ge\ \frac{3\beta}{32}\,\frac{\|x_1-x^*\|^2}{(t+1)^2}.1≤s≤tmin​f(xs​)−f(x∗) ≥ 323β​(t+1)2∥x1​−x∗∥2​.

The constant 3/323/323/32 is the book's.

Milestones (proof of Theorem 3.14, pp. 282–283)

  1. 0⪯Ak⪯4In0\preceq A_k\preceq4I_n0⪯Ak​⪯4In​, through x⊤Akx=x(1)2+x(k)2+∑i=1k−1(x(i)−x(i+1))2x^\top A_kx=x(1)^2+x(k)^2+\sum_{i=1}^{k-1}(x(i)-x(i+1))^2x⊤Ak​x=x(1)2+x(k)2+∑i=1k−1​(x(i)−x(i+1))2.
  2. For f=f2t+1f=f_{2t+1}f=f2t+1​ and any procedure satisfying (3.15), xs∈Span(e1,…,es−1)x_s\in\mathrm{Span}(e_1,\dots,e_{s-1})xs​∈Span(e1​,…,es−1​); hence f(xs)=fs(xs)f(x_s)=f_s(x_s)f(xs​)=fs​(xs​) for s≤ts\le ts≤t.
  3. xk∗(i)=1−ik+1x_k^*(i)=1-\frac i{k+1}xk∗​(i)=1−k+1i​ solves Akx=e1A_kx=e_1Ak​x=e1​, minimizes fkf_kfk​, and fk∗=−β8(1−1k+1)f_k^*=-\frac\beta8\bigl(1-\frac1{k+1}\bigr)fk∗​=−8β​(1−k+11​).
  4. ∥xk∗∥2≤k+13\|x_k^*\|^2\le\frac{k+1}3∥xk∗​∥2≤3k+1​.
  5. ft∗−f2t+1∗=β8(1t+1−12t+2)≥3β32∥x2t+1∗∥2(t+1)2f_t^*-f_{2t+1}^*=\frac\beta8\bigl(\frac1{t+1}-\frac1{2t+2}\bigr)\ge\frac{3\beta}{32}\frac{\|x^*_{2t+1}\|^2}{(t+1)^2}ft∗​−f2t+1∗​=8β​(t+11​−2t+21​)≥323β​(t+1)2∥x2t+1∗​∥2​.

Companion: Theorem 3.13 (p. 280)

For 1≤t≤n1\le t\le n1≤t≤n and L,R>0L,R>0L,R>0 there are a convex fff, LLL-Lipschitz on B2(R)\mathrm B_2(R)B2​(R), and a first-order oracle for it such that every procedure satisfying (3.15) has min⁡s≤tf(xs)−min⁡B2(R)f≥RL2(1+t)\min_{s\le t}f(x_s)-\min_{\mathrm B_2(R)}f\ge\frac{RL}{2(1+\sqrt t)}mins≤t​f(xs​)−minB2​(R)​f≥2(1+t​)RL​; and for α>0\alpha>0α>0 there are an α\alphaα-strongly convex fff, LLL-Lipschitz on B2(L2α)\mathrm B_2(\frac L{2\alpha})B2​(2αL​), and an oracle with gap at least L28αt\frac{L^2}{8\alpha t}8αtL2​ over that ball.

Significance

The upper bounds of Chapter 3 (projected subgradient descent at rate RL/tRL/\sqrt tRL/t​, accelerated gradient descent at rate β∥x1−x∗∥2/t2\beta\|x_1-x^*\|^2/t^2β∥x1​−x∗∥2/t2) become optimal statements only through these lower bounds: no method in the class (3.15) can be faster by more than a constant factor while ttt is below the dimension. The restriction to t≲nt\lesssim nt≲n is necessary, since Chapter 2's cutting-plane methods converge exponentially once the number of queries exceeds the dimension.

The results are classical and proved. Formalizing them yields machine-checked versions of the quadratic-form computation for the tridiagonal matrix, of the Krylov-type support argument under (3.15), and of the explicit minimizer of fkf_kfk​, each reusable in other lower-bound arguments (Theorem 3.15 in ℓ2\ell_2ℓ2​, lower bounds for strongly convex smooth functions, conjugate gradient analyses). The platform held no formal statement of these oracle lower bounds when this mission was drafted.

Difficulty

Each analytic step is elementary; the difficulty is in the bookkeeping. The span argument is an induction that must track, at each step, that the gradient of a tridiagonal quadratic at a vector supported on the first s−1s-1s−1 coordinates is supported on the first sss, and that the span hypothesis transfers this to the next query. The minimizer computation requires solving Akx=e1A_kx=e_1Ak​x=e1​ on the leading block and showing that the coordinates beyond kkk do not affect fkf_kfk​. A natural first attempt, choosing the hard function after seeing the procedure, proves a much weaker statement and is excluded by the quantifier order: the function is fixed first and must defeat every procedure.

Formalization scope

Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n). Coordinates in Lean are 0-based; the definitions provide the book's 1-based coordinate coord x i and basis vector basisVec n i, and the matrix tridiag n k translates book index iii to Fin n index i−1i-1i−1. The query sequence starts at index 111. The oracle is a fixed map ggg, so (3.15) reads xt+1∈Span(g(x1),…,g(xt))x_{t+1}\in\mathrm{Span}(g(x_1),\dots,g(x_t))xt+1​∈Span(g(x1​),…,g(xt​)) with x1=0x_1=0x1​=0. In Theorem 3.14 the oracle is the gradient, given as a map with HasGradientAt everywhere; β\betaβ-smoothness is the Lipschitz bound on that map. In Theorem 3.13 the oracle is part of what is constructed, because the book's proof uses a specific "resisting" subgradient selection and the claim fails for an arbitrary one.

Committed conventions, each stated in the item's Formalization Note: the minimum over 1≤s≤t1\le s\le t1≤s≤t is the bound for every such sss, and t≥1t\ge1t≥1 is required; t≤(n−1)/2t\le(n-1)/2t≤(n−1)/2 is 2t+1≤n2t+1\le n2t+1≤n; the minimizer x∗x^*x∗ is existentially chosen together with fff (the hard function has many minimizers when 2t+1<n2t+1<n2t+1<n, and the bound is false for some of them); the minimum over a ball is the bound against every point of the ball; fk∗f_k^*fk∗​ is the real infimum, asserted to be attained.

A formalization that let the function depend on the procedure, dropped x1=0x_1=0x1​=0, or took the span over gradients at points other than the queries would state a different and weaker theorem; the statements here keep fff (and the oracle) before the universally quantified procedure.

Infrastructure needed: quadratic forms of explicit matrices on EuclideanSpace, gradients of quadratics, and span/support lemmas for EuclideanSpace.single-type vectors. Contributions of proofs for any milestone, and of the strongly convex ℓ2\ell_2ℓ2​ lower bound (Theorem 3.15, not included here), are welcome.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. https://arxiv.org/abs/1405.4980
  • A. Nemirovski and D. Yudin, Problem Complexity and Method Efficiency in Optimization, Wiley, 1983.
  • Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer, 2004. https://doi.org/10.1007/978-1-4419-8853-9
8 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity IX: Nesterov's Accelerated Gradient Descent on a Convex β-Smooth Function Has Rate 2β‖x₁ − x*‖²/t²Textbook

Motivation

Minimizing a convex function f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R whose gradient is Lipschitz is the basic problem of first-order optimization, and it is the regime in which most large-scale methods of machine learning, signal processing and operations research are analysed. The natural method, gradient descent, reaches accuracy ε\varepsilonε after O(1/ε)O(1/\varepsilon)O(1/ε) gradient evaluations. In 1983 Nesterov showed that a method using the same oracle, but combining the current and the previous iterate, reaches accuracy ε\varepsilonε after only O(1/ε)O(1/\sqrt\varepsilon)O(1/ε​) evaluations, and that no method using only gradient information can do better by more than a constant factor. This accelerated gradient descent and its proximal variants (FISTA) are now the default fast first-order methods for smooth and composite convex problems.

This mission formalizes the accelerated rate as presented in S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning, 2015; arXiv:1405.4980v2), §3.7.2, Theorem 3.19, whose proof follows Beck and Teboulle (2009).

Timeline.

  • 1983: Nesterov introduces the accelerated method with rate O(1/t2)O(1/t^2)O(1/t2) for convex functions with Lipschitz gradient (Soviet Math. Dokl. 27).
  • 1983: Nemirovski and Yudin's black-box lower bounds show that Ω(1/t2)\Omega(1/t^2)Ω(1/t2) is the best possible rate for this class (Theorem 3.14 of the book gives 3β∥x1−x∗∥2/(32(t+1)2)3\beta\|x_1-x^*\|^2/(32(t+1)^2)3β∥x1​−x∗∥2/(32(t+1)2)).
  • 2009: Beck and Teboulle's FISTA extends the method, with the same step sequence λt\lambda_tλt​, to composite problems with a simple nonsmooth term (SIAM J. Imaging Sci. 2(1)).
  • 2008: Tseng gives a unified treatment with simpler step sizes (manuscript).

Setting

Let Rn\mathbb R^nRn carry the Euclidean inner product x⊤yx^\top yx⊤y and norm ∥⋅∥\|\cdot\|∥⋅∥. A differentiable function f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R is β-smooth if its gradient is β\betaβ-Lipschitz:

∥∇f(x)−∇f(y)∥≤β∥x−y∥(x,y∈Rn).\|\nabla f(x)-\nabla f(y)\|\le\beta\|x-y\|\qquad(x,y\in\mathbb R^n).∥∇f(x)−∇f(y)∥≤β∥x−y∥(x,y∈Rn).

Assume fff is convex and β\betaβ-smooth with β>0\beta>0β>0, and let x∗x^*x∗ be a minimizer of fff.

Define the step sequences λ0=0\lambda_0=0λ0​=0,

λt=1+1+4λt−122(t≥1),γt=1−λtλt+1.\lambda_t=\frac{1+\sqrt{1+4\lambda_{t-1}^2}}{2}\quad(t\ge1),\qquad\gamma_t=\frac{1-\lambda_t}{\lambda_{t+1}}.λt​=21+1+4λt−12​​​(t≥1),γt​=λt+1​1−λt​​.

Then λ1=1\lambda_1=1λ1​=1, λt≥1\lambda_t\ge1λt​≥1 for t≥1t\ge1t≥1, and γt≤0\gamma_t\le0γt​≤0. Nesterov's accelerated gradient descent for the smooth case starts from an arbitrary point x1=y1x_1=y_1x1​=y1​ and sets, for t≥1t\ge1t≥1,

yt+1=xt−1β∇f(xt),xt+1=(1−γt) yt+1+γt yt.y_{t+1}=x_t-\frac1\beta\nabla f(x_t),\qquad x_{t+1}=(1-\gamma_t)\,y_{t+1}+\gamma_t\,y_t.yt+1​=xt​−β1​∇f(xt​),xt+1​=(1−γt​)yt+1​+γt​yt​.

The primary sequence (yt)(y_t)(yt​) consists of gradient steps of length 1/β1/\beta1/β; the sequence (xt)(x_t)(xt​), at which the gradient is queried, moves past yt+1y_{t+1}yt+1​ away from yty_tyt​ (since γt≤0\gamma_t\le0γt​≤0). Write δt=f(yt)−f(x∗)\delta_t=f(y_t)-f(x^*)δt​=f(yt​)−f(x∗) for the optimality gap.

Formalization targets

Goal: Theorem 3.19

For every run of the method and every t≥1t\ge1t≥1,

f(yt)−f(x∗)≤2β∥x1−x∗∥2t2.f(y_t)-f(x^*)\le\frac{2\beta\|x_1-x^*\|^2}{t^2}.f(yt​)−f(x∗)≤t22β∥x1​−x∗∥2​.

The constant 222 and the exponent 222 are those printed in the book.

Milestones

In the order of the book's proof (pp. 294–295):

  1. Lemma 3.6, unconstrained (p. 270): f(x−1β∇f(x))−f(y)≤∇f(x)⊤(x−y)−12β∥∇f(x)∥2f(x-\tfrac1\beta\nabla f(x))-f(y)\le\nabla f(x)^\top(x-y)-\tfrac1{2\beta}\|\nabla f(x)\|^2f(x−β1​∇f(x))−f(y)≤∇f(x)⊤(x−y)−2β1​∥∇f(x)∥2 for all x,yx,yx,y.
  2. (3.23): f(ys+1)−f(ys)≤β(xs−ys+1)⊤(xs−ys)−β2∥xs−ys+1∥2f(y_{s+1})-f(y_s)\le\beta(x_s-y_{s+1})^\top(x_s-y_s)-\tfrac\beta2\|x_s-y_{s+1}\|^2f(ys+1​)−f(ys​)≤β(xs​−ys+1​)⊤(xs​−ys​)−2β​∥xs​−ys+1​∥2.
  3. (3.24): f(ys+1)−f(x∗)≤β(xs−ys+1)⊤(xs−x∗)−β2∥xs−ys+1∥2f(y_{s+1})-f(x^*)\le\beta(x_s-y_{s+1})^\top(x_s-x^*)-\tfrac\beta2\|x_s-y_{s+1}\|^2f(ys+1​)−f(x∗)≤β(xs​−ys+1​)⊤(xs​−x∗)−2β​∥xs​−ys+1​∥2.
  4. The λ identity: λs−12=λs2−λs\lambda_{s-1}^2=\lambda_s^2-\lambda_sλs−12​=λs2​−λs​ for s≥1s\ge1s≥1.
  5. (3.25): λs2δs+1−λs−12δs≤β2(∥λsxs−(λs−1)ys−x∗∥2−∥λsys+1−(λs−1)ys−x∗∥2)\lambda_s^2\delta_{s+1}-\lambda_{s-1}^2\delta_s\le\tfrac\beta2\bigl(\|\lambda_sx_s-(\lambda_s-1)y_s-x^*\|^2-\|\lambda_sy_{s+1}-(\lambda_s-1)y_s-x^*\|^2\bigr)λs2​δs+1​−λs−12​δs​≤2β​(∥λs​xs​−(λs​−1)ys​−x∗∥2−∥λs​ys+1​−(λs​−1)ys​−x∗∥2).
  6. (3.26): λs+1xs+1−(λs+1−1)ys+1=λsys+1−(λs−1)ys\lambda_{s+1}x_{s+1}-(\lambda_{s+1}-1)y_{s+1}=\lambda_sy_{s+1}-(\lambda_s-1)y_sλs+1​xs+1​−(λs+1​−1)ys+1​=λs​ys+1​−(λs​−1)ys​.
  7. Telescoped bound: δt≤β2λt−12∥u1∥2\delta_t\le\frac{\beta}{2\lambda_{t-1}^2}\|u_1\|^2δt​≤2λt−12​β​∥u1​∥2 for t≥2t\ge2t≥2, with u1=λ1x1−(λ1−1)y1−x∗u_1=\lambda_1x_1-(\lambda_1-1)y_1-x^*u1​=λ1​x1​−(λ1​−1)y1​−x∗.
  8. Growth of λ: λt−1≥t/2\lambda_{t-1}\ge t/2λt−1​≥t/2 for t≥2t\ge2t≥2.

Significance

The result. Theorem 3.19 is the upper half of the statement that first-order methods on smooth convex functions have complexity Θ(β∥x1−x∗∥2/ε)\Theta(\sqrt{\beta\|x_1-x^*\|^2/\varepsilon})Θ(β∥x1​−x∗∥2/ε​): together with the black-box lower bound of Theorem 3.14 it shows that the accelerated method is optimal up to a constant factor, while plain gradient descent (Theorem 3.3, rate 2β∥x1−x∗∥2/(t−1)2\beta\|x_1-x^*\|^2/(t-1)2β∥x1​−x∗∥2/(t−1)) is not. The same potential-function argument, with the gradient step replaced by a proximal step, yields the rate of FISTA for composite objectives, and the identity λs−12=λs2−λs\lambda_{s-1}^2=\lambda_s^2-\lambda_sλs−12​=λs2​−λs​ is the algebraic core of most later analyses of accelerated and momentum methods.

Formalizing it. The theorem is classical and fully proved on paper. What this mission adds is a machine-checked proof of the O(1/t2)O(1/t^2)O(1/t2) rate for the exact step sequence of the book, with every intermediate inequality of the proof stated separately so that each can be closed and reused. A Lean development of accelerated gradient descent with this rate is not, to our knowledge, in Mathlib.

Difficulty

The rate does not follow from monotone decrease of f(yt)f(y_t)f(yt​), which is the engine of the analysis of plain gradient descent: the accelerated iterates are not monotone, and no single-step inequality on δt\delta_tδt​ alone gives 1/t21/t^21/t2. The proof instead tracks a weighted potential λs−12δs+β2∥us∥2\lambda_{s-1}^2\delta_s+\frac\beta2\|u_s\|^2λs−12​δs​+2β​∥us​∥2 and needs three exact algebraic coincidences: the identity λs−12=λs2−λs\lambda_{s-1}^2=\lambda_s^2-\lambda_sλs−12​=λs2​−λs​, the completion of a square 2a⊤b−∥a∥2=∥b∥2−∥b−a∥22a^\top b-\|a\|^2=\|b\|^2-\|b-a\|^22a⊤b−∥a∥2=∥b∥2−∥b−a∥2, and the rewriting (3.26) of the update rule, which makes consecutive potentials match exactly. In Lean the work is in this bookkeeping (vector identities in an inner-product space, a real recursion with square roots, and a telescoping sum indexed from 111), not in any deep analytic fact.

Formalization scope

  • Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n); x⊤yx^\top yx⊤y is ⟪x, y⟫_ℝ.
  • The gradient is an explicit map ggg with HasGradientAt f (g x) x for every xxx; β-smoothness is the Lipschitz bound on ggg (not the quadratic upper bound (3.4), which is a consequence). Convexity is ConvexOn ℝ Set.univ f.
  • λ\lambdaλ is a real sequence defined by recursion on N\mathbb NN with λ0=0\lambda_0=0λ0​=0; γt=(1−λt)/λt+1\gamma_t=(1-\lambda_t)/\lambda_{t+1}γt​=(1−λt​)/λt+1​ never divides by zero because λt+1≥1\lambda_{t+1}\ge1λt+1​≥1.
  • A run is a predicate on two sequences x,y:N→Rnx,y:\mathbb N\to\mathbb R^nx,y:N→Rn, indexed from 111 as in the book; index 000 is unconstrained. Every theorem quantifies over all runs.
  • Standing assumptions: x∗x^*x∗ is a minimizer of fff (book, p. 242); β>0\beta>0β>0 (implicit in the step 1/β1/\beta1/β) is a stated hypothesis.
  • The page prints the update as xt+1=(1−γs)yt+1+γtytx_{t+1}=(1-\gamma_s)y_{t+1}+\gamma_ty_txt+1​=(1−γs​)yt+1​+γt​yt​; the formalization uses γt\gamma_tγt​ in both places, as the proof's (3.26) requires. A run predicate with a fixed coefficient γs\gamma_sγs​ would describe a different (non-accelerated) method and is ruled out.
  • The goal is stated for all t≥1t\ge1t≥1, including t=1t=1t=1, which the printed proof does not cover but which follows from (3.4). The growth bound λt−1≥t/2\lambda_{t-1}\ge t/2λt−1​≥t/2 is stated for t≥2t\ge2t≥2, since λ0=0\lambda_0=0λ0​=0. The telescoping display prints δs2\delta_s^2δs2​ for δs\delta_sδs​; the formalization uses δs\delta_sδs​.

Contributions of every kind are welcome: proofs of the λ facts and of (3.26) are pure algebra; (3.23)–(3.25) need the descent lemma for β-smooth functions and first-order convexity, both of which are reusable well beyond this mission.

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, A method of solving a convex programming problem with convergence rate O(1/k²), Soviet Mathematics Doklady 27:372–376, 1983. mathnet
  • A. Beck, M. Teboulle, A fast iterative shrinkage-thresholding algorithm for linear inverse problems, SIAM J. Imaging Sciences 2(1):183–202, 2009. doi:10.1137/080716542
  • P. Tseng, On accelerated proximal gradient methods for convex-concave optimization, manuscript, 2008. pdf
  • A. Nemirovski, D. Yudin, Problem Complexity and Method Efficiency in Optimization, Wiley, 1983.
10 thms3 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity X: Nesterov's Accelerated Gradient Descent on a β-Smooth α-Strongly Convex Function Has Rate ((α + β)/2)‖x₁ − x*‖² exp(−(t − 1)/√κ)Textbook

Why accelerated rates matter

First-order methods, which query only function values and gradients, are the workhorse of large-scale optimization in machine learning, signal processing and operations research, because each step costs little more than one gradient evaluation. For a function that is both strongly convex and smooth, plain gradient descent converges geometrically, but the number of steps needed to reach accuracy ε\varepsilonε scales with the condition number κ\kappaκ of the problem. In 1983 Nesterov showed that a gradient method with a carefully chosen momentum term needs a number of steps proportional to κ\sqrt\kappaκ​ instead, and that this is optimal for black-box first-order methods. On ill-conditioned problems, where κ\kappaκ is in the thousands or millions, the difference between κ\kappaκ and κ\sqrt\kappaκ​ is the difference between practical and impractical.

This mission is the tenth of a series formalizing S. Bubeck's monograph Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning, 2015; arXiv:1405.4980v2). It covers §3.7.1, the smooth and strongly convex case of Nesterov's accelerated gradient descent, and its main result, Theorem 3.18.

Setting

Work in Rn\mathbb R^nRn with the Euclidean inner product x⊤yx^\top yx⊤y and norm ∥⋅∥\|\cdot\|∥⋅∥. Let f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R be differentiable with gradient ∇f\nabla f∇f.

  • fff is β\betaβ-smooth if its gradient is β\betaβ-Lipschitz: ∥∇f(x)−∇f(y)∥≤β∥x−y∥\|\nabla f(x)-\nabla f(y)\|\le\beta\|x-y\|∥∇f(x)−∇f(y)∥≤β∥x−y∥ for all x,yx,yx,y.
  • fff is α\alphaα-strongly convex (α>0\alpha>0α>0) if for all x,yx,yx,y
f(y)≥f(x)+∇f(x)⊤(y−x)+α2∥y−x∥2.f(y)\ge f(x)+\nabla f(x)^\top(y-x)+\frac\alpha2\|y-x\|^2 .f(y)≥f(x)+∇f(x)⊤(y−x)+2α​∥y−x∥2.
  • The condition number is κ=β/α\kappa=\beta/\alphaκ=β/α; for n≥1n\ge1n≥1 one always has κ≥1\kappa\ge1κ≥1.
  • x∗x^*x∗ denotes a minimizer of fff on Rn\mathbb R^nRn.

Nesterov's accelerated gradient descent starts at an arbitrary point x1=y1x_1=y_1x1​=y1​ and iterates, for t≥1t\ge1t≥1,

yt+1=xt−1β∇f(xt),xt+1=(1+κ−1κ+1)yt+1−κ−1κ+1 yt.y_{t+1}=x_t-\frac1\beta\nabla f(x_t),\qquad x_{t+1}=\Big(1+\frac{\sqrt\kappa-1}{\sqrt\kappa+1}\Big)y_{t+1}-\frac{\sqrt\kappa-1}{\sqrt\kappa+1}\,y_t .yt+1​=xt​−β1​∇f(xt​),xt+1​=(1+κ​+1κ​−1​)yt+1​−κ​+1κ​−1​yt​.

The point yt+1y_{t+1}yt+1​ is a gradient step from xtx_txt​, and xt+1x_{t+1}xt+1​ moves beyond yt+1y_{t+1}yt+1​ in the direction yt+1−yty_{t+1}-y_tyt+1​−yt​ by the fixed momentum factor (κ−1)/(κ+1)(\sqrt\kappa-1)/(\sqrt\kappa+1)(κ​−1)/(κ​+1).

The analysis in the book uses auxiliary quadratic functions Φs\Phi_sΦs​ (an estimate sequence), defined from the points xsx_sxs​ by

Φ1(x)=f(x1)+α2∥x−x1∥2,Φs+1(x)=(1−1κ)Φs(x)+1κ(f(xs)+∇f(xs)⊤(x−xs)+α2∥x−xs∥2),\Phi_1(x)=f(x_1)+\frac\alpha2\|x-x_1\|^2,\qquad \Phi_{s+1}(x)=\Big(1-\frac1{\sqrt\kappa}\Big)\Phi_s(x)+\frac1{\sqrt\kappa}\Big(f(x_s)+\nabla f(x_s)^\top(x-x_s)+\frac\alpha2\|x-x_s\|^2\Big),Φ1​(x)=f(x1​)+2α​∥x−x1​∥2,Φs+1​(x)=(1−κ​1​)Φs​(x)+κ​1​(f(xs​)+∇f(xs​)⊤(x−xs​)+2α​∥x−xs​∥2),

together with their centres vsv_svs​ (with v1=x1v_1=x_1v1​=x1​ and the recursion (3.21) of the book) and their minimum values Φs∗\Phi^*_sΦs∗​.

Formalization targets

Goal: Theorem 3.18

For every run of the method and every t≥1t\ge1t≥1,

f(yt)−f(x∗)≤α+β2 ∥x1−x∗∥2exp⁡(−t−1κ).f(y_t)-f(x^*)\le\frac{\alpha+\beta}2\,\|x_1-x^*\|^2\exp\Big(-\frac{t-1}{\sqrt\kappa}\Big).f(yt​)−f(x∗)≤2α+β​∥x1​−x∗∥2exp(−κ​t−1​).

Milestones, from the book's proof

  1. (3.18): Φs+1(x)≤f(x)+(1−1/κ)s(Φ1(x)−f(x))\Phi_{s+1}(x)\le f(x)+(1-1/\sqrt\kappa)^s(\Phi_1(x)-f(x))Φs+1​(x)≤f(x)+(1−1/κ​)s(Φ1​(x)−f(x)) for all xxx.
  2. (3.19): f(ys)≤min⁡x∈RnΦs(x)f(y_s)\le\min_{x\in\mathbb R^n}\Phi_s(x)f(ys​)≤minx∈Rn​Φs​(x).
  3. The geometric rate: f(yt)−f(x∗)≤α+β2∥x1−x∗∥2(1−1/κ)t−1f(y_t)-f(x^*)\le\frac{\alpha+\beta}2\|x_1-x^*\|^2(1-1/\sqrt\kappa)^{t-1}f(yt​)−f(x∗)≤2α+β​∥x1​−x∗∥2(1−1/κ​)t−1.
  4. The form Φs(x)=Φs∗+α2∥x−vs∥2\Phi_s(x)=\Phi^*_s+\frac\alpha2\|x-v_s\|^2Φs​(x)=Φs∗​+2α​∥x−vs​∥2 with vsv_svs​ given by (3.21).
  5. The identity (3.22) for Φs+1∗\Phi^*_{s+1}Φs+1∗​.
  6. The inequality (3.20), the inductive step of (3.19).
  7. The coupling vs−xs=κ (xs−ys)v_s-x_s=\sqrt\kappa\,(x_s-y_s)vs​−xs​=κ​(xs​−ys​).

The geometric form in milestone 3 is slightly stronger than the goal, which follows from 1−u≤e−u1-u\le e^{-u}1−u≤e−u.

Significance

Theorem 3.18 gives ε\varepsilonε-accuracy after O(κlog⁡(1/ε))O(\sqrt\kappa\log(1/\varepsilon))O(κ​log(1/ε)) gradient evaluations. Projected gradient descent with step 1/β1/\beta1/β on the same class contracts only at the rate exp⁡(−t/κ)\exp(-t/\kappa)exp(−t/κ) (Theorem 3.10 of the book). The lower bound of Theorem 3.15 shows that no black-box first-order method can do better than ((κ−1)/(κ+1))2(t−1)((\sqrt\kappa-1)/(\sqrt\kappa+1))^{2(t-1)}((κ​−1)/(κ​+1))2(t−1), so the accelerated rate is optimal up to constants. The estimate-sequence argument is the template for many later accelerated methods: proximal, stochastic and variance-reduced variants such as Katyusha, and accelerated coordinate descent.

The result is classical and fully proved on paper. No machine-checked proof of the accelerated rate for strongly convex smooth functions is known to exist in Lean's Mathlib. This mission produces one, with the estimate sequence Φs\Phi_sΦs​, its centres and its minimum values as reusable objects, and with every algebraic identity of the book's proof stated separately.

Difficulty

The algorithm is two lines, but its analysis is not a one-step contraction: neither ∥xt−x∗∥\|x_t-x^*\|∥xt​−x∗∥ nor f(yt)−f(x∗)f(y_t)-f(x^*)f(yt​)−f(x∗) decreases by the factor 1−1/κ1-1/\sqrt\kappa1−1/κ​ at every step. A Lyapunov argument for gradient descent, applied directly to yty_tyt​, gives only the rate 1−1/κ1-1/\kappa1−1/κ. The book obtains the rate through the auxiliary functions Φs\Phi_sΦs​. The inequality (3.18) is easy, but (3.19), that the minimum of Φs\Phi_sΦs​ never drops below f(ys)f(y_s)f(ys​), depends on the exact choice of the momentum factor. It holds only through the identity vs−xs=κ(xs−ys)v_s-x_s=\sqrt\kappa(x_s-y_s)vs​−xs​=κ​(xs​−ys​), which ties the centre of Φs\Phi_sΦs​ to the iterates. Formally, the obstacles are the bookkeeping of the recursive quadratics on Rn\mathbb R^nRn and the algebra in κ\sqrt\kappaκ​, 1/κ1/\sqrt\kappa1/κ​ and 1/(ακ)=κ/β1/(\alpha\sqrt\kappa)=\sqrt\kappa/\beta1/(ακ​)=κ​/β.

Formalization scope

  • Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n). The gradient is an explicit map g with HasGradientAt f (g x) x at every point, which is part of the smoothness predicate IsBetaSmooth f g β. Strong convexity is the published definition OnlineConvexOpt.ConvexBasics.StronglyConvexOn Set.univ f g α, which is the book's (3.13).
  • A run of the method is a predicate IsNesterovSCRun g α β x y on two sequences indexed from 111, with x1=y1x_1=y_1x1​=y1​ arbitrary. Every theorem holds for every run, that is, every starting point.
  • κ\kappaκ is kappa α β = β / α. All theorems assume α>0\alpha>0α>0 and β>0\beta>0β>0. The second is implied by the other hypotheses for n≥1n\ge1n≥1; no hypothesis α≤β\alpha\le\betaα≤β is added.
  • The existence of a minimizer x∗x^*x∗ is the book's standing assumption, written as a hypothesis.
  • Φs\Phi_sΦs​, vsv_svs​ and Φs∗=Φs(vs)\Phi^*_s=\Phi_s(v_s)Φs∗​=Φs​(vs​) are explicit recursive definitions. The book's Φs∗=min⁡Φs\Phi^*_s=\min\Phi_sΦs∗​=minΦs​ is recovered by milestone 4, and no real infimum is used. The minimum in (3.19) is stated as f(ys)≤Φs(x)f(y_s)\le\Phi_s(x)f(ys​)≤Φs​(x) for every xxx.
  • The identities of milestones 4, 5 and 7 are algebraic and are stated without convexity or smoothness, for arbitrary sequences or runs.
  • Ruled out as trivializing: a run predicate that drops x1=y1x_1=y_1x1​=y1​ breaks (3.19) at s=1s=1s=1 and is not used. A minimum value Φs∗\Phi^*_sΦs∗​ defined through (3.22) would make that identity a tautology, so Φs∗\Phi^*_sΦs∗​ is defined as a value of Φs\Phi_sΦs​.
  • The definitions are local to the namespace ConvexOptAlg.NesterovStrong. β-smoothness duplicates the predicate of other missions of the series and will be merged afterwards. Contributions welcome: proofs of the milestones, and general lemmas on quadratics z↦c+α2∥z−v∥2z\mapsto c+\frac\alpha2\|z-v\|^2z↦c+2α​∥z−v∥2 that the algebraic milestones need.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. arXiv:1405.4980v2, §3.7.1, Theorem 3.18, pp. 290–293.
  • Y. Nesterov, A method of solving a convex programming problem with convergence rate O(1/k²), Soviet Mathematics Doklady 27:372–376, 1983.
  • Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer, 2004. doi:10.1007/978-1-4419-8853-9
10 thms2 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity XI: Mirror Descent with a ρ-Strongly Convex Mirror Map Has Rate RL√(2/(ρt))Textbook

Motivation

The projected subgradient method of Chapter 3 measures distances in the Euclidean norm, and its rate RL/tRL/\sqrt tRL/t​ depends on the geometry only through a Euclidean radius RRR and a Euclidean bound LLL on the subgradients. For many constraint sets that occur in practice this is the wrong geometry. On the probability simplex Δn\Delta_nΔn​ with subgradients bounded in ℓ∞\ell_\inftyℓ∞​, as for linear losses with bounded coefficients, the Euclidean constants make RLRLRL grow like n\sqrt nn​, while the problem itself is nearly dimension-free.

Mirror descent, introduced by Nemirovski and Yudin (1983), repairs this. The gradient step is taken in the dual space, after mapping the current point through the gradient of a strictly convex mirror map Φ\PhiΦ, and feasibility is restored by a projection in the Bregman divergence of Φ\PhiΦ rather than in the Euclidean distance. Beck and Teboulle (2003, doi:10.1016/S0167-6377(02)00231-6) recast it as a nonlinear projected subgradient method and showed that with the negative entropy on the simplex its rate is O(Llog⁡n/t)O(L\sqrt{\log n/t})O(Llogn/t​). Nesterov (2009, doi:10.1007/s10107-007-0149-x) introduced dual averaging, a lazy variant that averages the subgradients in the dual space and maps back only when a primal point is needed. Both methods underlie online learning (exponential weights is mirror descent with the entropy), stochastic optimization, and saddle-point methods such as mirror prox.

This mission is the eleventh of a series formalizing S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning, 2015, arXiv:1405.4980). It covers the preamble of Chapter 4, Sections 4.1 and 4.2, and Section 4.4.

Setting

Let EEE be a finite-dimensional real vector space (the book's Rn\mathbb R^nRn) with an arbitrary norm ∥⋅∥\|\cdot\|∥⋅∥. A linear functional ggg on EEE acts on xxx by g⊤xg^\top xg⊤x and has dual norm ∥g∥∗=sup⁡∥x∥≤1g⊤x\|g\|_*=\sup_{\|x\|\le1}g^\top x∥g∥∗​=sup∥x∥≤1​g⊤x. Let X⊆E\mathcal X\subseteq EX⊆E be compact and convex.

Let D⊆E\mathcal D\subseteq ED⊆E be a convex open set with X⊆D‾\mathcal X\subseteq\overline{\mathcal D}X⊆D and X∩D≠∅\mathcal X\cap\mathcal D\neq\emptysetX∩D=∅. A function Φ:D→R\Phi:\mathcal D\to\mathbb RΦ:D→R is a mirror map if it is strictly convex and differentiable, its gradient takes every value in the dual space, and ∥∇Φ(x)∥→+∞\|\nabla\Phi(x)\|\to+\infty∥∇Φ(x)∥→+∞ as xxx tends to the boundary of D\mathcal DD. Its Bregman divergence is

DΦ(x,y)=Φ(x)−Φ(y)−∇Φ(y)⊤(x−y),D_\Phi(x,y)=\Phi(x)-\Phi(y)-\nabla\Phi(y)^\top(x-y),DΦ​(x,y)=Φ(x)−Φ(y)−∇Φ(y)⊤(x−y),

and the Bregman projection of y∈Dy\in\mathcal Dy∈D is ΠXΦ(y)=argmin⁡x∈X∩DDΦ(x,y)\Pi^\Phi_{\mathcal X}(y)=\operatorname{argmin}_{x\in\mathcal X\cap\mathcal D}D_\Phi(x,y)ΠXΦ​(y)=argminx∈X∩D​DΦ​(x,y). The mirror map is ρ\rhoρ-strongly convex on X∩D\mathcal X\cap\mathcal DX∩D if Φ(x)−Φ(y)≤∇Φ(x)⊤(x−y)−ρ2∥x−y∥2\Phi(x)-\Phi(y)\le\nabla\Phi(x)^\top(x-y)-\frac\rho2\|x-y\|^2Φ(x)−Φ(y)≤∇Φ(x)⊤(x−y)−2ρ​∥x−y∥2 for all x,y∈X∩Dx,y\in\mathcal X\cap\mathcal Dx,y∈X∩D.

Let fff be convex on X\mathcal XX with a minimizer x∗∈Xx^*\in\mathcal Xx∗∈X. A linear functional ggg is a subgradient of fff at xxx if f(x)−f(y)≤g⊤(x−y)f(x)-f(y)\le g^\top(x-y)f(x)−f(y)≤g⊤(x−y) for all y∈Xy\in\mathcal Xy∈X, and fff is LLL-Lipschitz if the subgradients satisfy ∥g∥∗≤L\|g\|_*\le L∥g∥∗​≤L.

Mirror descent with step η\etaη starts at x1∈argmin⁡X∩DΦx_1\in\operatorname{argmin}_{\mathcal X\cap\mathcal D}\Phix1​∈argminX∩D​Φ and, for t≥1t\ge1t≥1, picks gt∈∂f(xt)g_t\in\partial f(x_t)gt​∈∂f(xt​) and

∇Φ(yt+1)=∇Φ(xt)−ηgt,yt+1∈D,xt+1=ΠXΦ(yt+1).(4.2–4.3)\nabla\Phi(y_{t+1})=\nabla\Phi(x_t)-\eta g_t,\quad y_{t+1}\in\mathcal D,\qquad x_{t+1}=\Pi^\Phi_{\mathcal X}(y_{t+1}).\tag{4.2–4.3}∇Φ(yt+1​)=∇Φ(xt​)−ηgt​,yt+1​∈D,xt+1​=ΠXΦ​(yt+1​).(4.2–4.3)

Dual averaging instead sets xt∈argmin⁡x∈X∩Dη∑s=1t−1gs⊤x+Φ(x)x_t\in\operatorname{argmin}_{x\in\mathcal X\cap\mathcal D}\eta\sum_{s=1}^{t-1}g_s^\top x+\Phi(x)xt​∈argminx∈X∩D​η∑s=1t−1​gs⊤​x+Φ(x) (4.6). Let R2R^2R2 bound Φ(x)−Φ(x1)\Phi(x)-\Phi(x_1)Φ(x)−Φ(x1​) over X∩D\mathcal X\cap\mathcal DX∩D.

Formalization targets

Goal: Theorem 4.2

With η=RL2ρt\eta=\frac RL\sqrt{\frac{2\rho}{t}}η=LR​t2ρ​​, every run of mirror descent satisfies

f(1t∑s=1txs)−f(x∗)≤RL2ρt.f\Big(\frac1t\sum_{s=1}^t x_s\Big)-f(x^*)\le RL\sqrt{\frac{2}{\rho t}}.f(t1​s=1∑t​xs​)−f(x∗)≤RLρt2​​.

Milestones

  1. The three-point identity (4.1): (∇f(x)−∇f(y))⊤(x−z)=Df(x,y)+Df(z,x)−Df(z,y)(\nabla f(x)-\nabla f(y))^\top(x-z)=D_f(x,y)+D_f(z,x)-D_f(z,y)(∇f(x)−∇f(y))⊤(x−z)=Df​(x,y)+Df​(z,x)−Df​(z,y).
  2. Lemma 4.1: for x∈X∩Dx\in\mathcal X\cap\mathcal Dx∈X∩D and y∈Dy\in\mathcal Dy∈D, DΦ(x,ΠXΦ(y))+DΦ(ΠXΦ(y),y)≤DΦ(x,y)D_\Phi(x,\Pi^\Phi_{\mathcal X}(y))+D_\Phi(\Pi^\Phi_{\mathcal X}(y),y)\le D_\Phi(x,y)DΦ​(x,ΠXΦ​(y))+DΦ​(ΠXΦ​(y),y)≤DΦ​(x,y), together with the first-order inequality that implies it.
  3. The per-step inequality: f(xs)−f(x)≤1η(DΦ(x,xs)+DΦ(xs,ys+1)−DΦ(x,xs+1)−DΦ(xs+1,ys+1))f(x_s)-f(x)\le\frac1\eta\big(D_\Phi(x,x_s)+D_\Phi(x_s,y_{s+1})-D_\Phi(x,x_{s+1})-D_\Phi(x_{s+1},y_{s+1})\big)f(xs​)−f(x)≤η1​(DΦ​(x,xs​)+DΦ​(xs​,ys+1​)−DΦ​(x,xs+1​)−DΦ​(xs+1​,ys+1​)).
  4. The stability term: DΦ(xs,ys+1)−DΦ(xs+1,ys+1)≤(ηL)2/(2ρ)D_\Phi(x_s,y_{s+1})-D_\Phi(x_{s+1},y_{s+1})\le(\eta L)^2/(2\rho)DΦ​(xs​,ys+1​)−DΦ​(xs+1​,ys+1​)≤(ηL)2/(2ρ).
  5. The summed bound: ∑s=1t(f(xs)−f(x))≤DΦ(x,x1)/η+ηL2t/(2ρ)\sum_{s=1}^t(f(x_s)-f(x))\le D_\Phi(x,x_1)/\eta+\eta L^2t/(2\rho)∑s=1t​(f(xs​)−f(x))≤DΦ​(x,x1​)/η+ηL2t/(2ρ).

Companion: Theorem 4.3

With η=RLρ2t\eta=\frac RL\sqrt{\frac{\rho}{2t}}η=LR​2tρ​​, dual averaging satisfies f(1t∑s=1txs)−f(x∗)≤2RL2ρtf\big(\frac1t\sum_{s=1}^tx_s\big)-f(x^*)\le2RL\sqrt{\frac2{\rho t}}f(t1​∑s=1t​xs​)−f(x∗)≤2RLρt2​​, with the two displays (4.7) and (4.8) of its proof as supporting items.

Significance

Theorem 4.2 is the basic rate of non-Euclidean first-order optimization. With the Euclidean mirror map Φ=12∥⋅∥22\Phi=\frac12\|\cdot\|_2^2Φ=21​∥⋅∥22​ it recovers the projected subgradient rate of Chapter 3; with the negative entropy on the simplex, which is 111-strongly convex for ℓ1\ell_1ℓ1​ by Pinsker's inequality and has R2=log⁡nR^2=\log nR2=logn, it gives L∞2log⁡n/tL_\infty\sqrt{2\log n/t}L∞​2logn/t​, so the dependence on the dimension becomes logarithmic. The same analysis, with fff replaced by a sequence of losses, is the regret bound of online mirror descent, and its per-step and stability inequalities are reused for stochastic mirror descent and mirror prox in later chapters of the book.

The results are classical and fully proved in the book. What this mission adds is a machine-checked version in a general finite-dimensional normed space, with gradients as linear functionals and the dual norm as the operator norm, for an arbitrary mirror map on an arbitrary open domain. On Prove2Me the three-point identity (4.1) is already posed, as Lemma 4.1 of the Beck–Teboulle formalization, and is reused here; a regret bound for online mirror descent with Legendre functions is proved in another library, but it is not this theorem and does not cover a domain D\mathcal DD distinct from the whole space or a Bregman projection onto X∩D\mathcal X\cap\mathcal DX∩D.

Difficulty

The algebra of the proof is short. The difficulty lies at the boundary of D\mathcal DD. The minimizer x∗x^*x∗ may lie on ∂D\partial\mathcal D∂D, where Φ\PhiΦ and ∇Φ\nabla\Phi∇Φ are not defined; this is the typical case for the entropy on the simplex, where x∗x^*x∗ is often a vertex. The per-step bound holds only for comparison points x∈X∩Dx\in\mathcal X\cap\mathcal Dx∈X∩D, so the bound at x∗x^*x∗ has to be recovered from points of X∩D\mathcal X\cap\mathcal DX∩D, where no regularity of fff beyond convexity on X\mathcal XX is available. Lemma 4.1 needs the first-order optimality condition of the Bregman projection over the convex set X∩D\mathcal X\cap\mathcal DX∩D, which is not closed. In the non-Euclidean setting the gradient step cannot be written in the primal space at all: the update (4.2) is an equation between linear functionals, and the norm and dual norm must be kept apart throughout.

Formalization scope

  • The space is a finite-dimensional real normed space E with an arbitrary norm; gradients and subgradients are elements of E →L[ℝ] ℝ, the dual norm is the operator norm, and ∇Φ\nabla\Phi∇Φ is an explicit map Φ' with HasFDerivAt Φ (Φ' x) x on D\mathcal DD.
  • The mirror map carries all three properties (i)–(iii) of the book; the standing setting (X\mathcal XX compact convex, X⊆D‾\mathcal X\subseteq\overline{\mathcal D}X⊆D, X∩D≠∅\mathcal X\cap\mathcal D\neq\emptysetX∩D=∅) is a single predicate.
  • Runs are predicates over the first ttt steps, quantified universally: any subgradient, any yt+1y_{t+1}yt+1​ solving (4.2), any minimizer. A run whose next iterate were an arbitrary point of X∩D\mathcal X\cap\mathcal DX∩D, rather than the Bregman projection, would make the theorem false; the projection is part of the run.
  • Sequences are indexed from 111; t≥1t\ge1t≥1, ρ>0\rho>0ρ>0, L>0L>0L>0, R>0R>0R>0 and η>0\eta>0η>0 are explicit, since the step and the bound divide by them.
  • R2R^2R2 is any upper bound of sup⁡X∩DΦ−Φ(x1)\sup_{\mathcal X\cap\mathcal D}\Phi-\Phi(x_1)supX∩D​Φ−Φ(x1​) (the book's RRR is the least one); when the supremum is infinite the hypotheses cannot be met, as on the page.
  • "fff is LLL-Lipschitz" is assumed for the subgradients the run uses. Subgradients relative to X\mathcal XX are unbounded at boundary points of X\mathcal XX, so the literal "every subgradient at every point" would make the hypothesis unsatisfiable.
  • The definitions (Bregman divergence, mirror map, Bregman projection, the two runs) are reusable by the later chapters on mirror prox and stochastic mirror descent. Proofs of any milestone, and of the general facts they need (nonnegativity of Bregman divergences, first-order optimality over a convex set that is not closed), are welcome.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. arXiv:1405.4980
  • A. Nemirovski and D. Yudin, Problem Complexity and Method Efficiency in Optimization, Wiley, 1983 (book; no DOI).
  • A. Beck and M. Teboulle, Mirror descent and nonlinear projected subgradient methods for convex optimization, Operations Research Letters 31(3):167–175, 2003. doi:10.1016/S0167-6377(02)00231-6
  • Y. Nesterov, Primal-dual subgradient methods for convex problems, Mathematical Programming 120(1):221–259, 2009. doi:10.1007/s10107-007-0149-x
7 thms3 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity XII: Mirror Prox on a Convex β-Smooth Function Has Rate βR²/(ρt)Textbook

Motivation

First-order methods for constrained convex optimization are usually analysed in the Euclidean norm, where gradient descent and its projected and accelerated variants have well-understood rates. Many constraint sets of practical interest, such as the probability simplex, the ℓ1\ell_1ℓ1​ ball and the spectrahedron, are poorly adapted to the Euclidean geometry: the Euclidean diameter and the Euclidean Lipschitz or smoothness constants grow with the dimension. Mirror descent, due to Nemirovski and Yudin, replaces the Euclidean step by a step taken in the dual space through a mirror map Φ\PhiΦ, so that the rate depends on constants measured in a norm matched to the set. For non-smooth Lipschitz functions it attains the rate 1/t1/\sqrt t1/t​ (Theorem 4.2 of the source, the subject of the previous mission of this series).

For smooth functions, a rate of order 1/t1/t1/t is available in any geometry by an extragradient-type modification. Mirror prox was introduced by Nemirovski in 2004 (Prox-method with rate of convergence O(1/t)) for variational inequalities with Lipschitz monotone operators and convex–concave saddle-point problems. It is the basis of smoothing approaches to non-smooth optimization and of stochastic saddle-point methods. This mission formalizes the version for minimizing a single smooth convex function, Theorem 4.4 of S. Bubeck, Convex Optimization: Algorithms and Complexity (Foundations and Trends in Machine Learning, 2015), §4.5.

Setting

Fix an arbitrary norm ∥⋅∥\|\cdot\|∥⋅∥ on a finite-dimensional real space EEE. Gradients are linear functionals on EEE; for a functional ggg write g⊤vg^\top vg⊤v for its value at vvv, and ∥g∥∗=sup⁡∥v∥≤1g⊤v\|g\|_*=\sup_{\|v\|\le1}g^\top v∥g∥∗​=sup∥v∥≤1​g⊤v for the dual norm. Let X⊆E\mathcal X\subseteq EX⊆E be compact and convex.

  • The Bregman divergence of a differentiable Φ\PhiΦ is DΦ(x,y)=Φ(x)−Φ(y)−∇Φ(y)⊤(x−y)D_\Phi(x,y)=\Phi(x)-\Phi(y)-\nabla\Phi(y)^\top(x-y)DΦ​(x,y)=Φ(x)−Φ(y)−∇Φ(y)⊤(x−y).
  • Let D\mathcal DD be a convex open set with X⊆D‾\mathcal X\subseteq\overline{\mathcal D}X⊆D and X∩D≠∅\mathcal X\cap\mathcal D\ne\emptysetX∩D=∅. A mirror map on D\mathcal DD is a function Φ\PhiΦ that is strictly convex and differentiable on D\mathcal DD, whose gradient takes every value (∇Φ(D)\nabla\Phi(\mathcal D)∇Φ(D) is the whole dual space), and whose gradient norm tends to +∞+\infty+∞ at the boundary of D\mathcal DD.
  • The Bregman projection of y∈Dy\in\mathcal Dy∈D is ΠXΦ(y)=argmin⁡x∈X∩DDΦ(x,y)\Pi^\Phi_{\mathcal X}(y)=\operatorname{argmin}_{x\in\mathcal X\cap\mathcal D}D_\Phi(x,y)ΠXΦ​(y)=argminx∈X∩D​DΦ​(x,y).
  • Φ\PhiΦ is ρ\rhoρ-strongly convex on X∩D\mathcal X\cap\mathcal DX∩D if Φ(x)−Φ(y)≤∇Φ(x)⊤(x−y)−ρ2∥x−y∥2\Phi(x)-\Phi(y)\le\nabla\Phi(x)^\top(x-y)-\frac\rho2\|x-y\|^2Φ(x)−Φ(y)≤∇Φ(x)⊤(x−y)−2ρ​∥x−y∥2 there; fff is β\betaβ-smooth on X\mathcal XX if ∥∇f(x)−∇f(y)∥∗≤β∥x−y∥\|\nabla f(x)-\nabla f(y)\|_*\le\beta\|x-y\|∥∇f(x)−∇f(y)∥∗​≤β∥x−y∥ for x,y∈Xx,y\in\mathcal Xx,y∈X.

Mirror prox with step size η\etaη generates, from xtx_txt​,

∇Φ(yt+1′)=∇Φ(xt)−η∇f(xt),yt+1∈argmin⁡x∈X∩DDΦ(x,yt+1′),\nabla\Phi(y'_{t+1})=\nabla\Phi(x_t)-\eta\nabla f(x_t),\qquad y_{t+1}\in\operatorname*{argmin}_{x\in\mathcal X\cap\mathcal D}D_\Phi(x,y'_{t+1}),∇Φ(yt+1′​)=∇Φ(xt​)−η∇f(xt​),yt+1​∈x∈X∩Dargmin​DΦ​(x,yt+1′​), ∇Φ(xt+1′)=∇Φ(xt)−η∇f(yt+1),xt+1∈argmin⁡x∈X∩DDΦ(x,xt+1′).\nabla\Phi(x'_{t+1})=\nabla\Phi(x_t)-\eta\nabla f(y_{t+1}),\qquad x_{t+1}\in\operatorname*{argmin}_{x\in\mathcal X\cap\mathcal D}D_\Phi(x,x'_{t+1}).∇Φ(xt+1′​)=∇Φ(xt​)−η∇f(yt+1​),xt+1​∈x∈X∩Dargmin​DΦ​(x,xt+1′​).

The first half is a mirror descent step from xtx_txt​ to yt+1y_{t+1}yt+1​; the second restarts from xtx_txt​ with the gradient evaluated at yt+1y_{t+1}yt+1​.

Formalization targets

Goal: Theorem 4.4

With η=ρ/β\eta=\rho/\betaη=ρ/β, x1∈argmin⁡X∩DΦx_1\in\operatorname{argmin}_{\mathcal X\cap\mathcal D}\Phix1​∈argminX∩D​Φ, R2≥sup⁡x∈X∩DΦ(x)−Φ(x1)R^2\ge\sup_{x\in\mathcal X\cap\mathcal D}\Phi(x)-\Phi(x_1)R2≥supx∈X∩D​Φ(x)−Φ(x1​), fff convex and β\betaβ-smooth and x∗x^*x∗ a minimizer of fff on X\mathcal XX, for every t≥1t\ge1t≥1

f(1t∑s=1tys+1)−f(x∗)≤βR2ρt.f\Bigl(\frac1t\sum_{s=1}^t y_{s+1}\Bigr)-f(x^*)\le\frac{\beta R^2}{\rho t}.f(t1​s=1∑t​ys+1​)−f(x∗)≤ρtβR2​.

Milestones

  1. Lemma 4.1 (Bregman projections): (∇Φ(ΠXΦ(y))−∇Φ(y))⊤(ΠXΦ(y)−x)≤0(\nabla\Phi(\Pi^\Phi_{\mathcal X}(y))-\nabla\Phi(y))^\top(\Pi^\Phi_{\mathcal X}(y)-x)\le0(∇Φ(ΠXΦ​(y))−∇Φ(y))⊤(ΠXΦ​(y)−x)≤0 and DΦ(x,ΠXΦ(y))+DΦ(ΠXΦ(y),y)≤DΦ(x,y)D_\Phi(x,\Pi^\Phi_{\mathcal X}(y))+D_\Phi(\Pi^\Phi_{\mathcal X}(y),y)\le D_\Phi(x,y)DΦ​(x,ΠXΦ​(y))+DΦ​(ΠXΦ​(y),y)≤DΦ​(x,y) for x∈X∩Dx\in\mathcal X\cap\mathcal Dx∈X∩D, y∈Dy\in\mathcal Dy∈D.
  2. First term: η∇f(yt+1)⊤(xt+1−x)≤DΦ(x,xt)−DΦ(x,xt+1)−DΦ(xt+1,xt)\eta\nabla f(y_{t+1})^\top(x_{t+1}-x)\le D_\Phi(x,x_t)-D_\Phi(x,x_{t+1})-D_\Phi(x_{t+1},x_t)η∇f(yt+1​)⊤(xt+1​−x)≤DΦ​(x,xt​)−DΦ​(x,xt+1​)−DΦ​(xt+1​,xt​).
  3. Second term, (4.9): η∇f(xt)⊤(yt+1−xt+1)≤DΦ(xt+1,xt)−ρ2∥xt+1−yt+1∥2−ρ2∥yt+1−xt∥2\eta\nabla f(x_t)^\top(y_{t+1}-x_{t+1})\le D_\Phi(x_{t+1},x_t)-\frac\rho2\|x_{t+1}-y_{t+1}\|^2-\frac\rho2\|y_{t+1}-x_t\|^2η∇f(xt​)⊤(yt+1​−xt+1​)≤DΦ​(xt+1​,xt​)−2ρ​∥xt+1​−yt+1​∥2−2ρ​∥yt+1​−xt​∥2.
  4. Third term: (∇f(yt+1)−∇f(xt))⊤(yt+1−xt+1)≤β2∥yt+1−xt∥2+β2∥yt+1−xt+1∥2(\nabla f(y_{t+1})-\nabla f(x_t))^\top(y_{t+1}-x_{t+1})\le\frac\beta2\|y_{t+1}-x_t\|^2+\frac\beta2\|y_{t+1}-x_{t+1}\|^2(∇f(yt+1​)−∇f(xt​))⊤(yt+1​−xt+1​)≤2β​∥yt+1​−xt​∥2+2β​∥yt+1​−xt+1​∥2.
  5. Per-step bound: f(yt+1)−f(x)≤(DΦ(x,xt)−DΦ(x,xt+1))/ηf(y_{t+1})-f(x)\le\bigl(D_\Phi(x,x_t)-D_\Phi(x,x_{t+1})\bigr)/\etaf(yt+1​)−f(x)≤(DΦ​(x,xt​)−DΦ​(x,xt+1​))/η for x∈X∩Dx\in\mathcal X\cap\mathcal Dx∈X∩D.

The three-point identity (4.1), (∇f(x)−∇f(y))⊤(x−z)=Df(x,y)+Df(z,x)−Df(z,y)(\nabla f(x)-\nabla f(y))^\top(x-z)=D_f(x,y)+D_f(z,x)-D_f(z,y)(∇f(x)−∇f(y))⊤(x−z)=Df​(x,y)+Df​(z,x)−Df​(z,y), is already posed on the platform as BeckTeboulleMD.EMDA.lemma_4_1 and is included by reference.

Significance

Theorem 4.4 shows that the 1/t1/t1/t rate of gradient descent on smooth functions survives in non-Euclidean geometries, with the dimension entering only through R2/ρR^2/\rhoR2/ρ. With the negative entropy on the simplex, R2/ρR^2/\rhoR2/ρ is of order log⁡n\log nlogn for the ℓ1\ell_1ℓ1​ norm, against a polynomial dependence for Euclidean methods. Beyond this single-function version, the same three-term argument is the core of the analysis of mirror prox for monotone variational inequalities and saddle-point problems (§4.6 and §5.2 of the source), where it gives the 1/t1/t1/t rate for smooth convex–concave games.

The result is proved in the source and in Nemirovski's paper. As far as this mission is aware it has no machine-checked proof; Mathlib has no Bregman divergences, mirror maps or mirror-descent-type algorithms. A formal proof would supply reusable infrastructure: the Bregman projection lemma (Lemma 4.1) and the three-point identity are used by every mirror-descent analysis, including the stochastic and online variants later in the source.

Difficulty

The obvious argument for mirror descent bounds f(xt)−f(x)f(x_t)-f(x)f(xt​)−f(x) by ∇f(xt)⊤(xt−x)\nabla f(x_t)^\top(x_t-x)∇f(xt​)⊤(xt​−x) and leaves a stability term of size η2∥∇f∥∗2\eta^2\|\nabla f\|_*^2η2∥∇f∥∗2​, which yields only 1/t1/\sqrt t1/t​. Smoothness has to be used to cancel that term, and a single gradient step does not do so in a general norm. Mirror prox evaluates the gradient at the extrapolated point yt+1y_{t+1}yt+1​; the analysis must show that the error created by using ∇f(xt)\nabla f(x_t)∇f(xt​) for the first step is paid for by the strong convexity of Φ\PhiΦ, which requires splitting ∇f(yt+1)⊤(yt+1−x)\nabla f(y_{t+1})^\top(y_{t+1}-x)∇f(yt+1​)⊤(yt+1​−x) into three terms and bounding each with matching constants.

In Lean the difficulty is in the infrastructure: first-order optimality of a Bregman projection over X∩D\mathcal X\cap\mathcal DX∩D, which need not be closed; the passage from x∈X∩Dx\in\mathcal X\cap\mathcal Dx∈X∩D to a minimizer x∗x^*x∗ that may lie on the boundary of D\mathcal DD; and Jensen's inequality for the average of the ys+1y_{s+1}ys+1​.

Formalization scope

  • EEE is a finite-dimensional real normed space. Gradients are continuous linear functionals E →L[ℝ] ℝ given by explicit maps Φ' and f'; the dual norm is the operator norm. Φ' is the Fréchet derivative of Φ\PhiΦ at each point of D\mathcal DD; f' is the derivative of fff relative to X\mathcal XX at each point of X\mathcal XX.
  • The mirror map definition keeps all three properties of §4.1, including surjectivity of the gradient and divergence at each frontier point of D\mathcal DD.
  • Projections and runs are relations: every admissible argmin choice is covered; no choice function is used. The run is indexed from t=1t=1t=1; index 000 is unused.
  • Standing and implicit hypotheses: X\mathcal XX compact convex; a minimizer x∗∈Xx^*\in\mathcal Xx∗∈X exists (the source's standing assumption); ρ>0\rho>0ρ>0 and β>0\beta>0β>0 (so that η=ρ/β\eta=\rho/\betaη=ρ/β and the division by ρt\rho tρt are meaningful); x1∈argmin⁡X∩DΦx_1\in\operatorname{argmin}_{\mathcal X\cap\mathcal D}\Phix1​∈argminX∩D​Φ (unspecified in §4.5, taken as in mirror descent, §4.2).
  • RRR is any real with Φ−Φ(x1)≤R2\Phi-\Phi(x_1)\le R^2Φ−Φ(x1​)≤R2 on X∩D\mathcal X\cap\mathcal DX∩D. The source's R2=sup⁡R^2=\supR2=sup is the case where the supremum is finite; when it is infinite, the source's bound is void.
  • A run in which the second step started from yt+1y_{t+1}yt+1​, or projected yt+1′y'_{t+1}yt+1′​ again, would be mirror descent, for which the claimed rate is false; the formal run follows the four equations of the source exactly. The average is over the extrapolated points ys+1y_{s+1}ys+1​, s=1,…,ts=1,\dots,ts=1,…,t.

Contributions welcome: proofs of Lemma 4.1 and the three-term bounds, which are reusable for the stochastic mirror descent and saddle-point missions of this series, and the final telescoping and limit argument.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. https://arxiv.org/abs/1405.4980v2
  • A. Nemirovski, Prox-method with rate of convergence O(1/t) for variational inequalities with Lipschitz continuous monotone operators and smooth convex-concave saddle point problems, SIAM Journal on Optimization 15(1):229–251, 2004. https://doi.org/10.1137/S1052623403425629
  • A. Beck and M. Teboulle, Mirror descent and nonlinear projected subgradient methods for convex optimization, Operations Research Letters 31(3):167–175, 2003. https://doi.org/10.1016/S0167-6377(02)00231-6
  • A. Nemirovski and D. Yudin, Problem Complexity and Method Efficiency in Optimization, Wiley, 1983.
8 thms3 active usersReviewed
🏆Completed
Convex OptimizationMachine LearningOptimization+1·Captain: mikedeng1

Convex Optimization: Algorithms and Complexity XVI: Random Coordinate Descent RCD(γ) on a Strongly Convex Coordinate-Smooth Function Has Rate (1 − 1/κ_γ)^tTextbook

Motivation

When a problem has millions of variables, even one full gradient can be too expensive to compute, while a single partial derivative ∂f/∂xi\partial f/\partial x_i∂f/∂xi​ is often cheap: in regularized regression, support vector machines and many structured problems, updating one coordinate costs a small fraction of a full gradient step. Coordinate descent methods exploit this by moving along one coordinate at a time. They are among the oldest optimization schemes and were for a long time analysed only for cyclic orders and only asymptotically.

Nesterov (2012) showed that choosing the coordinate at random, with probabilities depending on the coordinate-wise smoothness constants, gives global, non-asymptotic rates that can beat full gradient descent in total work. This mission formalizes that analysis as presented in §6.4 of S. Bubeck, Convex Optimization: Algorithms and Complexity (arXiv:1405.4980v2), pp. 338–342, and in particular its linear rate for strongly convex functions (Theorem 6.8).

Setting

Let f:Rn→Rf:\mathbb R^n\to\mathbb Rf:Rn→R be differentiable, write ∇if(x)=∂f∂xi(x)\nabla_i f(x)=\frac{\partial f}{\partial x_i}(x)∇i​f(x)=∂xi​∂f​(x) and let eie_iei​ be the iii-th standard basis vector. The function is directionally smooth with constants β1,…,βn>0\beta_1,\dots,\beta_n>0β1​,…,βn​>0 if

∣∇if(x+uei)−∇if(x)∣≤βi∣u∣for all i∈[n], x∈Rn, u∈R,|\nabla_i f(x+ue_i)-\nabla_i f(x)|\le\beta_i|u|\qquad\text{for all } i\in[n],\ x\in\mathbb R^n,\ u\in\mathbb R,∣∇i​f(x+uei​)−∇i​f(x)∣≤βi​∣u∣for all i∈[n], x∈Rn, u∈R,

equivalently, each one-variable restriction u↦f(x+uei)u\mapsto f(x+ue_i)u↦f(x+uei​) is βi\beta_iβi​-smooth.

For a real exponent ccc, the weighted norms are

∥x∥[c]=∑iβicxi2,∥x∥[c]∗=∑iβi−cxi2.\|x\|_{[c]}=\sqrt{\textstyle\sum_{i}\beta_i^{c}x_i^2},\qquad \|x\|^*_{[c]}=\sqrt{\textstyle\sum_{i}\beta_i^{-c}x_i^2}.∥x∥[c]​=∑i​βic​xi2​​,∥x∥[c]∗​=∑i​βi−c​xi2​​.

For α>0\alpha>0α>0, fff is α\alphaα-strongly convex w.r.t. a norm ∥⋅∥\|\cdot\|∥⋅∥ if f(x)−f(y)≤∇f(x)⊤(x−y)−α2∥x−y∥2f(x)-f(y)\le\nabla f(x)^\top(x-y)-\frac{\alpha}{2}\|x-y\|^2f(x)−f(y)≤∇f(x)⊤(x−y)−2α​∥x−y∥2 for all x,yx,yx,y. The point x∗x^*x∗ is a minimizer of fff.

For γ≥0\gamma\ge0γ≥0, RCD(γ\gammaγ) starts at x1∈Rnx_1\in\mathbb R^nx1​∈Rn and iterates

xs+1=xs−1βis∇isf(xs) eis,x_{s+1}=x_s-\frac{1}{\beta_{i_s}}\nabla_{i_s}f(x_s)\,e_{i_s},xs+1​=xs​−βis​​1​∇is​​f(xs​)eis​​,

where i1,i2,…i_1,i_2,\dotsi1​,i2​,… are drawn independently from pγ(i)=βiγ/∑jβjγp_\gamma(i)=\beta_i^\gamma/\sum_{j}\beta_j^\gammapγ​(i)=βiγ​/∑j​βjγ​. The case γ=0\gamma=0γ=0 is uniform sampling; γ=1\gamma=1γ=1 samples proportionally to βi\beta_iβi​.

Formalization targets

Goal: Theorem 6.8 (p. 341)

Let γ≥0\gamma\ge0γ≥0, let fff be α\alphaα-strongly convex w.r.t. ∥⋅∥[1−γ]\|\cdot\|_{[1-\gamma]}∥⋅∥[1−γ]​ and directionally smooth with constants βi\beta_iβi​, and let κγ=∑iβiγ/α\kappa_\gamma=\sum_i\beta_i^\gamma/\alphaκγ​=∑i​βiγ​/α. Then for every t≥0t\ge0t≥0

Ef(xt+1)−f(x∗)≤(1−1κγ)t(f(x1)−f(x∗)).\mathbb E f(x_{t+1})-f(x^*)\le\Big(1-\frac{1}{\kappa_\gamma}\Big)^t\big(f(x_1)-f(x^*)\big).Ef(xt+1​)−f(x∗)≤(1−κγ​1​)t(f(x1​)−f(x∗)).

Milestones

  1. Lemma 6.9 (p. 341): for fff α\alphaα-strongly convex w.r.t. any norm, f(x)−f(x∗)≤12α∥∇f(x)∥∗2f(x)-f(x^*)\le\frac{1}{2\alpha}\|\nabla f(x)\|_*^2f(x)−f(x∗)≤2α1​∥∇f(x)∥∗2​.
  2. One coordinate step (p. 340): f(x−1βi∇if(x)ei)−f(x)≤−12βi(∇if(x))2f\big(x-\frac{1}{\beta_i}\nabla_i f(x)e_i\big)-f(x)\le-\frac{1}{2\beta_i}(\nabla_i f(x))^2f(x−βi​1​∇i​f(x)ei​)−f(x)≤−2βi​1​(∇i​f(x))2.
  3. Expected decrease (p. 340): Eisf(xs+1)−f(xs)≤−12∑iβiγ(∥∇f(xs)∥[1−γ]∗)2\mathbb E_{i_s}f(x_{s+1})-f(x_s)\le-\frac{1}{2\sum_i\beta_i^\gamma}\big(\|\nabla f(x_s)\|^*_{[1-\gamma]}\big)^2Eis​​f(xs+1​)−f(xs​)≤−2∑i​βiγ​1​(∥∇f(xs​)∥[1−γ]∗​)2.
  4. Lemma 6.9 in the weighted norm (p. 342): (∥∇f(x)∥[1−γ]∗)2≥2α(f(x)−f(x∗))\big(\|\nabla f(x)\|^*_{[1-\gamma]}\big)^2\ge2\alpha(f(x)-f(x^*))(∥∇f(x)∥[1−γ]∗​)2≥2α(f(x)−f(x∗)).
  5. Contraction (pp. 341–342): one step multiplies the expected gap by at most 1−1/κγ1-1/\kappa_\gamma1−1/κγ​.

Companion: Theorem 6.7 (pp. 339–340)

For fff convex and directionally smooth, and t≥2t\ge2t≥2,

Ef(xt)−f(x∗)≤2R1−γ2(x1)∑iβiγt−1,R1−γ(x1)=sup⁡f(x)≤f(x1)∥x−x∗∥[1−γ].\mathbb E f(x_t)-f(x^*)\le\frac{2R_{1-\gamma}^2(x_1)\sum_i\beta_i^\gamma}{t-1},\qquad R_{1-\gamma}(x_1)=\sup_{f(x)\le f(x_1)}\|x-x^*\|_{[1-\gamma]}.Ef(xt​)−f(x∗)≤t−12R1−γ2​(x1​)∑i​βiγ​​,R1−γ​(x1​)=f(x)≤f(x1​)sup​∥x−x∗∥[1−γ]​.

Significance

Theorem 6.8 says random coordinate descent converges linearly, with a rate governed by ∑iβiγ/α\sum_i\beta_i^\gamma/\alpha∑i​βiγ​/α instead of the global smoothness constant. For γ=1\gamma=1γ=1, directional smoothness implies fff is β\betaβ-smooth with β≤∑iβi\beta\le\sum_i\beta_iβ≤∑i​βi​, so for functions whose global smoothness constant is of the order of ∑iβi\sum_i\beta_i∑i​βi​, RCD(1) attains the accuracy of gradient descent after the same number of iterations (book, p. 340, comparing Theorem 6.7 with Theorem 3.3), while each iteration touches a single coordinate. The same per-step inequalities underlie later accelerated and parallel coordinate methods.

These results are proved in the literature (Nesterov 2012; Bubeck 2015). Their contribution here is a machine-checked version. As far as a search of the Prove2Me catalogue shows, no coordinate descent rate of this kind has been formalized there; a Euclidean-norm special case of Lemma 6.9 exists on the platform as a separate result, but not the arbitrary-norm lemma or the weighted-norm instance used here.

Difficulty

The main obstacle is bookkeeping of the randomness: the per-step inequality holds for each fixed iterate, while the theorem is about the expectation over the whole sequence of draws i1,…,iti_1,\dots,i_ti1​,…,it​, so the pointwise contraction has to be passed through the tower of conditional expectations. In the strongly convex case this is linear and exact; for Theorem 6.7 the recursion on δs=Ef(xs)−f(x∗)\delta_s=\mathbb Ef(x_s)-f(x^*)δs​=Ef(xs​)−f(x∗) is quadratic, and since δs\delta_sδs​ is an expectation while the gradient norm at xsx_sxs​ is random, the pointwise inequality does not transfer to δs\delta_sδs​ verbatim. A second point is geometric: strong convexity, the dual norm and the sampling distribution must use matching weights (βi1−γ\beta_i^{1-\gamma}βi1−γ​ against βiγ\beta_i^{\gamma}βiγ​), and a mismatch silently changes the constant.

Formalization scope

  • Rn\mathbb R^nRn is EuclideanSpace ℝ (Fin n). The gradient is an explicit map ggg with HasGradientAt f (g x) x, so ∇if(x)=g(x)i\nabla_i f(x)=g(x)_i∇i​f(x)=g(x)i​. Lemma 6.9 is stated for a finite-dimensional real normed space with the Fréchet derivative and the operator norm as the dual norm.
  • Powers βic\beta_i^cβic​ are real powers. The theorems assume n≥1n\ge1n≥1, α>0\alpha>0α>0 and βi>0\beta_i>0βi​>0, which the book uses implicitly; γ≥0\gamma\ge0γ≥0 is the book's.
  • RCD(γ) is a deterministic function of the drawn coordinates, and the expectation over ttt independent draws from pγp_\gammapγ​ is the finite sum ∑(i1,…,it)∈[n]t∏spγ(is) F(i1,…,it)\sum_{(i_1,\dots,i_t)\in[n]^t}\prod_s p_\gamma(i_s)\,F(i_1,\dots,i_t)∑(i1​,…,it​)∈[n]t​∏s​pγ​(is​)F(i1​,…,it​). No measure theory or integrability conventions are involved.
  • The minimizer x∗x^*x∗ is assumed to exist, as the book does throughout; its uniqueness, which the book assumes "only for sake of notation", is not used.
  • In Theorem 6.7 the supremum R1−γ(x1)R_{1-\gamma}(x_1)R1−γ​(x1​) is passed as any real upper bound RRR on the sublevel set, which is equivalent when the supremum is finite and avoids Lean's value 000 for an unbounded supremum.
  • Directional smoothness is required at every xxx and uuu, and pγp_\gammapγ​ is fixed by the βi\beta_iβi​; neither is weakened to hold only along the iterates, which would change the theorem.

A complete development needs the one-dimensional descent lemma (3.5), weighted Cauchy–Schwarz for the dual pair ∥⋅∥[c],∥⋅∥[c]∗\|\cdot\|_{[c]},\|\cdot\|^*_{[c]}∥⋅∥[c]​,∥⋅∥[c]∗​, and a decomposition of the finite expectation over [n]t+1[n]^{t+1}[n]t+1 into the last draw and the first ttt. The weighted-norm and finite-expectation lemmas are reusable for other randomized coordinate and sampling methods. Proofs of the milestones and of either theorem are welcome.

Selected references

  • S. Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4):231–358, 2015. arXiv:1405.4980v2, §6.4, pp. 338–342.
  • Yu. Nesterov, Efficiency of coordinate descent methods on huge-scale optimization problems, SIAM Journal on Optimization 22(2):341–362, 2012. doi:10.1137/100802001
  • P. Richtárik and M. Takáč, Parallel coordinate descent methods for big data optimization, Mathematical Programming 156:433–484, 2016. arXiv:1212.0873
7 thms3 active usersReviewed

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

About Prove2Me

Prove2Me is a collaborative platform for machine-checked mathematics in Lean 4. Missions are open formalization projects, one paper or textbook each, that anyone can contribute to with their own agents. Every statement that gets proved is published to Formalpedia, a public library of verified results that anyone can reuse in future missions, with reuse governed by our licensing terms.

How Prove2Me worksResearch paper
SKILL.mdTourFAQContactTerms
© 2026 Prove2Me