Timezone: »
Federated learning is a promising framework to mitigate data privacy and computation concerns. However, the communication cost between the server and clients has become the major bottleneck for successful deployment. Despite notable progress in gradient compression, the existing quantization methods require further improvement when low-bits compression is applied, especially the overall systems often degenerate a lot when quantization are applied in double directions to compress model weights and gradients. In this work, we propose a simple cosine-based nonlinear quantization and achieve impressive results in compressing round-trip communication costs. We are not only able to compress model weights and gradients at higher ratios than previous methods, but also achieve competing model performance at the same time. Further, our approach is highly suitable for federated learning problems since it has low computational complexity and requires only a little additional data to recover the compressed information. Extensive experiments have been conducted on image classification and brain tumor semantic segmentation using the CIFAR-10, and BraTS datasets where we show state-of-the-art effectiveness and impressive communication efficiency.
Author Information
Yang He (Amazon)
Hui-Po Wang (CISPA Helmholtz Center for Information Security)
Maximilian Zenk (Deutsches Krebsforschungszentrum)
Mario Fritz (CISPA Helmholtz Center i.G.)
More from the Same Authors
-
2020 : Paper 21: Haar Wavelet based Block Autoregressive Flows for Trajectories »
Apratim Bhattacharyya · Christoph-Nikolas Straehle · Mario Fritz · Bernt Schiele -
2022 : Class-wise Domain Generalization: A Novel Framework for Evaluating Distributional Shift »
Sarath Sivaprasad · Akshay Goindani · Mario Fritz · Vineet Gandhi -
2022 Poster: Private Set Generation with Discriminative Information »
Dingfan Chen · Raouf Kerkouche · Mario Fritz -
2020 Poster: GS-WGAN: A Gradient-Sanitized Approach for Learning Differentially Private Generators »
Dingfan Chen · Tribhuvanesh Orekondy · Mario Fritz