Note
This is the documentation for the current state of the development branch of Qiskit Experiments. The documentation or APIs here can change prior to being released.
Quantum State Tomography¶
Quantum tomography is an experimental procedure to reconstruct a description of part of a quantum system from the measurement outcomes of a specific set of experiments. In particular, quantum state tomography reconstructs the density matrix of a quantum state by preparing the state many times and measuring them in a tomographically complete basis of measurement operators.
Note
This tutorial requires the qiskit-aer and qiskit-ibm-runtime
packages to run simulations. You can install them with python -m pip
install qiskit-aer qiskit-ibm-runtime
.
We first initialize a simulator to run the experiments on.
from qiskit_aer import AerSimulator
from qiskit_ibm_runtime.fake_provider import FakePerth
backend = AerSimulator.from_backend(FakePerth())
To run a state tomography experiment, we initialize the experiment with a circuit to
prepare the state to be measured. We can also pass in an
Operator
or a Statevector
to describe the preparation circuit.
import qiskit
from qiskit_experiments.framework import ParallelExperiment
from qiskit_experiments.library import StateTomography
# GHZ State preparation circuit
nq = 2
qc_ghz = qiskit.QuantumCircuit(nq)
qc_ghz.h(0)
qc_ghz.s(0)
for i in range(1, nq):
qc_ghz.cx(0, i)
# QST Experiment
qstexp1 = StateTomography(qc_ghz)
qstdata1 = qstexp1.run(backend, seed_simulation=100).block_for_results()
# Print results
for result in qstdata1.analysis_results():
print(result)
AnalysisResult
- name: state
- value: DensityMatrix([[ 0.47607422+0.j , 0.01139323-0.01074219j,
-0.00634766-0.00846354j, -0.0078125 -0.44238281j],
[ 0.01139323+0.01074219j, 0.03336589+0.j ,
0.00488281-0.00488281j, -0.00537109-0.01041667j],
[-0.00634766+0.00846354j, 0.00488281+0.00488281j,
0.03499349+0.j , -0.00716146-0.00488281j],
[-0.0078125 +0.44238281j, -0.00537109+0.01041667j,
-0.00716146+0.00488281j, 0.45556641+0.j ]],
dims=(2, 2))
- quality: unknown
- extra: <9 items>
- device_components: ['Q0', 'Q1']
- verified: False
AnalysisResult
- name: state_fidelity
- value: 0.9082031249999992
- quality: unknown
- extra: <9 items>
- device_components: ['Q0', 'Q1']
- verified: False
AnalysisResult
- name: positive
- value: True
- quality: unknown
- extra: <9 items>
- device_components: ['Q0', 'Q1']
- verified: False
Tomography Results¶
The main result for tomography is the fitted state, which is stored as a
DensityMatrix
object:
state_result = qstdata1.analysis_results("state")
print(state_result.value)
DensityMatrix([[ 0.47607422+0.j , 0.01139323-0.01074219j,
-0.00634766-0.00846354j, -0.0078125 -0.44238281j],
[ 0.01139323+0.01074219j, 0.03336589+0.j ,
0.00488281-0.00488281j, -0.00537109-0.01041667j],
[-0.00634766+0.00846354j, 0.00488281+0.00488281j,
0.03499349+0.j , -0.00716146-0.00488281j],
[-0.0078125 +0.44238281j, -0.00537109+0.01041667j,
-0.00716146+0.00488281j, 0.45556641+0.j ]],
dims=(2, 2))
We can also visualize the density matrix:
from qiskit.visualization import plot_state_city
plot_state_city(qstdata1.analysis_results("state").value, title='Density Matrix')
The state fidelity of the fitted state with the ideal state prepared by
the input circuit is stored in the "state_fidelity"
result field.
Note that if the input circuit contained any measurements the ideal
state cannot be automatically generated and this field will be set to
None
.
fid_result = qstdata1.analysis_results("state_fidelity")
print("State Fidelity = {:.5f}".format(fid_result.value))
State Fidelity = 0.90820
Additional state metadata¶
Additional data is stored in the tomography under the
"state_metadata"
field. This includes
eigvals
: the eigenvalues of the fitted statetrace
: the trace of the fitted statepositive
: Whether the eigenvalues are all non-negativepositive_delta
: the deviation from positivity given by 1-norm of negative eigenvalues.
If trace rescaling was performed this dictionary will also contain a raw_trace
field
containing the trace before rescaling. Futhermore, if the state was rescaled to be
positive or trace 1 an additional field raw_eigvals
will contain the state
eigenvalues before rescaling was performed.
state_result.extra
{'trace': 1.0000000000000016,
'eigvals': array([0.90868683, 0.0518398 , 0.02711263, 0.01236075]),
'raw_eigvals': array([0.90868683, 0.0518398 , 0.02711263, 0.01236075]),
'rescaled_psd': False,
'fitter_metadata': {'fitter': 'linear_inversion',
'fitter_time': 0.004186868667602539},
'conditional_probability': 1.0,
'positive': True,
'experiment': 'StateTomography',
'run_time': None}
To see the effect of rescaling, we can perform a “bad” fit with very low counts:
# QST Experiment
bad_data = qstexp1.run(backend, shots=10, seed_simulation=100).block_for_results()
bad_state_result = bad_data.analysis_results("state")
# Print result
print(bad_state_result)
# Show extra data
bad_state_result.extra
AnalysisResult
- name: state
- value: DensityMatrix([[ 0.32460624+0.j , -0.05798543+0.02158558j,
0.0397827 -0.02707905j, -0.08580301-0.31159306j],
[-0.05798543-0.02158558j, 0.01602264+0.j ,
0.00637975-0.00914923j, 0.00929544+0.03274001j],
[ 0.0397827 +0.02707905j, 0.00637975+0.00914923j,
0.09280437+0.j , 0.14533747-0.10943242j],
[-0.08580301+0.31159306j, 0.00929544-0.03274001j,
0.14533747+0.10943242j, 0.56656675+0.j ]],
dims=(2, 2))
- quality: unknown
- extra: <9 items>
- device_components: ['Q0', 'Q1']
- verified: False
{'trace': 1.0000000000000002,
'eigvals': array([0.83555322, 0.16444678, 0. , 0. ]),
'raw_eigvals': array([ 0.93548763, 0.26438119, -0.05588552, -0.14398329]),
'rescaled_psd': True,
'fitter_metadata': {'fitter': 'linear_inversion',
'fitter_time': 0.0033676624298095703},
'conditional_probability': 1.0,
'positive': True,
'experiment': 'StateTomography',
'run_time': None}
Tomography Fitters¶
The default fitters is linear_inversion
, which reconstructs the
state using dual basis of the tomography basis. This will typically
result in a non-positive reconstructed state. This state is rescaled to
be positive-semidefinite (PSD) by computing its eigen-decomposition and
rescaling its eigenvalues using the approach from Ref. [1].
There are several other fitters are included (See API documentation for
details). For example, if cvxpy
is installed we can use the
cvxpy_gaussian_lstsq()
fitter, which allows constraining the fit to be
PSD without requiring rescaling.
try:
import cvxpy
# Set analysis option for cvxpy fitter
qstexp1.analysis.set_options(fitter='cvxpy_gaussian_lstsq')
# Re-run experiment
qstdata2 = qstexp1.run(backend, seed_simulation=100).block_for_results()
state_result2 = qstdata2.analysis_results("state")
print(state_result2)
print("\nextra:")
for key, val in state_result2.extra.items():
print(f"- {key}: {val}")
except ModuleNotFoundError:
print("CVXPY is not installed")
AnalysisResult
- name: state
- value: DensityMatrix([[ 0.47131771+0.j , 0.00457409-0.00152989j,
0.00325319+0.0164979j , -0.0018478 -0.44487636j],
[ 0.00457409+0.00152989j, 0.02113254+0.j ,
0.00567847-0.006379j , -0.00808356-0.00260133j],
[ 0.00325319-0.0164979j , 0.00567847+0.006379j ,
0.03707474+0.j , -0.01347399+0.00611815j],
[-0.0018478 +0.44487636j, -0.00808356+0.00260133j,
-0.01347399-0.00611815j, 0.47047501+0.j ]],
dims=(2, 2))
- quality: unknown
- extra: <9 items>
- device_components: ['Q0', 'Q1']
- verified: False
extra:
- trace: 1.00000000029414
- eigvals: [0.91634535 0.04182533 0.03019168 0.01163764]
- raw_eigvals: [0.91634535 0.04182533 0.03019168 0.01163764]
- rescaled_psd: False
- fitter_metadata: {'fitter': 'cvxpy_gaussian_lstsq', 'cvxpy_solver': 'SCS', 'cvxpy_status': ['optimal'], 'psd_constraint': True, 'trace_preserving': True, 'fitter_time': 0.0193936824798584}
- conditional_probability: 1.0
- positive: True
- experiment: StateTomography
- run_time: None
Parallel Tomography Experiment¶
We can also use the ParallelExperiment
class to
run subsystem tomography on multiple qubits in parallel.
For example if we want to perform 1-qubit QST on several qubits at once:
from math import pi
num_qubits = 5
gates = [qiskit.circuit.library.RXGate(i * pi / (num_qubits - 1))
for i in range(num_qubits)]
subexps = [
StateTomography(gate, physical_qubits=(i,))
for i, gate in enumerate(gates)
]
parexp = ParallelExperiment(subexps)
pardata = parexp.run(backend, seed_simulation=100).block_for_results()
for result in pardata.analysis_results():
print(result)
AnalysisResult
- name: state
- value: DensityMatrix([[0.97558594+0.j , 0.03125 -0.0078125j],
[0.03125 +0.0078125j, 0.02441406+0.j ]],
dims=(2,))
- quality: unknown
- extra: <9 items>
- device_components: ['Q0']
- verified: False
AnalysisResult
- name: state_fidelity
- value: 0.9755859374999997
- quality: unknown
- extra: <9 items>
- device_components: ['Q0']
- verified: False
AnalysisResult
- name: positive
- value: True
- quality: unknown
- extra: <9 items>
- device_components: ['Q0']
- verified: False
AnalysisResult
- name: state
- value: DensityMatrix([[ 0.81054688+0.j , -0.00976562+0.31347656j],
[-0.00976562-0.31347656j, 0.18945313+0.j ]],
dims=(2,))
- quality: unknown
- extra: <9 items>
- device_components: ['Q1']
- verified: False
AnalysisResult
- name: state_fidelity
- value: 0.9412512042755902
- quality: unknown
- extra: <9 items>
- device_components: ['Q1']
- verified: False
AnalysisResult
- name: positive
- value: True
- quality: unknown
- extra: <9 items>
- device_components: ['Q1']
- verified: False
AnalysisResult
- name: state
- value: DensityMatrix([[0.47558594+0.j , 0.00976562+0.46679688j],
[0.00976562-0.46679688j, 0.52441406+0.j ]],
dims=(2,))
- quality: unknown
- extra: <9 items>
- device_components: ['Q2']
- verified: False
AnalysisResult
- name: state_fidelity
- value: 0.9667968750000002
- quality: unknown
- extra: <9 items>
- device_components: ['Q2']
- verified: False
AnalysisResult
- name: positive
- value: True
- quality: unknown
- extra: <9 items>
- device_components: ['Q2']
- verified: False
AnalysisResult
- name: state
- value: DensityMatrix([[ 0.16699219+0.j , -0.00585938+0.33007812j],
[-0.00585938-0.33007812j, 0.83300781+0.j ]],
dims=(2,))
- quality: unknown
- extra: <9 items>
- device_components: ['Q3']
- verified: False
AnalysisResult
- name: state_fidelity
- value: 0.9688725629156892
- quality: unknown
- extra: <9 items>
- device_components: ['Q3']
- verified: False
AnalysisResult
- name: positive
- value: True
- quality: unknown
- extra: <9 items>
- device_components: ['Q3']
- verified: False
AnalysisResult
- name: state
- value: DensityMatrix([[0.02539063+0.j , 0.00292969-0.00976562j],
[0.00292969+0.00976562j, 0.97460937+0.j ]],
dims=(2,))
- quality: unknown
- extra: <9 items>
- device_components: ['Q4']
- verified: False
AnalysisResult
- name: state_fidelity
- value: 0.9746093749999994
- quality: unknown
- extra: <9 items>
- device_components: ['Q4']
- verified: False
AnalysisResult
- name: positive
- value: True
- quality: unknown
- extra: <9 items>
- device_components: ['Q4']
- verified: False
View component experiment analysis results:
for i, expdata in enumerate(pardata.child_data()):
state_result_i = expdata.analysis_results("state")
fid_result_i = expdata.analysis_results("state_fidelity")
print(f'\nPARALLEL EXP {i}')
print("State Fidelity: {:.5f}".format(fid_result_i.value))
print("State: {}".format(state_result_i.value))
References¶
See also¶
API documentation:
StateTomography