Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Critical Parameters for Scalable Distributed Learning with Large Batches and Asynchronous Updates
 
Loading...
Thumbnail Image
conference paper

Critical Parameters for Scalable Distributed Learning with Large Batches and Asynchronous Updates

Stich, Sebastian Urban  
•
Mohtashami, Amirkeivan  
•
Jaggi, Martin  
2021
24Th International Conference On Artificial Intelligence And Statistics (Aistats)
24th International Conference on Artificial Intelligence and Statistics (AISTATS)

It has been experimentally observed that the efficiency of distributed training with stochastic gradient (SGD) depends decisively on the batch size and—in asynchronous implementations—on the gradient staleness. Especially, it has been observed that the speedup saturates beyond a certain batch size and/or when the delays grow too large. We identify a data-dependent parameter that explains the speedup saturation in both these settings. Our comprehensive theoretical analysis, for strongly convex, convex and non-convex settings, unifies and generalized prior work directions that often focused on only one of these two aspects. In particular, our approach allows us to derive improved speedup results under frequently considered sparsity assumptions. Our insights give rise to theoretically based guidelines on how the learning rates can be adjusted in practice. We show that our results are tight and illustrate key findings in numerical experiments.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

stich21a.pdf

Type

Publisher's Version

Access type

openaccess

License Condition

Copyright

Size

2.79 MB

Format

Adobe PDF

Checksum (MD5)

d2c7339be14ad874b09277760fd5be4d

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés