Timezone: »

 
Poster
ErrorCompensatedX: error compensation for variance reduced algorithms
Hanlin Tang · Yao Li · Ji Liu · Ming Yan

Fri Dec 10 08:30 AM -- 10:00 AM (PST) @
in Poster Session 8 »

Communication cost is one major bottleneck for the scalability for distributed learning. One approach to reduce the communication cost is to compress the gradient during communication. However, directly compressing the gradient decelerates the convergence speed, and the resulting algorithm may diverge for biased compression. Recent work addressed this problem for stochastic gradient descent by adding back the compression error from the previous step. This idea was further extended to one class of variance reduced algorithms, where the variance of the stochastic gradient is reduced by taking a moving average over all history gradients. However, our analysis shows that just adding the previous step's compression error, as done in existing work, does not fully compensate the compression error. So, we propose ErrorCompensateX, which uses the compression error from the previous two steps. We show that ErrorCompensateX can achieve the same asymptotic convergence rate with the training without compression. Moreover, we provide a unified theoretical analysis framework for this class of variance reduced algorithms, with or without error compensation.

Keywords: Optimization

Author Information

Hanlin Tang (University of Rochester)
Yao Li (Michigan State University)
Ji Liu (Kwai Inc.)
Ming Yan (Michigan State University)

More from the Same Authors