AESMOTE: Adversarial Reinforcement Learning With SMOTE for Anomaly Detection

被引:79
作者
Ma, Xiangyu [1 ]
Shi, Wei [1 ]
机构
[1] Carleton Univ, Sch Informat Technol, Ottawa, ON K1S 5B6, Canada
来源
IEEE TRANSACTIONS ON NETWORK SCIENCE AND ENGINEERING | 2021年 / 8卷 / 02期
基金
加拿大自然科学与工程研究理事会;
关键词
adversarial strategy; anomaly detection; class-imbalance; deep reinforcement learning; dynamic training; feature-selection;
D O I
10.1109/TNSE.2020.3004312
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Intrusion Detection Systems (IDSs) play a vital role in securing today's Data-Centric Networks. In a dynamic environment such as the Internet of Things (IoT), which is vulnerable to various types of attacks, fast and robust solutions are in demand to handle fast-changing threats and thus the everincreasing difficulty of detection. In this paper, we present a novel framework for the detection of anomalies, which, in particular, supports intrusion detection. The anomaly-detection framework we propose combines reinforcement learning with class-imbalance techniques. Our goal is not only to exploit the autolearning ability of the reinforcement-learning loop but also to address the dataset imbalance problem, which is pervasive in existing learning-based solutions. We introduce an adapted SMOTE to address the class-imbalance problem while remodelling the behaviors of the environment agent for better performance. Experiments are conducted on NSL-KDD datasets. Comparative evaluations and their results are presented and analyzed. Using techniques such as SMOTE, ROS, NearMiss1 and NearMiss2, performance measures obtained from our simulations have led us to recognize specific performance trends. In particular, the proposed model AESMOTE outperforms AE-RL in several cases. Experiment results show an Accuracy greater than 0.82 and a F1 greater than 0.824.
引用
收藏
页码:943 / 956
页数:14
相关论文
共 36 条
[1]  
[Anonymous], 2011, CLASS IMBALANCE PROB
[2]  
[Anonymous], 2012, GUIDE INTRUSION DETE
[3]  
[Anonymous], 2019, CORR
[4]   Boosted Near-miss Under-sampling on SVM ensembles for concept detection in large-scale imbalanced datasets [J].
Bao, Lei ;
Juan, Cao ;
Li, Jintao ;
Zhang, Yongdong .
NEUROCOMPUTING, 2016, 172 :198-206
[5]   Network Anomaly Detection: Methods, Systems and Tools [J].
Bhuyan, Monowar H. ;
Bhattacharyya, D. K. ;
Kalita, J. K. .
IEEE COMMUNICATIONS SURVEYS AND TUTORIALS, 2014, 16 (01) :303-336
[6]   SMOTE for high-dimensional class-imbalanced data [J].
Blagus, Rok ;
Lusa, Lara .
BMC BIOINFORMATICS, 2013, 14
[7]   Adversarial environment reinforcement learning algorithm for intrusion detection [J].
Caminero, Guillermo ;
Lopez-Martin, Manuel ;
Carro, Belen .
COMPUTER NETWORKS, 2019, 159 :96-109
[8]   SMOTE: Synthetic minority over-sampling technique [J].
Chawla, Nitesh V. ;
Bowyer, Kevin W. ;
Hall, Lawrence O. ;
Kegelmeyer, W. Philip .
2002, American Association for Artificial Intelligence (16)
[9]  
Dhanabal L., 2015, Appl. Sci.
[10]   Intrusion detection systems for IoT-based smart environments: a survey [J].
Elrawy, Mohamed Faisal ;
Awad, Ali Ismail ;
Hamed, Hesham F. A. .
JOURNAL OF CLOUD COMPUTING-ADVANCES SYSTEMS AND APPLICATIONS, 2018, 7