Skip to content
nonconvexcard

Derive the constant-step SGD noise floor

ProofGraduate⏱ 1 hourSGDsgdnoise floorconstant steprecursionflagship

Problem

Let ff be μ\mu-strongly convex and LL-smooth, and let SGD use a constant step η≤1/(2L)\eta\le1/(2L) with an unbiased oracle of variance at most σ2\sigma^2. Prove

E∥xT−x⋆∥22≤(1−ημ)T∥x0−x⋆∥22+ησ2μ,\E\|x_T-x^\star\|_2^2 \le (1-\eta\mu)^T\|x_0-x^\star\|_2^2 + \frac{\eta\sigma^2}{\mu},

and interpret each term. Then determine the η\eta minimising the bound for a fixed budget TT.

Write a complete argument. Every inequality you use should be one you could state precisely.

Hints

Hints — 0 of 3 revealed

Solution

Opening this before you have a real attempt on paper is the fastest way to feel like you understand something you cannot reproduce.

Related problems

All problems
SolvedProblemFieldTypeDifficultyTime
Mini-batching and the compute budgetSGDNumericalIntermediate30 min
Reading the Robbins–Monro conditions off the proofSGDProofIntermediate30 min
SGD on convex Lipschitz objectivesSGDProofIntermediate30 min
Expected error does not give a high-probability guaranteeSGDProofAdvanced30 min
Implement SGD and locate the noise floor runSGDCodingAdvanced3 hours