arXiv:2206.01273
Reconstructing Quantum States Using Basis-Enhanced Born Machines
Abigail McClain Gomez, Susanne F. Yelin, Najafi, Khadijeh
Abstract
Rapid improvement in quantum hardware has opened the door to complex problems, but the precise characterization of quantum systems itself remains a challenge. To address this obstacle, novel tomography schemes have been developed that employ generative machine learning models, enabling quantum state reconstruction from limited classical data. In particular, quantum-inspired Born machines provide a natural way to encode measured data into a model of a quantum state. Born machines have shown great success in learning from classical data; however, the full potential of a Born machine in learning from quantum measurement has thus far been unrealized. To this end, we devise a complex-valued basis-enhanced Born machine and show that it can reconstruct pure quantum states using projective measurements from only two Pauli measurement bases. We implement the basis-enhanced Born machine to learn the ground states across the phase diagram of a 1D chain of Rydberg atoms, reconstructing quantum states deep in ordered phases and even at critical points with quantum fidelities surpassing 99%. The model accurately predicts quantum correlations and different observables, and system sizes as large as 37 qubits are considered. Quantum states across the phase diagram of a 1D XY spin chain are also successfully reconstructed using this scheme. Our method only requires simple Pauli measurements with a sample complexity that scales quadratically with system size, making it amenable to experimental implementation.