Parameter optimization of Measurement-Device-Independent Quantum Key Distribution based on regression decision tree

Abstract

The parameter configuration of Quantum Key Distribution (QKD) has a great impact on the communication effect, and in the practical application of the QKD network in the future, it is necessary to quickly realize the parameter configuration optimization of the asymmetric channel Measurement-Device-Independent QKD according to the communication state, so as to ensure the good communication effect of the mobile users, which is an inevitable requirement for real-time quantum communication. Aiming at the problem that the traditional QKD parameter optimization configuration scheme cannot guarantee real-time, this paper proposes to apply the supervised machine learning algorithm to the QKD parameter optimization configuration, and predict the optimal parameters of TF-QKD and MDI-QKD under different conditions through the machine learning model. First, we delineated the range of system parameters and evenly spaced (linear or logarithmic) values through experimental experience. Then, use the traditional Local Search Algorithm(LSA) to obtain the optimal parameters and take them as the optimal parameters in this paper. Finally, we train various machine learning models based on the above data and compare their performance. We compare the supervised regression learning models such as Neural Network, KNeighbors, Random Forest, Gradient Tree Boosting and Classification And Regression Tree (CART), and the results show that the CART decision tree model has the best performance on the regression evaluation index, and the average value of the key rate (of the prediction parameters) and the optimal key rate ratio is about 0.995, which can meet the communication needs in the actual environment. At the same time, the CART decision tree model shows good environmental robustness in the residual analysis of asymmetric QKD protocol. In addition, compared with the traditional scheme, the new scheme based on CART decision tree has greatly improved the real-time performance of computing, shortening the single prediction time of the optimal parameters of different environments to the order of microseconds, which well meets the real-time communication needs of the communicator in the mobile state. This paper mainly focuses on the parameter optimization of Discrete Variable QKD (DV QKD). In recent years, the development of Continuous Variable QKD (CV QKD) is also rapid. At the end of the paper, we briefly introduce the academic attempts to apply machine learning to the parameter optimization of CV QKD system. And discusses the applicability of the scheme in this paper to the CV QKD system.