Dynamic Backup Workers for Parallel Machine Learning - Archive ouverte HAL
Article Dans Une Revue Computer Networks Année : 2021

Dynamic Backup Workers for Parallel Machine Learning

Chuan Xu
Giovanni Neglia

Résumé

The most popular framework for distributed training of machine learning models is the (synchronous) parameter server (PS). This paradigm consists of n workers, which iteratively compute updates of the model parameters, and a stateful PS, which waits and aggregates all updates to generate a new estimate of model parameters and sends it back to the workers for a new iteration. Transient computation slowdowns or transmission delays can intolerably lengthen the time of each iteration. An efficient way to mitigate this problem is to let the PS wait only for the fastest n − b updates, before generating the new parameters. The slowest b workers are called backup workers. The correct choice of the number b of backup workers depends on the cluster configuration and workload, but also (as we show in this paper) on the hyper-parameters of the learning algorithm and the current stage of the training. We propose DBW, an algorithm that dynamically decides the number of backup workers during the training process to maximize the convergence speed at each iteration. Our experiments show that DBW 1) removes the necessity to tune b by preliminary time-consuming experiments, and 2) makes the training up to a factor 3 faster than the optimal static configuration.
Fichier principal
Vignette du fichier
revised_com_net.pdf (1.83 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03364773 , version 1 (04-10-2021)

Identifiants

Citer

Chuan Xu, Giovanni Neglia, Nicola Sebastianelli. Dynamic Backup Workers for Parallel Machine Learning. Computer Networks, 2021, 188, pp.107846. ⟨10.1016/j.comnet.2021.107846⟩. ⟨hal-03364773⟩
49 Consultations
84 Téléchargements

Altmetric

Partager

More