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 条
[11]   Temporal Prediction of Multiapplication Consolidated Workloads in Distributed Clouds [J].
Bi, Jing ;
Yuan, Haitao ;
Zhou, Mengchu .
IEEE TRANSACTIONS ON AUTOMATION SCIENCE AND ENGINEERING, 2019, 16 (04) :1763-1773
[12]   Workload Prediction Using ARIMA Model and Its Impact on Cloud Applications' QoS [J].
Calheiros, Rodrigo N. ;
Masoumi, Enayat ;
Ranjan, Rajiv ;
Buyya, Rajkumar .
IEEE TRANSACTIONS ON CLOUD COMPUTING, 2015, 3 (04) :449-458
[13]   Machine learning techniques applied to detect cyber attacks on web applications [J].
Choras, Michal ;
Kozik, Rafal .
LOGIC JOURNAL OF THE IGPL, 2015, 23 (01) :45-56
[14]   ARIMA models to predict next-day electricity prices [J].
Contreras, J ;
Espínola, R ;
Nogales, FJ ;
Conejo, AJ .
IEEE TRANSACTIONS ON POWER SYSTEMS, 2003, 18 (03) :1014-1020
[15]   A Survey of Cyber Attacks on Cyber Physical Systems: Recent Advances and Challenges [J].
Duo, Wenli ;
Zhou, Meng Chu ;
Abusorrah, Abdullah .
IEEE-CAA JOURNAL OF AUTOMATICA SINICA, 2022, 9 (05) :784-800
[16]   Resilience Against Replay Attacks: A Distributed Model Predictive Control Scheme for Networked Multi-Agent Systems [J].
Franze, Giuseppe ;
Tedesco, Francesco ;
Famularo, Domenico .
IEEE-CAA JOURNAL OF AUTOMATICA SINICA, 2021, 8 (03) :628-640
[17]   Learning-Based Simultaneous Detection and Characterization of Time Delay Attack in Cyber-Physical Systems [J].
Ganesh, Prakhar ;
Lou, Xin ;
Chen, Yao ;
Tan, Rui ;
Yau, David K. Y. ;
Chen, Deming ;
Winslett, Marianne .
IEEE TRANSACTIONS ON SMART GRID, 2021, 12 (04) :3581-3593
[18]   Denoising Nonlinear Time Series by Adaptive Filtering and Wavelet Shrinkage: A Comparison [J].
Gao, Jianbo ;
Sultan, Hussain ;
Hu, Jing ;
Tung, Wen-Wen .
IEEE SIGNAL PROCESSING LETTERS, 2010, 17 (03) :237-240
[19]   Novel Transformer Based on Gated Convolutional Neural Network for Dynamic Soft Sensor Modeling of Industrial Processes [J].
Geng, Zhiqiang ;
Chen, Zhiwei ;
Meng, Qingchao ;
Han, Yongming .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2022, 18 (03) :1521-1529
[20]   Genetic Learning Particle Swarm Optimization [J].
Gong, Yue-Jiao ;
Li, Jing-Jing ;
Zhou, Yicong ;
Li, Yun ;
Chung, Henry Shu-Hung ;
Shi, Yu-Hui ;
Zhang, Jun .
IEEE TRANSACTIONS ON CYBERNETICS, 2016, 46 (10) :2277-2290