无人机群多任务联邦学习,动态共享知识提升灾后救援效率
Efficient UAV Swarm-Based Multi-Task Federated Learning with Dynamic Task Knowledge Sharing
- 无人机与地面车辆协同,动态分配任务优先级和资源
- 引入任务亲和度指标,提升多任务训练速度与泛化能力
- 兼顾能效与性能,适合应急通信等资源受限场景
无人机群广泛应用于应急通信、区域监控和灾害救援。由控制中心协调的无人机群适合作为联邦学习(FL)框架的载体。然而,现有无人机辅助的联邦学习方法主要聚焦单任务,忽视了多任务训练需求。在灾害救援场景中,无人机需执行人群检测、道路可行性分析、灾害评估等任务,这些任务具有时变需求且存在潜在关联。为应对任务的时变需求,并在资源受限条件下高效完成多任务,本文提出一种基于无人机群的多任务联邦学习框架,通过地面应急车辆(EVs)与无人机协同,在能量与带宽受限条件下高效完成多任务。通过理论分析,识别影响任务性能的关键因素,引入任务注意力机制以动态评估任务重要性,实现高效资源分配。此外,提出任务亲和度(TA)度量来捕捉任务间的动态相关性,促进任务知识共享,加速训练并提升模型在不同场景下的泛化能力。为优化资源分配,构建两层优化问题,联合优化无人机传输功率、计算频率、带宽分配及无人机-车辆关联。针对内层问题,推导出传输功率、计算频率和带宽分配的闭式解,并采用块坐标下降法求解;针对外层问题,设计两阶段算法确定最优无人机-车辆关联。理论分析揭示了无人机能耗与多任务性能之间的权衡关系。
原文摘要 · Abstract (English)
UAV swarms are widely used in emergency communications, area monitoring, and disaster relief. Coordinated by control centers, they are ideal for federated learning (FL) frameworks. However, current UAV-assisted FL methods primarily focus on single tasks, overlooking the need for multi-task training. In disaster relief scenarios, UAVs perform tasks such as crowd detection, road feasibility analysis, and disaster assessment, which exhibit time-varying demands and potential correlations. In order to meet the time-varying requirements of tasks and complete multiple tasks efficiently under resource constraints, in this paper, we propose a UAV swarm based multi-task FL framework, where ground emergency vehicles (EVs) collaborate with UAVs to accomplish multiple tasks efficiently under constrained energy and bandwidth resources. Through theoretical analysis, we identify key factors affecting task performance and introduce a task attention mechanism to dynamically evaluate task importance, thereby achieving efficient resource allocation. Additionally, we propose a task affinity (TA) metric to capture the dynamic correlation among tasks, thereby promoting task knowledge sharing to accelerate training and improve the generalization ability of the model in different scenarios. To optimize resource allocation, we formulate a two-layer optimization problem to jointly optimize UAV transmission power, computation frequency, bandwidth allocation, and UAV-EV associations. For the inner problem, we derive closed-form solutions for transmission power, computation frequency, and bandwidth allocation and apply a block coordinate descent method for optimization. For the outer problem, a two-stage algorithm is designed to determine optimal UAV-EV associations. Furthermore, theoretical analysis reveals a trade-off between UAV energy consumption and multi-task performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。