Machine learning-based network intrusion detection for big and imbalanced data using oversampling, stacking feature embedding and feature extraction

被引:34
作者
Talukder, Md. Alamin [1 ,7 ]
Islam, Md. Manowarul [1 ]
Uddin, Md Ashraf [2 ]
Hasan, Khondokar Fida [3 ]
Sharmin, Selina [1 ]
Alyami, Salem A. [4 ]
Moni, Mohammad Ali [5 ,6 ]
机构
[1] Jagannath Univ, Dept Comp Sci & Engn, Dhaka, Bangladesh
[2] Deakin Univ, Sch Informat Technol, Waurn Ponds Campus, Geelong, Australia
[3] Univ New South Wales UNSW, Sch Profess Studies, 37 Constitut Ave, Canberra, ACT 2601, Australia
[4] Imam Mohammad Ibn Saud Islamic Univ IMSIU, Fac Sci, Dept Math & Stat, Riyadh 11432, Saudi Arabia
[5] Charles Sturt Univ, Artificial Intelligence & Cyber Futures Inst, AI & Digital Hlth Technol, Bathurst, NSW 2795, Australia
[6] Charles Sturt Univ, Rural Hlth Res Inst, AI & Digital Hlth Technol, Orange, NSW 2800, Australia
[7] Int Univ Business Agr & Technol, Dept Comp Sci & Engn, Dhaka, Bangladesh
关键词
Intrusion detection system; Feature extraction; Random oversampling; Principal component analysis; Machine learning; FEATURE-SELECTION; DETECTION SYSTEM;
D O I
10.1186/s40537-024-00886-w
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Cybersecurity has emerged as a critical global concern. Intrusion Detection Systems (IDS) play a critical role in protecting interconnected networks by detecting malicious actors and activities. Machine Learning (ML)-based behavior analysis within the IDS has considerable potential for detecting dynamic cyber threats, identifying abnormalities, and identifying malicious conduct within the network. However, as the number of data grows, dimension reduction becomes an increasingly difficult task when training ML models. Addressing this, our paper introduces a novel ML-based network intrusion detection model that uses Random Oversampling (RO) to address data imbalance and Stacking Feature Embedding based on clustering results, as well as Principal Component Analysis (PCA) for dimension reduction and is specifically designed for large and imbalanced datasets. This model's performance is carefully evaluated using three cutting-edge benchmark datasets: UNSW-NB15, CIC-IDS-2017, and CIC-IDS-2018. On the UNSW-NB15 dataset, our trials show that the RF and ET models achieve accuracy rates of 99.59% and 99.95%, respectively. Furthermore, using the CIC-IDS2017 dataset, DT, RF, and ET models reach 99.99% accuracy, while DT and RF models obtain 99.94% accuracy on CIC-IDS2018. These performance results continuously outperform the state-of-art, indicating significant progress in the field of network intrusion detection. This achievement demonstrates the efficacy of the suggested methodology, which can be used practically to accurately monitor and identify network traffic intrusions, thereby blocking possible threats.
引用
收藏
页数:44
相关论文
共 67 条
  • [61] Machine learning-based lung and colon cancer detection using deep feature extraction and ensemble learning
    Talukder, Md Alamin
    Islam, Md Manowarul
    Uddin, Md Ashraf
    Akhter, Arnisha
    Hasan, Khondokar Fida
    Moni, Mohammad Ali
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2022, 205
  • [62] Uddin MA, 2023, Biomed Mater Dev, P1
  • [63] Uddin N., 2023, Int J Cogn Comput Eng, V4, P327
  • [64] StAR: a simple tool for the statistical comparison of ROC curves
    Vergara, Ismael A.
    Norambuena, Tomas
    Ferrada, Evandro
    Slater, Alex W.
    Melo, Francisco
    [J]. BMC BIOINFORMATICS, 2008, 9 (1)
  • [65] Improving AdaBoost-based Intrusion Detection System (IDS) Performance on CIC IDS 2017 Dataset
    Yulianto, Arif
    Sukarno, Parman
    Suwastika, Novian Anggis
    [J]. 2ND INTERNATIONAL CONFERENCE ON DATA AND INFORMATION SCIENCE, 2019, 1192
  • [66] An effective convolutional neural network based on SMOTE and Gaussian mixture model for intrusion detection in imbalanced dataset
    Zhang, Hongpo
    Huang, Lulu
    Wu, Chase Q.
    Li, Zhanbo
    [J]. COMPUTER NETWORKS, 2020, 177
  • [67] Sparse principal component analysis
    Zou, Hui
    Hastie, Trevor
    Tibshirani, Robert
    [J]. JOURNAL OF COMPUTATIONAL AND GRAPHICAL STATISTICS, 2006, 15 (02) : 265 - 286