Network Attack Prediction With Hybrid Temporal Convolutional Network and Bidirectional GRU

被引:13
作者
Bi, Jing [1 ]
Xu, Kangyuan [1 ]
Yuan, Haitao [2 ]
Zhang, Jia [3 ]
Zhou, Mengchu [4 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Sch Software Engn, Beijing 100124, Peoples R China
[2] Beihang Univ, Sch Automat Sci & Elect Engn, Beijing 100191, Peoples R China
[3] Southern Methodist Univ, Lyle Sch Engn, Dept Comp Sci, Dallas, TX 75205 USA
[4] New Jersey Inst Technol, Dept Elect & Comp Engn, Newark, NJ 07102 USA
基金
北京市自然科学基金;
关键词
Feature extraction; Time series analysis; Bidirectional control; Correlation; Predictive models; Logic gates; Recurrent neural networks; Gated recurrent unit (GRU); multihead self-attention; network attack prediction; Savitzky-Golay (SG) filter; temporal convolutional network (TCN); NEURAL-NETWORK; MODEL; IMPACT;
D O I
10.1109/JIOT.2023.3334912
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Precise and real-time prediction of future network attacks can not only prompt cloud infrastructures to fast respond and protect network security but also prevents economic and business losses. In recent years, neural networks, e.g., bidirectional gated recurrent unit (Bi-GRU) network and temporal convolutional network (TCN), have been proven to be suitable for predicting time-series data. Attention mechanisms are also widely used for the prediction of the time series of network attacks. This work proposes a hybrid deep learning prediction method that combines the capabilities of Savitzky-Golay (SG) filter, TCN, multihead self-attention, and Bi-GRU (STMB) for the prediction of network attacks. This work first adopts an SG filter to smooth possible outliers and noise in network attack traffic data. It applies TCN to extract abstract features from 1-D time series to make full use of data. It then adopts multihead self-attention to capture internal correlations among multidimensional features, by increasing the weights of key features and reducing those weight of non-key features, making that STMB captures important features adaptively. Finally, this work adopts Bi-GRU to extract bidirectional and long-term correlations in the time series to improve the prediction accuracy. This work also utilizes a hybrid algorithm named genetic simulated-annealing-based particle swarm optimizer to determine the hyperparameter setting of STMB. Experimental results with real-life data sets show that STMB outperforms several commonly used algorithms in terms of prediction accuracy.
引用
收藏
页码:12619 / 12630
页数:12
相关论文
共 40 条
[1]   An Ensemble Deep Learning-Based Cyber-Attack Detection in Industrial Control System [J].
Al-Abassi, Abdulrahman ;
Karimipour, Hadis ;
Dehghantanha, Ali ;
Parizi, Reza M. .
IEEE ACCESS, 2020, 8 :83965-83973
[2]  
Bai SJ, 2018, Arxiv, DOI [arXiv:1803.01271, 10.48550/arXiv.1803.01271, DOI 10.48550/ARXIV.1803.01271]
[3]   Identifying Attack Propagation Patterns in Honeypots using Markov Chains Modeling and Complex Networks Analysis [J].
Bar, Ariel ;
Shapira, Bracha ;
Rokach, Lior ;
Unger, Moshe .
2016 IEEE INTERNATIONAL CONFERENCE ON SOFTWARE SCIENCE, TECHNOLOGY AND ENGINEERING (SWSTE 2016), 2016, :28-36
[4]   Support vector regression for link load prediction [J].
Bermolen, Paola ;
Rossi, Dario .
COMPUTER NETWORKS, 2009, 53 (02) :191-201
[5]  
Bi Jing, 2022, 2022 IEEE International Conference on Systems, Man, and Cybernetics (SMC), P544, DOI 10.1109/SMC53654.2022.9945189
[6]  
Bi J, 2018, IEEE INT C NETW SENS
[7]   Green Energy Forecast-Based Bi-Objective Scheduling of Tasks Across Distributed Clouds [J].
Bi, Jing ;
Yuan, Haitao ;
Zhang, Jia ;
Zhou, MengChu .
IEEE TRANSACTIONS ON SUSTAINABLE COMPUTING, 2022, 7 (03) :619-630
[8]   A Hybrid Prediction Method for Realistic Network Traffic With Temporal Convolutional Network and LSTM [J].
Bi, Jing ;
Zhang, Xiang ;
Yuan, Haitao ;
Zhang, Jia ;
Zhou, MengChu .
IEEE TRANSACTIONS ON AUTOMATION SCIENCE AND ENGINEERING, 2022, 19 (03) :1869-1879
[9]   Energy-Optimized Partial Computation Offloading in Mobile-Edge Computing With Genetic Simulated-Annealing-Based Particle Swarm Optimization [J].
Bi, Jing ;
Yuan, Haitao ;
Duanmu, Shuaifei ;
Zhou, MengChu ;
Abusorrah, Abdullah .
IEEE INTERNET OF THINGS JOURNAL, 2021, 8 (05) :3774-3785
[10]   Integrated deep learning method for workload and resource prediction cloud systems [J].
Bi, Jing ;
Li, Shuang ;
Yuan, Haitao ;
Zhou, MengChu .
NEUROCOMPUTING, 2021, 424 :35-48