Residential College | false |
Status | 已發表Published |
Ranking-based Client Imitation Selection for Efficient Federated Learning | |
Tian, Chunlin1; Shi, Zhan2; Qin, Xinpeng3; Li, Li1; Xu, Chengzhong1 | |
2024 | |
Conference Name | 41st International Conference on Machine Learning, ICML 2024 |
Source Publication | Proceedings of Machine Learning Research |
Volume | 235 |
Pages | 48211-48225 |
Conference Date | 21 July 2024through 27 July 2024 |
Conference Place | Vienna |
Publisher | ML Research Press |
Abstract | Federated Learning (FL) enables multiple devices to collaboratively train a shared model while ensuring data privacy. The selection of participating devices in each training round critically affects both the model performance and training efficiency, especially given the vast heterogeneity in training capabilities and data distribution across devices. To deal with these challenges, we introduce a novel device selection solution called FedRank, which is based on an end-to-end, ranking-based model that is pre-trained by imitation learning against state-of-the-art analytical approaches. It not only considers data and system heterogeneity at runtime but also adaptively and efficiently chooses the most suitable clients for model training. Specifically, FedRank views client selection in FL as a ranking problem and employs a pairwise training strategy for the smart selection process. Additionally, an imitation learning-based approach is designed to counteract the cold-start issues often seen in state-of-the-art learning-based approaches. Experimental results reveal that FedRank boosts model accuracy by 5.2% to 56.9%, accelerates the training convergence up to 2.01× and saves the energy consumption up to 40.1%. |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85203822522 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Affiliation | 1.University of Macau, Macao 2.University of Texas, Austin, United States 3.University of Electronic Science and Technology of China, China |
First Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Tian, Chunlin,Shi, Zhan,Qin, Xinpeng,et al. Ranking-based Client Imitation Selection for Efficient Federated Learning[C]:ML Research Press, 2024, 48211-48225. |
APA | Tian, Chunlin., Shi, Zhan., Qin, Xinpeng., Li, Li., & Xu, Chengzhong (2024). Ranking-based Client Imitation Selection for Efficient Federated Learning. Proceedings of Machine Learning Research, 235, 48211-48225. |
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