English

Stochastic subgradient method converges on tame functions

Optimization and Control 2018-05-29 v3 Machine Learning

Abstract

This work considers the question: what convergence guarantees does the stochastic subgradient method have in the absence of smoothness and convexity? We prove that the stochastic subgradient method, on any semialgebraic locally Lipschitz function, produces limit points that are all first-order stationary. More generally, our result applies to any function with a Whitney stratifiable graph. In particular, this work endows the stochastic subgradient method, and its proximal extension, with rigorous convergence guarantees for a wide class of problems arising in data science---including all popular deep learning architectures.

Keywords

Cite

@article{arxiv.1804.07795,
  title  = {Stochastic subgradient method converges on tame functions},
  author = {Damek Davis and Dmitriy Drusvyatskiy and Sham Kakade and Jason D. Lee},
  journal= {arXiv preprint arXiv:1804.07795},
  year   = {2018}
}

Comments

32 pages, 1 figure

R2 v1 2026-06-23T01:30:28.715Z