arXiv:2503.06145cs.LG2025-03中稿 · IEEE Transactions …被引 8

无人机辅助分层联邦学习,优化能耗、延迟与鲁棒性。

Adaptive UAV-Assisted Hierarchical Federated Learning: Optimizing Energy, Latency, and Resilience for Dynamic Smart IoT

  • 分三步优化:学习配置、带宽分配与设备-无人机关联
  • 降低训练成本,提升通信中断下的聚合效率
  • 适合动态智能物联网场景,如战场监控与远程感知

分层联邦学习(HFL)通过引入中间聚合层,扩展了传统联邦学习在地理分散环境中的应用,特别适用于蜂窝网络受限的智能物联网系统,如远程监测与战场作业。在此类场景中,无人机作为移动聚合器,动态连接地面物联网设备。本文研究了能量受限、动态部署且易受通信中断影响的无人机辅助HFL架构。提出一种联合优化方法,同时考虑学习配置、带宽分配与设备-无人机关联,确保在无人机断连或重部署前完成全局聚合,以最小化全局训练成本。该问题具有动态设备与间歇性连接特性,属于NP-hard问题。为此,将其分解为三个子问题:(i) 基于增强拉格朗日法优化学习配置与带宽分配以降低训练成本;(ii) 提出基于数据异质性(用KL散度衡量)、设备-无人机距离与计算资源的设备适配度评分,采用TD3算法实现自适应设备-无人机分配;(iii) 设计低复杂度两阶段贪心策略用于无人机重部署与全局聚合器选择,保障在无人机断连情况下的高效聚合。在多个真实世界数据集上的实验验证了该方法的有效性,展现出显著的成本降低与强鲁棒性。

原文摘要 · Abstract (English)

Hierarchical Federated Learning (HFL) extends conventional Federated Learning (FL) by introducing intermediate aggregation layers, enabling distributed learning in geographically dispersed environments, particularly relevant for smart IoT systems, such as remote monitoring and battlefield operations, where cellular connectivity is limited. In these scenarios, UAVs serve as mobile aggregators, dynamically connecting terrestrial IoT devices. This paper investigates an HFL architecture with energy-constrained, dynamically deployed UAVs prone to communication disruptions. We propose a novel approach to minimize global training costs by formulating a joint optimization problem that integrates learning configuration, bandwidth allocation, and device-to-UAV association, ensuring timely global aggregation before UAV disconnections and redeployments. The problem accounts for dynamic IoT devices and intermittent UAV connectivity and is NP-hard. To tackle this, we decompose it into three subproblems: \textit{(i)} optimizing learning configuration and bandwidth allocation via an augmented Lagrangian to reduce training costs; \textit{(ii)} introducing a device fitness score based on data heterogeneity (via Kullback-Leibler divergence), device-to-UAV proximity, and computational resources, using a TD3-based algorithm for adaptive device-to-UAV assignment; \textit{(iii)} developing a low-complexity two-stage greedy strategy for UAV redeployment and global aggregator selection, ensuring efficient aggregation despite UAV disconnections. Experiments on diverse real-world datasets validate the approach, demonstrating cost reduction and robust performance under communication disruptions.

联邦学习无人机智能物联网优化

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