Abstract
Classical shadows provide a noise-resilient and sample-efficient method for learning quantum system properties, relying on a user-specified unitary ensemble. What is the weakest assumption on this ensemble that can still yield meaningful results? To address this, we focus on Pauli-invariant unitary ensembles—those invariant under multiplication by Pauli operators. For these ensembles, we present explicit formulas for the reconstruction map and sample complexity bounds and extend our results to the case when noise impacts the protocol implementation. Two applications are explored: one for locally scrambled unitary ensembles, where we present formulas for the reconstruction map and sample complexity bounds that circumvent the need to solve an exponential-sized linear system, and another for the classical shadows of quantum channels. Our results establish a unified framework for classical shadows with Pauli-invariant unitary ensembles, applicable to both noisy and noiseless scenarios for states and channels and primed for implementation on near-term quantum devices.
Similar content being viewed by others
Introduction
Learning the properties of an unknown but physically accessible quantum system is a fundamental task in quantum information processing1,2,3. A standard tool for this task is quantum tomography, a process by which one recovers a classical description of a quantum system through performing measurements on it. Unfortunately, finding the full description of a quantum system by quantum tomography is computationally intensive; it requires an exponential number of copies of the system4,5,6.
Recently, Huang, Kueng, and Preskill introduced a novel method—the classical shadow paradigm1—to circumvent the above limitation. A key insight behind classical shadows rests on the fact that in many cases one does not need to learn a complete description of a quantum system; one can learn its most useful properties from a minimal sketch of the quantum system, the classical shadow. The classical shadow paradigm, with the sample efficiency it touts, has attracted considerable attention over the last couple of years7. Several applications have been proposed1,8,9,10,11,12,13,14,15,16,17,18,19,20,21, ranging from the estimation of properties of quantum states and gates to quantum chaos in quantum evolution. Moreover, several improved versions of this protocol have been developed1,8,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36, such as noise-resilient versions20,37.
The performance of the classical shadow protocol depends on several factors. In particular, it depends on an ensemble of unitary operators from which an operation is chosen randomly to be applied to the unknown quantum state. A user must choose this ensemble in advance, according to some desiderata, such as the need for the shadow channel to be invertible and for one to have an efficient algorithm for sampling a unitary from the ensemble. Several unitary ensembles have been considered by previous authors, including the local and global Clifford ensembles1, fermionic Gaussian unitaries28, chaotic Hamiltonian evolutions27, locally scrambled unitary ensembles26 and other short-depth quantum circuits. One motivation for studying short-depth quantum circuits is that random unitaries based on these random circuits could be more powerful in predicting properties of quantum systems38 than the original proposals.
One can ask: what is the weakest assumption on the unitary ensemble that would still yield universal, meaningful and interesting results? Our candidate solution to this question is the assumption that the unitary ensemble is invariant under multiplication by a Pauli operator. Ensembles satisfying this assumption—namely the Pauli-invariant unitary ensembles—include a wide range of ensembles, including the Pauli group, the local and global Clifford groups, locally scrambled unitary ensembles, and short-depth 2-qubit Clifford circuits. The goal of this work is to provide a unified framework for the classical shadow protocol with Pauli-invariant unitary ensembles in both the noiseless and noisy settings.
The rest of our paper is structured as follows. In Section ("General framework for Pauli-invariant unitary ensembles”), we introduce the framework of classical shadows with Pauli-invariant unitary ensembles and provide an explicit formula for the shadow channel and reconstruction map, which are key elements in the classical shadow protocol described in Section ("Preliminaries: Classical shadows”). In addition, we establish a connection between the reconstruction map and the entanglement features of the dynamics. We give upper bounds on the sample complexity in terms of the average shadow norm for the task of expectation estimation using the classical shadow protocol. Considering the fact that noise is inevitable in the noisy intermediate-scale quantum (NISQ) era, we consider classical shadows for Pauli-invariant unitary ensembles in the presence of noise in Section ("Classical shadows with noise”). This generalizes our results from the noiseless case to the noisy case.
In Sections ("Application to locally scrambled unitary ensembles”) and ("Application to quantum channels”), we provide two applications of our main results. First, we investigate locally scrambled unitary ensembles, which are a special example of Pauli-invariant unitary ensembles. We provide an explicit reconstruction map, addressing a crucial gap in a previous approach provided by26, which specifies the map only in terms of a solution of an exponential-sized system of linear equations; by contrast, using our explicit formula, ours circumvents the need to solve such an exponential-sized linear system. Second, we apply our results to the shadow process tomography of quantum channels using the Choi-Jamiołkowski isomorphism, thereby generalizing the results of refs. 29,30 to the broad class of Pauli-invariant unitary ensembles.
Results
Preliminaries: classical shadows
Let us begin by describing the classical shadow protocol that was introduced in ref. 1. A user will have to pre-decide on a unitary ensemble \({{{\mathcal{E}}}}={\{(U,P(U))\}}_{U\in {{{\mathcal{U}}}}}\), where \({{{\mathcal{U}}}}\) is a set of unitary operators and P is a probability distribution on \({{{\mathcal{U}}}}\). The first step is choosing a random unitary U from the unitary ensemble \({{{\mathcal{E}}}}\) according to the specified distribution P(U). The unitary U is then applied to ρ and a computational basis measurement is performed on the resultant state to obtain an n-bit string \(\overrightarrow{b}={b}_{1}{b}_{2}\ldots {b}_{n}\in {\{0,1\}}^{n}\). We construct the state
which is stored in classical memory. Note that this produces an ensemble of states
where \(P(U,\overrightarrow{b})=P(U)P(\overrightarrow{b}| U)\) and \(P(\overrightarrow{b}| U)={{{\rm{Tr}}}}[{\hat{\sigma }}_{U,\overrightarrow{b}}\rho ]\). The expected value of this ensemble is given by
where \({{{\mathcal{M}}}}\)—called the shadow channel—is a completely positive and trace-preserving map. To construct the classical shadow, the shadow channel needs to be invertible. If the conditions for invertibility are met, the inverse of the shadow channel \({{{{\mathcal{M}}}}}^{-1}\), called the reconstruction map, is applied to the classically stored \({\hat{\sigma }}_{U,\overrightarrow{b}}\) to obtain the classical snapshot \(\hat{\rho }={{{{\mathcal{M}}}}}^{-1}[{\hat{\sigma }}_{U,\overrightarrow{b}}]\), which is called the classical shadow. As required, \(\hat{\rho }\) is an unbiased estimator of ρ:
As illustrated above, the reconstruction map \({{{{\mathcal{M}}}}}^{-1}\) plays a pivotal role in the classical shadow protocol. However, for arbitrary unitary ensembles, no general closed-form analytic formula is known for \({{{{\mathcal{M}}}}}^{-1}\). Instead, prior to this study, it was only for a handful of unitary ensembles (e.g., the local and global Clifford ensembles) that analytic expressions have been derived. In this study, we address this gap by presenting an explicit formula for the reconstruction map for the large class of Pauli-invariant unitary ensembles, thereby enlarging the class of ensembles for which such expressions are known.
General framework for Pauli-invariant unitary ensembles
In this paper, we consider classical shadows with Pauli-invariant unitary ensembles. We start by introducing these ensembles, before deriving our main results characterizing the performance of classical shadow protocols utilizing these ensembles.
We denote the set of Pauli operators on n qubits by \({{{{\mathcal{P}}}}}_{n}=\{{P}_{\overrightarrow{a}}={\otimes }_{i}{P}_{{a}_{i}}:\overrightarrow{a}\in {V}^{n}\}\), where \(V:= {{\mathbb{Z}}}_{2}\times {{\mathbb{Z}}}_{2}=\{(0,0),(0,1),(1,0),(1,1)\}\), and P(x, z) = ixzXxZz, i.e., \({P}_{(0,0)}={\mathbb{I}},{P}_{(0,1)}=Z,{P}_{(1,0)}=X,{P}_{(1,1)}=Y\). A given unitary ensemble \({{{\mathcal{E}}}}={\{(U,P(U))\}}_{U}\), is called Pauli-invariant, if the probability distribution P(U) satisfies the following (right) Pauli-invariant condition:
This is a weak assumption, and many unitary ensembles satisfy this condition, for example, (1) the Pauli group \(\{\pm 1,\pm i\}\times {{{{\mathcal{P}}}}}_{n}\); (2) D-dimensional local, random quantum circuits39; (3) local Clifford unitaries; (4) global Clifford unitaries; and (5) short-depth 2-qubit Clifford circuits.
For Pauli-invariant unitary ensembles, we will now provide an explicit formula for the shadow channel \({{{\mathcal{M}}}}\) and the corresponding reconstruction map \({{{{\mathcal{M}}}}}^{-1}\), if it exists. The key idea is to use the Pauli coefficients in the decomposition in terms of Pauli operators, noting that the Pauli operators form an orthonormal basis with respect to the scaled Hilbert-Schmidt inner product \(\langle A,B\rangle =\frac{1}{{2}^{n}}{{{\rm{Tr}}}}[{A}^{{\dagger} }B]\) for an n-qubit system. These Pauli coefficients can be thought of as quantum Fourier coefficients, which have found extensive uses in a myriad of applications, including quantum Boolean functions40, quantum circuit complexity41, quantum scrambling42 and quantum convolutions43,44,45. For clarity of presentation, the detailed proofs of all the results in this section are provided in Supplementary Note 2. This follows Supplementary Note 1, which introduces the symplectic inner product used in these proofs.
Theorem 1
(Shadow channel and reconstruction map) For a Pauli-invariant unitary ensemble \({{{\mathcal{E}}}}\), the shadow channel can be written as
where \({W}_{{{{\mathcal{E}}}}}[\overrightarrow{a}]\) is the average squared Pauli coefficient of the classical shadow, defined as
where \({{\mathbb{E}}}_{\overrightarrow{b}}:= \frac{1}{{2}^{n}}{\sum }_{\overrightarrow{b}}\) denotes the average with respect to the uniform distribution over n-bit strings. For a state σ, we have used the notation \({W}_{\sigma }[\overrightarrow{a}]:= | {{{\rm{Tr}}}}[\sigma {P}_{\overrightarrow{a}}]{| }^{2}\) to denote the square of the \(\overrightarrow{a}\)-th Pauli coefficient of σ.
For Pauli-invariant unitary ensembles, the reconstruction map \({{{{\mathcal{M}}}}}^{-1}\) exists if and only if \({W}_{{{{\mathcal{E}}}}}[\overrightarrow{a}] \,> \,0\) for all \(\overrightarrow{a}\). If it exists, the reconstruction map, which is diagonal in the Pauli basis, is completely specified by its action on the Pauli basis elements as follows:
Now, let us discuss the connection between the coefficients of the reconstruction map \({W}_{{{{\mathcal{E}}}}}\) and the 2nd entanglement feature46,47, which has been used to describe the entangling power of unitary ensembles and is defined as follows:
where A is a subset of \([n],{S}_{A}^{(2)}(\sigma )=-\log {{{\rm{Tr}}}}[{\sigma }_{A}^{2}]\) denotes the 2nd Rényi entanglement entropy of the state σ on the subset A, and \({\sigma }_{A}={{{\mbox{Tr}}}}_{{A}^{c}}[\sigma ]\). For any subset S ⊂ [n], let us define \({W}_{{{{\mathcal{E}}}}}[S]\) as the sum of the coefficients whose support is S, i.e.,
where \({{{\rm{supp}}}}(\overrightarrow{a})\) denotes the support of the vector \(\overrightarrow{a}\). The following proposition expresses \({W}_{{{{\mathcal{E}}}}}[S]\) in terms of the 2nd entanglement feature.
Proposition 2
(Connection with entanglement feature for Pauli-invariant unitary ensemble) The coefficients of the reconstruction map can be expressed as follows:
Now, let us consider the sample complexity for the task of expectation estimation using classical shadows with Pauli-invariant unitary ensembles. We shall make use of the result from ref. 1 that the sample complexity of classical shadows with the input state ρ and an observable O is upper bounded by the following (squared) shadow norm:
where the estimator \(\hat{o}\) of the observable O is given by \(\hat{o}={{{\rm{Tr}}}}[O{{{{\mathcal{M}}}}}^{-1}[\hat{\sigma }]]\).
Lemma 3
(Huang et al.1) The sample complexity \({{{\mathscr{S}}}}\) needed to accurately predict a collection of N linear target functions \({\{{{{\rm{Tr}}}}[{O}_{i}\rho ]\}}_{i = 1}^{N}\) with error ϵ and failure probability δ is
In general, the shadow norm is hard to compute, though for special cases, closed-form expressions for the shadow norm can be derived. To this end, let us consider the widely used Pauli operators O as observables. For this case, we provide a unified formula for the shadow norm corresponding to all Pauli-invariant unitary ensembles.
Proposition 4
(Shadow norm for Pauli-invariant unitary ensembles for Pauli observables) If the observable O is the Pauli operator \({P}_{\overrightarrow{a}}\), then the (squared) shadow norm for a Pauli-invariant unitary ensemble is equal to
Hence the sample complexity is bounded above by
Note that the shadow norm \({\left\Vert O\right\Vert }_{{{{\mathcal{E}}}}}\) usually depends on the input state ρ. If we are interested in the expectation of the shadow norm over some unitary ensemble instead of some specific state, we will need a notion of an average shadow norm. To this end, let us consider the (squared) average shadow norm over the Pauli group, defined as \({\left\Vert O\right\Vert }_{{{{\mathcal{E}}}}}^{2}={{\mathbb{E}}}_{V\in {{{{\mathcal{P}}}}}_{n}}{\left\Vert O\right\Vert }_{{{{{\mathcal{E}}}}}_{V\rho {V}^{{\dagger} }}}^{2}.\) The following proposition gives an expression for the average shadow norm.
Proposition 5
(Average shadow norm for Pauli-invariant unitary ensemble) The (squared) average shadow norm over the Pauli group \({\left\Vert O\right\Vert }_{{{{\mathcal{E}}}}}^{2}\) can be expressed as follows:
where \({W}_{O}[\overrightarrow{a}]=| {{{\rm{Tr}}}}[O{P}_{\overrightarrow{a}}]{| }^{2}\). Therefore, given a set of N traceless observables \({\{{O}_{i}\}}_{i = 1}^{N}\), the average shadow norm provides the following lower bound for the sample complexity:
where the (normalized) inner product is defined as \(\langle {\overrightarrow{W}}_{{{{\mathcal{E}}}}}^{-1},{\overrightarrow{W}}_{{O}_{i}}\rangle := \frac{1}{{4}^{n}}{\sum }_{\overrightarrow{a}\in {V}^{n}}{W}_{{{{\mathcal{E}}}}}{[\overrightarrow{a}]}^{-1}{W}_{{O}_{i}}[\overrightarrow{a}]\).
Classical shadows with noise
Considering the fact that noise is unavoidable in real-world experiments, it is necessary to employ error mitigation techniques to make classical shadows useful in the presence of noise. In this subsection, let us consider noisy classical shadows with a Pauli-invariant unitary ensemble. Similar to the setting in20,37,48, we shall assume that a noise channel Λ acts on the pre-measurement state UρU† just before the measurement is performed. Such an assumption is obeyed by gate-independent, time-stationary, and Markovian noise20,49. Similar to the noiseless case (2), the ensemble of states is given by \({{{{\mathcal{E}}}}}_{{{\Lambda }},\rho }={\left\{\left({\hat{\sigma }}_{U,\overrightarrow{b}},P(U,\overrightarrow{b})\right)\right\}}_{U,\overrightarrow{b}}\), where \({P}_{{{\Lambda }}}(\overrightarrow{b}| U)={{{\rm{Tr}}}}[\left\vert \overrightarrow{b}\right\rangle \left\langle \overrightarrow{b}\right\vert {{\Lambda }}[U\rho {U}^{{\dagger} }]]\). Hence, by taking the average of the post-measurement states, we have \(\sigma ={{\mathbb{E}}}_{U}{\sum }_{\overrightarrow{b}}{\hat{\sigma }}_{U,\overrightarrow{b}}{{{\rm{Tr}}}}[\vert \overrightarrow{b}\rangle \langle \overrightarrow{b}\vert {{\Lambda }}[U\rho {U}^{{\dagger} }]]={{{{\mathcal{M}}}}}_{{{\Lambda }}}[\rho ],\) where \({{{{\mathcal{M}}}}}_{{{\Lambda }}}\) is the noisy shadow channel, and \(\rho ={{{{\mathcal{M}}}}}_{{{\Lambda }}}^{-1}[\sigma ]\), where \({{{{\mathcal{M}}}}}_{{{\Lambda }}}^{-1}\) is the noisy reconstruction map. We will now provide an explicit form for the noisy reconstruction map \({{{{\mathcal{M}}}}}_{{{\Lambda }}}^{-1}\). For clarity of presentation, the detailed proofs of all the results in this subsection are provided in Supplementary Note 5.
Theorem 6
(Shadow channel and reconstruction map in noisy case) Given a Pauli-invariant unitary ensemble \({{{\mathcal{E}}}}\) and a noise channel Λ, the noisy shadow channel is given by
where \({W}_{{{{{\mathcal{E}}}}}_{{{\Lambda }}}}[\overrightarrow{a}]\) is the average Pauli coefficient of the noisy classical shadow, and is defined as
For the Pauli-invariant unitary ensemble, \({{{{\mathcal{M}}}}}_{{{\Lambda }}}^{-1}\) exists if and only if \({W}_{{{{{\mathcal{E}}}}}_{{{\Lambda }}}}[\overrightarrow{a}]\,>\,0\) for all \(\overrightarrow{a}\). If it exists, the reconstruction map is defined by
Now, let us consider the sample complexity of the noisy classical shadow protocol. Similar to the noiseless case, the sample complexity of the noisy classical shadow with the input state ρ and observable O is upper bounded by the (squared) noisy shadow norm \({\left\Vert O\right\Vert }_{{{{{\mathcal{E}}}}}_{{{\Lambda }}},\rho }^{2}={{\mathbb{E}}}_{\hat{\sigma }\in {{{{\mathcal{E}}}}}_{{{\Lambda }}}}\hat{o}{[\hat{\sigma }]}^{2},\) where the estimator of the observable is taken to be \(\hat{o}={{{\rm{Tr}}}}[O{{{{\mathcal{M}}}}}_{{{\Lambda }}}^{-1}[\hat{\sigma }]]\). Let us also define the (squared) average shadow norm over the Pauli group for the noisy classical shadow to be \({\left\Vert O\right\Vert }_{{{{{\mathcal{E}}}}}_{{{\Lambda }}}}^{2}={{\mathbb{E}}}_{V\in {{{{\mathcal{P}}}}}_{n}}{\left\Vert O\right\Vert }_{{{{{\mathcal{E}}}}}_{{{\Lambda }},V\rho {V}^{{\dagger} }}}^{2}\).
Proposition 7
If the observable O is taken to be the Pauli operator \({P}_{\overrightarrow{a}}\), the shadow norm is equal to
where \({W}_{{{{{\mathcal{E}}}}}_{{{\Lambda }}}}^{u}[\overrightarrow{a}]\) is defined as
Hence, if Λ is unital, then \({W}_{{{{{\mathcal{E}}}}}_{{{\Lambda }}}}^{u}[\overrightarrow{a}]={W}_{{{{\mathcal{E}}}}}[\overrightarrow{a}]\). Moreover, the average shadow norm in the noisy classical shadow protocol with the noise channel Λ can be expressed as follows
Let us now provide an example of noisy classical shadows and demonstrate how the performance of the protocol depends on the noise rate.
Example 1
Let us consider the case where the noise channel is the global depolarizing channel, defined as \({D}_{p}[\cdot ]=(1-p)[\cdot ]+p\frac{{{{\rm{Tr}}}}[\cdot ]{\mathbb{I}}}{{2}^{n}}.\) The coefficients of the shadow channel can be expressed as
and the shadow norm for a Pauli operator \({P}_{\overrightarrow{a}}\) obeys the following identity
where \({\delta }_{\overrightarrow{a},\overrightarrow{0}}\) denotes the Kronecker delta function. Hence the sample complexity for any non-identity Pauli operator is bounded by
From the above equation, we see that depolarizing noise increases the number of samples needed for expectation estimation, with an increase that is proportional to the noise rate.
Application to locally scrambled unitary ensembles
First, let us apply our results to locally scrambled unitary ensembles, which are a special case of Pauli-invariant unitary ensembles. A unitary ensemble is said to be locally scrambled26 if the probability distribution P(U) satisfies local basis invariance, that is,
It is easy to see that the Pauli-invariance assumption is weaker than the locally scrambled assumption, as locally scrambled unitary ensembles are Pauli-invariant. Not all Pauli-invariant unitary ensembles are locally scrambled though—a counterexample is the Pauli group.
Classical shadows with locally scrambled unitary ensembles were previously considered in ref. 26. However, a crucial limitation of the results presented therein is that the reconstruction map was left specified in terms of the solution of a linear system of size O(2n) without any explicit formula given for the map. By contrast, for our study, since locally scrambled unitary ensembles are a special case of the Pauli-invariant unitary ensembles, we get as a consequence of Theorem 1 an explicit formula for the reconstruction map that circumvents the need to solve an exponential-sized linear system. Our next proposition makes this explicit. For clarity of presentation, the detailed proofs of all the results in this subsection are provided in Supplementary Note 3.
Proposition 8
Given a locally scrambled unitary ensemble \({{{\mathcal{E}}}}\), the shadow channel is
where \({\bar{W}}_{{{{\mathcal{E}}}}}[S]\) is defined as \({\bar{W}}_{{{{\mathcal{E}}}}}[S]={{\mathbb{E}}}_{\overrightarrow{a}:{{{\rm{supp}}}}(\overrightarrow{a}) = S}{W}_{{{{\mathcal{E}}}}}[\overrightarrow{a}],\) and the reconstruction map is given by
Our expression for the reconstruction map above may be compared with that given in ref. 26, which expressed the reconstruction map as \({{{{\mathcal{M}}}}}^{-1}[\sigma ]={\sum }_{S\subset [n]}{r}_{S}{D}^{S}[\sigma ]\), where DS denotes the ∣S∣-fold Kronecker product of the single-qubit erasure channel \(D[\cdot ]={{{\rm{Tr}}}}[\cdot ]{\mathbb{I}}/2\) with itself acting on all the qubits indexed by S, and the coefficients rs are given as the solution of a linear system whose coefficients are the entanglement features. In the next proposition, we find an explicit formula for rS in terms of the coefficients \({\bar{W}}_{{{{\mathcal{E}}}}}[S]\).
Proposition 9
Given a reconstruction map written as \({{{{\mathcal{M}}}}}^{-1}[\sigma ]={\sum }_{S}{r}_{S}{D}^{S}[\sigma ]\), the coefficients rS can be expressed in terms of \({\bar{W}}_{{{{\mathcal{E}}}}}[A]\) as follows
where Ac denotes the complement of A in [n].
Using Proposition 2 and the fact that \({W}_{{{{\mathcal{E}}}}}[S]={3}^{| S| }{\bar{W}}_{{{{\mathcal{E}}}}}[S]\), our next corollary shows how the coefficients rS can be expressed in terms of the entanglement feature.
Corollary 10
Given a reconstruction map written as \({{{{\mathcal{M}}}}}^{-1}[\sigma ]={\sum }_{S}{r}_{S}{D}^{S}[\sigma ]\), the coefficients rS can be expressed as follows
For the locally scrambled unitary ensembles, the average (squared) shadow norm is defined in ref. 26 as \({\left\Vert O\right\Vert }_{{{{\mathcal{E}}}}}^{2}={{\mathbb{E}}}_{V\in U{(2)}^{n}}{\left\Vert O\right\Vert }_{{{{{\mathcal{E}}}}}_{V\rho {V}^{{\dagger} }}}^{2}\), which provides a typical lower bound for the sample complexity of expectation estimation using classical shadows with locally scrambled unitary ensembles.
Just as we have provided an explicit formula for the reconstruction map in terms of the entanglement feature, so too can we provide an explicit formula for the shadow norm using the entanglement feature.
Proposition 11
Given a locally scrambled unitary ensemble \({{{\mathcal{E}}}}\), the (squared) average shadow norm is
where \({W}_{O}[S]={\sum }_{\overrightarrow{a}:{{{\rm{supp}}}}(\overrightarrow{a}) = S}{W}_{O}[\overrightarrow{a}]\).
Based on Propositions 2 and 11, we can express the (squared) average shadow norm in terms of the entanglement feature as follows:
In summary, this section saw an application of our results to classical shadows with locally scrambled unitary ensembles, where we obtained an explicit formula for the reconstruction map and the average shadow norm in terms of the entanglement features, thus circumventing the need to solve the exponential-sized system of linear equations in ref. 26. These explicit formulae may be helpful for the further analysis of the role of entanglement in the classical shadow protocol, which we shall leave for future work.
Application to quantum channels
The classical shadow paradigm was recently extended to the shadow tomography of quantum channels29,30. In these works, the unitary ensembles considered were the local and global Clifford ensembles. In this study, we extend their results to the case of Pauli-invariant unitary ensembles. For clarity of presentation, the detailed proofs of all the results in this subsection are provided in Supplementary Note 4.
Next, we shall outline the procedure proposed in refs. 29,30 for constructing the classical shadows for quantum channels. The unitary ensembles used will be assumed to be Pauli-invariant.
-
(1)
Prepare \(\left\vert {\overrightarrow{b}}_{i}\right\rangle\) with \({\overrightarrow{b}}_{i}\in {\{0,1\}}^{n}\) chosen uniformly randomly.
-
(2)
Apply a unitary Ui chosen from the locally Pauli-invariant ensemble \({{{{\mathcal{E}}}}}_{i}\).
-
(3)
Apply the quantum channel \({{{\mathcal{T}}}}\).
-
(4)
Apply a unitary U chosen form the locally Pauli-invariant ensemble \({{{{\mathcal{E}}}}}_{o}\), where the unitary ensemble \({{{{\mathcal{E}}}}}_{o}\) may be different from \({{{{\mathcal{E}}}}}_{i}\).
-
(5)
Measure in the Pauli Z basis to get the output \({\overrightarrow{b}}_{o}\in {\{0,1\}}^{n}\).
The post-measurement state is given by \({\hat{\sigma }}_{i,o}={U}_{i}^{T}\vert {\overrightarrow{b}}_{i}\rangle \langle {\overrightarrow{b}}_{i}\vert {U}_{i}^{* }\otimes {U}_{o}^{{\dagger} }\vert {\overrightarrow{b}}_{o}\rangle \langle{\overrightarrow{b}}_{o}\vert {U}_{o}\). Hence, given \({\overrightarrow{b}}_{i},{U}_{i},{U}_{o}\), the probability of getting the outcome \({\overrightarrow{b}}_{o}\) is given by \(P({\overrightarrow{b}}_{o}| {\overrightarrow{b}}_{i},{U}_{i},{U}_{o})={2}^{n}{{{\rm{Tr}}}}[{\hat{\sigma }}_{i,o}{{{\mathcal{J}}}}({{{\mathcal{T}}}})],\) where \({{{\mathcal{J}}}}({{{\mathcal{T}}}})={\mathbb{I}}\otimes {{{\mathcal{T}}}}(\left\vert {{\Psi }}\right\rangle \left\langle {{\Psi }}\right\vert )\) is the Choi-Jamiolkowski state of \({{{\mathcal{T}}}}\), where \(\left\vert {{\Psi }}\right\rangle =\frac{1}{\sqrt{{2}^{n}}}{\sum }_{\overrightarrow{i}}\left\vert \overrightarrow{i}\right\rangle \left\vert \overrightarrow{i}\right\rangle\) is the Bell state. It is easy to verify that \({\sum }_{{\overrightarrow{b}}_{o}}P({\overrightarrow{b}}_{o}| {\overrightarrow{b}}_{i},{U}_{i},{U}_{o})=1\). Also, the probability of obtaining \({\overrightarrow{b}}_{i},{U}_{i},{U}_{o}\) is equal to \(P({\overrightarrow{b}}_{i},{U}_{i},{U}_{o})=P({\overrightarrow{b}}_{i})P({U}_{i})P({U}_{o})\). Hence, the ensemble of states is described by
where \(P({\overrightarrow{b}}_{i},{U}_{i},{U}_{o},{\overrightarrow{b}}_{o}):= P({\overrightarrow{b}}_{i},{U}_{i},{U}_{o})P({\overrightarrow{b}}_{o}| {\overrightarrow{b}}_{i},{U}_{i},{U}_{o})\). Taking the average of the classical shadow, we obtain \(\sigma ={{\mathbb{E}}}_{\hat{\sigma }\in {{{{\mathcal{E}}}}}_{i,o}}\hat{\sigma }={{{{\mathcal{M}}}}}_{i,o}[{{{\mathcal{J}}}}({{{\mathcal{T}}}})].\) To implement classical shadow tomography, the inverse of the shadow channel \({{{{\mathcal{M}}}}}_{i,o}^{-1}\) would need to be implemented. If the inverse exists, then \({{{\mathcal{J}}}}({{{\mathcal{T}}}})={{{{\mathcal{M}}}}}_{i,o}^{-1}[\sigma ]={{\mathbb{E}}}_{\hat{\sigma }\in {{{{\mathcal{E}}}}}_{i,o}}{{{{\mathcal{M}}}}}_{i,o}^{-1}[\hat{\sigma }].\) Given a quantum state ρ and an observable O, the estimator \(\hat{o}\) is defined as \(\hat{o}={{{\rm{Tr}}}}[{{{{\mathcal{M}}}}}_{i,o}^{-1}[{\hat{\sigma }}_{i,o}]{\rho }^{T}\otimes O].\) Therefore, \({\mathbb{E}}[\hat{o}]={{{\rm{Tr}}}}[{{{\mathcal{T}}}}[\rho ]O].\)
Proposition 12
Given two Pauli-invariant unitary ensembles \({{{{\mathcal{E}}}}}_{i}\) and \({{{{\mathcal{E}}}}}_{o}\), the shadow channel \({{{{\mathcal{M}}}}}_{i,o}\) is
where \({W}_{{{{{\mathcal{E}}}}}_{i}}[{P}_{{\overrightarrow{a}}_{i}}]={{\mathbb{E}}}_{{b}_{i}}{{\mathbb{E}}}_{{U}_{i}}| {{{\rm{Tr}}}}[{\hat{\sigma }}_{i}{P}_{{\overrightarrow{a}}_{i}}]{| }^{2}\) and \({W}_{{{{{\mathcal{E}}}}}_{o}}[{P}_{{\overrightarrow{a}}_{o}}]={{\mathbb{E}}}_{{b}_{o}}{{\mathbb{E}}}_{{U}_{o}}| {{{\rm{Tr}}}}[{\hat{\sigma }}_{o}{P}_{{\overrightarrow{a}}_{o}}]{| }^{2}\). Hence, for the Pauli-invariant unitary ensemble, \({{{{\mathcal{M}}}}}_{i,o}^{-1}\) exists iff \({W}_{{{{{\mathcal{E}}}}}_{i}}[{\overrightarrow{a}}_{i}]\,>\,0,{W}_{{{{{\mathcal{E}}}}}_{o}}[{\overrightarrow{a}}_{o}]\,>\,0\) for all \(\overrightarrow{a}\), and the reconstruction map is defined by
Now, for the classical shadow tomography for a quantum channel, the sample complexity is upper bounded by the following shadow norm:
Next, let us consider the shadow norm in the case where the observable is some Pauli operator and apply this result to the problem of estimating Pauli channels.
Proposition 13
If the observable is taken to be a Pauli operator \({P}_{\overrightarrow{a}}\), then the (squared) shadow norm is equal to
Now, we provide an example of a class of channels for which our results on the classical shadow tomography for quantum channels can be applied.
Example 2
(Estimation of Pauli channels) A quantum channel \({{{\mathcal{T}}}}\) is called a Pauli channel if it can be written as \({{{\mathcal{T}}}}[\cdot ]={\sum }_{\overrightarrow{a}\in {V}^{n}}{p}_{\overrightarrow{a}}{P}_{\overrightarrow{a}}[\cdot ]{P}_{\overrightarrow{a}}^{{\dagger} }\) with \({\sum }_{\overrightarrow{a}}{p}_{\overrightarrow{a}}=1\). Equivalently, Pauli channels \({{{\mathcal{T}}}}\) can be written as \({{{\mathcal{T}}}}[\cdot ]=1/{2}^{n}{\sum }_{\overrightarrow{b}}{\lambda }_{\overrightarrow{b}}{{{\rm{Tr}}}}[\cdot {P}_{\overrightarrow{b}}]{P}_{\overrightarrow{b}}\). The task of estimating the coefficients \({\{{\lambda }_{\overrightarrow{b}}\}}_{\overrightarrow{b}}\) of the Pauli channel has been investigated in ref. 49,50,51,52,53,54,55,56,57. Here, we consider the classical shadow protocol for estimating coefficients \({\lambda }_{\overrightarrow{b}}\) with Pauli-invariant unitary ensembles \({{{{\mathcal{E}}}}}_{i}\) and \({{{{\mathcal{E}}}}}_{o}\). The classical shadow is taken to be \(\{{{{{\mathcal{M}}}}}_{i,o}^{-1}[{\hat{\sigma }}_{i,o}]\}\), and the reconstruction map is given in Proposition 12. To estimate the coefficients \({\lambda }_{\overrightarrow{b}}\), let us consider the observable \({P}_{\overrightarrow{b}}\otimes {P}_{\overrightarrow{b}}\), and take the estimator of the observable to be \({\hat{o}}_{\overrightarrow{b}}={{{\rm{Tr}}}}[{{{{\mathcal{M}}}}}_{i,o}^{-1}[{\hat{\sigma }}_{i,o}]{P}_{\overrightarrow{b}}\otimes {P}_{\overrightarrow{b}}]\), where it is easy to verify that \({\mathbb{E}}[{\hat{o}}_{\overrightarrow{b}}]={{{\rm{Tr}}}}[{{{\mathcal{J}}}}[{{{\mathcal{T}}}}]{P}_{\overrightarrow{b}}\otimes {P}_{\overrightarrow{b}}]={\lambda }_{\overrightarrow{b}}\). Then the sample complexity \({{{\mathscr{S}}}}\) needed to accurately predict a collection of 4n linear target functions \({\left\{{\lambda }_{\overrightarrow{b}} = {{{\rm{Tr}}}}[{{{\mathcal{J}}}}[{{{\mathcal{T}}}}]{P}_{\overrightarrow{b}}\otimes {P}_{\overrightarrow{b}}\left.\right]]\right\}}_{\overrightarrow{b}\in {V}^{n}}\) with error ϵ and failure probability δ is
Discussion
In this study, we investigated the classical shadow protocol with Pauli-invariant unitary ensembles. First, we provided an explicit formula for the reconstruction map corresponding to the shadow channel and established a connection between the coefficients of the reconstruction map and the entanglement features of the dynamics. Using the shadow norm, we gave explicit sample complexity upper bounds for the estimation task that the classical shadow protocol solves. Finally, we presented two applications of our results. First, we applied our results to locally scrambled unitary ensembles, where we presented explicit formulas for the reconstruction map and the sample complexity bounds. Second, we applied our results to the shadow process tomography of quantum channels with Pauli-invariant unitary ensembles and provided an example where we considered the task of estimating Pauli channels. Our results provide a general framework for classical shadows with a weak assumption on the unitary ensemble in both the noiseless and noisy cases, which can be utilized to predict pertinent properties of quantum states in NISQ devices, including their fidelity, entanglement entropy, and quantum Fisher information.
There are still several interesting unresolved problems: (1) Can one generalize our results to ensembles beyond the Pauli-invariant unitary ensembles? We anticipate that this would be challenging, as a crucial ingredient that we utilize in our proof is that the Pauli ensemble forms a 1-design. This is a very weak assumption, and a more general unitary ensemble may fail to satisfy the 1-design property. That will likely impede a straightforward generalization of our methods. (2) How can one utilize the entanglement feature of the unitary ensemble to improve the performance of the classical shadow across various tasks, such as fidelity estimation?
Methods
Diagonalization of the shadow channel in the Pauli basis
This section provides a bird’s-eye view of the proof of our main result, which centers on the shadow channel \({{{\mathcal{M}}}}(\cdot )\) being diagonal in the Pauli basis. A comprehensive description of our proof is available in Supplementary Note 2. The primary method employed in our proof relies on Fourier analysis with respect to the Pauli basis. The crucial point to establish is as follows:
This equation is derived using the following two facts: (i) the set of Pauli operators forms an orthonormal basis for the space of n-qubit linear operators, (ii) the Pauli operators obey the following commutation property: \({P}_{\overrightarrow{b}}{P}_{\overrightarrow{a}}{P}_{\overrightarrow{b}}={(-1)}^{{\langle \overrightarrow{a},\overrightarrow{b}\rangle }_{s}}{P}_{\overrightarrow{a}}\), where 〈⋅, ⋅〉s is the symplectic inner product defined in Supplementary Note 1. The fact that \({{{\mathcal{M}}}}(\cdot )\) is diagonal in the Pauli basis allows us to build the reconstruction map \({{{{\mathcal{M}}}}}^{-1}\) by simply taking the inverse of the coefficients of \({{{\mathcal{M}}}}(\cdot )\) in the Pauli basis.
Data availability
Data sharing is not applicable to this article as no datasets were generated or analyzed during the current study.
References
Huang, H.-Y., Kueng, R. & Preskill, J. Predicting many properties of a quantum system from very few measurements. Nat. Phys. 16, 1050–1057 (2020).
Cotler, J. & Wilczek, F. Quantum overlapping tomography. Phys. Rev. Lett. 124, 100401 (2020).
Evans, T. J., Harper, R. & Flammia, S. T. Scalable Bayesian Hamiltonian learning. Preprint at https://arxiv.org/abs/1912.07636 (2019).
Aaronson, S. Shadow tomography of quantum states. SIAM J. Comput. 49, STOC18–368 (2019).
Haah, J., Harrow, A. W., Ji, Z., Wu, X. & Yu, N. Sample-optimal tomography of quantum states. IEEE Trans. Inf. Theory 63, 5628–5641 (2017).
O’Donnell, R. & Wright, J. Efficient quantum tomography. In Proc. forty-eighth annual ACM symposium on Theory of Computing, 899–912 (2016).
Huang, H.-Y. Learning quantum states from their classical shadows. Nat. Rev. Phys. 4, 81 (2022).
Hadfield, C., Bravyi, S., Raymond, R. & Mezzacapo, A. Measurements of Quantum Hamiltonians with Locally-biased Classical Shadows. Commun. Math. Phys. 391, 951–967 (2022).
Hadfield, C. Adaptive Pauli shadows for energy estimation. Preprint at https://arxiv.org/abs/2105.12207 (2021).
Elben, A. et al. Mixed-state entanglement from local randomized measurements. Phys. Rev. Lett. 125, 200501 (2020).
Neven, A. et al. Symmetry-resolved entanglement detection using partial transpose moments. npj Quantum Inf. 7, 1–12 (2021).
Rath, A., Branciard, C., Minguzzi, A. & Vermersch, B. Quantum Fisher information from randomized measurements. Phys. Rev. Lett. 127, 260501 (2021).
Li, G., Song, Z. & Wang, X. VSQL: Variational shadow quantum learning for classification. Proc. AAAI Conf. Artif. Intell. 35, 8357–8365 (2021).
Garcia, R. J., Zhou, Y. & Jaffe, A. Quantum scrambling with classical shadows. Phys. Rev. Res. 3, 033155 (2021).
McGinley, M., Leontica, S., Garratt, S. J., Jovanovic, J. & Simon, S. H. Quantifying information scrambling via classical shadow tomography on programmable quantum simulators. Phys. Rev. A 106, 012441 (2022).
Huggins, W. J. et al. Unbiasing fermionic quantum Monte Carlo with a quantum computer. Nature 603, 416–420 (2022).
Huang, H.-Y., Kueng, R., Torlai, G., Albert, V. V. & Preskill, J. Provably efficient machine learning for quantum many-body problems. Science 377, eabk3333 (2022).
Notarnicola, S. et al. A randomized measurement toolbox for an interacting Rydberg-atom quantum simulator. New J. Phys. 25, 103006 (2023).
Helsen, J. et al. Shadow estimation of gate-set properties from random sequences. Nat. Commun. 14, 1–11 (2023).
Chen, S., Yu, W., Zeng, P. & Flammia, S. T. Robust shadow estimation. PRX Quantum 2, 030348 (2021).
Flammia, S. T. Averaged circuit eigenvalue sampling. Preprint at https://arxiv.org/abs/2108.05803 (2021).
Huang, H.-Y., Kueng, R. & Preskill, J. Efficient estimation of Pauli observables by derandomization. Phys. Rev. Lett. 127, 030503 (2021).
Hillmich, S., Hadfield, C., Raymond, R., Mezzacapo, A. & Wille, R. Decision diagrams for quantum measurements with shallow circuits. In Proc. IEEE International Conference on Quantum Computing and Engineering (QCE), 24–34 (2021).
Wu, B., Sun, J., Huang, Q. & Yuan, X. Overlapped grouping measurement: a unified framework for measuring quantum states. Quantum 7, 896 (2023).
Acharya, A., Saha, S. & Sengupta, A. M. Shadow tomography based on informationally complete positive operator-valued measure. Phys. Rev. A 104, 052418 (2021).
Hu, H.-Y., Choi, S. & You, Y.-Z. Classical shadow tomography with locally scrambled quantum dynamics. Phys. Rev. Res. 5, 023027 (2023).
Hu, H.-Y. & You, Y.-Z. Hamiltonian-driven shadow tomography of quantum states. Phys. Rev. Res. 4, 013054 (2022).
Zhao, A., Rubin, N. C. & Miyake, A. Fermionic partial tomography via classical shadows. Phys. Rev. Lett. 127, 110504 (2021).
Levy, R., Luo, D. & Clark, B. K. Classical shadows for quantum process tomography on near-term quantum computers. Preprint at https://arxiv.org/abs/2110.02965 (2021).
Kunjummen, J., Tran, M. C., Carney, D. & Taylor, J. M. Shadow process tomography of quantum channels. Phys. Rev. A 107, 042403 (2023).
Lukens, J. M., Law, K. J. H. & Bennink, R. S. Classical shadows and Bayesian mean estimation: a comparison. In Proc. Conference on Lasers and Electro-Optics, FW3N.3 (Optical Society of America, 2021). http://www.osapublishing.org/abstract.cfm?URI=CLEO_QELS-2021-FW3N.3.
Lukens, J. M., Law, K. J. H. & Bennink, R. S. A Bayesian analysis of classical shadows. npj Quantum Inf. 7, 1–10 (2021).
Chen, S., Cotler, J., Huang, H.-Y. & Li, J. Exponential separations between learning with and without quantum memory. In Proc. IEEE 62nd Annual Symposium on Foundations of Computer Science (FOCS), 574–585 (2022).
Struchalin, G., Zagorovskii, Y. A., Kovlakov, E., Straupe, S. & Kulik, S. Experimental estimation of quantum state properties from classical shadows. PRX Quantum 2, 010307 (2021).
Zhang, T. et al. Experimental quantum state measurement with classical shadows. Phys. Rev. Lett. 127, 200501 (2021).
Liu, L., Zhang, T., Yuan, X. & Lu, H. Experimental investigation of quantum uncertainty relations with classical shadows. Front. Phys. 10, 873810 (2022).
Koh, D. E. & Grewal, S. Classical shadows with noise. Quantum 6, 776 (2022).
Ippoliti, M., Li, Y., Rakovszky, T. & Khemani, V. Operator relaxation and the optimal depth of classical shadows. Phys. Rev. Lett. 130, 230403 (2023).
Harrow, A. W. & Mehraban, S. Approximate unitary t-designs by short random quantum circuits using nearest-neighbor and long-range gates. Commun. Math. Phys. 401, 1531–1626 (2023).
Montanaro, A. & Osborne, T. J. Quantum boolean functions. Chicago Journal of Theoretical Computer Science 2010, 1–45 (2010).
Bu, K., Garcia, R. J., Jaffe, A., Koh, D. E. & Li, L. Complexity of quantum circuits via sensitivity, magic, and coherence. Preprint at https://arxiv.org/abs/2204.12051 (2022).
Garcia, R. J., Bu, K. & Jaffe, A. Resource theory of quantum scrambling. Proc. Natl. Acad. Sci. 120, e2217031120 (2023).
Bu, K., Gu, W. & Jaffe, A. Quantum entropy and central limit theorem. Proc. Natl. Acad. Sci. 120, e2304589120 (2023).
Bu, K., Gu, W. & Jaffe, A. Discrete Quantum Gaussians and Central Limit Theorem. Preprint at https://arxiv.org/abs/2302.08423 (2023).
Bu, K., Gu, W. & Jaffe, A. Stabilizer testing and magic entropy. Preprint at https://arxiv.org/abs/2306.09292 (2023).
You, Y.-Z., Yang, Z. & Qi, X.-L. Machine learning spatial geometry from entanglement features. Phys. Rev. B 97, 045153 (2018).
You, Y.-Z. & Gu, Y. Entanglement features of random Hamiltonian dynamics. Phys. Rev. B 98, 014309 (2018).
Wu, B. & Koh, D. E. Error-mitigated fermionic classical shadows on noisy quantum devices. Preprint at https://arxiv.org/abs/2310.12726 (2023).
Flammia, S. T. & Wallman, J. J. Efficient estimation of Pauli channels. ACM Trans. Quantum Comput. 1, 1 (2020).
Fujiwara, A. & Imai, H. Quantum parameter estimation of a generalized Pauli channel. J. Phys. A: Math. Gen. 36, 8093–8103 (2003).
Hayashi, M. Quantum channel estimation and asymptotic bound. J. Phys. Conf. Ser. 233, 012016 (2010).
Chiuri, A. et al. Experimental realization of optimal noise estimation for a general Pauli channel. Phys. Rev. Lett. 107, 253602 (2011).
Ruppert, L., Virosztek, D. & Hangos, K. Optimal parameter estimation of Pauli channels. J. Phys. A: Math. Theor. 45, 265305 (2012).
Collins, D. Mixed-state Pauli-channel parameter estimation. Phys. Rev. A 87, 032301 (2013).
Harper, R., Yu, W. & Flammia, S. T. Fast estimation of sparse quantum noise. PRX Quantum 2, 010322 (2021).
Flammia, S. T. & O’Donnell, R. Pauli error estimation via population recovery. Quantum 5, 549 (2021).
Chen, S., Zhou, S., Seif, A. & Jiang, L. Quantum advantages for Pauli channel estimation. Phys. Rev. A 105, 032435 (2022).
Acknowledgements
K.B. thanks Hong-Ye Hu for the discussion on classical shadows and its connection with quantum entanglement. This work was supported in part by the ARO Grant W911NF-19-1-0302, ARO MURI Grant W911NF-20-1-0082, and NSF Eager Grant 2037687. D.E.K. acknowledges funding support from the National Research Foundation, Singapore and the Agency for Science, Technology and Research (A*STAR) under its Quantum Engineering Programme (NRF2021-QEP2-02-P03), A*STAR C230917003, and A*STAR under the Central Research Fund (CRF) Award for Use-Inspired Basic Research (UIBR).
Author information
Authors and Affiliations
Contributions
K.B., D.E.K., R.J.G. and A.J. designed the study, performed the research, and wrote the manuscript.
Corresponding authors
Ethics declarations
Competing interests
The authors declare no competing interests.
Additional information
Publisher’s note Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Supplementary information
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/.
About this article
Cite this article
Bu, K., Koh, D.E., Garcia, R.J. et al. Classical shadows with Pauli-invariant unitary ensembles. npj Quantum Inf 10, 6 (2024). https://doi.org/10.1038/s41534-023-00801-w
Received:
Accepted:
Published:
DOI: https://doi.org/10.1038/s41534-023-00801-w
This article is cited by
-
Error-mitigated fermionic classical shadows on noisy quantum devices
npj Quantum Information (2024)
-
Improved machine learning algorithm for predicting ground state properties
Nature Communications (2024)