arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.12990cs.LGcs.DCcs.ITmath.IT

Lightweight Projective Derivative Codes for Compressed Asynchronous Gradient Descent

The Graduate Center, CUNY(纽约市立大学研究生中心);Icahn School of Medicine at Mount Sinai(西奈山伊坎医学院);CUNY Queens College & Graduate Center(纽约市立大学皇后学院及研究生中心)

Pedro Soto, Ilia Ilmer, Haibin Guan, Jun Li

英文摘要:

Coded distributed computation has become common practice for performing gradient descent on large datasets to mitigate stragglers and other faults. This paper proposes a novel algorithm that encodes the partial derivatives themselves and furthermore optimizes the codes by performing lossy compression on the derivative codewords by maximizing the information contained in the codewords while minimizing the information between the codewords. The utility of this application of coding theory is a geometrical consequence of the observed fact in optimization research that noise is tolerable, sometimes even helpful, in gradient descent based learning algorithms since it helps avoid overfitting and local minima. This stands in contrast with much current conventional work on distributed coded computation which focuses on recovering all of the data from the workers. A second further contribution is that the low-weight nature of the coding scheme allows for asynchronous gradient updates since the code can be iteratively decoded; i.e., a worker's task can immediately be updated into the larger gradient. The directional derivative is always a linear function of the direction vectors; thus, our framework is robust since it can apply linear coding techniques to general machine learning frameworks such as deep neural networks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑