无人机群协同追踪目标,用联邦学习提升精度与效率。
DroneFL: Federated Learning for Multi-UAV Visual Target Tracking
- 轻量化模型+自适应归一化,解决数据异构问题。
- 预测误差降低6%-83%,追踪距离减少0.4%-4.6%。
- 适合资源受限的无人机集群,支持实时部署。
多机器人目标追踪是精准农业、环境监测、应急响应和安全监控等应用中的基础问题。尽管联邦学习(FL)可在不集中数据的情况下提升多机器人协同学习能力,但在多无人机(UAV)目标追踪中的应用仍较少被探索。主要挑战包括机载计算资源有限、因目标和视角差异导致的数据异构性,以及轨迹预测与多机规划之间的紧密耦合需求。本文提出DroneFL,首个专为高效多无人机目标追踪设计的联邦学习框架。通过采用冻结的YOLO主干网络与浅层Transformer构建轻量本地模型,实现传感器输入下的目标轨迹预测,并周期性地将更新模型上传至云端进行全局知识聚合。为缓解数据异构对FL收敛的负面影响,DroneFL引入位置无关的模型架构及基于高度的自适应实例归一化。最终在云端融合多架无人机预测结果,生成兼顾预测精度与整体追踪性能的最优轨迹。实验表明,相比分布式非联邦学习框架,DroneFL将预测误差降低6%-83%,追踪距离减少0.4%-4.6%;同时可在Raspberry Pi 5上实时运行,平均仅需1.56 KBps云通信带宽。
原文摘要 · Abstract (English)
Multi-robot target tracking is a fundamental problem that requires coordinated monitoring of dynamic entities in applications such as precision agriculture, environmental monitoring, disaster response, and security surveillance. While Federated Learning (FL) has the potential to enhance learning across multiple robots without centralized data aggregation, its use in multi-Unmanned Aerial Vehicle (UAV) target tracking remains largely underexplored. Key challenges include limited onboard computational resources, significant data heterogeneity in FL due to varying targets and the fields of view, and the need for tight coupling between trajectory prediction and multi-robot planning. In this paper, we introduce DroneFL, the first federated learning framework specifically designed for efficient multi-UAV target tracking. We design a lightweight local model to predict target trajectories from sensor inputs, using a frozen YOLO backbone and a shallow transformer for efficient onboard training. The updated models are periodically aggregated in the cloud for global knowledge sharing. To alleviate the data heterogeneity that hinders FL convergence, DroneFL introduces a position-invariant model architecture with altitude-based adaptive instance normalization. Finally, we fuse predictions from multiple UAVs in the cloud and generate optimal trajectories that balance target prediction accuracy and overall tracking performance. Our results show that DroneFL reduces prediction error by 6%-83% and tracking distance by 0.4%-4.6% compared to a distributed non-FL framework. In terms of efficiency, DroneFL runs in real time on a Raspberry Pi 5 and has on average just 1.56 KBps data rate to the cloud.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。