Flexible learning of quantum states with generative query neural
  networks
  Deep neural networks are a powerful tool for characterizing quantum states.
In this task, neural networks are typically trained with measurement data
gathered from the quantum state to be characterized. But is it possible to
train a neural network in a general-purpose way, which makes it applicable to
multiple unknown quantum states? Here we show that learning across multiple
quantum states and different measurement settings can be achieved by a
generative query neural network, a type of neural network originally used in
the classical domain for learning 3D scenes from 2D pictures. Our network can
be trained offline with classically simulated data, and later be used to
characterize unknown quantum states from real experimental data. With little
guidance of quantum physics, the network builds its own data-driven
representation of quantum states, and then uses it to predict the outcome
probabilities of requested quantum measurements on the states of interest. This
approach can be applied to state learning scenarios where quantum measurement
settings are not informationally complete and predictions must be given in real
time, as experimental data become available, as well as to adversarial
scenarios where measurement choices and prediction requests are designed to
expose learning inaccuracies. The internal representation produced by the
network can be used for other tasks beyond state characterization, including
clustering of states and prediction of physical properties. The features of our
method are illustrated on many-qubit ground states of Ising model and
continuous-variable non-Gaussian states.
