Fast and noise-robust quantum state tomography based on ELM

Abstract

This paper proposed a quantum state tomography approach based on the extreme learning machine (ELM), which is available in the reconstruction of quantum states via a lightweight neural network. The key step of the proposed tomography approach is to employ the ELM to approximate the complex mapping between the measurement values sequence and the real density matrix. After obtaining the output of the ELM-based estimator, a matrix transformation technique is used to make the network outputs satisfy quantum state constraints. Compared with deep learning-based tomography approaches, our proposed ELM-based approach enables both high-fidelity and high-efficiency quantum state tomography with only one training process under the condition of very few numbers of training samples, network layers and hidden layer nodes. In addition, the proposed tomography approach is robust to noisy measurement values, since the ELM-based estimator is quite lightweight. Simulations on the tomography of eigenstates, superposition states and mixed states are presented to verify our theoretical findings. Also, the superiority of the ELM-based tomography approach is demonstrated in comparison with that based on the radial basis function network, convolutional neural network and maximum likelihood estimation approach.