Context. Online SGD estimating a unit vector from a stream of samples starts from a random point, whose correlation with the truth is about : on the equator, where per-sample gradient noise dwarfs the population gradient. The ODE method assumes a warm start and scaling limits fix , so it was open how many samples SGD needs just to leave the equator, and what sets that power of . Ben Arous et al. (2021) answer for online SGD, and Damian et al. (2023) show how to do better. Only the population loss near the equator matters. If has its first nonzero derivative at in order (the information exponent), the drift at the start is against order-one noise per step, and escaping costs about samples for , to for , and for ; once the correlation is macroscopic the descent is an ODE and takes a vanishing fraction of the data. Smoothing the loss over a cap of radius about around the iterate picks up signal from the far side of the cap and reduces the noise faster than the signal, which lowers the cost to at , the correlational-query lower bound.
Setting. Samples are i.i.d. from , of them, with population loss , . Spherical online SGD takes and renormalizes, one fresh sample per step. Assumption A: on . Assumption B: the sample-wise error has gradient noise of order one along and order in norm ( moments), as for an isotropic subgaussian vector. For the single-index model with Gaussian and square loss, the noise operator is diagonal on Hermite polynomials, so and is the index of the first nonzero Hermite coefficient with . Linear, ReLU, sigmoid, and canonical-link GLMs have ; phase retrieval, online PCA and the two-Gaussian mixture have ; the spiked -tensor has ; has . With a misspecified link the sign of decides whether SGD moves toward at all. Damian et al. write for , for , for , a known link , and the correlation loss .
Ben Arous et al.'s theorems. Let for . Recovery (Theorem 1.3): under A and B, with and , , respectively, SGD from a uniform start on the upper hemisphere has in probability; in samples that is , and . Refutation (Theorem 1.4): if and () or (), then , so the thresholds are sharp up to ; this is about online SGD in these step ranges only, since information-theoretically order samples suffice. Descent (Theorems 3.2 and 1.5): from with , SGD tracks projected gradient descent on , plays no role, and the climb to takes order samples, small next to the search.
Damian et al.'s theorems. Smoothing averages over the tangent sphere, with uniform on , and the algorithm is online spherical SGD on . Theorem 1: from , with and step for steps, then and , the last iterate has loss with high probability, so up to polylogs; recovers the Ben Arous et al. rate. Theorem 2: any correlational statistical query (CSQ) algorithm making queries of tolerance that reaches loss below needs , which with means .
The escape time is set by a drift budget and a noise budget. Track only . To first order, a drift from the Taylor expansion of plus the increment of the directional error martingale. In ODE time the drift from needs time , and for , and the run has ODE time . The martingale over all steps is by Doob, and it must stay below the lead , so and the available ODE time is at most ; escape requires or , which is up to logs. During the search the noise is as large as , so the trajectory does not concentrate, and the proof works without a law of large numbers.
The normalization costs less than half the drift as long as . Since , the spherical step is the Step 1 increment divided by , The division shrinks by about per step, second order but with of order one, and it stays below the drift iff . At this is the step cap , a second reason small steps are forced for . The rigorous version truncates and uses Markov and Freedman's inequality with a union bound.
The martingale stays below half the initial correlation, which gives a one-sided comparison. The proof does not need the noise to average out, only to stay below half the initial correlation, and the drift then compounds on the remaining half. Doob's maximal inequality keeps the directional martingale below up to , so with Step 2, for , with probability , The condition is the noise budget of Step 1.
Solving the inequality gives the three escape times. It is a discrete version of the drift ODE started at . For growth is linear, ; for discrete Grönwall gives , so ; for the discrete Bihari-LaSalle inequality compares with the finite-time blow-up of , The budget checks and become, with , Theorem 1.3's conditions; for the is exactly . A uniform start has with probability near 1.
After weak recovery, a law of large numbers by Grönwall. From the drift is order one per unit ODE time, so noise of total size cannot matter. Linearizing SGD and the population iterate, the dropped normalization terms sum to , Doob controls the martingale, and Lipschitz continuity of the drift with discrete Grönwall gives on a window long enough for the population iterate to reach , after which Assumption A keeps SGD close. The strong Markov property at connects this to Step 4.
The lower bound runs the same comparison upward, excursion by excursion. Upward the normalization only helps, since , so while Restart the comparison at every positive excursion: Chebyshev and a union bound keep each noise increment below , so every excursion starts below , and reverse Grönwall or Bihari-LaSalle bounds keep below for order , and steps, which exceeds under the hypotheses. The step ranges come from the choice of : for exponential growth sees the start only through , so covers every ; for a larger start shortens the blow-up, which forces .
Online SGD moves the overlap at the rate signal/noise. Damian et al. reread Step 1 as a signal-to-noise ratio. With the negative spherical gradient and renormalization, and with the largest step whose drag stays below the drift the overlap follows , . Unsmoothed, the signal is of order by Hermite orthogonality and the noise of order , so and escape from takes . The rest of the proof changes the two halves of this ratio.
Below , smoothing gives the loss a floor of signal. The cap mixes in the random tangent step , whose overlap with is about . Expanding binomially, odd moments of vanish and even ones are : The sum is dominated by its first term when and by its last, the far side of the cap, below. Smoothing nearly commutes with differentiation (a spherical Stein identity), so the smoothed drift is above the threshold and, below it, the constant for odd or linear in for even : near the equator the smoothed landscape looks like information exponent 1 or 2, with a small slope.
Smoothing shrinks the gradient noise by , as long as . The per-sample loss depends on all of , so Step 8 does not reach the noise. Hermite tensors make smoothing linear, , and their orthogonality gives the noise in a direction is the smoothed power at the overlap of the cap center with a typical cap point. That sits in Step 8's first regime exactly when , and then . Above the threshold signal and noise shrink alike; below it the signal has its floor, a gain of in SNR for odd . Past the noise falls only like , no faster than the squared signal, so the gain stops at .
Stage 1 escapes from to . A one-step progress bound makes Step 7 rigorous, and an induction with the Rosenthal-Burkholder-Pinelis inequality keeps , as in Step 3. The step keeps the drag below half the drift, and for odd the constant drift covers the distance in steps, at ; for even a Bihari-type comparison gives exponential growth in the same number of steps up to logs.
The rest of the trajectory is the unsmoothed problem. Above smoothing no longer changes the SNR, so Ben Arous et al.'s analysis at the same step carries to , and a contraction for brings it to by . Then and : near the loss is locally strongly convex, and the Robbins-Monro induction gives , the term.
The CSQ lower bound comes from many nearly orthogonal directions. There are unit vectors with pairwise overlaps at most , and Hermite orthogonality makes their models nearly uncorrelated, Szörényi's CSQ lemma then demands about queries, and gives the bound. The same that flattens the landscape at the equator makes directions indistinguishable to correlational queries.
Consequences and limits. For online SGD the difficulty of a rank-one estimation problem is one integer read off the population loss at the equator, for single-index models the degree of the link's lowest nonzero Hermite coefficient. Small changes to the activation move the cost by powers of ( needs samples, about ), while from a warm start every problem is linear, and almost all of the samples used before strong recovery go into the search. The barrier is specific to online SGD on the raw loss: smoothing reaches the CSQ rate, and experiments with fit exponents 1.5, 2.1 and 2.5 for . For the spiked tensor, matches the threshold conjectured for first-order methods without spectral initialization, and as the smoothed gradient becomes the partial-trace estimator of tensor PCA, a link the paper sketches without proof. Damian et al. conjecture, without proof, that minibatch SGD gets the same benefit implicitly. Limits: rank one, known norm, one pass, a start at the CLT scale ( can be trapped), to gaps, and a refutation specific to online SGD in its step range; smoothing needs a known link and isotropic Gaussian inputs, an initial overlap of the right sign (constant probability from a random start), and is optimal only within CSQ, since non-correlational methods (Chen and Meka) learn polynomial links with about samples.
Terms
- correlation (latitude) with the truth; a random start has , the equator
- Information exponent
- order of the first nonzero derivative of at ; drift near the equator is
- Hermite rule
- single-index square loss: , so is the first with
- ; sample complexity up to
- Directional error martingale
- ; size , so keeps it below
- Step cap
- : normalization shrinkage must stay below the drift
- average over , uniform on the tangent sphere
- overlap reached by the random part of a cap point; below it smoothing adds signal
- smoothed gradient noise, valid for
- CSQ bound
- , so with
Quiz
- Gaussian single-index model, square loss: has , but has . Why?The noise operator gives , whose first nonzero -derivative at 0 sits at the first nonzero . Subtracting removes exactly the degree-1 Hermite component of .
- What does Theorem 1.4 (the refutation) rule out?The proof bounds above by the drift ODE plus a martingale it controls only in these step ranges, so the statement is about this algorithm with these steps. Information-theoretically order samples suffice, and smoothing the loss beats .
- For , where does the dimension enter the escape cost?The drift ODE blows up after time , which is from a random start, and the noise budget turns that into . From a warm start the cost is linear.
- Why is the smoothing radius capped at ?The noise variance is the smoothed power at the self-overlap , which is in the high-overlap regime only while . Past that point the noise falls only like , no faster than the squared signal, so more smoothing gives no gain while the threshold to be crossed keeps moving out.
Sources
- Gerard Ben Arous, Reza Gheissari, and Aukosh Jagannath. Online stochastic gradient descent on non-convex losses from high-dimensional inference. Journal of Machine Learning Research, 2021.
- Alex Damian, Eshaan Nichani, Rong Ge, and Jason D. Lee. Smoothing the Landscape Boosts the Signal for SGD: Optimal Sample Complexity for Learning Single Index Models. NeurIPS, 2023.