Worker Disagreement Reveals Sharp Directions in Local SGD
Not on the map. Why?
Abstract (arXiv)
Deep neural network training often exhibits highly anisotropic loss geometry, where a few sharp dominant Hessian directions coexist with a large flatter bulk. Gradients tend to align disproportionately with these dominant directions, although stable progress often requires movement through flatter bulk directions. Estimating the dominant subspace is therefore useful but costly with direct Hessian-based methods. We show that standard Local SGD exposes this geometry through worker disagreement. We theoretically show that the worker-average gap covariance is shaped by stochastic-gradient noise and Hessian curvature, causing workers to disagree along sharp, curvature-sensitive directions. Thus, worker-average gaps provide a cheap Hessian-free estimator of the dominant subspace. Experiments on MLPs, CNNs, and Transformers show that subspaces formed by worker-average gaps capture a substantial fraction of the gradient component lying in the dominant Hessian eigenspace.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 21 references that are in the map. First posted on arXiv 26 May 2026.
Builds on
11 of 21 references.
References · 11
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2018Local SGD Converges Fast and Communicates Littlecited by 595
- 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 289
- 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 283
- 2018Gradient Descent Happens in a Tiny Subspacecited by 208
- 2016Eigenvalues of the Hessian in Deep Learning: Singularity and Beyondcited by 156
- 2019Local SGD with Periodic Averaging: Tighter Analysis and Adaptive Synchronizationcited by 106
- 2024Does SGD really happen in tiny subspaces?cited by 35
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptationcited by 26
- 2022On the Overlooked Structure of Stochastic Gradientscited by 13
- 2026On the Superlinear Relationship between SGD Noise Covariance and Loss Landscape Curvaturecited by 2
Built on by
Not cited yet.