A Swarm Anomaly Detection Model for IoT UAVs Based on a Multi-Modal Denoising Autoencoder and Federated Learning

Yu Lu,Tao Yang,Chong Zhao,Wen Chen,Rong Zeng
DOI: https://doi.org/10.1016/j.cie.2024.110454
IF: 7.18
2024-01-01
Computers & Industrial Engineering
Abstract:The widespread application of unmanned aerial vehicle (UAV) swarms has posed unique challenges for anomaly detection. Multi-modal noise from multi-source heterogeneous sensors during UAV swarm communication affects data quality, and limited data sharing between different UAV organisations restricts training a unified anomaly detection model. To address these problems, this study proposes a UAV swarm anomaly detection model based on a multi-modal denoising autoencoder and federated learning (L-MDAE). First, L-MDAE simulates noise by adding perturbations to the original data during UAV swarm communication. Second, according to the characteristics of UAV data noise, this study designs a new MSE loss function (normalised mean square error, NMSE) based on the normalised correlation coefficient. Furthermore, heterogeneous neural networks with NMSE are constructed to enhance the multi-modal noise-removal capability of the model. Finally, this study considers the UAV control node as the client and the ground control station as the server. Using a federated learning mechanism, L-MDAE is trained on a client dataset, and its parameters are integrated and distributed on the server. In this way, each UAV can effectively detect abnormal data using L-MDAE. Experimental results on five datasets, including ALFA, TLM and ITS, demonstrate that L-MDAE outperforms baseline and related models. When using ALFA, L-MDAE achieved an accuracy of 0.9919 and a swarm anomaly detection accuracy of 0.9901, approximately 2% higher than that of the baseline model.
What problem does this paper attempt to address?