Physical Review Research
- Welcome
- Recent
- Subjects
- Accepted
- Collections
- Authors
- Referees
- Search
- About
- Scope
- Editorial Team
- Open Access
Neural-shadow quantum state tomography
Victor Wei, W. A. Coish, Pooya Ronagh, and Christine A. Muschik
Phys. Rev. Research 6, 023250 – Published 6 June 2024
- Article
- References
- No Citing Articles
PDFHTMLExport Citation
Abstract
Quantum state tomography (QST) is the art of reconstructing an unknown quantum state through measurements. It is a key primitive for developing quantum technologies. Neural network quantum state tomography (NNQST), which aims to reconstruct the quantum state via a neural network ansatz, is often implemented via a basis-dependent cross-entropy loss function. State-of-the-art implementations of NNQST are often restricted to characterizing a particular subclass of states, to avoid an exponential growth in the number of required measurement settings. To provide a more broadly applicable method for efficient state reconstruction, we present “neural-shadow quantum state tomography” (NSQST)—an alternative neural network-based QST protocol that uses infidelity as the loss function. The infidelity is estimated using the classical shadows of the target state. Infidelity is a natural choice for training loss, benefiting from the proven measurement sample efficiency of the classical shadow formalism. Furthermore, NSQST is robust against various types of noise without any error mitigation. We numerically demonstrate the advantage of NSQST over NNQST at learning the relative phases of three target quantum states of practical interest, as well as the advantage over direct shadow estimation. NSQST greatly extends the practical reach of NNQST and provides a novel route to effective quantum state tomography.
7 More
- Received 8 May 2023
- Accepted 27 March 2024
DOI:https://doi.org/10.1103/PhysRevResearch.6.023250
![Neural-shadow quantum state tomography (8) Neural-shadow quantum state tomography (8)](https://i0.wp.com/cdn.journals.aps.org/files/icons/creativecommons.png)
Published by the American Physical Society under the terms of the Creative Commons Attribution 4.0 International license. Further distribution of this work must maintain attribution to the author(s) and the published article's title, journal citation, and DOI.
Published by the American Physical Society
Physics Subject Headings (PhySH)
- Research Areas
Quantum circuitsQuantum information processingQuantum measurementsQuantum simulationQuantum tomography
Quantum Information, Science & Technology
Authors & Affiliations
Victor Wei1,2,*, W. A. Coish1,†, Pooya Ronagh2,3,4,5,‡, and Christine A. Muschik2,3,4,§
- 1Department of Physics, McGill University, Montreal, Quebec H3A 2T8, Canada
- 2Institute for Quantum Computing, University of Waterloo, Waterloo, Ontario N2L 3G1, Canada
- 3Department of Physics & Astronomy, University of Waterloo, Waterloo, Ontario N2L 3G1, Canada
- 4Perimeter Institute for Theoretical Physics, Waterloo, Ontario N2L 2Y5, Canada
- 51QB Information Technologies (1QBit), Vancouver, British Columbia V6E 4B1, Canada
- *victor.wei203@gmail.com
- †william.coish@mcgill.ca
- ‡pooya.ronagh@uwaterloo.ca
- §christine.muschik@uwaterloo.ca
Article Text
Click to Expand
References
Click to Expand
Issue
Vol. 6, Iss. 2 — June - August 2024
Subject Areas
- Quantum Physics
- Quantum Information
![Neural-shadow quantum state tomography (12) Neural-shadow quantum state tomography (12)](https://i0.wp.com/cdn.journals.aps.org/development/journals/images/author-services-placard.png)
Images
Figure 1
Overview of the NNQST and NSQST protocols. Panel (a)shows the NNQST protocol with the cross-entropy loss function . The training data determine , the measured probability distribution of measurement outcomes for measurements of the target state performed in the local Pauli basis . The feedback loop on the right-hand side indicates the iterative first-order optimization for neural network training. Panel (b)displays the NSQST protocol described in Sec.2d, where the training data set consists of classical shadows only and where the network parameters are trained via an infidelity loss function . The expression is the stored classical shadow of the target state with the Clifford unitary and bit-string .
Figure 2
Overview of the NSQST with pretraining protocol. The neural networks learning and are separately parameterized, and is pretrained using NNQST with training data derived from measurements only in the computational basis.
Figure 3
Tomography of the quantum state following a one-dimensional QCD time evolution. Two Trotter steps are used for a total evolution time of . Panel (a)displays the average final-state infidelity for each of the three protocols. In each trial, we extract the (exactly calculated) average state infidelity , averaged over the last 100 iterations (and further averaged over ten trials). The error bar is the standard error in the mean calculated over the ten trials. The embedded schematic shows the qubit encoding for a unit cell, containing up to three quarks (filled circles) and up to three antiquarks (striped circles). In panel (b), the plot compares the expectation value of the kinetic energy, evaluated for the neural network quantum state found in the last iteration of each trial, and averaged over ten trials for each protocol. In panel c, the optimization progress curves are displayed for a typical trial, where the adjusted iteration refers to epochs for NNQST, but rather indicates increments of ten iterations for the two NSQST protocols (a total of 2000 iterations were run in these cases). Panel (c)shows the NNQST (blue) loss in the top plot, the estimated NSQST (infidelity) loss function (with and without pretraining, green and red, respectively) in the middle plot (fluctuations are dominated by the finite number of classical shadows taken for each estimate), and the exact infidelity is shown in every (adjusted) iteration for all three protocols in the lower plot.
Figure 4
Typical neural network quantum states following optimization, approximating the state after a one-dimensional QCD time evolution. Panel (a)displays a typical state found in the last iteration of NNQST training: the left plot shows the square root of the probability of the final neural network quantum state compared to the exact target state and the right plot shows the phase output of the state over the set of computational basis states . To highlight the dominant contributions, the phase output has been truncated for states with . For the purpose of better visualization, the overall (global) phase of the neural network quantum state is chosen by aligning the phase of the most probable computational-basis state to that of the target state. The dashed line corresponds to a phase of , since we choose our phase predictions to be in the range . Panels (b)and (c)show typical final states from NSQST and NSQST with pretraining. We observe that both NSQST protocols succeed at learning the phase structure while NNQST fails at the same task.
Figure 5
Training and results for a simulation of tomography on the time-evolved state for a one-dimensional AFH model. Four Trotter steps are used for a total evolution time of . Panel (a)compares the final state infidelity, averaged over ten trials for the three protocols, following the same procedure used for one-dimensional QCD time evolution. Panel (b)compares the predicted mean staggered magnetization in the direction (where ), following a Trotterized time evolution under the AFH model, for all the three protocols. In panel (c), we show optimization progress curves for a typical run, with the NNQST (blue) loss in the top plot, NSQST loss functions (with and without pretraining, green and red, respectively) in the middle plot, and the exact infidelity in every adjusted iteration for all three protocols in the lower plot.
Figure 6
Typical final neural network quantum states, trained on the time-evolved state of a one-dimensional AFH model. Panel (a)displays a typical final state in the last iteration of NNQST optimization, generated using the same procedure from Fig.4. Panels (b)and (c)show the typical final states from NSQST and from NSQST with pretraining, respectively.
Figure 7
Simulation of tomography on a six-qubit phase-shifted GHZ state. Panel (a)compares the final state infidelity, averaged over ten trials for each of the three protocols. Panel (b)shows typical optimization progress curves for NNQST (blue), NSQST (red), and NSQST with pretraining (green).
Figure 8
Comparison of NSQST with pretraining to direct shadow estimation. In each trial of NSQST with pretraining, 200 Clifford shadows are used as training data without resampling in every iteration and 1000 measurements in the computational basis are used in pretraining. For direct shadow estimation, 1200 Clifford shadows and 1200 Pauli shadows are used. Panel (a)compares the absolute error in the predicted kinetic energy, averaged over ten trials for each of the four protocols. Panel (b)compares the absolute error in the predicted fidelity to the ideal time-evolved state, averaged over ten trials for each of the four protocols. Panel (c)compares the absolute error in the predicted expectation value of a Pauli string observable of various weight in the Pauli-X basis, averaged over ten trials for each of the four protocols. The data points are slightly shifted relative to the ticks of the axis for a better display of error bars.
Figure 9
Infidelity and predicted expectation value of the phase-shifted GHZ state as the system size grows. For each system size, we generate 3000 computational-basis measurements and 200 Clifford shadows. In panel (a), with independently sampled measurement data and 5000 Monte Carlo samples, we run NSQST with pretraining using the improved strategy for five trials and report the individual final infidelities. Note that the number of Monte Carlo samples used during training is much less than the number of basis states, which is not an issue if the target state is sufficiently sparse, as in the case of the multiqubit GHZ state. In panel (b), we plot the predicted expectation value of the Pauli string , which is one of the target state's stabilizers. In comparison, the expectation values predicted from five trials of direct shadow estimation are plotted, each with 3200 independently sampled Clifford or Pauli shadows. The data points are slightly shifted relative to the ticks of the axis for a better display.
Figure 10
Simulation of tomography for a phase-shifted GHZ state in the presence of noise. Panel (a)displays the average loss function (red) defined in Eq.(23) and the exact infidelity (blue) for the amplitude damping channel (the loss function is averaged over the last 100 iterations for each trial and then the average is taken over ten trials). The strength of the noise increases with increasing . The noiseless infidelity loss function is then transformed into an estimated infidelity for the noisy case using Eq.(22) for the amplitude damping channel to obtain the transformed cost function. The error bars represent the standard error in the mean over ten trials. In panel (b), we show the average loss function and exact infidelity for the local depolarizing noise model with a two-qubit depolarizing channel applied after every CNOT gate in the appended random Clifford circuit . The channel parameter characterizes the growing strength of the noise. We do not plot the transformed loss function in this case because the CNOT-dependent local depolarizing noise model does not have an analytic noisy shadow expression.
Figure 11
Simulation circuit for the one-dimensional QCD model. The initial state preparation circuit (before the barrier) and a single Trotter step (after the barrier) are drawn using Qiskit [69]. In our numerical experiments an evolution for time is decomposed into two Trotter steps.
Figure 12
Simulation circuit for the one-dimensional AFH model. The initial state preparation circuit (before the barrier) and a single Trotter step (after the barrier) are drawn using Qiskit [69]. In our numerical experiments an evolution for time is decomposed into four Trotter steps.
Figure 13
Additional plots for the quantum state following a one-dimensional QCD time evolution. Panel (a)displays the expectation values of the total energy, evaluated for the neural network quantum state found in the last iteration of each trial, and averaged over ten trials for each protocol. Panel (b)displays the expectation values of the mass Hamiltonian.
Figure 14
Typical optimization progress curve from NSQST with pretraining and fixed Clifford shadows. Unlike the other plots, the iteration number on the axis is not adjusted and corresponds to every gradient update during optimization.