# Practical phase-modulation stabilization in quantum key distribution via machine learning

## TL;DR

This paper introduces a machine learning approach using LSTM networks to predict and control physical parameters in quantum key distribution systems, significantly improving efficiency and maintaining low error rates.

## Contribution

It is the first to apply LSTM-based prediction for real-time control in QKD, reducing calibration time and enhancing key transmission efficiency.

## Key findings

- LSTM model maintains quantum-bit error rate comparable to traditional methods.
- Reduces calibration time, increasing key transmission efficiency.
- Applicable to various QKD schemes and protocols.

## Abstract

In practical implementation of quantum key distributions (QKD), it requires efficient, real-time feedback control to maintain system stability when facing disturbance from either external environment or imperfect internal components. Usually, a "scanning-and-transmitting" program is adopted to compensate physical parameter variations of devices, which can provide accurate compensation but may cost plenty of time in stopping and calibrating processes, resulting in reduced efficiency in key transmission. Here we for the first propose to employ a well known machine learning model, i.e., the Long Short-Term Memory Network (LSTM), to predict those physical parameter variations in advance and actively perform real-time control on corresponding QKD devices. Experimentally, we take the phase-coding scheme as an example and run the LSTM model based QKD system for more than 10 days. Experimental results show that we can keep the same level of quantum-bit error rate as the traditional "scanning-and-transmitting" program by employing our new machine learning method, but dramatically reducing the scanning time and resulting in significantly enhanced key transmission efficiency. Furthermore, our present machine learning model should also be applicable to any other QKD systems using any coding scheme or QKD protocols, and thus seems a very promising candidate in large-scale application of quantum communication network in the near future.