Machine Learning assisted noise classification with Quantum Key Distribution protocols

We propose a hybrid protocol to classify quantum noises using supervised classical machine learning models and simple quantum key distribution protocols. We consider the quantum bit error rates (QBERs) generated in QKD schemes under consideration of different noises, and identify the noise channels with high accuracy for both training and test data. Our protocol classifies quantum noises with high accuracy under the assumption of two different scenarios; in one case we assume two remotely located parties share keys through noisy quantum channels, whereas, in the second case, we simulate the QKD protocols on a gate-based quantum computer, where the gates are afflicted with noise. Alongside efficient classification, our work also throws light on the difference in distribution characteristics of QBERs generated in these two scenarios. Finally, our method is based on classical post processing of data generated from very simplistic quantum protocols, making it readily implementable in the current era of noisy quantum computing with low number of qubits.