Gradient coding with dynamic clustering for straggler-tolerant distributed learning
File(s)BOUG_TCOM22.pdf (7.34 MB)
Accepted version
Author(s)
Buyukates, Baturalp
Ozfatura, Emre
Ulukus, Sennur
Gunduz, Deniz
Type
Journal Article
Abstract
Distributed implementations are crucial in speeding up large scale machine learning applications. Distributed gradient descent (GD) is widely employed to parallelize the learning task by distributing the dataset across multiple workers. A significant performance bottleneck for the per-iteration completion time in distributed synchronous GD is straggling workers. Coded distributed computation techniques have been introduced recently to mitigate stragglers and to speed up GD iterations by assigning redundant computations to workers. In this paper, we introduce a novel paradigm of dynamic coded computation, which assigns redundant data to workers to acquire the flexibility to dynamically choose from among a set of possible codes depending on the past straggling behavior. In particular, we propose gradient coding (GC) with dynamic clustering, called GC-DC, and regulate the number of stragglers in each cluster by dynamically forming the clusters at each iteration. With time-correlated straggling behavior, GC-DC adapts to the straggling behavior over time; in particular, at each iteration, GC-DC aims at distributing the stragglers across clusters as uniformly as possible based on the past straggler behavior. For both homogeneous and heterogeneous worker models, we numerically show that GC-DC provides significant improvements in the average per-iteration completion time without an increase in the communication load compared to the original GC scheme.
Date Issued
2023-06-01
Date Acceptance
2022-04-01
Citation
IEEE Transactions on Communications, 2023, 71 (6), pp.3317-3332
ISSN
0090-6778
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Start Page
3317
End Page
3332
Journal / Book Title
IEEE Transactions on Communications
Volume
71
Issue
6
Copyright Statement
© 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Identifier
https://ieeexplore.ieee.org/document/9755943
Publication Status
Published
Date Publish Online
2022-04-12