CBFL: A Communication-Efficient Federated Learning Framework From Data Redundancy Perspective
- 29 October 2021
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Systems Journal
- Vol. 16 (4) , 5572-5583
Abstract
Federated learning (FL) is an emerging machinelearning framework, which enables multiple mobile users to collaboratively train a global model without uploading their local sensitive data. Due to limited network bandwidth, communication efficiency has become a significant bottleneck for the implementation of FL. Existing works attempt to improve this situation by reducing the total bits transferred for each client update via data compression. However, these research works are only from the perspective of update parameters to reduce the transmission of redundant parameters. They do not explore the intrinsic reasons for redundant parameters. In this article, we propose a coreset-based FL (CBFL) framework. Instead of training model on full datasets with a regular network model, CBFL uses a much smaller well-matched evolutionary network model on coreset. CBFL indirectly reduces the total transmission bits for each client while achieving a similar accuracy as training with full datasets. CBFL includes a novel distributed coreset construction and adaptive model evolution algorithms. The network model is adaptively adjusted during this process, which dynamically removes the least important connections from the current model. Experimental results with various datasets and models show that CBFL is able to find an optimized evolutionary model that has about 10% of the total number of connections in the original regular model while only about 2% degradation in model accuracy.Keywords
Funding Information
- National Natural Science Foundation of China (61772094)
- Entrepreneurship and Innovation Support Plan for overseas Students in Chongqing (CX2019094)
- Central University Fund (2020CDJQY-A019)
This publication has 15 references indexed in Scilit:
- Robust Coreset Construction for Distributed Machine LearningIEEE Journal on Selected Areas in Communications, 2020
- GGS: General Gradient Sparsification for Federated Learning in Edge Computing2020
- Federated Learning: Challenges, Methods, and Future DirectionsIEEE Signal Processing Magazine, 2020
- Federated Learning in Mobile Edge Networks: A Comprehensive SurveyIEEE Communications Surveys & Tutorials, 2020
- Communication-Efficient Federated Deep Learning With Layerwise Asynchronous Model Update and Temporally Weighted AggregationIEEE Transactions on Neural Networks and Learning Systems, 2019
- Robust and Communication-Efficient Federated Learning From Non-i.i.d. DataIEEE Transactions on Neural Networks and Learning Systems, 2019
- Sparse Binary Compression: Towards Distributed Deep Learning with minimal Communication2019
- Communication Efficient Coreset Sampling for Distributed Learning2018
- Sparse Communication for Distributed Gradient Descent2017
- Evolving Neural Networks through Augmenting TopologiesEvolutionary Computation, 2002