arXiv · 2404.05919
AdaGossip: Adaptive Consensus Step-size for Decentralized Deep Learning with Communication Compression
Abstract
Decentralized learning is crucial in supporting on-device learning over large distributed datasets, eliminating the need for a central server. However, the communication overhead remains a major bottleneck for the practical realization of such decentralized setups. To tackle this issue, several algorithms for decentralized training with compressed communication have been proposed in the literature. Most of these algorithms introduce an additional hyper-parameter referred to as consensus step-size which is tuned based on the compression ratio at the beginning of the training. In this work, we propose AdaGossip, a novel technique that adaptively adjusts the consensus step-size based on the compressed model differences between neighboring agents. We demonstrate the effectiveness of the proposed method through an exhaustive set of experiments on various Computer Vision datasets (CIFAR-10, CIFAR-100, Fashion MNIST, Imagenette, and ImageNet), model architectures, and network topologies. Our experiments show that the proposed method achieves superior performance ($0-2\%$ improvement in test accuracy) compared to the current state-of-the-art method for decentralized learning with communication compression.
Explore related subjects
Keep this discovery
Sai Aparna Aketi, Abolfazl Hashemi, Kaushik Roy. 2024-04-09. AdaGossip: Adaptive Consensus Step-size for Decentralized Deep Learning with Communication Compression. https://arxiv.org/abs/2404.05919
Cite the original work for its findings. Save a collection to share your selection of sources.