arXiv:2412.07428eess.SPcs.LG2024-12被引 23

用无人机当移动服务器,降低物联网联邦学习延迟

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

  • 无人机飞到设备附近,通过最优路径和资源分配提升通信效率
  • 相比现有方案,延迟降低最高达15.29%,接近理想情况下的训练效率
  • 适合需要低延迟、高可靠通信的智能物联网场景

联邦学习(FL)已成为无线网络中分布式机器学习的变革性范式。然而,资源受限的物联网(IoT)设备与中心服务器之间的不可靠通信链路常制约其性能。为此,我们提出一种新框架,利用无人机(UAV)作为移动服务器以增强联邦学习训练过程。借助无人机的机动性,建立与物联网设备间的强视距连接,从而提升通信可靠性与容量。为最大化训练效率,我们构建了一个联合优化带宽分配、计算频率、收发功率及无人机飞行轨迹的延迟最小化问题。随后,分析了设备训练轮次与无人机聚合所需的轮数,并基于收敛约束将问题分解为三个子问题,设计了一种高效的交替优化算法。此外,提供了算法收敛性与计算复杂度的详细分析。大量数值结果表明,所提方案不仅使延迟降低最多达15.29%,且训练效率几乎逼近理想情形。

原文摘要 · Abstract (English)

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.

联邦学习无人机延迟优化资源分配

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。