Skip to main content

Communication-computation efficient gradient coding

Author(s): Ye, M; Abbe, Emmanuel

To refer to this page use:
Abstract: This paper develops coding techniques to reduce the running time of distributed learning tasks. It characterizes the fundamental tradeoff to compute gradients in terms of three parameters: computation load, straggler tolerance and communication cost. It further gives an explicit coding scheme that achieves the optimal tradeoff based on recursive polynomial constructions, coding both across data subsets and vector components. As a result, the proposed scheme allows to minimize the running time for gradient computations. Implementations are made on Amazon EC2 clusters using Python with mpi4py package. Results show that the proposed scheme maintains the same generalization error while reducing the running time by 32% compared to uncoded schemes and 23% compared to prior coded schemes focusing only on stragglers (Tandon et al., ICML 2017).
Publication Date: 2018
Citation: Ye, M, Abbe, E. (2018). Communication-computation efficient gradient coding. 12 (9716p - 9716p
Pages: 9716p - 9716p
Type of Material: Conference Article
Journal/Proceeding Title: 35th International Conference on Machine Learning, ICML 2018
Version: Author's manuscript

Items in OAR@Princeton are protected by copyright, with all rights reserved, unless otherwise indicated.