Deep neural network-assisted improvement of quantum compressed sensing tomography

Quantum compressed sensing is a fundamental technique for tomographic reconstruction of low-rank density matrices in informationally incomplete scenarios. However, when the available measurement data is insufficient, standard compressed sensing, like other algorithmic estimators such as maximum-likelihood, can yield imprecise or noisy estimates. To mitigate this, we propose a deep neural network-based post-processing to enhance the initial reconstruction. By treating the compressed sensing-estimated quantum state as a noisy input, the network performs a supervised denoising task. After the network is applied, a projection onto the space of feasible density matrices is performed to obtain an improved final state estimation. We demonstrate the effectiveness of our method through numerical experiments. Additionally, we explore iterative application of the inference process to further enhance performance, and we analyze the network's generalization ability by testing its performance on states and noise models not seen during training.