End-Edge Collaborative Lightweight Secure Federated Learning for Anomaly Detection of Wireless Industrial Control Systems

Abstract

With the wide applications of industrial wireless network technologies, the industrial control system (ICS) is evolving from wired and centralized to wireless and distributed, during which eavesdropping and attacking become serious problems. To guarantee the security of wireless and distributed ICS, this article establishes an end-edge collaborative lightweight secure federated learning (LSFL) architecture and proposes an LSFL anomaly detection strategy. Specifically, we first design a residual multihead self-attention convolutional neural network for local feature learning, where the variability and dependence of spatial-temporal features can be sufficiently evaluated. Then, to reduce the wireless communication cost for parameter exchange and edge federal learning, we propose a dynamic parameter pruning algorithm by evaluating the contribution of each parameter based on the information entropy gain. Furthermore, to ensure the parameter security during wireless transmission in the open radio environment, we propose an adaptive key generation algorithm for parameter encryption. Finally, the proposed strategy is experimentally validated on representative datasets, including Smart Meter, NSL-KDD, and UNSW-NB15. Experimental results demonstrate that the proposed strategy achieves 99% accuracy on different datasets, where at least 89.6% wireless communication cost is reduced and tampering/injecting attacks are defended.

Keywords

• Industrial control system (ICS)
• anomaly detection
• federated learning
• convolutional neural network (CNN)