Federation Learning for Intrusion Detection Methods by Parse Convolutional Neural Network
Jiechen Luo, Yang Xuelan, Muamer N. Mohammed · 2022 Second International Conference on Advances in Electrical, Computing, Communication and Sustainable Technologies (ICAECT) · 2022
Intrusion detection algorithms based on deep learning are currently a hot topic in the field of intrusion detection research, but most of the research focuses on how to improve the algorithms to improve the accuracy of intrusion detection, while ignoring the problem that the limited labeled data generated by a single organization is not enough to train a deep model with high accuracy in practice. The paper proposes a federal learning and convolutional neural network-based intrusion detection method that can expand the data volume by jointly training the model with data sets from multiple participants. The method uses a federation learning framework to design a deep learning-based intrusion detection model. The data dimensionality is first reconstructed by data padding to form two-dimensional data, then the DCNN network is used for feature extraction and learning under the mechanism of federation learning, and finally the model is trained with a softmax classifier for detection. In addition, aiming at the over-fitting problem caused by the compression and over-parameterization of deep neural networks, this paper proposes a model pruning method based on sparse convolutional neural network according to the weight parameters of the convolutional layer and the Batch Normalization layer (BN). This section will introduce the implementation details of this method. The experimental results show that this method largely reduces the training time and maintains a high detection rate. In addition, the model ensures data security and privacy compared to typical intrusion detection models.