用联邦学习让无人机群协作训练目标检测模型,数据本地保留
Federated Learning for Object Detection: Enabling Collaborative Drone Learning Without Centralizing Data

- 通过联邦学习在各无人机上本地训练,共享模型参数而不传原始图像
- YOLO26 nano模型在[email protected]和[email protected]:0.95上分别提升52.89%和67.80%
- 适合边缘设备部署,保障隐私且无需集中数据
目标检测是安全关键型无人机与边缘视觉系统(如灾害响应、安防监控、基础设施巡检及国防应用)中人工智能感知的核心能力。高性能模型依赖大规模、持续更新的数据集,但传统训练需集中航拍图像,带来隐私、监管、存储与带宽挑战,尤其在分布式无人机部署中,图像常无法或不宜上传至中心平台。本文提出基于联邦学习(FL)的目标检测框架,使无人机在不共享原始数据的前提下协同优化共享模型。我们在Sherpa.ai FL平台上使用KIIT-MiTA数据集实现该流程,并对比单机训练与集中式训练的mAP表现(IoU阈值分别为0.50和0.50–0.95)。实验表明,所提方法接近集中式训练性能,显著优于单机训练:最优轻量模型YOLO26 nano在[email protected]和[email protected]:0.95上相对提升52.89%与67.80%,适用于极低算力边缘设备。结果证明,联邦学习可实现无数据集中化的可扩展、高性能、隐私保护目标检测。
原文摘要 · Abstract (English)
Object detection is a fundamental capability for AI-driven perception in safety-critical drone and edge-vision systems, including disaster response, operational security environments, infrastructure monitoring and defense applications. Robust model performance in such environments depends on large, continuously updated datasets. However, training high-performing detectors typically requires centralizing aerial imagery, which raises privacy, regulatory, storage, and bandwidth challenges. This is especially problematic in distributed drone deployments, where visual data is generated onboard and is often impractical or undesirable to transfer to a centralized infrastructure. In this work, we apply Federated Learning (FL) for object detection, enabling drones to improve a shared model while keeping image data local and private. We implement a federated object detection pipeline using the Sherpa.ai FL platform on the KIIT-MiTA dataset, and compare it with Single-drone and Centralized baselines using mean Average Precision (mAP) at IoU thresholds of 0.50 and 0.50-0.95. In our experiments, the proposed FL approach remains close to Centralized training while dramatically improving over Single-drone training, with the best lightweight model (YOLO26 nano), suitable for deployment even on very limited edge infrastructure, achieving relative gains of 52.89% and 67.80% in [email protected] and [email protected]:0.95, respectively. These results show that FL enables scalable, high-performing, and privacy-preserving object detection across distributed drone fleets without data centralization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。