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. Refined Convergence and Topology Learning for Decentralized SGD with Heterogeneous Data
 
conference paper

Refined Convergence and Topology Learning for Decentralized SGD with Heterogeneous Data

Le Bars, Batiste
•
Bellet, Aurélien
•
Tommasi, Marc
Show more
2023
Proceedings of the 26th International Conference on Artificial Intelligence and Statistics (AISTATS)
26th International Conference on Artificial Intelligence and Statistics (AISTATS)

One of the key challenges in decentralized and federated learning is to design algorithms that efficiently deal with highly heterogeneous data distributions across agents. In this paper, we revisit the analysis of the popular Decentralized Stochastic Gradient Descent algorithm (D-SGD) under data heterogeneity. We exhibit the key role played by a new quantity, called neighborhood heterogeneity, on the convergence rate of D-SGD. By coupling the communication topology and the heterogeneity, our analysis sheds light on the poorly understood interplay between these two concepts. We then argue that neighborhood heterogeneity provides a natural criterion to learn data-dependent topologies that reduce (and can even eliminate) the otherwise detrimental effect of data heterogeneity on the convergence time of D-SGD. For the important case of classification with label skew, we formulate the problem of learning such a good topology as a tractable optimization problem that we solve with a Frank-Wolfe algorithm. As illustrated over a set of simulated and real-world experiments, our approach provides a principled way to design a sparse topology that balances the convergence speed and the per-iteration communication costs of D-SGD under data heterogeneity.

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

le-bars23a.pdf

Type

Main Document

Version

Published version

Access type

openaccess

License Condition

CC BY

Size

1.4 MB

Format

Adobe PDF

Checksum (MD5)

c040989ad32e71074909e31e51c366ac

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