journal article·2024-07-30Open access

Prediction of Network Security Situation Based on Attention Mechanism and Convolutional Neural Network–Gated Recurrent Unit

Yuan Feng, Hongying Zhao, Jianwei Zhang, Zengyu Cai, Liang Zhu, Ran Zhang

Applied Sciences· Volume 14· Issue 15

Abstract

Network-security situation prediction is a crucial aspect in the field of network security. It is primarily achieved through monitoring network behavior and identifying potential threats to prevent and respond to network attacks. In order to enhance the accuracy of situation prediction, this paper proposes a method that combines a convolutional neural network (CNN) and a gated recurrent unit (GRU), while also incorporating an attention mechanism. The model can simultaneously handle the spatial and temporal features of network behavior and optimize the weight allocation of features through the attention mechanism. Firstly, the CNN’s powerful feature extraction ability is utilized to extract the spatial features of the network behavior. Secondly, time-series features of network behavior are processed through the GRU layer. Finally, to enhance the model’s performance further, we introduce attention mechanisms, which can dynamically adjust the importance of different features based on the current context information; this enables the model to focus more on critical information for accurate predictions. The experimental results show that the network-security situation prediction method, which combines a CNN and a GRU and introduces an attention mechanism, performs well in terms of the fitting effect and can effectively enhance the accuracy of situation prediction.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Network Security and Intrusion Detection · Anomaly Detection Techniques and Applications · Internet Traffic Analysis and Secure E-voting

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

School of Electronics and Information, Zhengzhou University of Light Industry, Zhengzhou 450003, China

School of Computer Science and Technology, Zhengzhou University of Light Industry, Zhengzhou 450003, China

School of Software Engineering, Zhengzhou University of Light Industry, Zhengzhou 450003, China

Research Institute of Industrial Technology, Zhengzhou University of Light Industry, Zhengzhou 450003, China

Funding

National Natural Science Foundation of China · Key Research and Development Special Project of Henan Province · Key Technologies R&D Program of Henan Province

References (32)

Show source-provided references
1. Network security concepts, dangers, and defense best practical
2. Survey of attack projection, prediction, and forecasting in cyber security
3. Swarm intelligence inspired intrusion detection systems—A systematic literature review10.1016/j.comnet.2021.108708 ↗
4. Deep learning with blockchain based cyber security threat intelligence and situational awareness system for intrusion alert prediction
5. Zhang, J., Feng, H., Liu, B., and Zhao, D. (2023). Survey of technology in network security situation awareness. Sensors, 23.10.3390/s23052608 ↗
6. Network security situation awareness forecasting based on statistical approach and neural networks10.1093/jigpal/jzac024 ↗
7. Shen, Y., Mariconti, E., Vervier, P.A., and Stringhini, G. (2018, January 15–19). Tiresias: Predicting security events through deep learning. Proceedings of the 2018 ACM SIGSAC Conference on Computer and Communications Security, Toronto, ON, Canada.
8. Zhang, H., Kang, C., and Xiao, Y. (2021). Research on network security situation awareness based on the LSTM-DT model. Sensors, 21.10.3390/s21144788 ↗
9. Network Security Situation Prediction Model Based on VMD Decomposition and DWOA Optimized BiGRU-ATTN Neural Network10.1109/ACCESS.2023.3333666 ↗
10. Security Situation Prediction Method of Industrial Control System Based on Self-Attention and GRU Neural Network
11. Neural network security situation prediction method based on attention-GRU
12. Li, X. (2023). CNN-GRU model based on attention mechanism for large-scale energy storage optimization in smart grid. Front. Energy Res., 11.10.3389/fenrg.2023.1228256 ↗
13. A hybrid approach for forecasting ship motion using CNN–GRU–AM and GCWOA10.1016/j.asoc.2021.108084 ↗
14. Water level prediction model based on GRU and CNN10.1109/ACCESS.2020.2982433 ↗
15. Research on network security situation prediction method based on DS evidence theory
16. Prediction of network security based on DS evidence theory10.4218/etrij.2019-0147 ↗
17. Network security situation prediction method based on grey theory
18. Leau, Y.B., and Manickam, S. (2016). A novel adaptive grey verhulst model for network security situation prediction. Int. J. Adv. Comput. Sci. Appl., 7.10.14569/IJACSA.2016.070113 ↗
19. Network security situation prediction based on MR-SVM10.1109/ACCESS.2019.2939490 ↗
20. Network security situation prediction method based on support vector machine optimized by artificial Bee colony algorithms
21. A security situation prediction algorithm based on HMM in mobile network10.1155/2018/5380481 ↗
22. Peshave, A., Ganesan, A., and Oates, T. (2022). Predicting network threat events using HMM ensembles. International Conference on Advanced Data Mining and Applications, Springer International Publishing.10.1007/978-3-030-95405-5_17 ↗
23. Li, Y., and Feng, W. (2023). Improved population intelligence algorithm and BP neural network for network security posture prediction. Int. J. Distrib. Sens. Netw., 9970205.10.1155/2023/9970205 ↗
24. Xiao, P., Xian, M., and Wang, H. (2017, January 9–10). Network security situation prediction method based on MEA-BP. Proceedings of the 2017 3rd International Conference on Computational Intelligence & Communication Technology (CICT), Ghaziabad, India.10.1109/CIACT.2017.7977327 ↗
25. Research on internet security situation awareness prediction stechnology based on improved RBF neural network algorithm
26. Shang, L., Zhao, W., Zhang, J., Fu, Q., Zhao, Q., and Yang, Y. (2019, January 18–20). Network security situation prediction based on long short-term memory network. Proceedings of the 2019 20th Asia-Pacific Network Operations and Management Symposium (APNOMS), Matsue, Japan.10.23919/APNOMS.2019.8893096 ↗
27. Xiao, K., Zhang, Y., He, Y., Xu, G., and Wang, C. (2022, January 10–13). Industrial IoT Network Security Situation Prediction Based on Improved SSA-BiLSTM. Proceedings of the China Conference on Wireless Sensor Networks, Guangzhou, China.10.1007/978-981-19-8350-4_17 ↗
28. GRU-based deep learning approach for network intrusion alert prediction10.1016/j.future.2021.09.040 ↗
29. Jacob, S., Qiao, Y., Jacob, P., and Lee, B. (2020, January 25–29). Using recurrent neural networks to predict future events in a case with application to cyber security. Proceedings of the BUSTECH 2020: The Tenth International Conference on Business Intelligence and Technology, Nice, France.
30. Gao, F., Xia, J., Wu, D., Wang, W., Wang, C., and Song, C. (2023, January 3–5). Network security situation prediction based on LSTM. Proceedings of the 2023 2nd International Conference on Cloud Computing, Big Data Application and Software Engineering (CBASE), Chengdu, China.10.1109/CBASE60015.2023.10439145 ↗
31. Du, X., Ding, X., and Tao, F. (2023). Network Security Situation Prediction Based on Optimized Clock-Cycle Recurrent Neural Network for Sensor-Enabled Networks. Sensors, 23.10.3390/s23136087 ↗
32. ALSNAP: Attention-based long and short-period network security situation prediction10.1016/j.adhoc.2023.103279 ↗