Mini-Batch Covariance, Diffusion Limits, and Oracle Complexity in Stochastic Gradient Descent: A Sampling-Design Perspective
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
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2015An Introduction to Matrix Concentration Inequalitiescited by 391
- 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 298
- 2018An Empirical Model of Large-Batch Trainingcited by 268
- 2018Measuring the Effects of Data Parallelism on Neural Network Trainingcited by 190
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 183
- 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 141
- 2022Homogenization of SGD in high-dimensions: Exact dynamics and generalization propertiescited by 26
Built on by
Not cited yet.