arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.07428eess.SPcs.LG

无人机赋能的联邦学习延迟最小化:轨迹设计与资源分配

Latency Minimization for UAV-Enabled Federated Learning: Trajectory Design and Resource Allocation

Shenzhen Future Network of Intelligence Institute (FNii-Shenzhen)(深圳未来智能网络研究院);School of Science and Engineering (SSE)(科学与工程学院);The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));Shenzhen Institute of Advanced Technology (SIAT)(深圳先进技术研究院);Chinese Academy of Sciences (CAS)(中国科学院);Southern University of Science and Technology(南方科技大学);Imperial College London(伦敦帝国学院);Central South University(中南大学);Shenzhen University of Advanced Technology(深圳理工大学)

Xuhui Zhang, Wenchao Liu, Jinke Ren, Huijun Xing, Gui Gui, Yanyan Shen, Shuguang Cui

AI总结:

本文提出无人机作为移动服务器增强联邦学习,通过联合优化带宽、计算频率、功率和轨迹最小化训练延迟,所提交替优化算法较基准方案降低延迟达15.29%。

摘要:

联邦学习(FL)已成为跨无线网络分布式机器学习的一种变革性范式。然而,FL的性能常常受到资源受限的物联网(IoT)设备与中心服务器之间不可靠通信链路的阻碍。为了克服这一挑战,我们提出了一种新颖的框架,采用无人机(UAV)作为移动服务器来增强FL训练过程。通过利用无人机的移动性,我们与物联网设备建立强视线连接,从而增强通信可靠性和容量。为了最大化训练效率,我们制定了一个延迟最小化问题,联合优化带宽分配、计算频率、无人机和物联网设备的发射功率以及无人机的飞行轨迹。随后,我们分析了物联网设备训练和无人机聚合达到FL收敛所需的轮数。基于收敛约束,我们将问题转化为三个子问题,并开发了一种高效的交替优化算法来有效解决该问题。此外,我们对算法的收敛性和计算复杂度进行了全面分析。大量数值结果表明,我们提出的方案不仅在减少延迟方面优于现有基准方案,最高可达15.29%,而且训练效率几乎接近理想场景。

英文摘要:

Federated learning (FL) has become a transformative paradigm for distributed machine learning across wireless networks. However, the performance of FL is often hindered by the unreliable communication links between resource-constrained Internet of Things (IoT) devices and the central server. To overcome this challenge, we propose a novel framework that employs an unmanned aerial vehicle (UAV) as a mobile server to enhance the FL training process. By capitalizing on the UAV's mobility, we establish strong line-of-sight connections with IoT devices, thereby enhancing communication reliability and capacity. To maximize training efficiency, we formulate a latency minimization problem that jointly optimizes bandwidth allocation, computing frequencies, transmit power for both the UAV and IoT devices, and the UAV's flight trajectory. Subsequently, we analyze the required rounds of the IoT devices training and the UAV aggregation for FL convergence. Based on the convergence constraint, we transform the problem into three subproblems and develop an efficient alternating optimization algorithm to solve this problem effectively. Additionally, we provide a thorough analysis of the algorithm's convergence and computational complexity. Extensive numerical results demonstrate that our proposed scheme not only surpasses existing benchmark schemes in reducing latency up to 15.29%, but also achieves training efficiency that nearly matches the ideal scenario.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑