FEDZIP: A Compression Framework for Communication-Efficient Federated Learning
arXiv:2102.01593
Abstract
Federated Learning marks a turning point in the implementation of decentralized machine learning (especially deep learning) for wireless devices by protecting users' privacy and safeguarding raw data from third-party access. It assigns the learning process independently to each client. First, clients locally train a machine learning model based on local data. Next, clients transfer local updates of model weights and biases (training data) to a server. Then, the server aggregates updates (received from clients) to create a global learning model. However, the continuous transfer between clients and the server increases communication costs and is inefficient from a resource utilization perspective due to the large number of parameters (weights and biases) used by deep learning models. The cost of communication becomes a greater concern when the number of contributing clients and communication rounds increases. In this work, we propose a novel framework, FedZip, that significantly decreases the size of updates while transferring weights from the deep learning model between clients and their servers. FedZip implements Top-z sparsification, uses quantization with clustering, and implements compression with three different encoding methods. FedZip outperforms state-of-the-art compression frameworks and reaches compression rates up to 1085x, and preserves up to 99% of bandwidth and 99% of energy for clients during communication.
References in corpus (16)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- Federated Learning: Strategies for Improving Communication Efficiency
- Accurate, Large Minibatch SGD: Training ImageNet in 1 Hour
- Federated Optimization: Distributed Machine Learning for On-Device Intelligence
- Towards Federated Learning at Scale: System Design
- QSGD: Communication-Efficient SGD via Gradient Quantization and Encoding
- Revisiting Distributed Synchronous SGD
- Federated Optimization:Distributed Optimization Beyond the Datacenter
- Differentially Private Federated Learning: A Client Level Perspective
- LEAF: A Benchmark for Federated Settings
- ATOMO: Communication-efficient Learning via Atomic Sparsification
- Variance-based Gradient Compression for Efficient Distributed Deep Learning
- CovidCTNet: An Open-Source Deep Learning Approach to Identify Covid-19 Using CT Image
- SEFR: A Fast Linear-Time Classifier for Ultra-Low Power Devices
- Nested Dithered Quantization for Communication Reduction in Distributed Training