First-Order and Stochastic Optimization Methods for Machine Learning III: Stochastic Mirror DescentTextbook
Motivation
Machine learning's canonical training objective — minimize an expected or empirical risk over a data distribution — is almost never observed exactly: at each step an algorithm sees only a noisy gradient sample (a minibatch gradient, a single-example gradient, a simulation draw). Stochastic mirror descent (Nemirovski, Juditsky, Lan & Shapiro 2009) is the modern, general-norm answer to "what happens to first-order convergence guarantees when the gradient itself is a random variable": it takes the deterministic mirror-descent scheme of the previous chapter and replaces the exact subgradient with an unbiased stochastic estimate, and asks for both an expected convergence rate and, when the noise is well-behaved, an explicit probability-of-large-deviation guarantee. This is the theoretical backbone of stochastic gradient descent as used in practice.
Setting
Fix a nonempty closed convex set in a real normed space , and a convex with and an arbitrary minimizer, exactly as in Chapter 3. A stochastic oracle , queried at a point with a fresh random sample , returns an estimate of a subgradient : (unbiasedness), (a dual-norm Lipschitz bound, Eq. (4.1.7)), and (a second-moment/variance bound). The stochastic mirror-descent update is exactly Chapter 3's mirror-descent update with in place of the deterministic : (Eq. (4.1.6)), where is the Bregman divergence of a fixed distance-generating function .
Formalization targets
Goal — Theorem 4.1
Supporting milestones, in attack order
- Lemma 3.4, invoked for the stochastic update: the same three-point inequality as the deterministic mirror-descent update, restated with the stochastic gradient functional in place of — the book's own remark ("It can be easily seen that the result in Lemma 3.4 holds with replaced by ") is exactly what licenses treating this as the same algebraic fact for a fixed sample path.
- Lemma 4.1: the martingale-difference deviation bound, a Chernoff-type concentration inequality for a conditionally sub-Gaussian martingale-difference sequence — the chapter's general-purpose probabilistic tool, proved independently of the optimization setting.
Every constant is exactly the book's; (not a generic ) is the goal's own noise-dependent constant, taken verbatim.
Significance
This is the first mission in the series to leave the purely deterministic, real-analytic setting of Chapters 2-3 and formalize a genuinely probabilistic convergence guarantee: an expectation taken over an entire random algorithm trajectory , not merely over a single random variable. Getting the goal theorem's statement right requires being explicit about exactly which quantities are random (the iterates , hence and ) and which are deterministic constants fixed in advance (), and about the precise mathematical content of "the stochastic gradient's bias vanishes after conditioning on the past" — Lemma 4.1 is included specifically because it is the general machine that makes that vanishing rigorous, independent of the optimization application.
No result matching stochastic mirror descent, Assumption 4's sub-Gaussian/light-tail condition, or
this martingale-difference concentration lemma exists on the platform as of 2026-09-18 (q= stochastic gradient, q=stochastic mirror descent, q=martingale, q=sub-Gaussian — see
Prior art below for what these queries actually returned).
Difficulty
The central difficulty is disentangling which facts in the chapter's proof genuinely need
measure theory and which do not. The per-step algorithmic relations — 's minimality,
's subgradient inequality at , the dual-norm bound on — hold for every sample path
individually and are formalized pointwise in , exactly as chunk 03-deterministic
formalizes its deterministic analogues; only the second-moment bound and the final expectation
inequality are genuine integrals. The one place this pointwise treatment cannot simply mirror the
deterministic case is the noise cross-term :
in the book's proof this vanishes because is conditionally mean-zero given
the past and is a function of the past (the martingale-difference property, via the tower
property of conditional expectation) — a genuinely non-pointwise fact. Rather than thread an
explicit filtration through the goal theorem's own statement (which Lemma 4.1 already does, as
the chapter's dedicated home for that machinery), the goal theorem takes this post-tower-property
consequence directly as a named hypothesis (hcross); see Formalization scope.
Formalization scope
stochastic_mirror_iterate_three_point and stochastic_mirror_descent_bound are stated over a
general real normed space [NormedAddCommGroup E] [NormedSpace ℝ E], matching chunk
03-deterministic's general-norm milestones (mirror_iterate_three_point/mirror_descent_bound)
rather than the Euclidean/inner-product specialization of that chunk's §3.1 items — Chapter 4's
own stochastic mirror descent is presented directly in the general-norm framework of §3.2, with no
Euclidean-only warm-up. is left a free two-point function (never hard-coded to a squared
Euclidean distance), and the stochastic gradient and the subgradient selector are
continuous linear functionals E →L[ℝ] ℝ, whose Mathlib operator norm supplies the dual norm
with no separate definition needed — the same trivializing formalization chunk
03-deterministic rules out (specializing to the Euclidean case) applies here and is ruled
out the same way.
martingale_difference_deviation_bound (Lemma 4.1) is a standalone probabilistic result,
formalized with Mathlib's MeasureTheory.Filtration and condExp machinery: the sequence
's generated filtration, 's -measurability, and the two
conditional-expectation hypotheses (conditional mean zero, conditional sub-Gaussian tail) are all
literal translations of the book's own E|ξ[t-1] notation.
Left out of scope, for time: Assumption 4 (the light-tail/sub-Gaussian oracle assumption),
Proposition 4.1 (the large-deviation bound under Assumption 4, which chains Lemma 4.1's
concentration bound with the constant stepsize policy (4.1.11) and a second Markov-inequality
argument on ), Lemma 4.2 and Theorem 4.2 (the smooth- case,
§4.1.2, requiring a separate recursion and averaging convention ). All four are natural
continuations reusing this mission's stochastic_mirror_iterate_three_point and/or
martingale_difference_deviation_bound; a later mission or an amendment to this one could add
them without touching what is here. Per Hard Rule 7 (faithfulness over coverage), a genuinely
faithful formalization of Proposition 4.1 in particular — which needs Assumption 4's own
conditional-MGF hypothesis threaded consistently with Lemma 4.1's, plus the constant-stepsize
substitution and a second concentration argument — was judged to need more time than this
session's budget allowed to do without shortcuts; it is named here rather than approximated.
Selected references
- G. Lan, First-Order and Stochastic Optimization Methods for Machine Learning, Springer Series in the Data Sciences, Springer 2020, Chapter 4, §4.1. https://doi.org/10.1007/978-3-030-39568-1
- A. Nemirovski, A. Juditsky, G. Lan, A. Shapiro, "Robust stochastic approximation approach to stochastic programming," SIAM Journal on Optimization, 19(4), 2009, pp. 1574-1609.
- H. Robbins, S. Monro, "A stochastic approximation method," Annals of Mathematical Statistics, 22(3), 1951, pp. 400-407 (origin of stochastic approximation).