Communication Dans Un Congrès Année : 2023

A2CiD2: Accelerating Asynchronous Communication in Decentralized Deep Learning

Résumé

Distributed training of Deep Learning models has been critical to many recent successes in the field. Current standard methods primarily rely on synchronous centralized algorithms which induce major communication bottlenecks and synchronization locks at scale. Decentralized asynchronous algorithms are emerging as a potential alternative but their practical applicability still lags. In order to mitigate the increase in communication cost that naturally comes with scaling the number of workers, we introduce a principled asynchronous, randomized, gossip-based optimization algorithm which works thanks to a continuous local momentum named A2CiD2. Our method allows each worker to continuously process mini-batches without stopping, and run a peer-to-peer averaging routine in parallel, reducing idle time. In addition to inducing a significant communication acceleration at no cost other than adding a local momentum variable, minimal adaptation is required to incorporate A2CiD2 to standard asynchronous approaches. Our theoretical analysis proves accelerated rates compared to previous asynchronous decentralized baselines and we empirically show that using our A2CiD2 momentum significantly decrease communication costs in poorly connected networks. In particular, we show consistent improvement on the ImageNet dataset using up to 64 asynchronous workers (A100 GPUs) and various communication network topologies.
Fichier principal
Vignette du fichier
ACiD.pdf (1) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04124318 , version 1 (13-06-2023)
hal-04124318 , version 2 (15-11-2023)

Identifiants

Citer

Adel Nabli, Eugene Belilovsky, Edouard Oyallon. A2CiD2: Accelerating Asynchronous Communication in Decentralized Deep Learning. Thirty-seventh Conference on Neural Information Processing Systems, Dec 2023, New Orleans, United States. ⟨hal-04124318v2⟩
246 Consultations
365 Téléchargements

Altmetric

Partager

More