Training a quantum annealing based restricted Boltzmann machine on cybersecurity data

Abstract

We present a real-world application that uses a quantum computer. Specifically, we train a RBM using QA for cybersecurity applications. The D-Wave 2000Q has been used to implement QA. RBMs are trained on the ISCX data, which is a benchmark dataset for cybersecurity. For comparison, RBMs are also trained using CD. CD is a commonly used method for RBM training. Our analysis of the ISCX data shows that the dataset is imbalanced. We present two different schemes to balance the training dataset before feeding it to a classifier. The first scheme is based on the undersampling of benign instances. The imbalanced training dataset is divided into five sub-datasets that are trained separately. A majority voting is then performed to get the result. Our results show the majority vote increases the classification accuracy up from 90.24% to 95.68%, in the case of CD. For the case of QA, the classification accuracy increases from 74.14% to 80.04%. In the second scheme, a RBM is used to generate synthetic data to balance the training dataset. We show that both QA and CD-trained RBM can be used to generate useful synthetic data. Balanced training data is used to evaluate several classifiers. Among the classifiers investigated, K-Nearest Neighbor (KNN) and Neural Network (NN) perform better than other classifiers. They both show an accuracy of 93%. Our results show a proof-of-concept that a QA-based RBM can be trained on a 64-bit binary dataset. The illustrative example suggests the possibility to migrate many practical classification problems to QA-based techniques. Further, we show that synthetic data generated from a RBM can be used to balance the original dataset.

arXiv commentin IEEE Transactions on Emerging Topics in Computational Intelligence

References in corpus (8)

• Quantum Annealing in the Transverse Ising Model
• Quantum Annealing: A New Method for Minimizing Multidimensional Functions
• Simple proof of equivalence between adiabatic quantum computation and the circuit model
• Estimation of effective temperatures in quantum annealers for sampling applications: A case study with possible applications in deep learning
• Support vector machines on the D-Wave quantum annealer
• Quantum Annealing for Combinatorial Clustering
• Adiabatic Quantum Linear Regression
• Training a quantum annealing based restricted Boltzmann machine on cybersecurity data

Cited by in corpus (5)

• Training a quantum annealing based restricted Boltzmann machine on cybersecurity data
• A hybrid quantum-classical approach for inference on restricted Boltzmann machines
• Hamiltonian learning from time dynamics using variational algorithms
• A Universal Quantum Circuit Design for Periodical Functions
• Diabatic quantum annealing for training energy-based generative models