Residential Collegefalse
Status已發表Published
Ranking-based Client Selection with Imitation Learning for Efficient Federated Learning
Chunlin Tian1; Zhan Shi2; Xinpeng Qin3; Li Li1; ChengZhong Xu1
2024-06
Source PublicationInternational Conference on Machine Learning
ISSN26403498
Volume235Pages:48211 - 48225
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%.

URLView the original
Language英語English
PublisherML Research Press
Scopus ID2-s2.0-85203822522
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionTHE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU)
Affiliation1.University of Macau
2.University of Texas, Austin, United States
3.University of Electronic Science and Technology of China, China
First Author AffilicationUniversity of Macau
Recommended Citation
GB/T 7714
Chunlin Tian,Zhan Shi,Xinpeng Qin,et al. Ranking-based Client Selection with Imitation Learning for Efficient Federated Learning[J]. International Conference on Machine Learning, 2024, 235, 48211 - 48225.
APA Chunlin Tian., Zhan Shi., Xinpeng Qin., Li Li., & ChengZhong Xu (2024). Ranking-based Client Selection with Imitation Learning for Efficient Federated Learning. International Conference on Machine Learning, 235, 48211 - 48225.
MLA Chunlin Tian,et al."Ranking-based Client Selection with Imitation Learning for Efficient Federated Learning".International Conference on Machine Learning 235(2024):48211 - 48225.
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
[Chunlin Tian]'s Articles
[Zhan Shi]'s Articles
[Xinpeng Qin]'s Articles
Baidu academic
Similar articles in Baidu academic
[Chunlin Tian]'s Articles
[Zhan Shi]'s Articles
[Xinpeng Qin]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Chunlin Tian]'s Articles
[Zhan Shi]'s Articles
[Xinpeng Qin]'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.