QShield-NS: A Variational Quantum Machine Learning Model for Zero-Day Cyber Threat Detection in National Security Systems
Abstract
: Machine-learning intrusion detection systems are regularly benchmarked against attack classes that also exist in their training data, a practice that inflates preparedness against the new types of threats most pertinent to national-security networks. We introduce QShield-NS, a variance-reduced quantum machine learning classifier capable of zero-day threat detection, and demonstrate its ability via strict Leave-One-Attack-Out (LOAO) evaluation on the UNSW-NB15 dataset, where each of nine different attack families are omitted from training in turn. Eight principal components are angle-encoded on eight qubits, processed by strongly entangling variational layers and read through Pauli-Z expectation values into a compact classical head that is trained end-to-end. QShield-NS is compared against a multilayer perceptron, a random forest and XGBoost on the same folds, with the same features and decision threshold. A rank inversion is what the results reveal. If however the training is known-attack control split QShield-NS ranks last (macro-F1 0.8793 v s 0.9158 random forest), and under the zero-day protocol first (0.7303 against 0.7044). Its generalisation gap, at 0.1489 to 0.2084–0.2302, is the smallest for the four models and it lowered the mean false-positive rate from 20.24 percent to 10.72 percent, scoring precision on eight of nine folds. It is a calibration, not discrimination advantage: the classical baselines get better ROC-AUC and PR-AUC in every fold but achieve distribution shift recall at the lower price of over-predicting the attack class. A depth ablation and a study of gradient-variance establish robust trainability up to 8 qubits. We do not claim any quantum advantage; all results were obtained by classical simulation
Article Information
Journal |
International Journal of Future Innovative Science and Technology (IJFIST) |
|---|---|
Volume (Issue) |
Vol. 5 No. 5 (2022): International Journal of Future Innovative Science and Technology (IJFIST) |
DOI |
|
Pages |
9266-9283 |
Published |
October 10, 2022 |
| Copyright |
All rights reserved |
Open Access |
This work is licensed under a Creative Commons Attribution 4.0 International License. |
How to Cite |
Md Sajedul Karim Chy, Salman Mohammad Abdullah, Mahbub Ahmed Nabil, Abidul Alam, Md Himeluzzaman, Tofayel Ahmed Onik, Shaown Mahamud Shakil, Hafiz Aziz Khan (2022). QShield-NS: A Variational Quantum Machine Learning Model for Zero-Day Cyber Threat Detection in National Security Systems. International Journal of Future Innovative Science and Technology (IJFIST) , Vol. 5 No. 5 (2022): International Journal of Future Innovative Science and Technology (IJFIST) , pp. 9266-9283. https://doi.org/10.15662/IJFIST.2022.0505009 |
References
[2] Moustafa, N.; Slay, J. The evaluation of Network Anomaly Detection Systems: Statistical analysis of the UNSW-NB15 data set and the comparison with the KDD99 data set. Information Security Journal: A Global Perspective 2016, 25, 18–31.
[3] Tavallaee, M.; Bagheri, E.; Lu, W.; Ghorbani, A.A. A detailed analysis of the KDD CUP 99 data set. In Proceedings of the 2009 IEEE Symposium on Computational Intelligence for Security and Defense Applications (CISDA), Ottawa, Canada, 2009; pp. 1–6.
[4] Sommer, R.; Paxson, V. Outside the Closed World: On Using Machine Learning for Network Intrusion Detection. In Proceedings of the 2010 IEEE Symposium on Security and Privacy, Oakland, CA, USA, 2010; pp. 305–316.
[5] Buczak, A.L.; Guven, E. A Survey of Data Mining and Machine Learning Methods for Cyber Security Intrusion Detection. IEEE Communications Surveys & Tutorials 2016, 18, 1153–1176.
[6] Khraisat, A.; Gondal, I.; Vamplew, P.; Kamruzzaman, J. Survey of intrusion detection systems: techniques, datasets and challenges. Cybersecurity 2019, 2, 20.
[7] Chandola, V.; Banerjee, A.; Kumar, V. Anomaly detection: A survey. ACM Computing Surveys 2009, 41, 1–58.
[8] Vinayakumar, R.; Alazab, M.; Soman, K.P.; Poornachandran, P.; Al-Nemrat, A.; Venkatraman, S. Deep Learning Approach for Intelligent Intrusion Detection System. IEEE Access 2019, 7, 41525–41550.
[9] Kasongo, S.M.; Sun, Y. Performance analysis of intrusion detection systems using a feature selection method on the UNSW-NB15 dataset. Journal of Big Data 2020, 7, 105.
[10] Moustafa, N.; Turnbull, B.; Choo, K.-K.R. An Ensemble Intrusion Detection Technique Based on Proposed Statistical Flow Features for Protecting Network Traffic of Internet of Things. IEEE Internet of Things Journal 2019, 6, 4815–4830.
[11] Sharafaldin, I.; Lashkari, A.H.; Ghorbani, A.A. Toward Generating a New Intrusion Detection Dataset and Intrusion Traffic Characterization. In Proceedings of the 4th International Conference on Information Systems Security and Privacy (ICISSP), Funchal, Portugal, 2018; pp. 108–116.
[12] Breiman, L. Random Forests. Machine Learning 2001, 45, 5–32.
[13] Chen, T.; Guestrin, C. XGBoost: A Scalable Tree Boosting System. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, San Francisco, CA, USA, 2016; pp. 785–794.
[14] Pedregosa, F.; Varoquaux, G.; Gramfort, A.; Michel, V.; Thirion, B.; Grisel, O.; et al. Scikit-learn: Machine Learning in Python. Journal of Machine Learning Research 2011, 12, 2825–2830.
[15] Paszke, A.; Gross, S.; Massa, F.; Lerer, A.; Bradbury, J.; Chanan, G.; et al. PyTorch: An Imperative Style, High-Performance Deep Learning Library. In Advances in Neural Information Processing Systems 32 (NeurIPS), Vancouver, Canada, 2019; pp. 8024–8035.
[16] Kingma, D.P.; Ba, J. Adam: A Method for Stochastic Optimization. In Proceedings of the 3rd International Conference on Learning Representations (ICLR), San Diego, CA, USA, 2015.
[17] Jolliffe, I.T. Principal Component Analysis, 2nd ed.; Springer: New York, NY, USA, 2002.
[18] Shor, P.W. Polynomial-Time Algorithms for Prime Factorization and Discrete Logarithms on a Quantum Computer. SIAM Journal on Computing 1997, 26, 1484–1509.
[19] Grover, L.K. A fast quantum mechanical algorithm for database search. In Proceedings of the 28th Annual ACM Symposium on Theory of Computing (STOC), Philadelphia, PA, USA, 1996; pp. 212–219.
[20] Preskill, J. Quantum Computing in the NISQ era and beyond. Quantum 2018, 2, 79.
[21] Mosca, M. Cybersecurity in an Era with Quantum Computers: Will We Be Ready? IEEE Security & Privacy 2018, 16, 38–41.
[22] Chen, L.; Jordan, S.; Liu, Y.-K.; Moody, D.; Peralta, R.; Perlner, R.; Smith-Tone, D. Report on Post-Quantum Cryptography; NISTIR 8105; National Institute of Standards and Technology: Gaithersburg, MD, USA, 2016.
[23] Alagic, G.; Alperin-Sheriff, J.; Apon, D.; Cooper, D.; Dang, Q.; Kelsey, J.; et al. Status Report on the Second Round of the NIST Post-Quantum Cryptography Standardization Process; NISTIR 8309; National Institute of Standards and Technology: Gaithersburg, MD, USA, 2020.
[24] Biamonte, J.; Wittek, P.; Pancotti, N.; Rebentrost, P.; Wiebe, N.; Lloyd, S. Quantum machine learning. Nature 2017, 549, 195–202.
[25] Schuld, M.; Sinayskiy, I.; Petruccione, F. An introduction to quantum machine learning. Contemporary Physics 2015, 56, 172–185.
[26] Havlíček, V.; Córcoles, A.D.; Temme, K.; Harrow, A.W.; Kandala, A.; Chow, J.M.; Gambetta, J.M. Supervised learning with quantum-enhanced feature spaces. Nature 2019, 567, 209–212.
[27] Schuld, M.; Killoran, N. Quantum Machine Learning in Feature Hilbert Spaces. Physical Review Letters 2019, 122, 040504.
[28] Schuld, M.; Bocharov, A.; Svore, K.M.; Wiebe, N. Circuit-centric quantum classifiers. Physical Review A 2020, 101, 032308.
[29] Mitarai, K.; Negoro, M.; Kitagawa, M.; Fujii, K. Quantum circuit learning. Physical Review A 2018, 98, 032309.
[30] Farhi, E.; Neven, H. Classification with Quantum Neural Networks on Near Term Processors. arXiv 2018, arXiv:1802.06002.
[31] Benedetti, M.; Lloyd, E.; Sack, S.; Fiorentini, M. Parameterized quantum circuits as machine learning models. Quantum Science and Technology 2019, 4, 043001.
[32] Cerezo, M.; Arrasmith, A.; Babbush, R.; Benjamin, S.C.; Endo, S.; Fujii, K.; et al. Variational quantum algorithms. Nature Reviews Physics 2021, 3, 625–644.
[33] McClean, J.R.; Boixo, S.; Smelyanskiy, V.N.; Babbush, R.; Neven, H. Barren plateaus in quantum neural network training landscapes. Nature Communications 2018, 9, 4812.
[34] Cerezo, M.; Sone, A.; Volkoff, T.; Cincio, L.; Coles, P.J. Cost function dependent barren plateaus in shallow parametrized quantum circuits. Nature Communications 2021, 12, 1791.
[35] Wang, S.; Fontana, E.; Cerezo, M.; Sharma, K.; Sone, A.; Cincio, L.; Coles, P.J. Noise-induced barren plateaus in variational quantum algorithms. Nature Communications 2021, 12, 6961.
[36] Grant, E.; Wossnig, L.; Ostaszewski, M.; Benedetti, M. An initialization strategy for addressing barren plateaus in parametrized quantum circuits. Quantum 2019, 3, 214.
[37] Abbas, A.; Sutter, D.; Zoufal, C.; Lucchi, A.; Figalli, A.; Woerner, S. The power of quantum neural networks. Nature Computational Science 2021, 1, 403–409.
[38] Huang, H.-Y.; Broughton, M.; Mohseni, M.; Babbush, R.; Boixo, S.; Neven, H.; McClean, J.R. Power of data in quantum machine learning. Nature Communications 2021, 12, 2631.
[39] Schuld, M.; Sweke, R.; Meyer, J.J. Effect of data encoding on the expressive power of variational quantum-machine-learning models. Physical Review A 2021, 103, 032430.
[40] Bergholm, V.; Izaac, J.; Schuld, M.; Gogolin, C.; Ahmed, S.; Ajith, V.; et al. PennyLane: Automatic differentiation of hybrid quantum-classical computations. arXiv 2018, arXiv:1811.04968.
[41] Lloyd, S.; Mohseni, M.; Rebentrost, P. Quantum principal component analysis. Nature Physics 2014, 10, 631–633.
[42] Rebentrost, P.; Mohseni, M.; Lloyd, S. Quantum Support Vector Machine for Big Data Classification. Physical Review Letters 2014, 113, 130503.
[43] Gouveia, A.; Correia, M. Towards Quantum-Enhanced Machine Learning for Network Intrusion Detection. In Proceedings of the 2020 IEEE 19th International Symposium on Network Computing and Applications (NCA), Cambridge, MA, USA, 2020; pp. 1–8.
[44] Payares, E.D.; Martinez-Santos, J.C. Quantum machine learning for intrusion detection of distributed denial of service attacks: a comparative overview. Proceedings of SPIE 11699, Quantum Computing, Communication, and Simulation, 116990B, 2021.
[45] Chen, J.; Qi, X.; Chen, L.; Chen, F.; Cheng, G. Quantum-inspired ant lion optimized hybrid k-means for cluster analysis and intrusion detection. Knowledge-Based Systems 2020, 203, 106167.
[46] Chen, S.Y.-C.; Yang, C.-H.H.; Qi, J.; Chen, P.-Y.; Ma, X.; Goan, H.-S. Variational Quantum Circuits for Deep Reinforcement Learning. IEEE Access 2020, 8, 141007–141024.
[47] Mitchell, R.; Chen, I.-R. A survey of intrusion detection techniques for cyber-physical systems. ACM Computing Surveys 2014, 46, 1–29.
[48] Ahmad, Z.; Shahid Khan, A.; Wai Shiang, C.; Abdullah, J.; Ahmad, F. Network intrusion detection system: A systematic study of machine learning and deep learning approaches. Transactions on Emerging Telecommunications Technologies 2021, 32, e4150.
[49] Jing, D.; Chen, H.-B. SVM Based Network Intrusion Detection for the UNSW-NB15 Dataset. In Proceedings of the 2019 IEEE 13th International Conference on ASIC (ASICON), Chongqing, China, 2019; pp. 1–4.
[50] Kumar, V.; Das, A.K.; Sinha, D. Statistical analysis of the UNSW-NB15 dataset for intrusion detection. In Computational Intelligence in Pattern Recognition; Springer: Singapore, 2020; pp. 279–294.