MMODELYST
Papers/Worker Disagreement Reveals Sharp Directions in Local SGD
PAP

Worker Disagreement Reveals Sharp Directions in Local SGD

May 26, 2026

arXiv
Abstract

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.

Select text to highlight · click a highlight to remove · saved in this browser only
Authors
Tolga Dimlioglu, Kristi Topollai, Anna Choromanska
Your notes (browser-local)
saved
arXiv:2605.27739