Skip to main content

Attention-based quantum tomography

Cornell Affiliated Author(s)

Author

P. Cha
P. Ginsparg
F. Wu
J. Carrasquilla
P.L. McMahon
Eun-Ah Kim

Abstract

With rapid progress across platforms for quantum systems, the problem of many-body quantum state reconstruction for noisy quantum states becomes an important challenge. There has been a growing interest in approaching the problem of quantum state reconstruction using generative neural network models. Here we propose the ‘attention-based quantum tomography’ (AQT), a quantum state reconstruction using an attention mechanism-based generative network that learns the mixed state density matrix of a noisy quantum state. AQT is based on the model proposed in ‘Attention is all you need’ by Vaswani et al (2017 NIPS) that is designed to learn long-range correlations in natural language sentences and thereby outperform previous natural language processing (NLP) models. We demonstrate not only that AQT outperforms earlier neural-network-based quantum state reconstruction on identical tasks but that AQT can accurately reconstruct the density matrix associated with a noisy quantum state experimentally realized in an IBMQ quantum computer. We speculate the success of the AQT stems from its ability to model quantum entanglement across the entire quantum system much as the attention model for NLP captures the correlations among words in a sentence. © 2021 The Author(s). Published by IOP Publishing Ltd

Date Published

Journal

Machine Learning: Science and Technology

Volume

3

Issue

1

URL

https://www.scopus.com/inward/record.uri?eid=2-s2.0-85120753676&doi=10.1088%2f2632-2153%2fac362b&partnerID=40&md5=58a81249e7f0a3c3b5fc8bb873984c91

DOI

10.1088/2632-2153/ac362b

Group (Lab)

Download citation