Basic tutorial: Gaussian transformation

This tutorial demonstrates the basic working principles of PennyLane for continuous-variable (CV) photonic devices. For more details about photonic quantum computing, the Strawberry Fields documentation is a great starting point.

The quantum circuit

For this basic tutorial, we will consider a special subset of CV operations: the Gaussian transformations. We work with the following simple Gaussian circuit:


../_images/gaussian_transformation.svg


What is this circuit doing?

  1. We begin with one wire (qumode) in the vacuum state. Note that we use the same notation \(|0\rangle\) for the initial state as the previous qubit tutorial. In a photonic CV system, this state is the vacuum state, i.e., the state with no photons in the wire.
  2. We displace the qumode. The displacement gate linearly shifts the state of the qumode in phase space. The vacuum state is centered at the origin in phase space, while the displaced state will be centered at the point \(\alpha\).
  3. We rotate the qumode. This is another linear transformation in phase space, albeit a rotation (by angle \(\phi\)) instead of a displacement.
  4. Finally, we measure the mean photon number \(\langle\hat{n}\rangle = \langle\ad\a\rangle\). This quantity, which tells us the average amount of photons in the final state, is proportional to the energy of the photonic system.

The aim of this tutorial is to optimize the circuit parameters \((\alpha, \phi)\) such that the mean photon number is equal to one. The rotation gate is actually a passive transformation, meaning that it does not change the energy of the system. The displacement gate is an active transformation, which modifies the energy of the photonic system.

Constructing the QNode

As before, we import PennyLane, as well as the wrapped version of NumPy provided by PennyLane:

import pennylane as qml
from pennylane import numpy as np

Next, we instantiate a device which will be used to evaluate the circuit. Because our circuit contains only Gaussian operations, we can make use of the built-in default.gaussian device.

dev_gaussian = qml.device('default.gaussian', wires=1)

After initializing the device, we can construct our quantum node. As before, we use the qnode decorator to convert our quantum function (encoded by the circuit above) into a quantum node running on the default.gaussian device.

@qml.qnode(dev_gaussian)
def mean_photon_gaussian(mag_alpha, phase_alpha, phi):
    qml.Displacement(mag_alpha, phase_alpha, wires=0)
    qml.Rotation(phi, wires=0)
    return qml.expval.MeanPhoton(0)

Notice that we have broken up the complex number \(\alpha\) into two real numbers mag_alpha and phase_alpha, which form a polar representation of \(\alpha\). This is so that the notion of a gradient is clear and well-defined.

Optimization

As in the qubit rotation tutorial, let’s now use one of the built-in PennyLane optimizers in order to optimize the quantum circuit towards the desired output. We want the mean photon number to be exactly one, so we will use a squared-difference cost function:

def cost(params):
    return (mean_photon_gaussian(params[0], params[1], params[2]) - 1.) ** 2

At the beginning of the optimization, we choose arbitrary small initial parameters:

>>> init_params = [0.015, 0.02, 0.005]
>>> cost(init_params)
0.9995500506249999

When the gate parameters are near to zero, the gates are close to the identity transformation, which leaves the initial state largely unchanged. Since the initial state contains no photons, the mean photon number of the circuit output is approximately zero, and the cost is close to one.

Note

We avoided initial parameters which are exactly zero because that corresponds to a critical point with zero gradient.

Now, let’s use the GradientDescentOptimizer, and update the circuit parameters over 100 optimization steps.

# initialise the optimizer
opt = qml.GradientDescentOptimizer(stepsize=0.1)

# set the number of steps
steps = 20
# set the initial parameter values
params = init_params

for i in range(steps):
    # update the circuit parameters
    params = opt.step(cost, params)

    print('Cost after step {:5d}: {:8f}'.format(i+1, cost(params)) )

print('Optimized mag_alpha:{:8f}'.format(params[0]))
print('Optimized phase_alpha:{:8f}'.format(params[1]))
print('Optimized phi:{:8f}'.format(params[2]))

Try this yourself — the optimization should converge after about 20 steps to a cost function value of zero, corresponding to the following final values for the parameters:

Optimized mag_alpha: 0.999994
Optimized phase_alpha: 0.020000
Optimized phi: 0.00500000

We observe that the two angular parameters phase_alpha and phi do not change during the optimization. Only the magnitude of the complex displacement \(|\alpha|\) affects the mean photon number of the circuit.

Continue on to the next tutorial, Plugins and Hybrid computation, to learn how to utilize the extensive plugin ecosystem of PennyLane, build continuous-variable (CV) quantum nodes, and to see an example of a hybrid qubit-CV-classical computation using PennyLane.