English

Convert, compress, correct: Three steps toward communication-efficient DNN training

Machine Learning 2022-03-18 v1

Abstract

In this paper, we introduce a novel algorithm, CO3\mathsf{CO}_3, for communication-efficiency distributed Deep Neural Network (DNN) training. CO3\mathsf{CO}_3 is a joint training/communication protocol, which encompasses three processing steps for the network gradients: (i) quantization through floating-point conversion, (ii) lossless compression, and (iii) error correction. These three components are crucial in the implementation of distributed DNN training over rate-constrained links. The interplay of these three steps in processing the DNN gradients is carefully balanced to yield a robust and high-performance scheme. The performance of the proposed scheme is investigated through numerical evaluations over CIFAR-10.

Keywords

Cite

@article{arxiv.2203.09044,
  title  = {Convert, compress, correct: Three steps toward communication-efficient DNN training},
  author = {Zhong-Jing Chen and Eduin E. Hernandez and Yu-Chih Huang and Stefano Rini},
  journal= {arXiv preprint arXiv:2203.09044},
  year   = {2022}
}
R2 v1 2026-06-24T10:16:33.736Z