Iterative phase estimation

Note

Click here to view this tutorial as an interactive Jupyter notebook in IBM Quantum Lab (requires sign-in).

The goal of this exercise is to understand how the Iterative Phase Estimation (IPE) algorithm uses dynamic circuits, and why would we use the IPE algorithm instead of the QPE (Quantum Phase Estimation) algorithm. We demonstrate how to perform this algorithm with Qiskit using circuits containing reset in order to apply gates conditioned by the values resulting from previous measurements and stored in a classical register.

If you want to learn more about iterative phase estimation, visit the references below.

References

Preamble

import os
from typing import Any, List, Dict, Union

import numpy as np
import matplotlib.pyplot as plt

from qiskit import Aer, QuantumCircuit, QuantumRegister, ClassicalRegister, transpile, execute
from qiskit.tools.visualization import plot_histogram
from qiskit_ibm_provider import IBMProvider

import warnings
warnings.filterwarnings("ignore")

pi = np.pi
# define backend to request
# Note: This can be any hub/group/project that has access to the required device and Qiskit Runtime.\n",
hub = "<hub>"
group = "<group>"
project = "<project>"
backend_name = "<your backend>"
hgp = f"{hub}/{group}/{project}"
# load backend
provider = IBMProvider()
backend = provider.get_backend(backend_name, instance=hgp)
backend_sim = Aer.get_backend('aer_simulator')
#backend = least_busy(provider.backends(dynamic_circuits=True))
print(f"Using backend {backend.name}")

Iterative phase estimation

The motivation for using the IPE algorithm is that the Quantum Phase Estimation algorithm works fine for short depth circuits, but when the circuit starts to grow, it doesn’t work properly due to gate noise and decoherence times.

The detailed explanation of how the algorithm works can be found in Iterative Phase Estimation (IPE) Algorithm. To understand QPE in depth, you can see also Ch.3.6 Quantum Phase Estimation.

IPE example with a 1-qubit gate for U

We want to apply the IPE algorithm to estimate the phase for a 1-qubit operator U . For example, here we use the S -gate.

Let’s apply the IPE algorithm to estimate the phase for the S -gate. Its matrix is

\begin{split}S =\begin{bmatrix} 1 & 0 \\ 0 & e^\frac{i\pi}{2} \end{bmatrix}\end{split}

That is, the S -gate adds a phase \pi/2 to the state |1\rangle , leaving unchanged the phase of the state |0\rangle

S|1\rangle = e^\frac{i\pi}{2}|1\rangle

In the following, we will use the notation and terms used in Section 2 of lab 4 of the Qiskit Textbook.

To estimate the phase \phi=\frac{\pi}{2} for the eigenstate |1\rangle , we should find \varphi=\frac{1}{4} (where \phi = 2 \pi \varphi ). Therefore, to estimate the phase we need exactly two phase bits, i.e.,  m=2 , since 1/2^2=1/4 . Therefore, \varphi=0.\varphi_1\varphi_2 .

Remember from the theory that for the IPE algorithm, m is also the number of iterations, so we need only 2 iterations or steps.

First, we initialize the circuit. IPE works with only one auxiliary qubit, instead of m counting qubits of the QPE algorithm. Therefore, we need two qubits: one auxiliary qubit, and one for the eigenstate of U -gate, as well as a classical register of two bits, for the phase bits \varphi_1 , \varphi_2 .

IPE - First step

Now we build the quantum circuit for the first step (that is, the first iteration of the algorithm), to estimate the least significant phase bit \varphi_m , which in this case is \varphi_2 . For the first step we have four sub-steps:

  • qubit initialization

  • initialization

  • application of the Controlled- U gates

  • measure of the auxiliary qubit in the x-basis

qubits = [0, 1]
shots = 1000

nq = 2
m = 2
qr = QuantumRegister(2,'q')
c0 = ClassicalRegister(1,'c0')
c1 = ClassicalRegister(1, 'c1')
qc_S = QuantumCircuit(qr, c0, c1)

Eigenstate initialization

The initialization consists of applying the Hadamard gate to the auxiliary qubit, and the preparation of the eigenstate |1\rangle .

def initialize_eigenstate(qc):
    """Initialize the eigenstate and prepare our ancilla qubit"""
    qc.h(0)
    qc.x(1)

initialize_eigenstate(qc_S)
qc_S.draw('mpl')
../../../../_images/output_26_0.png

Application of the Controlled- U gates

We now have to apply 2^t times the Controlled- U operators (see also in the Qiskit documentation Two-qubit gates), that, in this example, is the Controlled- S gate ( CS for short).

To implement CS in the circuit, since S is a phase gate, we can use the controlled phase gate \text{CP}(\theta) , with \theta=\pi/2 .

theta = 1 * np.pi / 2
cu_circ = QuantumCircuit(2)
cu_circ.cp(theta,0,1)
cu_circ.draw('mpl')
../../../../_images/output_29_0.png

Let’s apply 2^t times \text{CP}(\pi/2) . Since for the first step t=m-1 , and m=2 , we have 2^t=2 .

for _ in range(2**(m-1)):
    qc_S.cp(theta,0,1)
qc_S.draw('mpl')
../../../../_images/output_31_0.png

Measure in the x-basis

Finally, we perform the measurement of the auxiliary qubit in the x-basis.

In this way we obtain the phase bit \varphi_2 and store it in the classical bit c_0 .

Note

The eigenvectors of the Hadamard gate are the basis vectors of the x-basis.

def x_measurement(qc, qubit, cbit):
    """Measure 'qubit' in the X-basis, and store the result in 'cbit'"""
    qc.h(qubit)
    qc.measure(qubit, cbit)

x_measurement(qc_S, qr[0], c0)
qc_S.draw('mpl')
../../../../_images/output_33_0.png

Subsequent steps - Second step

Now we build the quantum circuit for the other remaining steps (in this example, only the second one). In these steps we have four sub-steps: the three sub-steps as in the first step and, in the middle, the additional step of the phase correction.

  • initialization with reset

  • phase correction

  • application of the Control- U gates

  • measurement of the auxiliary qubit in x-basis

Reset the auxiliary qubit

As we want to perform an iterative algorithm in the same circuit, we need to reset the auxiliary qubit q0 after the measurement gate and initialize it again as before, to recycle the qubit.

We use the built-in qc.reset operation to perform a reset of qubit 0, followed by the Hadamard operation.

def reset_auxiliary(qc):
    qc.reset(0)
    qc.h(0)

reset_auxiliary(qc_S)
qc_S.draw('mpl')
../../../../_images/output_36_1.png

Phase correction (for step 2)

As seen in the theory, in order to extract the phase bit \varphi_{1} , we perform a phase correction of -\pi\varphi_2/2 . Of course, we need to apply the phase correction in the circuit only if the phase bit \varphi_2=1 , i.e., we have to apply the phase correction of -\pi/2 only if the classical bit c_0 is 1.

So, after the reset we apply the phase gate P(\theta) with phase \theta=-\pi/2 conditioned by the classical bit c_0 ( =\varphi_2 )

with qc_S.if_test((c0, 1)):
    qc_S.p(-np.pi/2,0)
qc_S.draw('mpl')
../../../../_images/output_38_0.png

Apply the controlled- U gates and x-measurement (for step 2)

We apply the CU operations as we did in the first step. For the second step we have t=m-2 , hence 2^t=1 . We therefore apply \text{CP}(\pi/2) once, and then we perform the x-measurement of the qubit q_0 , storing the result, the phase bit \varphi_1 , in bit c_1 of the classical register.

## 2^t c-U operations (with t=m-2)
for _ in range(2**(m-2)):
    qc_S.cp(theta,0,1)

x_measurement(qc_S, qr[0], c1)

We now have our final circuit!

qc_S.draw('mpl')
../../../../_images/output_42_1.png

Let’s execute the circuit with the qasm_simulator, the simulator without noise that runs locally.

count = execute(qc_S, backend_sim).result().get_counts()
count
{'0 1': 1024}
count0 = execute(qc_S, backend_sim).result().get_counts()
plot_histogram(count0)
../../../../_images/output_45_1.png
count0
{'0 1': 1024}

In the picture we have the same histograms, but on the left we have on the x-axis the string with phase bits \varphi_1 , \varphi_2 , and on the right the actual phase \varphi in decimal representation.

As we expected, we have found \varphi=\frac{1}{4}=0.25 with a 100\% probability.

Run on real hardware

qc_S_backend = transpile(qc_S, backend, initial_layout=qubits, optimization_level=1)
qc_S_backend.draw(output="mpl")
../../../../_images/output_49_0.png
real_ipe_job = backend.run(qc_S_backend, dynamic=True, shots=shots)
print(f"IPE job id: {real_ipe_job.job_id()}")
IPE job id: cdlf4v6h8o8scshbikqg
real_counts0 = real_ipe_job.result().get_counts()
plot_histogram(real_counts0)
../../../../_images/output_51_1.png

As expected, we estimate the correct result of 01 the majority of the time.

Now that we have estimated the phase of the S gate, let’s try finding the phase of another gate. Recall that we implemented IPE for S using the P(\theta) gate. Choose a new instance of the P gate that can be estimated with two bits of precision and rerun.

After you have mastered IPE for two bits of precision, try adding a third bit. Modify the code of this exercise to add another round of IPE.

import qiskit.tools.jupyter
%qiskit_version_table
%qiskit_copyright

Version Information

Qiskit SoftwareVersion
qiskit-terra0.23.0
qiskit-aer0.11.4
qiskit-ignis0.7.0
qiskit-ibmq-provider0.20.0.dev0+4f1f8c6
qiskit0.36.1
System information
Python version3.9.5
Python compilerClang 10.0.0
Python builddefault, May 18 2021 12:31:01
OSDarwin
CPUs8
Memory (Gb)32.0
Tue Nov 08 19:22:46 2022 EST

This code is a part of Qiskit

© Copyright IBM 2017, 2022.

This code is licensed under the Apache License, Version 2.0. You may
obtain a copy of this license in the LICENSE.txt file in the root directory
of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.

Any modifications or derivative works of this code must retain this
copyright notice, and modified files need to carry a notice indicating
that they have been altered from the originals.