Residential College | false |
Status | 已發表Published |
UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning | |
Wang Tianshun1; Huang Xumin2; Wu Yuan3; Qian Liping4; Lin Bin5; Su Zhou6 | |
2024-01 | |
Source Publication | IEEE Transactions on Network Science and Engineering |
ISSN | 2327-4697 |
Volume | 11Issue:1Pages:943-956 |
Abstract | Federated Learning (FL) enables the distributed machine learning (ML) without violating the privacy of local users. In the scenario wireless FL, it is challenging for some local clients to establish reliable connections with the parameter server due to the potential long-distance transmission. To address this issue, unmanned aerial vehicle (UAV) can be leveraged as a relay between the FL parameter server and local clients for efficiently forwarding the ML models. In this work, we propose a two-tier hierarchical FL scheme assisted by a UAV swarm. During the local training phase, the UAVs offload their own data to the base station (BS). For the remaining time, the UAVs act as the relays to assist the parameter server and local clients in forwarding ML models. To optimize the FL convergence and the UAVs' data transmissions, we formulate a joint optimization of the matching between the UAVs and local clients, the time allocation of the hierarchical FL, and the number of iterations for the local model training. To solve this optimization problem, we design an efficient algorithm that integrates a subgradient-based method with the cross entropy-based genetic algorithm. Numerical results are provided to demonstrate the advantages of our proposed two-tier hierarchical FL scheme with the UAV swarm and our proposed algorithm. |
Keyword | Autonomous Aerial Vehicles Computational Modeling Data Models Hierarchical Federated Learning Optimization Relays Servers Training Uav Swarm Wireless Federated Learning |
DOI | 10.1109/TNSE.2023.3311024 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Engineering ; Mathematics |
WOS Subject | Engineering, Multidisciplinary ; Mathematics, Interdisciplinary Applications |
WOS ID | WOS:001139144400041 |
Scopus ID | 2-s2.0-85169668330 |
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 | Wu Yuan |
Affiliation | 1.School of Communication and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing, 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, Macao, China 4.College of Information Engineering, Zhejiang University of Technology, Hangzhou, China 5.Department of Communication Engineering, Dalian Maritime University, Dalian, China 6.School of Cyber Science and Engineering, Xi'an Jiaotong University, Xi'an, China |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Wang Tianshun,Huang Xumin,Wu Yuan,et al. UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning[J]. IEEE Transactions on Network Science and Engineering, 2024, 11(1), 943-956. |
APA | Wang Tianshun., Huang Xumin., Wu Yuan., Qian Liping., Lin Bin., & Su Zhou (2024). UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning. IEEE Transactions on Network Science and Engineering, 11(1), 943-956. |
MLA | Wang Tianshun,et al."UAV Swarm-Assisted Two-Tier Hierarchical Federated Learning".IEEE Transactions on Network Science and Engineering 11.1(2024):943-956. |
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