First-Order and Stochastic Optimization Methods for Machine Learning II: Subgradient Descent, Mirror Descent and Accelerated Gradient DescentTextbook
Motivation
Gradient descent's convergence rate for a general smooth convex problem is in the function-value gap; Nemirovski and Yudin (1983) proved that no first-order method can do better than is achievable, and Nesterov (1983, 1988, 2004) constructed the first method attaining it — the accelerated (or "fast") gradient method. For thirty years this was the standard route to -rate solvers in convex optimization, and the technique underlies essentially every modern accelerated first-order method used at scale in machine learning (accelerated SGD, momentum methods, Nesterov-style extensions of Adam). The two building blocks this mission formalizes on the way there — subgradient descent (Polyak, 1960s) and mirror descent (Nemirovski & Yudin, 1983) — are themselves the default tools whenever the objective is nonsmooth or the constraint set's natural geometry is not Euclidean (e.g. the probability simplex, where mirror descent with the entropic distance-generating function beats projected subgradient descent by a factor).
Setting
Fix a nonempty closed convex set (in Lean: a normed real vector space , X : Set E) and a
convex ; write and for an arbitrary
minimizer. The projected-subgradient update is for a subgradient and stepsize
. Its generalization, mirror descent, replaces the Euclidean proximal term with a
Bregman divergence built from a
1-strongly-convex distance-generating function with respect to a general norm
(dual norm ): ,
where is now a continuous linear functional (a subgradient in the dual space, since the
norm need not come from an inner product). Choosing recovers and the plain subgradient update as a special case.
The accelerated gradient method additionally assumes has -Lipschitz gradient () and is -generalized-strongly-convex w.r.t. ( for ), and tracks three coupled sequences from :
Formalization targets
Goal — Theorem 3.6, closed-form rate
With , and :
Supporting milestones, in attack order
- Lemma 3.1 / Theorem 3.1 (Euclidean case): the three-point inequality for the plain projected-subgradient step, and the resulting bound under -Lipschitz .
- Lemma 3.4 / Theorem 3.5 (general-norm mirror descent): the same two results with the squared Euclidean distance replaced by and the Euclidean norm by a general dual pair .
- Proposition 3.1: the one-step accelerated-method recursion .
- Theorem 3.6, general form: Proposition 3.1's recursion telescoped across (with ) into a single two-term bound relating step to step .
Every constant here is exactly the book's; no milestone hides an behind an unspecified absolute constant.
Significance
The chain culminates in an explicit, non-asymptotic certificate for accelerated gradient descent — the theoretically optimal rate for smooth convex minimization by a first-order method (matching the Nemirovski–Yudin lower bound, not re-derived here). Formalizing it forces every implicit convention in a standard optimization-course derivation to become explicit: which of the three sequences a given quantity refers to, exactly which inequality (3.3.7)-(3.3.9) each specific stepsize schedule needs to satisfy, and the precise index range over which the chapter's own stated hypotheses actually get used in its own proof (see Difficulty below).
None of these six results (or their strongly-convex counterpart, Theorem 3.7, left for future
work — see Formalization scope) has a machine-checked proof on Prove2Me. The one theorem with
the same name as this mission's subject, BanditAlgorithm.mirror_descent_regret_bound
(Lattimore & Szepesvári, Theorem 28.4), is a different object: an online, adversarial regret
bound against a changing sequence of loss vectors , not an offline function-value gap for a
single fixed ; not reused. Likewise OnlineConvexOpt.FirstOrder.online_gradient_descent_regret
(Hazan) and OnlineConvexOpt.ConvexBasics.constrained_gd_well_conditioned_convergence are,
respectively, an online-regret bound and a plain-gradient-descent (non-accelerated) linear-rate
result — checked and confirmed not reusable per the mission brief.
Difficulty
The three-point inequalities (Lemmas 3.1/3.4) are routine consequences of a strongly-convex minimizer's optimality condition. The real difficulty is bookkeeping across three coupled sequences in the accelerated method: a formalization using only and (dropping , the point at which the gradient is actually evaluated) is not Lan's algorithm and proves either a false or a different bound — is what lets the method use a gradient computed at a point between and , which is exactly the extrapolation step that makes acceleration work.
A second, subtler difficulty is that Theorem 3.6's own stated hypothesis — "(3.3.15) for any " — is not quite what its proof uses. Telescoping Proposition 3.1's per-step bound via (3.3.15) requires the previous step's constants ; at these would be , values the recursion (3.3.4)-(3.3.6) never defines (it only ever uses for ). The book's own proof, read closely, invokes (3.3.15) only for , with handled directly by Proposition 3.1's conclusion connecting to the given base data . Formalizing the literal hypothesis range would either be unstatable (no exist) or vacuous (adding unused ghost parameters); this mission states the range the proof actually needs.
Formalization scope
Chapter 3's own §3.1/§3.2 split (Euclidean vs. general norm) is preserved rather than collapsed:
subgradient_iterate_three_point/subgradient_descent_bound are stated over a real inner
product space with the vector subgradient and the Euclidean norm, exactly matching
§3.1; mirror_iterate_three_point/mirror_descent_bound and the two accelerated-method
milestones are stated over a general real normed space [NormedAddCommGroup E] [NormedSpace ℝ E], with subgradients as continuous linear functionals E →L[ℝ] ℝ (whose Mathlib operator norm
is already the dual norm , needing no separate definition) and the Bregman
divergence left as a free two-point function — but, following a
2026-09-19 revision, no longer a totally free function. V is now required to satisfy the two
facts (3.2.2)/(3.2.3)/(3.2.6) actually establish and every downstream proof (Lemma 3.4, Theorem
3.5, Proposition 3.1, Theorem 3.6) uses: nonnegativity ( for ) and the
three-point/cosine identity ,
the latter made explicit via an added parameter dV : E → E → (E →L[ℝ] ℝ) read as "the gradient
of at ." Without these two hypotheses the five items that use an abstract V
(mirror_iterate_three_point, mirror_descent_bound, accelerated_one_step_recursion,
accelerated_gradient_recursion_bound, accelerated_gradient_rate) are false as stated — a
constant V satisfies the bare pointwise-minimality hypotheses while violating the conclusion, as
two worked counterexamples confirmed. This mission does not derive V/dV from an explicit
distance-generating function (the heavier, fully book-literal route (3.2.1)-(3.2.2) would);
it takes the two facts the proofs actually consume as hypotheses directly, which is lighter and
sufficient. Satisfiability is witnessed by the Euclidean case already in §3.1:
, , , exactly how
subgradient_iterate_three_point/subgradient_descent_bound already handle the Euclidean
special case. A trivializing formalization this mission rules out: specializing to the
Euclidean squared distance in mirror_iterate_three_point/mirror_descent_bound would make
those two milestones restatements of the §3.1 Euclidean results rather than genuine
generalizations, exactly the pitfall the chapter brief flags.
Every argmin-defined iterate ( in each of the three update rules) is represented by
its defining pointwise-minimality property rather than by an IsMinOn/argmin term, so no
existence or uniqueness lemma for the underlying minimization problem is needed anywhere in this
mission — matching how the book's own proofs use these updates (via their first-order optimality
condition, never via an explicit formula for the minimizer).
Left out of scope, for time: Theorem 3.7 (the strongly-convex, linear-rate
companion to Theorem 3.6, sharing Proposition 3.1 as its own base lemma) and Corollary 3.5 (the
composite-objective extension ). Both are natural continuations reusing this
mission's accelerated_one_step_recursion; a later mission or an amendment to this one could add
them as additional milestones/goals without touching what is here.
Selected references
- G. Lan, First-Order and Stochastic Optimization Methods for Machine Learning, Springer Series in the Data Sciences, Springer 2020, Chapter 3. https://doi.org/10.1007/978-3-030-39568-1
- Y. Nesterov, "A method for solving the convex programming problem with convergence rate ," Doklady AN SSSR, 269, 1983, pp. 543–547.
- Y. Nesterov, Introductory Lectures on Convex Optimization, Springer, 2004.
- A. Nemirovski and D. Yudin, Problem Complexity and Method Efficiency in Optimization, Wiley, 1983 (source of the mirror-descent method and the lower bound for smooth convex optimization).