Coded Distributed Computing with Partial Recovery
This work addresses the challenge of straggling workers in distributed computing for machine learning and optimization, offering an incremental improvement over existing coded computation methods.
The paper tackles the problem of straggling workers in distributed computing by introducing a coded computation scheme with partial recovery, which reduces computation time and decoding complexity by allowing a trade-off between accuracy and speed, as confirmed by numerical simulations on a large linear regression task.
Coded computation techniques provide robustness against straggling workers in distributed computing. However, most of the existing schemes require exact provisioning of the straggling behaviour and ignore the computations carried out by straggling workers. Moreover, these schemes are typically designed to recover the desired computation results accurately, while in many machine learning and iterative optimization algorithms, faster approximate solutions are known to result in an improvement in the overall convergence time. In this paper, we first introduce a novel coded matrix-vector multiplication scheme, called coded computation with partial recovery (CCPR), which benefits from the advantages of both coded and uncoded computation schemes, and reduces both the computation time and the decoding complexity by allowing a trade-off between the accuracy and the speed of computation. We then extend this approach to distributed implementation of more general computation tasks by proposing a coded communication scheme with partial recovery, where the results of subtasks computed by the workers are coded before being communicated. Numerical simulations on a large linear regression task confirm the benefits of the proposed distributed computation scheme with partial recovery in terms of the trade-off between the computation accuracy and latency.