Residential College | false |
Status | 已發表Published |
Snowball: Energy Efficient and Accurate Federated Learning with Coarse-to-Fine Compression over Heterogeneous Wireless Edge Devices | |
Li Peichun1; Cheng Guoliang2; Huang Xumin1; Kang Jiawen2; Yu Rong2; Wu Yuan3; Pan Miao4; Niyato Dusit5 | |
2023-02 | |
Source Publication | IEEE Transactions on Wireless Communications |
ISSN | 1536-1276 |
Volume | 22Issue:10Pages:6778-6792 |
Abstract | Model update compression is a widely used technique to alleviate the communication cost in federated learning (FL). However, there is evidence indicating that the compression-based FL system often suffers the following two issues, i) the implicit learning performance deterioration of the global model due to the inaccurate update, ii) the limitation of sharing the same compression rate over heterogeneous edge devices. In this paper, we propose an energy-efficient learning framework, named Snowball, that enables edge devices to incrementally upload their model updates in a coarse-to-fine compression manner. To this end, we first design a fine-grained compression scheme that enables a nearly continuous compression rate. After that, we investigate the Snowball optimization problem to minimize the energy consumption of parameter transmission with learning performance constraints. By leveraging the theoretical insights of the convergence analysis, the optimization problem is transformed into a tractable form. Following that, a water-filling algorithm is designed to solve the problem, where each device is assigned a personalized compression rate according to the status of the locally available resource. Experiments indicate that, compared to state-of-the-art FL algorithms, our learning framework can save five times the required energy of uplink communication to achieve a good global accuracy. |
Keyword | Federated Learning Gradient Compression Wireless Resource Management |
DOI | 10.1109/TWC.2023.3245601 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85149413319 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Yu Rong; Wu Yuan |
Affiliation | 1.Guangdong University of Technology, Guangzhou, China 2.School of Automation, Guangdong University of Technology, Guangzhou, China 3.State Key Laboratory of Internet of Things for Smart City, University of Macau, Macau, China 4.Department of Electrical and Computer Engineering, University of Houston, Houston, USA 5.School of Computer Science and Engineering, Nanyang Technological University, Singapore, Block N4-02a-32, Nanyang Avenue, Singapore |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Li Peichun,Cheng Guoliang,Huang Xumin,et al. Snowball: Energy Efficient and Accurate Federated Learning with Coarse-to-Fine Compression over Heterogeneous Wireless Edge Devices[J]. IEEE Transactions on Wireless Communications, 2023, 22(10), 6778-6792. |
APA | Li Peichun., Cheng Guoliang., Huang Xumin., Kang Jiawen., Yu Rong., Wu Yuan., Pan Miao., & Niyato Dusit (2023). Snowball: Energy Efficient and Accurate Federated Learning with Coarse-to-Fine Compression over Heterogeneous Wireless Edge Devices. IEEE Transactions on Wireless Communications, 22(10), 6778-6792. |
MLA | Li Peichun,et al."Snowball: Energy Efficient and Accurate Federated Learning with Coarse-to-Fine Compression over Heterogeneous Wireless Edge Devices".IEEE Transactions on Wireless Communications 22.10(2023):6778-6792. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment