Convex Optimization: Algorithms and Complexity XV: SVRG with η = 1/(10β) and k = 20κ Contracts the Expected Optimality Gap by 0.9 per EpochTextbook
Motivation
Many optimization problems in machine learning minimize an average of losses, one loss for each observation. A full gradient step examines every observation, while a stochastic gradient step examines one. The latter is cheaper per step, but its sampled gradient can remain noisy even near the optimum. Section 6.3 of Bubeck's monograph studies stochastic variance reduced gradient descent (SVRG), which periodically computes a full gradient at an anchor point and uses it to correct subsequent sampled gradients. The question for this mission is whether that correction gives a geometric reduction of the expected objective gap at the constants printed in Theorem 6.5.
Bubeck places this method alongside full gradient descent and stochastic gradient descent for finite sums. The section records that earlier stochastic average gradient and dual coordinate ascent methods attain a gradient-computation cost of order for the same regime, where is the number of components and is a condition number. The target here is the precise SVRG convergence statement in the book, rather than a comparison of implementation costs. The source's discussion on pp. 334–336 gives the context and the algorithm.
Setting
Let be differentiable convex functions, with , and define the finite-sum objective and its gradient by
Each component is -smooth when its gradient is -Lipschitz in the Euclidean norm: for all . The average is -strongly convex, meaning that for all it lies at least above its first-order affine approximation at . The constants and are positive, minimizes over , and .
An epoch begins at an anchor . Its first inner iterate is . For , draw uniformly from , independently across steps and epochs, and update
The next anchor is the average . In particular, this average uses through , while the last updated point is excluded. Starting from an arbitrary and repeating the epoch produces . The expectation of is over all sampled indices in the first epochs.
Formalization targets
Goal: geometric contraction across epochs
Theorem 6.5 sets and and asserts, for every ,
The epoch length is a count, so the statement takes and explicitly requires . The goal uses exactly the book's step size, epoch length, and contraction factor.
Milestones: second moments and a single epoch
Lemma 6.4 bounds by . Equation (6.3) bounds the second moment of the corrected sampled direction by the objective gaps at the current point and the anchor. Equation (6.2), the unbiased-direction display, and the one-step display express how that direction changes squared distance to . The later display on p. 338 bounds one epoch for any positive step size with . Finally, equation (6.1) substitutes the stated constants to obtain the factor for one epoch. These seven source claims form the milestone list in reading order.
Significance
The theorem gives an explicit accuracy guarantee after a specified number of epochs: an initial gap falls below in expectation. Because each epoch uses a full gradient at its anchor as well as sampled component gradients, the result makes clear which quantity contracts and which operations are counted. It is a concrete linear-rate statement for a method whose individual stochastic gradients need not approach zero at the optimum. Bubeck, §6.3 discusses this issue when introducing the correction term.
The mathematical result is already proved in the monograph. The remaining task is to produce machine-checked proofs of its precise finite-sum model, the single-index estimates, the epoch inequality, and the full repeated-epoch guarantee. The mission drafts those statements and definitions; no proof is claimed for the open theorem items. The finite uniform-average representation and the separation between a conditional one-step average and the full multi-epoch average can be reused in other finite-sum stochastic algorithms.
Difficulty
The sampled component gradient need not be small when is near , so a bound using only its norm does not yield the desired fixed-step contraction. The correction has mean zero relative to the full gradient at the current iterate, but its second moment still depends on both and . The proof must control those two gaps while respecting the fact that depends on earlier samples. A single-index estimate with held fixed and an expectation over complete sample histories are different statements; confusing them would make the goal weaker or false.
Formalization scope
The carrier is EuclideanSpace ℝ (Fin n) with its usual inner product and norm. The Fin m components and every sample array are finite. A real-valued uniform average is an ordinary finite sum divided by the number of arrays, and and prevent an empty average. Independent uniform sampling is represented by averaging over every function from step positions to component indices. The multi-epoch sample space has one such block for every epoch. There are no integrals or measurability side conditions.
The component assumptions include differentiability with an explicit gradient map, convexity on all of , and the book's gradient-Lipschitz version of smoothness. Strong convexity is imposed on the average objective alone, using the published OnlineConvexOpt.ConvexBasics.StronglyConvexOn definition on the whole space. The book's standing notation assumes a minimizing exists; this is explicit. Positivity of and , and integrality of , make the displayed divisions and epoch length meaningful. The general epoch bound also requires and . Dimension zero is allowed: the theorem remains a statement about the unique point of and its zero objective gap.
The direction always contains the sampled difference and the full anchor gradient . Replacing that direction with would define gradient descent and would not satisfy this mission's algorithm. Contributions are welcome for the finite averaging identities, the component-gradient estimate, the conditional one-step calculation, the epoch inequality, and the induction across epochs.
Selected references
- Sébastien Bubeck, Convex Optimization: Algorithms and Complexity, Foundations and Trends in Machine Learning 8(3–4), 2015, pp. 231–358. arXiv:1405.4980v2
- Rie Johnson and Tong Zhang, Accelerating Stochastic Gradient Descent using Predictive Variance Reduction, Advances in Neural Information Processing Systems 26 (NIPS), 2013 (the origin of SVRG, cited by Bubeck on p. 335). https://proceedings.neurips.cc/paper/2013/hash/ac1dd209cbcc5e5d1c6e28598e8cbbe8-Abstract.html