This notebook demonstrates some convenience functions for coming up with the experiments needed to do GST on a new gateset.
import pygsti
import pygsti.algorithms.germselection as germsel
import pygsti.algorithms.fiducialselection as fidsel
import pygsti.construction as constr
Let's construct a 1-qubit $X(\pi/2)$, $Y(\pi/2)$, $I$ gateset for which we will need to find germs and fiducials.
gs_target = constr.build_gateset([2], [('Q0',)], ['Gi', 'Gx', 'Gy'],
["I(Q0)", "X(pi/2,Q0)", "Y(pi/2,Q0)"],
prepLabels=['rho0'], prepExpressions=["0"],
effectLabels=['E0'], effectExpressions=["1"],
spamdefs={'plus': ('rho0', 'E0'),
'minus': ('rho0', 'remainder')})
We begin by demonstrating the most hands-off approach.
We can generate a germ set simply by providing the target gateset.
germs = germsel.generate_germs(gs_target)
Using greedy algorithm. Constructed germ set: ['Gi', 'Gx', 'Gy', 'GxGy', 'GiGiGiGiGy', 'GiGiGiGyGiGx', 'GiGiGiGxGxGy', 'GiGiGyGyGx', 'GxGxGyGxGyGy', 'GiGiGxGyGy'] Score: 128.25897662711938
In the same way we can generate preparation and measurement fiducials.
prepFiducials, measFiducials = fidsel.generate_fiducials(gs_target)
Using GRASP algorithm. Preparation fiducials: ['{}', 'Gx', 'Gy', 'GxGx'] Score: 31.999999999999986 Measurement fiducials: ['{}', 'Gx', 'Gy', 'GxGx'] Score: 31.999999999999996
Now that we have germs and fiducials, we can construct the list of experiments we need to perform in order to do GST. The only new things to provide at this point are the sizes for the experiments we want to perform (in this case we want to perform between 0 and 256 gates between fiducial pairs, going up by a factor of 2 at each stage).
maxLengths = [0] + [2**n for n in range(8 + 1)]
listOfExperiments = constr.make_lsgst_experiment_list(gs_target.gates.keys(), prepFiducials,
measFiducials, germs, maxLengths)
The list of GateString
that the previous function gave us isn't necessarily the most readable
form to present the information in, so we can write the experiment list out to an empty data
file to be filled in after the experiments are performed.
pygsti.io.write_empty_dataset("tutorial_files/EasyDataTemplate.txt", listOfExperiments)
That's it! You can now take the data this file is asking for and come back to analyze the data.
There are many ways you can assume more control over the experiment design process. We'll only demonstrate a few here, but all options are discussed in the documentation for the various functions we've used.
There are a number of different algorithms available for germ selection. You can choose a non-default
algorithm by specifying the algorithm
keyword argument.
Each of the available algorithms has a set of keyword arguments of its own with which you
can more precisely specify how you want it to behave. These keyword arguments can be passed
as a dictionary to generate_germs
through the keyword argument algorithm_kwargs
.
graspGerms = germsel.generate_germs(gs_target, algorithm='grasp', algorithm_kwargs={'iterations': 1})
Using GRASP algorithm. Constructed germ set: ['Gi', 'Gx', 'Gy', 'GxGyGy', 'GiGiGyGxGx', 'GiGiGiGiGiGx', 'GiGiGiGiGiGy', 'GiGiGiGxGxGy', 'GiGiGiGyGxGy', 'GxGxGyGxGyGy'] Score: 118.49799767889849
Fiducial selection can be controlled in much the same way.
slackPrepFids, slackMeasFids = fidsel.generate_fiducials(gs_target, algorithm='slack',
algorithm_kwargs={'slackFrac': 0.25})
Using slack algorithm. Preparation fiducials: ['{}', 'GxGy', 'GyGx', 'GyGy'] Score: 32.00000000000001 Measurement fiducials: ['{}', 'GxGy', 'GyGx', 'GyGy'] Score: 32.0
We can also adjust some algorithm-independent parameters for germ and fiducial selection. For instance, all of the algorithms currently rely on having a pool of gatestring from which they construct germs and fiducials. The size of this pool is set by specifying the longest germ or fiducial to include in this pool.
For germ selection, the default maximum germ length is 6, and we can see that our original construction indeed makes use of germs of length 6.
max([len(germ) for germ in germs])
6
We can try and set the maximum germ length to 5 and see what we get.
germsMaxLength5 = germsel.generate_germs(gs_target, maxGermLength=5)
Using greedy algorithm. Constructed germ set: ['Gi', 'Gx', 'Gy', 'GiGiGiGxGy', 'GxGy', 'GiGiGiGyGx', 'GiGiGiGiGy', 'GxGyGy', 'GiGiGiGiGx', 'GiGiGyGxGy'] Score: 43996.058419589564
max([len(germ) for germ in germsMaxLength5])
5
Sure enough, we now have a germ set with a shorter longest germ. If we get too ambitious in shortening the maximum
germ length, germ selection won't be able to find an amplificationally complete germ set. It will send a warning
message to stderr
if this happens and return None
.
germsMaxLength3 = germsel.generate_germs(gs_target, maxGermLength=3)
print(germsMaxLength3)
Using greedy algorithm. None
WARNING: Complete initial germ set FAILS on gateset 0. WARNING: Aborting search.
Fiducial selection defaults to a maximum fiducial length of 2. This allows us to construct an informationally
complete set of states and measurements, but for this gateset we know that there is a uniformly
informationally complete set of states and measurements that require fiducials of up to length 3. We can find
that set of fiducials by telling generate_fiducials
to consider longer fiducials.
uniformPrepFids, uniformMeasFids = fidsel.generate_fiducials(gs_target, maxFidLength=3,
algorithm='grasp',
algorithm_kwargs={'iterations': 100})
Using GRASP algorithm. Preparation fiducials: ['{}', 'Gx', 'Gy', 'GyGy', 'GyGyGx', 'GyGyGy'] Score: 20.0 Measurement fiducials: ['{}', 'Gx', 'Gy', 'GxGx', 'GxGyGy', 'GyGyGy'] Score: 20.0
As was the case with germ selection, if you are too aggressive in limiting fiducial length you may
constrain the algorithm to the extent that it cannot even find a set of fiducials to generate an
informationally complete set of states and measurements. In that case, it will also send a warning
message to stderr
and return None
for the preparation and measurement fiducial sets.
incompletePrepFids, incompleteMeasFids = fidsel.generate_fiducials(gs_target, maxFidLength=1)
Using GRASP algorithm.
WARNING: Complete initial fiducial set FAILS. WARNING: Aborting search. WARNING: Complete initial fiducial set FAILS. WARNING: Aborting search.
print(incompleteMeasFids, incompletePrepFids)
None None
There are several natural things to require of the returned germ and fiducial sets. For germ sets, you will usually
want the individual gates to be included as germs. If for some reason you don't want this, you can set the
forceSingletons keyword argument to False
.
nonSingletonGerms = germsel.generate_germs(gs_target, force=None, maxGermLength=4,
algorithm='grasp', algorithm_kwargs={'iterations': 5})
Using GRASP algorithm. Constructed germ set: ['Gi', 'Gy', 'GxGy', 'GiGiGy', 'GiGiGiGx', 'GiGiGiGy', 'GiGiGxGy', 'GiGiGyGx', 'GiGxGiGy', 'GiGxGxGy', 'GiGxGyGx', 'GiGxGyGy', 'GiGyGxGx', 'GiGyGxGy', 'GiGyGyGx'] Score: 124360.17388048557
In fiducial selection, it is likewise natural to require the empty gate string to be in the
fiducial set. This requirement may be disabled by setting forceEmpty to False
. It is also
often desireable for identity gates to be left out of fiducials, since they add no diversity
to the set of states and measurements generated. You can allow identity gates in fiducials by
setting omitIdentity to False
.
A more common modification to the fiducial set requirements is to leave out additional gates from fiducials. This might be desireable if you have a multi-qubit system and you expect your 2-qubit gates to be of lower fidelity than your single-qubit gates. In this case you might want to construct fiducials from only single-qubit gates. A list of gates that you would like to omit from your fiducials can be provided as a list of gate labels to the gatesToOmit keyword argument.
Our gateset doesn't have multi-qubit gates, but we can demonstrate several pieces of this
functionality by setting omitIdentity to False
and omitting the identity manually using
gatesToOmit.
omitIdentityPrepFids, omitIdentityMeasFids = fidsel.generate_fiducials(gs_target, omitIdentity=False,
gatesToOmit=['Gi'])
Using GRASP algorithm. Preparation fiducials: ['{}', 'Gx', 'Gy', 'GxGx'] Score: 31.999999999999986 Measurement fiducials: ['{}', 'Gx', 'Gy', 'GxGx'] Score: 31.999999999999996
omitIdentityPrepFids
[GateString({}), GateString(Gx), GateString(Gy), GateString(GxGx)]
omitIdentityMeasFids
[GateString({}), GateString(Gx), GateString(Gy), GateString(GxGx)]
The various algorithms can tell you something of what's going on with them while they're running. By default, this output is silenced, but it can be turned on using the verbosity keyword argument.
A verbosity level of 1 is the default. This prints out what algorithm is being used, the returned set, and the score of that set.
A verbosity level of 0 silences all output (other than warnings that things have gone wrong).
A verbosity level of $n+1$ where $n\geq0$ prints the output of verbosity level 1 in addition to the output that the current algorithm displays when its own verbosity is set to $n$.
verbosePrepFids, verboseMeasFids = fidsel.generate_fiducials(gs_target, verbosity=2)
Using GRASP algorithm. Complete initial fiducial set succeeds. Now searching for best fiducial set. Starting fiducial list optimization. Lower score is better. Starting iteration 1 of 5. Initial construction: ['{}', 'GxGx', 'GyGx', 'GxGy'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx'] Finished iteration 1 of 5. Starting iteration 2 of 5. Initial construction: ['{}', 'GxGy', 'GyGx', 'GyGy'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx'] Finished iteration 2 of 5. Starting iteration 3 of 5. Initial construction: ['{}', 'GxGx', 'Gy', 'GyGx', 'GxGy'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx', 'GyGy'] Finished iteration 3 of 5. Starting iteration 4 of 5. Initial construction: ['{}', 'GyGy', 'GxGx', 'GyGx', 'Gx'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx', 'GyGy'] Finished iteration 4 of 5. Starting iteration 5 of 5. Initial construction: ['{}', 'Gx', 'Gy', 'GxGy', 'GyGx', 'GxGx'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGy', 'GyGx', 'GyGy'] Finished iteration 5 of 5. Preparation fiducials: ['{}', 'Gx', 'Gy', 'GxGx'] Score: 31.999999999999986 Complete initial fiducial set succeeds. Now searching for best fiducial set. Starting fiducial list optimization. Lower score is better. Starting iteration 1 of 5. Initial construction: ['{}', 'Gx', 'GxGy', 'GyGy'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx'] Finished iteration 1 of 5. Starting iteration 2 of 5. Initial construction: ['{}', 'GxGy', 'GxGx', 'Gx'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx'] Finished iteration 2 of 5. Starting iteration 3 of 5. Initial construction: ['{}', 'GyGy', 'GxGx', 'GxGy', 'GyGx'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx', 'GyGy'] Finished iteration 3 of 5. Starting iteration 4 of 5. Initial construction: ['{}', 'Gx', 'GyGy', 'GyGx', 'GxGx', 'Gy'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx', 'GxGy', 'GyGy'] Finished iteration 4 of 5. Starting iteration 5 of 5. Initial construction: ['{}', 'Gy', 'Gx', 'GyGy'] Local optimum: ['{}', 'Gx', 'Gy', 'GxGx'] Finished iteration 5 of 5. Measurement fiducials: ['{}', 'Gx', 'Gy', 'GxGx'] Score: 31.999999999999996
silentGerms = germsel.generate_germs(gs_target, algorithm='slack', algorithm_kwargs={'maxIter': 5},
verbosity=0)