Few-Shot network intrusion detection based on prototypical capsule network with attention mechanism

Handi Sun,Liang Wan,Mengying Liu,Bo Wang
DOI: https://doi.org/10.1371/journal.pone.0284632
IF: 3.7
2023-04-21
PLoS ONE
Abstract:Network intrusion detection plays a crucial role in ensuring network security by distinguishing malicious attacks from normal network traffic. However, imbalanced data affects the performance of intrusion detection system. This paper utilizes few-shot learning to solve the data imbalance problem caused by insufficient samples in network intrusion detection, and proposes a few-shot intrusion detection method based on prototypical capsule network with the attention mechanism. Our method is mainly divided into two parts, a temporal-spatial feature fusion method using capsules for feature extraction and a prototypical network classification method with attention and vote mechanisms. The experimental results demonstrate that our proposed model outperforms state-of-the-art methods on imbalanced datasets.
multidisciplinary sciences
What problem does this paper attempt to address?