First-Order and Stochastic Optimization Methods for Machine Learning IV: Variance-Reduced Mirror Descent for Finite-Sum ProblemsTextbook
Motivation
Empirical-risk-minimization objectives in machine learning are finite sums: , one smooth term per training example (or per worker, in a distributed setting), plus a simple nonsmooth regularizer . Chapter 4's basic stochastic mirror descent handles this by sampling a single random component gradient as an unbiased estimator of — but that estimator's variance is a constant throughout the algorithm, which caps the achievable convergence rate. Variance-reduced mirror descent asks a sharper question: can an unbiased finite-sum gradient estimator be built whose variance itself vanishes as the algorithm approaches the optimum? The answer — periodic full-gradient snapshots combined with single-component corrections — is the SVRG-style idea this mission formalizes in Lan's general-norm mirror-descent framework, with an explicit, sampling-distribution-dependent constant rather than a generic .
Setting
Fix a closed convex set in a real normed space , and the finite-sum composite problem (Eq. (5.3.1)), where is the average of smooth convex component functions, each with -Lipschitz gradient (), and is a simple, possibly nondifferentiable convex function. is possibly -strongly convex, (Eq. (5.3.2)); this mission's goal takes (§5.3.1, "Smooth Problems Without Strong Convexity"). A fixed probability distribution on the component indices governs the algorithm's random sampling, and
is the section's key aggregate smoothness constant (Eq. (5.3.4)), replacing the plain average wherever component-wise variance enters the analysis. Variance-reduced mirror descent (Algorithm 5.6) is a multi-epoch method: each epoch of length recomputes a full gradient at a snapshot point , then runs inner iterations using the estimator and the mirror-descent-with-composite-term update , where is the Bregman divergence of a fixed distance-generating function, exactly as in Chapters 3-4.
Formalization targets
Goal — Corollary 5.8
With , , and the doubling epoch schedule , (Eq. (5.3.17)),
for every epoch count , where is the weighted average of the epoch snapshots (Eq. (5.3.16)).
Supporting milestones, in attack order
- Lemma 5.12 — the per-component gradient-variation bound , the basic smoothness consequence from which the estimator's variance bound is built.
- Lemma 5.13 — unbiasedness () and two variance bounds ( and ) for the variance-reduced estimator's error .
- Lemma 5.14 — the one-step progress bound combining Lemma 5.13's variance control with the mirror-descent update's three-point inequality.
- Theorem 5.6 — the general epoch-level convergence bound (with an arbitrary epoch-length schedule and stepsize satisfying ) that Corollary 5.8 instantiates.
Every constant is exactly the book's: 's own sampling-distribution-dependent definition (never specialized to uniform ), and Corollary 5.8's explicit , , — not a generic — are all taken verbatim.
Significance
This is the series' first genuinely finite-sum result: unlike Chapters 3-4's single abstract objective , here is structurally a named average of component functions, and the sampling distribution over those components is a first-class free parameter of both the algorithm and the analysis (not fixed to uniform sampling) — itself depends on this choice, and a formalization that hard-codes would understate what Lemma 5.12's own proof needs. Getting Theorem 5.6/Corollary 5.8 right also requires keeping two nested indices straight: inner iterations within an epoch, and outer epoch counts , with the convergence bound stated in terms of the epoch count alone — and keeping the two "gap" quantities (an objective-value gap) and (a Bregman-divergence gap) distinct throughout, since they enter Corollary 5.8's final bound with different explicit coefficients ( vs. ) and neither generically bounds the other.
No result on the platform models a finite-sum objective with named component functions
sampled by a general index distribution , a variance-reduction snapshot/anchor point, or
this specific SVRG-style estimator, as of 2026-09-18 (q=finite sum, q=variance reduction,
q=SVRG, q=component function, q=variance reduced gradient, q=mirror descent finite sum —
see Prior art below).
Difficulty
The central difficulty is Theorem 5.6's own epoch-weight sequence : the book defines
explicitly only for (Eq. (5.3.14)), yet the
displayed sums in (5.3.15)-(5.3.16) run from . A 2026-09-19 revision found
that this, combined with the epoch snapshot being constrained only by membership in
and not tied to the algorithm's own dynamics, made the originally drafted statements false,
not merely incomplete: an adversarial, unboundedly-large-, -independent
together with violates the stated conclusion. The fix restores the connection via an
auxiliary epoch-boundary sequence and the per-epoch progress inequality Theorem 5.6's own proof
derives from Lemma 5.14 (see epoch_convergence_bound's hepoch hypothesis), and resolves
by extending (5.3.14)'s domain to via a fixed "epoch 0" length — w_1 is no longer
left free beyond positivity. finite_sum_variance_reduced_rate instantiates
concretely, reproducing the arithmetic Corollary 5.8's own proof is internally consistent with
(, matching the closed form ) — this was
previously only a documented-but-unresolved observation, not yet a stated hypothesis.
Formalization scope
All five items are stated over a general real normed space [NormedAddCommGroup E] [NormedSpace ℝ E], matching the mirror-descent chunks' general-norm convention (never specialized to Euclidean
space or squared distance) — is a free two-point function throughout, and each ,
, are continuous linear functionals E →L[ℝ] ℝ, whose Mathlib operator norm
supplies the dual norm with no separate definition needed. This is the trivializing
formalization this mission rules out: hard-coding (uniform sampling) or (Euclidean Bregman divergence) would understate both 's dependence on the sampling
distribution (the whole point of Lemma 5.12's bound) and the general-norm apparatus the rest of
this book series shares.
Ψ(x_0)-Ψ(x^*) and V(x_0,x^*) are kept as two syntactically distinct terms throughout — never
conflated or bounded one by the other — matching Corollary 5.8's own two separate coefficients.
Corollary 5.8's own explicit constants (, , ) are stated verbatim rather
than left as an unspecified , per Hard Rule 6.
Left out of scope, for time: the gradient-computation-count complexity bound (Eq. (5.3.19), an
statement about total oracle calls, not a convergence-rate inequality on ) and
§5.3.2's strongly-convex case (Theorem 5.7, a geometric-decay bound
under ) are natural continuations reusing this mission's variance_reduced_progress_bound
milestone, not attempted here.
Prior art
q=finite sum, q=variance reduction, q=SVRG, q=component function, q=variance reduced gradient, and q=mirror descent finite sum were all searched on 2026-09-18. The only
topically-adjacent hit across all six queries is ShiOptRates.Stochastic.variance_purchase_ classical ("Classical variance reduction is cost-neutral..."), which models plain minibatch SGD
on a smooth objective with an i.i.d.-noise oracle characterized by a single scalar variance
and a minibatch-size trade-off — no finite-sum structure with named component
functions, no sampling distribution , no snapshot/anchor point , and a
different question (cost-neutrality of minibatch size vs. this mission's convergence rate for a
fixed variance-reduction scheme). Not reused; every item in this mission is drafted fresh.
Selected references
- G. Lan, First-Order and Stochastic Optimization Methods for Machine Learning, Springer Series in the Data Sciences, Springer 2020, Chapter 5, §5.3. https://doi.org/10.1007/978-3-030-39568-1
- R. Johnson, T. Zhang, "Accelerating stochastic gradient descent using predictive variance reduction," Advances in Neural Information Processing Systems (NeurIPS), 2013 (the SVRG estimator this section's gradient estimator generalizes to the composite mirror-descent setting).
- 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.