Decentralized Learning Made Practical with Client Sampling

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: de Vos, Martijn, Dhasade, Akash, Kermarrec, Anne-Marie, Lavoie, Erick, Pouwelse, Johan, Sharma, Rishi
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913343564939264
author de Vos, Martijn
Dhasade, Akash
Kermarrec, Anne-Marie
Lavoie, Erick
Pouwelse, Johan
Sharma, Rishi
author_facet de Vos, Martijn
Dhasade, Akash
Kermarrec, Anne-Marie
Lavoie, Erick
Pouwelse, Johan
Sharma, Rishi
contents Decentralized learning (DL) leverages edge devices for collaborative model training while avoiding coordination by a central server. Due to privacy concerns, DL has become an attractive alternative to centralized learning schemes since training data never leaves the device. In a round of DL, all nodes participate in model training and exchange their model with some other nodes. Performing DL in large-scale heterogeneous networks results in high communication costs and prolonged round durations due to slow nodes, effectively inflating the total training time. Furthermore, current DL algorithms also assume all nodes are available for training and aggregation at all times, diminishing the practicality of DL. This paper presents Plexus, an efficient, scalable, and practical DL system. Plexus (1) avoids network-wide participation by introducing a decentralized peer sampler that selects small subsets of available nodes that train the model each round and, (2) aggregates the trained models produced by nodes every round. Plexus is designed to handle joining and leaving nodes (churn). We extensively evaluate Plexus by incorporating realistic traces for compute speed, pairwise latency, network capacity, and availability of edge devices in our experiments. Our experiments on four common learning tasks empirically show that Plexus reduces time-to-accuracy by 1.2-8.3x, communication volume by 2.4-15.3x and training resources needed for convergence by 6.4-370x compared to baseline DL algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2302_13837
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Decentralized Learning Made Practical with Client Sampling
de Vos, Martijn
Dhasade, Akash
Kermarrec, Anne-Marie
Lavoie, Erick
Pouwelse, Johan
Sharma, Rishi
Distributed, Parallel, and Cluster Computing
Decentralized learning (DL) leverages edge devices for collaborative model training while avoiding coordination by a central server. Due to privacy concerns, DL has become an attractive alternative to centralized learning schemes since training data never leaves the device. In a round of DL, all nodes participate in model training and exchange their model with some other nodes. Performing DL in large-scale heterogeneous networks results in high communication costs and prolonged round durations due to slow nodes, effectively inflating the total training time. Furthermore, current DL algorithms also assume all nodes are available for training and aggregation at all times, diminishing the practicality of DL. This paper presents Plexus, an efficient, scalable, and practical DL system. Plexus (1) avoids network-wide participation by introducing a decentralized peer sampler that selects small subsets of available nodes that train the model each round and, (2) aggregates the trained models produced by nodes every round. Plexus is designed to handle joining and leaving nodes (churn). We extensively evaluate Plexus by incorporating realistic traces for compute speed, pairwise latency, network capacity, and availability of edge devices in our experiments. Our experiments on four common learning tasks empirically show that Plexus reduces time-to-accuracy by 1.2-8.3x, communication volume by 2.4-15.3x and training resources needed for convergence by 6.4-370x compared to baseline DL algorithms.
title Decentralized Learning Made Practical with Client Sampling
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2302.13837