arrow
Return

DSAG: A Mixed Synchronous-Asynchronous Iterative Method for Straggler-Resilient Learning

delete2023-02-01
delete0
delete
OA
AI
A
Albin Severinson
E
Eirik Rosnes *
S
Salim El Rouayheb
A
Alexandre Graell i Amat
DOI:10.1109/TCOMM.2022.3227286delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
We consider straggler-resilient learning. In many previous works, e.g., in the coded computing literature, straggling is modeled as random delays that are independent and identically distributed between workers. However, in many practical scenarios, a given worker may straggle over an extended period of time. We propose a latency model that captures this behavior and is substantiated by traces collected on Microsoft Azure, Amazon Web Services (AWS), and a small local cluster. Building on this model, we propose DSAG, a mixed synchronous-asynchronous iterative optimization method, based on the stochastic average gradient (SAG) method, that combines timely and stale results. We also propose a dynamic load-balancing strategy to further reduce the impact of straggling workers. We evaluate DSAG for principal component analysis, cast as a finite-sum optimization problem, of a large genomics dataset, and for logistic regression on a cluster composed of 100 workers on AWS, and find that DSAG is up to about 50% faster than SAG, and more than twice as fast as coded computing methods, for the particular scenario that we consider.
Keywords:
Radio frequency
Computational modeling
Principal component analysis
Iterative methods
Stochastic processes
Logistics
Convergence
Coded computing
iterative optimization
load-balancing
principal component analysis (PCA)
stochastic average gradient (SAG)
straggler mitigation
variance reduction

Journal

IEEE Transactions on Communications cover
IEEE Transactions on Communications
IF:
8.3
Papers:
1.2W
Citations:
3.6W

Organization

U
university of bergen
Scholars:
2.0W
Papers: 1.7W
Citations: 19
C
chalmers university of technology
Scholars:
1.5W
Papers: 1.6W
Citations: 10