A Quantum Feature Selection Method for Network Intrusion Detection

被引:2
作者
Li, Mingze [1 ,2 ]
Zhang, Hongliang [3 ]
Fan, Lei [2 ]
Han, Zhu [1 ]
机构
[1] Univ Houston, Dept Elect & Comp Engn, Houston, TX 77204 USA
[2] Univ Houston, Dept Engn Technol, Houston, TX 77204 USA
[3] Princeton Univ, Dept Elect & Comp Engn, Princeton, NJ 08544 USA
来源
2022 IEEE 19TH INTERNATIONAL CONFERENCE ON MOBILE AD HOC AND SMART SYSTEMS (MASS 2022) | 2022年
关键词
Quantum annealing; quantum machine learning; feature selection; QUBO;
D O I
10.1109/MASS56207.2022.00048
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Feature selection (FS) approaches rank features based on the score of the coefficients with labels. However, these selected features usually lead to a suboptimal solution to classification problems as they are selected independently. Moreover, with large dimensional feature sets, the computational complexity of FS algorithms can be prohibitively high. In this paper, we first formulate the feature selection problem as an integer programming model. Then we propose to utilize the strong computation ability of quantum annealers to solve the discrete integer programming problems, where the quantum annealer has the potential to be significantly faster than classical solvers to solve discrete optimization problems. We also design a wrapper algorithm to choose the optimal parameters of QUBO. Experiments show that our proposed strategy can select the representative features in the NSL-KDD dataset. Compared with HHO, WOA, PSO, and other algorithms, our strategy retains the least features to minimize the detection time, while the accuracy increase to 89.2%. Our algorithm also shows a good performance in computation time, detection rate and precision.
引用
收藏
页码:281 / 289
页数:9
相关论文
共 26 条
[1]   A feature selection algorithm for intrusion detection system based on Pigeon Inspired Optimizer [J].
Alazzam, Hadeel ;
Sharieh, Ahmad ;
Sabri, Khair Eddin .
EXPERT SYSTEMS WITH APPLICATIONS, 2020, 148
[2]   Integer programming models for feature selection: New extensions and a randomized solution algorithm [J].
Bertolazzi, P. ;
Felici, G. ;
Festa, P. ;
Fiscon, G. ;
Weitschek, E. .
EUROPEAN JOURNAL OF OPERATIONAL RESEARCH, 2016, 250 (02) :389-399
[3]  
Cohen I., 2009, The SAGE Encyclopedia of Educational Research, Measurement, and Evaluation, P1, DOI [DOI 10.4135/9781506326139, 10.1007/978-3-642-00296-0]
[4]  
Cristianini N, 2000, An introduction to support vector machines and other kernel-based learning methods
[5]  
Daum F, 2003, AEROSP CONF PROC, P1979
[6]  
E. National Academies of Sciences Medicine, 2019, QUANT COMP PROGR PRO
[7]   Normalized Mutual Information Feature Selection [J].
Estevez, Pablo. A. ;
Tesmer, Michel ;
Perez, Claudio A. ;
Zurada, Jacek A. .
IEEE TRANSACTIONS ON NEURAL NETWORKS, 2009, 20 (02) :189-201
[8]  
Glover F, 2019, Arxiv, DOI [arXiv:1811.11538, 10.48550/arXiv.1811.11538]
[9]  
Gurobi, MIX INT PROGR MIP PR
[10]  
Isola P, 2014, LECT NOTES COMPUT SC, V8691, P799, DOI 10.1007/978-3-319-10578-9_52