Residential Collegefalse
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 PublicationIEEE Transactions on Wireless Communications
ISSN1536-1276
Volume22Issue: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.

KeywordFederated Learning Gradient Compression Wireless Resource Management
DOI10.1109/TWC.2023.3245601
URLView the original
Language英語English
Scopus ID2-s2.0-85149413319
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionTHE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU)
Corresponding AuthorYu Rong; Wu Yuan
Affiliation1.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 AffilicationUniversity 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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Li Peichun]'s Articles
[Cheng Guoliang]'s Articles
[Huang Xumin]'s Articles
Baidu academic
Similar articles in Baidu academic
[Li Peichun]'s Articles
[Cheng Guoliang]'s Articles
[Huang Xumin]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Li Peichun]'s Articles
[Cheng Guoliang]'s Articles
[Huang Xumin]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.