We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Stochastic Training is Not Necessary for Generalization

Abstract: It is widely believed that the implicit regularization of SGD is fundamental to the impressive generalization behavior we observe in neural networks. In this work, we demonstrate that non-stochastic full-batch training can achieve comparably strong performance to SGD on CIFAR-10 using modern architectures. To this end, we show that the implicit regularization of SGD can be completely replaced with explicit regularization even when comparing against a strong and well-researched baseline. Our observations indicate that the perceived difficulty of full-batch training may be the result of its optimization properties and the disproportionate time and effort spent by the ML community tuning optimizers and hyperparameters for small-batch training.
Comments: 25 pages, 6 figures. Code published at github.com/JonasGeiping/fullbatchtraining. Decompressed version of paper published at ICLR 2022
Subjects: Machine Learning (cs.LG); Optimization and Control (math.OC)
Cite as: arXiv:2109.14119 [cs.LG]
  (or arXiv:2109.14119v2 [cs.LG] for this version)

Submission history

From: Jonas Geiping [view email]
[v1] Wed, 29 Sep 2021 00:50:00 GMT (553kb,D)
[v2] Tue, 19 Apr 2022 22:01:13 GMT (1007kb,D)

Link back to: arXiv, form interface, contact.