Computation scheduling for distributed machine learning with straggling workers
File(s)AG_ICASSP_19.pdf (241.4 KB)
Accepted version
Author(s)
Mohammadi Amiri, Mohammad
Gunduz, Deniz
Type
Conference Paper
Abstract
We study scheduling of computation tasks acrossnworkers in a large scale distributed learning problem. Computa-tion speeds of the workers are assumed to be heterogeneous andunknown to the master, and redundant computations are assignedto the workers in order to tolerate straggling workers. We con-sider sequential computation and instantaneous communicationfrom each worker to the master, and each computation round,which can model a single iteration of the stochastic gradientdescent (SGD) algorithm, iscompletedonce the master receivesk≤ndistinct computations, referred to as thecomputationtarget. Our goal is to characterize theaverage completion timeas a function of thecomputation load, which denotes the portionof the dataset available at each worker, and the computationtarget. We propose two computation scheduling schemes thatspecify the computation tasks assigned to each worker, as wellas their order of execution. We also establish a lower bound onthe minimum average completion time. Numerical results showa significant reduction in the average computation time over theexisting coded and uncoded computing schemes.
Date Issued
2019-05
Date Acceptance
2019-02-01
Citation
ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings, 2019, pp.8177-8181
ISBN
978-1-4799-8131-1
ISSN
2379-190X
Publisher
Institute of Electrical and Electronics Engineers
Start Page
8177
End Page
8181
Journal / Book Title
ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings
Copyright Statement
© 2019 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.
Source
ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing
Subjects
Science & Technology
Technology
Acoustics
Engineering, Electrical & Electronic
Engineering
Machine learning
distributed computation
Publication Status
Published
Start Date
2019-05-12
Finish Date
2019-05-17
Coverage Spatial
Brighton, UK
Date Publish Online
2020-04-17