mlsanityData to 7 Oct 2026

Mini-Batch Covariance, Diffusion Limits, and Oracle Complexity in Stochastic Gradient Descent: A Sampling-Design Perspective

arXiv 2603.02417 ↗First posted 2 Mar 2026stat.ML cs.LG math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

Stochastic gradient descent (SGD) is central to simulation optimization, stochastic programming, and online M-estimation, where sampling effort is a decision variable. We study the mini-batch gradient noise as a sampling-design object. Under exchangeable fresh-sampling mini-batches, the conditional covariance given the de Finetti directing measure mu is b^{-1} G_mu(theta), and under identifiability the projected population object is b^{-1} G*(theta) -- projected Fisher information for correctly specified likelihoods, the sandwich partner of the Hessian otherwise. This identification fixes the noise matrix entering the diffusion analysis of constant-step SGD: the raw iterate path has a deterministic fluid limit, and the sqrt(b/eta)-scaled fluctuations satisfy a functional CLT with noise covariance G*; near a nondegenerate optimum the limit is Ornstein-Uhlenbeck, and its Lyapunov covariance scaled by eta/b matches the linearized discrete recursion at leading order. Under a curvature-noise compatibility condition mu_F > 0, we prove 1/N mean-square upper bounds and an i.i.d. parametric Fisher van Trees lower bound of the same rate order, with oracle-complexity guarantees depending on an effective dimension d_eff and condition number kappa_F. Numerical experiments verify the identification and confirm the Lyapunov predictions in direct SGD.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 33 references that are in the map. First posted on arXiv 2 Mar 2026.

Builds on

10 of 33 references.

References · 10

  1. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
  2. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
  3. 2014New insights and perspectives on the natural gradient methodcited by 435
  4. 2015An Introduction to Matrix Concentration Inequalitiescited by 391
  5. 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 298
  6. 2018An Empirical Model of Large-Batch Trainingcited by 268
  7. 2018Measuring the Effects of Data Parallelism on Neural Network Trainingcited by 190
  8. 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 183
  9. 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 141
  10. 2022Homogenization of SGD in high-dimensions: Exact dynamics and generalization propertiescited by 26

Built on by

Not cited yet.