Robust federated learning for UAV object detection: a joint self-distillation and drift compensation approach

面向无人机目标检测的鲁棒联邦学习:一种联合自蒸馏和漂移补偿方法

阅读:1

Abstract

The rapid advancement of unmanned aerial vehicles (UAVs) in disaster response and environmental monitoring has underscored the growing importance of real-time object detection within UAV swarm networks. However, the non-independent and identically distributed (non-IID) characteristics of data in UAV networks present significant challenges to model convergence and adaptability. To tackle these challenges, this study introduces a robust federated UAV object detection framework tailored for non-IID data distributions. The framework aims to enhance adaptability across clients, thereby improving both detection performance and convergence speed. Our approach includes a self-distillation mechanism that leverages personalized knowledge from local model historical states to guide current local training, striking a balance between specialization and adaptability. Additionally, we propose a drift compensation mechanism to synchronize local and global model updates, mitigating model drift. We conducted extensive experiments on the VisDrone2019-DET dataset, comparing our method to baseline models. Results demonstrate that our approach accelerates convergence speed by approximately 2.2 times and enhances detection performance by around 3%, offering an efficient and robust solution for UAV-based object detection under non-IID conditions.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。