Magnitude Matters: Fixing SIGNSGD Through Magnitude-Aware Sparsification in the Presence of Data Heterogeneity
Not on the map. Why?
Abstract (arXiv)
Communication overhead has become one of the major bottlenecks in the distributed training of deep neural networks. To alleviate the concern, various gradient compression methods have been proposed, and sign-based algorithms are of surging interest. However, SIGNSGD fails to converge in the presence of data heterogeneity, which is commonly observed in the emerging federated learning (FL) paradigm. Error feedback has been proposed to address the non-convergence issue. Nonetheless, it requires the workers to locally keep track of the compression errors, which renders it not suitable for FL since the workers may not participate in the training throughout the learning process. In this paper, we propose a magnitude-driven sparsification scheme, which addresses the non-convergence issue of SIGNSGD while further improving communication efficiency. Moreover, the local update scheme is further incorporated to improve the learning performance, and the convergence of the proposed method is established. The effectiveness of the proposed scheme is validated through experiments on Fashion-MNIST, CIFAR-10, and CIFAR-100 datasets.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 36 of its 46 references that are in the map. First posted on arXiv 19 Feb 2023.
Builds on
36 of 46 references.
References · 36
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2018Federated Optimization in Heterogeneous Networkscited by 2,987
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 851
- 2018Local SGD Converges Fast and Communicates Littlecited by 605
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
- 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 478
- 2019Robust and Communication-Efficient Federated Learning from Non-IID Datacited by 419
- 2018Sparsified SGD with Memorycited by 376
- 2019Error Feedback Fixes SignSGD and other Gradient Compression Schemescited by 310
- 2017TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep Learningcited by 256
- 2018The Convergence of Sparsified Gradient Methodscited by 229
- 2017Gradient Sparsification for Communication-Efficient Distributed Optimizationcited by 201
24 more
- 2019Qsparse-local-SGD: Distributed SGD with Quantization, Sparsification, and Local Computationscited by 181
- 2020FetchSGD: Communication-Efficient Federated Learning with Sketchingcited by 179
- 2020Federated Learning with Compression: Unified Analysis and Sharp Guaranteescited by 160
- 2017Revisiting Distributed Synchronous SGDcited by 142
- 2020Understanding Gradient Clipping in Private SGD: A Geometric Perspectivecited by 142
- 2018ATOMO: Communication-efficient Learning via Atomic Sparsificationcited by 136
- 2019Enabling Spike-based Backpropagation for Training Deep Neural Network Architecturescited by 123
- 2019Distributed Learning with Compressed Gradient Differencescited by 112
- 2020UVeQFed: Universal Vector Quantization for Federated Learningcited by 107
- 2019DoubleSqueeze: Parallel Stochastic Gradient Descent with Double-Pass Error-Compensated Compressioncited by 99
- 2020Acceleration for Compressed Gradient Descent in Distributed and Federated Optimizationcited by 90
- 2018Error Compensated Quantized SGD and its Applications to Large-scale Distributed Optimizationcited by 76
- 2021Understanding Clipping for Federated Learning: Convergence and Client-Level Differential Privacycited by 69
- 2019Stochastic Distributed Learning with Gradient Quantization and Variance Reductioncited by 65
- 2018Sparse Binary Compression: Towards Distributed Deep Learning with minimal Communicationcited by 51
- 2019Communication-Efficient Distributed Blockwise Momentum SGD with Error-Feedbackcited by 50
- 2019Stochastic Sign Descent Methods: New Algorithms and Better Theorycited by 49
- 2020Bidirectional compression in heterogeneous settings for distributed or federated learning with partial participation: tight convergence guaranteescited by 46
- 2020A Better Alternative to Error Feedback for Communication-Efficient Distributed Learningcited by 43
- 2021Rethinking gradient sparsification as total error minimizationcited by 41
- 2020Stochastic-Sign SGD for Federated Learning with Theoretical Guaranteescited by 35
- 2019A Double Residual Compression Algorithm for Efficient Distributed Learningcited by 31
- 2019Distributed Training with Heterogeneous Data: Bridging Median- and Mean-Based Algorithmscited by 29
- 2018signSGD with Majority Vote is Communication Efficient And Fault Tolerantcited by 11
Built on by
Citing papers per year: 2023: 1; 2024: 0; 2025: 0; 2026: 1.
2 papers.