journal article·2019-08-19Open access

An LSTM-Based Deep Learning Approach for Classifying Malicious Traffic at the Packet Level

Ren-Hung Hwang, Min-Chun Peng, Van-Linh Nguyen, Yu-Lun Chang

Applied Sciences· Volume 9· Issue 16

Abstract

Recently, deep learning has been successfully applied to network security assessments and intrusion detection systems (IDSs) with various breakthroughs such as using Convolutional Neural Networks (CNN) and Long Short-Term Memory (LSTM) to classify malicious traffic. However, these state-of-the-art systems also face tremendous challenges to satisfy real-time analysis requirements due to the major delay of the flow-based data preprocessing, i.e., requiring time for accumulating the packets into particular flows and then extracting features. If detecting malicious traffic can be done at the packet level, detecting time will be significantly reduced, which makes the online real-time malicious traffic detection based on deep learning technologies become very promising. With the goal of accelerating the whole detection process by considering a packet level classification, which has not been studied in the literature, in this research, we propose a novel approach in building the malicious classification system with the primary support of word embedding and the LSTM model. Specifically, we propose a novel word embedding mechanism to extract packet semantic meanings and adopt LSTM to learn the temporal relation among fields in the packet header and for further classifying whether an incoming packet is normal or a part of malicious traffic. The evaluation results on ISCX2012, USTC-TFC2016, IoT dataset from Robert Gordon University and IoT dataset collected on our Mirai Botnet show that our approach is competitive to the prior literature which detects malicious traffic at the flow level. While the network traffic is booming year by year, our first attempt can inspire the research community to exploit the advantages of deep learning to build effective IDSs without suffering significant detection delay.
Source: crossref · Quality: baseline

Research context

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

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

Affiliations

Department of Computer Science and Information Engineering, National Chung Cheng University, Chiayi 62102, Taiwan

Funding

Ministry of Science and Technology, Taiwan

References (20)

Show source-provided references
1. Javaid, A., Niyaz, Q., Sun, W., and Alam, M. (2016, January 3–5). A deep learning approach for network intrusion detection system. Proceedings of the 9th EAI International Conference on Bio-Inspired Information and Communications Technologies, New York, NY, USA.10.4108/eai.3-12-2015.2262516 ↗
2. N-BaIoT—Network-Based Detection of IoT Botnet Attacks Using Deep Autoencoders10.1109/MPRV.2018.03367731 ↗
3. Wang, W., Zhu, M., Zeng, X., Ye, X., and Sheng, Y. (2017, January 11–13). Malware Traffic Classification Using Convolutional Neural Networks for Representation Learning. Proceedings of the 31st International Conference on Information Networking, Da Nang, Vietnam.
4. Chen, Y.C., Li, Y.J., Tseng, A., and Lin, T. (2017, January 8–13). Deep Learning for Malicious Flow Detection. Proceedings of the IEEE 28th Annual International Symposium on Personal, Indoor, and Mobile Radio Communications, Montreal, QC, Canada.10.1109/PIMRC.2017.8292316 ↗
5. Yuan, X., Li, C., and Li, X. (2017, January 29–31). DeepDefense: Identifying DDoS Attack via Deep Learning. Proceedings of the IEEE International Conference on Smart Computing, Hong Kong, China.10.1109/SMARTCOMP.2017.7946998 ↗
6. A deep learning approach for intrusion detection using recurrent neural networks10.1109/ACCESS.2017.2762418 ↗
7. Using a Recurrent Neural Network and Restricted Boltzmann Machines for Malicious Traffic Detection
8. Hwang, R.H., Peng, M.C., and Huang, C.W. (2019). Detecting IoT Malicious Traffic based on Autoencoder and Convolutional Neural Network. IEEE Globecom Conf., submitted.10.1109/GCWkshps45667.2019.9024425 ↗
9. Radford, B.J., Apolonio, L.M., Trias, A.J., and Simpson, J.A. (2018). Network Traffic Anomaly Detection Using Recurrent Neural Networks. arXiv.
10. Cui, J., Long, J., Min, E., and Mao, Y. (2018, January 15–18). WEDL-NIDS: Improving Network Intrusion Detection Using Word Embedding-Based Deep Learning Method. Proceedings of the International Conference on Modeling Decisions for Artificial Intelligence, Palma de Mallorce, Spain.10.1007/978-3-030-00202-2_23 ↗
11. Kim, M.S., Kong, H.J., Hong, S.C., Chung, S.H., and Hong, J.W. (2004, January 23). A flow-based method for abnormal network traffic detection. Proceedings of the 2004 IEEE/IFIP Network Operations and Management Symposium, Seoul, Korea.
12. Fu, R., Zhang, Z., and Li, L. (2016, January 11–13). Using LSTM and GRU neural network methods for traffic flow prediction. Proceedings of the 2016 31st Youth Academic Annual Conference of Chinese Association of Automation, Wuhan, China.10.1109/YAC.2016.7804912 ↗
13. Goldberg, Y., and Levy, O. (2013). word2vec Explained: Deriving Mikolov et al.’s negative-sampling word-embedding method. arXiv.
14. Pennington, J., Socher, R., and Manning, C.D. (2014, January 25–29). GloVe: Global vectors for word representation. Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing, Doha, Qatar.10.3115/v1/D14-1162 ↗
15. Ronen, R., Radu, M., Feuerstein, C., Yom-Tov, E., and Ahmadi, M. (2019, May 29). Microsoft Malware Classification Challenge. Available online: https://arxiv.org/pdf/1802.10135.pdf.
16. McDermott, C.D., Majdani, F., and Petrovski, A.V. (2018, January 8–13). Botnet Detection in the Internet of Things using Deep Learning Approaches. Proceedings of the 2018 International Joint Conference on Neural Networks (IJCNN), Rio de Janeiro, Brazil.10.1109/IJCNN.2018.8489489 ↗
17. Toward developing a systematic approach to generate benchmark datasets for intrusion detection10.1016/j.cose.2011.12.012 ↗
18. Mikolov, T., tau Yih, W., and Zweig, G. (2013, January 9–14). Linguistic Regularities in Continuous Space Word Representations. Proceedings of the 2013 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Atlanta, GA, USA.
19. A Deep Learning Approach to Network Intrusion Detection10.1109/TETCI.2017.2772792 ↗
20. HAST-IDS: Learning Hierarchical Spatial-Temporal Features Using Deep Neural Networks to Improve Intrusion Detection10.1109/ACCESS.2017.2780250 ↗