After applying the photon-number cutoff assumption, the optimization problem in Eq. (REF ) can be solved by applying the numerical method in Refs. [1]}, [2]}, but this is very time consuming. In this work, to reduce the time to predict secure key rates, we use the key rates obtained by the numerical method in Refs. [1]}, [2]} as labels to train our neural networks.
