arXiv:2501.10644cs.LGcs.MA2025-01中稿 · IEEE International…被引 3

无人机协同多任务联邦学习,提升训练效率与性能

UAV-Assisted Multi-Task Federated Learning with Task Knowledge Sharing

  • 多无人机共享特征提取器,通过任务注意力机制协同训练多个相关任务
  • 理论证明算法收敛性,优化带宽分配降低通信开销
  • 适合需要多任务并行的无人机智能系统,如灾后监测与区域巡检

无人机技术快速发展,广泛应用于应急通信、区域监控和灾害救援。由于电池容量和算力有限,复杂任务常需多架无人机协作,此时控制中心对协调其行为至关重要,契合联邦学习(FL)框架。然而传统联邦学习通常仅聚焦单一任务,忽视多任务联合训练潜力。本文提出一种无人机辅助的多任务联邦学习方案,允许多架无人机采集的数据同时用于多个相关任务的训练。该方案通过共享特征提取器促进知识迁移,并引入任务注意力机制平衡各任务性能,增强知识共享。进一步进行了收敛性分析以提供训练性能的理论描述。针对带宽受限场景,推导出最优带宽分配策略以最小化通信时间。同时,提出基于联盟博弈的无人机-电动车关联策略。仿真结果验证了该方案在提升多任务性能和训练速度方面的有效性。

原文摘要 · Abstract (English)

The rapid development of Unmanned aerial vehicles (UAVs) technology has spawned a wide variety of applications, such as emergency communications, regional surveillance, and disaster relief. Due to their limited battery capacity and processing power, multiple UAVs are often required for complex tasks. In such cases, a control center is crucial for coordinating their activities, which fits well with the federated learning (FL) framework. However, conventional FL approaches often focus on a single task, ignoring the potential of training multiple related tasks simultaneously. In this paper, we propose a UAV-assisted multi-task federated learning scheme, in which data collected by multiple UAVs can be used to train multiple related tasks concurrently. The scheme facilitates the training process by sharing feature extractors across related tasks and introduces a task attention mechanism to balance task performance and encourage knowledge sharing. To provide an analytical description of training performance, the convergence analysis of the proposed scheme is performed. Additionally, the optimal bandwidth allocation for UAVs under limited bandwidth conditions is derived to minimize communication time. Meanwhile, a UAV-EV association strategy based on coalition formation game is proposed. Simulation results validate the effectiveness of the proposed scheme in enhancing multi-task performance and training speed.

联邦学习无人机多任务学习边缘计算

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