Drag analysis

qiskit_experiments/curve_analysis/curve_analysis.py

@@ -515,6 +515,17 @@ def _extra_database_entry(self, fit_data: FitData) -> List[AnalysisResultData]:
         """
         return []
 
+    def _post_process_fit_result(self, fit_result: FitData) -> FitData:
+        """A hook that sub-classes can override to manipulate the result of the fit.
+
+        Args:
+            fit_result: A result from the fitting.
+
+        Returns:
+            A fit result that might be post-processed.
+        """
+        return fit_result
+
     # pylint: disable=unused-argument
     def _evaluate_quality(self, fit_data: FitData) -> Union[str, None]:
         """Evaluate quality of the fit result.
@@ -857,6 +868,7 @@ def _run_analysis(
             fit_result = None
         else:
             fit_result = sorted(fit_results, key=lambda r: r.reduced_chisq)[0]
+            fit_result = self._post_process_fit_result(fit_result)
 
         #
         # 5. Create database entry

qiskit_experiments/library/characterization/analysis/drag_analysis.py

@@ -25,27 +25,31 @@ class DragCalAnalysis(curve.CurveAnalysis):
 
     # section: fit_model
 
-        Analyse a Drag calibration experiment by fitting three series each to a cosine function.
-        The three functions share the phase parameter (i.e. beta) but each have their own amplitude,
-        baseline, and frequency parameters (which therefore depend on the number of repetitions of
-        xp-xm). Several initial guesses are tried if the user does not provide one.
+        Analyse a Drag calibration experiment by fitting three series each to a cosine
+        function. The three functions share the phase parameter (i.e. beta), amplitude, and
+        baseline. The frequencies of the oscillations are related through the number of
+        repetitions of the Drag gates. Several initial guesses are tried if the user
+        does not provide one. The fit function is
 
         .. math::
 
-            y_i = {\rm amp} \cos\left(2 \pi\cdot {\rm freq}_i\cdot x -
-            2 \pi\cdot {\rm freq}_i\cdot \beta\right) + {\rm base}
+            y_i = {\rm amp} \cos\left(2 \pi\cdot {\rm reps}_i \cdot {\rm freq}\cdot x -
+            2 \pi\cdot {\rm reps}_i \cdot {\rm freq}\cdot \beta\right) + {\rm base}
 
-        Note that the aim of the Drag calibration is to find the :math:`\beta` that minimizes the
-        phase shifts. This implies that the optimal :math:`\beta` occurs when all three :math:`y`
-        curves are minimum, i.e. they produce the ground state. Therefore,
+        Here, the fit parameter :math:`freq` is the frequency of the oscillation of a
+        single pair of Drag plus and minus rotations and :math:`{\rm reps}_i` is the number
+        of times that the Drag plus and minus rotations are repeated in curve :math:`i`.
+        Note that the aim of the Drag calibration is to find the :math:`\beta` that
+        minimizes the phase shifts. This implies that the optimal :math:`\beta` occurs when
+        all three :math:`y` curves are minimum, i.e. they produce the ground state. Therefore,
 
         .. math::
 
             y_i = 0 \quad \Longrightarrow \quad -{\rm amp} \cos(2 \pi\cdot X_i) = {\rm base}
 
-        Here, we abbreviated :math:`{\rm freq}_i\cdot x - {\rm freq}_i\cdot \beta` by :math:`X_i`.
-        For a signal between 0 and 1 the :math:`{\rm base}` will typically fit to 0.5. However, the
-        equation has an ambiguity if the amplitude is not properly bounded. Indeed,
+        Here, we abbreviated :math:`{\rm reps}_i\cdot{\rm freq}(x - \beta)` by :math:`X_i`.
+        For a signal between 0 and 1 the :math:`{\rm base}` will typically fit to 0.5. However,
+        the equation has an ambiguity if the amplitude is not properly bounded. Indeed,
 
         - if :math:`{\rm amp} < 0` then we require :math:`2 \pi\cdot X_i = 0` mod :math:`2\pi`, and
         - if :math:`{\rm amp} > 0` then we require :math:`2 \pi\cdot X_i = \pi` mod :math:`2\pi`.
@@ -56,17 +60,19 @@ class DragCalAnalysis(curve.CurveAnalysis):
     # section: fit_parameters
         defpar \rm amp:
             desc: Amplitude of all series.
-            init_guess: The maximum y value less the minimum y value. 0.5 is also tried.
-            bounds: [-2, 2] scaled to the maximum signal value.
+            init_guess: The maximum y value scaled by -1, -0.5, and -0.25.
+            bounds: [-2, 0] scaled to the maximum signal value.
 
         defpar \rm base:
             desc: Base line of all series.
-            init_guess: The average of the data. 0.5 is also tried.
-            bounds: [-1, 1] scaled to the maximum signal value.
-
-        defpar {\rm freq}_i:
-            desc: Frequency of the :math:`i` th oscillation.
-            init_guess: The frequency with the highest power spectral density.
+            init_guess: Half the maximum y-value of the data.
+            bounds: [-1, 1] scaled to the maximum y-value.
+
+        defpar {\rm freq}:
+            desc: Frequency of oscillation as a function of :math:`\beta` for a single pair
+                of DRAG plus and minus pulses.
+            init_guess: For the curve with the most Drag pulse repetitions, the peak frequency
+                of the power spectral density is found and then divided by the number of repetitions.
             bounds: [0, inf].
 
         defpar \beta:
@@ -77,8 +83,8 @@ class DragCalAnalysis(curve.CurveAnalysis):
 
     __series__ = [
         curve.SeriesDef(
-            fit_func=lambda x, amp, freq0, freq1, freq2, beta, base: cos(
-                x, amp=amp, freq=freq0, phase=-2 * np.pi * freq0 * beta, baseline=base
+            fit_func=lambda x, amp, freq, reps0, reps1, reps2, beta, base: cos(
+                x, amp=amp, freq=reps0 * freq, phase=-2 * np.pi * reps0 * freq * beta, baseline=base
             ),
             plot_color="blue",
             name="series-0",
@@ -88,8 +94,8 @@ class DragCalAnalysis(curve.CurveAnalysis):
             r"- 2 \pi\cdot {\rm freq}_0\cdot \beta\right) + {\rm base}",
         ),
         curve.SeriesDef(
-            fit_func=lambda x, amp, freq0, freq1, freq2, beta, base: cos(
-                x, amp=amp, freq=freq1, phase=-2 * np.pi * freq1 * beta, baseline=base
+            fit_func=lambda x, amp, freq, reps0, reps1, reps2, beta, base: cos(
+                x, amp=amp, freq=reps1 * freq, phase=-2 * np.pi * reps1 * freq * beta, baseline=base
             ),
             plot_color="green",
             name="series-1",
@@ -99,8 +105,8 @@ class DragCalAnalysis(curve.CurveAnalysis):
             r"- 2 \pi\cdot {\rm freq}_1\cdot \beta\right) + {\rm base}",
         ),
         curve.SeriesDef(
-            fit_func=lambda x, amp, freq0, freq1, freq2, beta, base: cos(
-                x, amp=amp, freq=freq2, phase=-2 * np.pi * freq2 * beta, baseline=base
+            fit_func=lambda x, amp, freq, reps0, reps1, reps2, beta, base: cos(
+                x, amp=amp, freq=reps2 * freq, phase=-2 * np.pi * reps2 * freq * beta, baseline=base
             ),
             plot_color="red",
             name="series-2",
@@ -117,11 +123,19 @@ def _default_options(cls):
 
         See :meth:`~qiskit_experiment.curve_analysis.CurveAnalysis._default_options` for
         descriptions of analysis options.
+
+        Analysis Options:
+            beta_integers_check (Sequence[int]): A sequence of integers that is used to
+                ensure that we return the beta closest to zero. See the method
+                :meth:`_post_process_fit_result`.
         """
         default_options = super()._default_options()
         default_options.result_parameters = ["beta"]
         default_options.xlabel = "Beta"
         default_options.ylabel = "Signal (arb. units)"
+        default_options.fixed_parameters = {"reps0": 1, "reps1": 3, "reps2": 5}
+        default_options.normalization = True
+        default_options.beta_integers_check = [-5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5]
 
         return default_options
 
@@ -140,39 +154,70 @@ def _generate_fit_guesses(
         x_data = self._data("series-0").x
         min_beta, max_beta = min(x_data), max(x_data)
 
-        freqs_guesses = {}
-        for i in range(3):
-            curve_data = self._data(f"series-{i}")
-            freqs_guesses[f"freq{i}"] = curve.guess.frequency(curve_data.x, curve_data.y)
-        user_opt.p0.set_if_empty(**freqs_guesses)
+        # Use the highest-frequency curve to estimate the oscillation frequency.
+        curve_data = self._data("series-2")
+        reps2 = self.options.fixed_parameters["reps2"]
+        freqs_guess = curve.guess.frequency(curve_data.x, curve_data.y) / reps2
+        user_opt.p0.set_if_empty(freq=freqs_guess)
 
-        max_abs_y, _ = curve.guess.max_height(self._data().y, absolute=True)
-        freq_bound = max(10 / user_opt.p0["freq0"], max(x_data))
+        avg_x = (max(x_data) + min(x_data)) / 2
+        span_x = max(x_data) - min(x_data)
+        beta_bound = max(5 / user_opt.p0["freq"], span_x)
 
+        ptp_y = np.ptp(self._data().y)
         user_opt.bounds.set_if_empty(
-            amp=(-2 * max_abs_y, 0),
-            freq0=(0, np.inf),
-            freq1=(0, np.inf),
-            freq2=(0, np.inf),
-            beta=(-freq_bound, freq_bound),
-            base=(-max_abs_y, max_abs_y),
+            amp=(-2 * ptp_y, 0),
+            freq=(0, np.inf),
+            beta=(avg_x - beta_bound, avg_x + beta_bound),
+            base=(-ptp_y, ptp_y),
         )
-        user_opt.p0.set_if_empty(base=0.5)
+        base_guess = (max(self._data().y) - min(self._data().y)) / 2
+        user_opt.p0.set_if_empty(base=(user_opt.p0["amp"] or base_guess))
 
         # Drag curves can sometimes be very flat, i.e. averages of y-data
         # and min-max do not always make good initial guesses. We therefore add
         # 0.5 to the initial guesses. Note that we also set amp=-0.5 because the cosine function
         # becomes +1 at zero phase, i.e. optimal beta, in which y data should become zero
         # in discriminated measurement level.
         options = []
-        for amp_guess in (0.5, -0.5):
+        for amp_factor in (-1, -0.5, -0.25):
             for beta_guess in np.linspace(min_beta, max_beta, 20):
                 new_opt = user_opt.copy()
-                new_opt.p0.set_if_empty(amp=amp_guess, beta=beta_guess)
+                new_opt.p0.set_if_empty(amp=ptp_y * amp_factor, beta=beta_guess)
                 options.append(new_opt)
 
         return options
 
+    def _post_process_fit_result(self, fit_result: curve.FitData) -> curve.FitData:
+        r"""Post-process the fit result from a Drag analysis.
+
+        The Drag analysis should return the beta value that is closest to zero.
+        Since the oscillating term is of the form
+
+        .. math::
+
+            \cos(2 \pi\cdot {\rm reps}_i \cdot {\rm freq}\cdot [x - \beta])
+
+        There is a periodicity in beta. This post processing finds the beta that is
+        closest to zero by performing the minimization
+
+        .. math::
+
+            n_\text{min} = \min_{n}|\beta_\text{fit} + n / {\rm freq}|
+
+        and assigning the new beta value to
+
+        .. math::
+
+            \beta = \beta_\text{fit} + n_\text{min} / {\rm freq}.
+        """
+        beta = fit_result.popt[2]
+        freq = fit_result.popt[1]
+        betas = [abs(beta + n / freq) for n in self.options.beta_integers_check]
+        n_min = self.options.beta_integers_check[np.argmin(betas)]
+        fit_result.popt[2] = beta + n_min / freq
+        return fit_result
+
     def _evaluate_quality(self, fit_data: curve.FitData) -> Union[str, None]:
         """Algorithmic criteria for whether the fit is good or bad.
 
@@ -182,11 +227,11 @@ def _evaluate_quality(self, fit_data: curve.FitData) -> Union[str, None]:
             - an error on the drag beta smaller than the beta.
         """
         fit_beta = fit_data.fitval("beta")
-        fit_freq0 = fit_data.fitval("freq0")
+        fit_freq = fit_data.fitval("freq")
 
         criteria = [
             fit_data.reduced_chisq < 3,
-            fit_beta.nominal_value < 1 / fit_freq0.nominal_value,
+            abs(fit_beta.nominal_value) < 1 / fit_freq.nominal_value / 2,
             curve.is_error_not_significant(fit_beta),
         ]
 

qiskit_experiments/library/characterization/drag.py

@@ -92,14 +92,6 @@ def _default_experiment_options(cls) -> Options:
 
         return options
 
-    @classmethod
-    def _default_analysis_options(cls) -> Options:
-        """Default analysis options."""
-        options = Options()
-        options.normalization = True
-
-        return options
-
     # pylint: disable=arguments-differ
     def set_experiment_options(self, reps: Optional[List] = None, **fields):
         """Raise if reps has a length different from three.
@@ -121,6 +113,11 @@ def set_experiment_options(self, reps: Optional[List] = None, **fields):
 
         super().set_experiment_options(reps=reps, **fields)
 
+        if isinstance(self.analysis, DragCalAnalysis):
+            self.analysis.set_options(
+                fixed_parameters={"reps0": reps[0], "reps1": reps[1], "reps2": reps[2]}
+            )
+
     def __init__(
         self,
         qubit: int,
@@ -143,7 +140,6 @@ def __init__(
         """
 
         super().__init__([qubit], analysis=DragCalAnalysis(), backend=backend)
-        self.analysis.set_options(**self._default_analysis_options.__dict__)
 
         if betas is not None:
             self.set_experiment_options(betas=betas)

qiskit_experiments/test/mock_iq_backend.py

@@ -291,25 +291,36 @@ def __init__(
         self,
         iq_cluster_centers: Tuple[float, float, float, float] = (1.0, 1.0, -1.0, -1.0),
         iq_cluster_width: float = 1.0,
-        error: float = 0.03,
+        freq: float = 0.005,
         ideal_beta=2.0,
         gate_name: str = "Rp",
         rng_seed: int = 0,
+        max_prob: float = 1.0,
+        offset_prob: float = 0.0,
     ):
         """Initialize the rabi backend."""
-        self._error = error
+        self._freq = freq
         self._gate_name = gate_name
         self.ideal_beta = ideal_beta
 
+        if max_prob + offset_prob > 1:
+            raise ValueError("Probabilities need to be between 0 and 1.")
+
+        self._max_prob = max_prob
+        self._offset_prob = offset_prob
+
         super().__init__(iq_cluster_centers, iq_cluster_width, rng_seed=rng_seed)
 
     def _compute_probability(self, circuit: QuantumCircuit) -> float:
         """Returns the probability based on the beta, number of gates, and leakage."""
-        n_gates = sum(circuit.count_ops().values())
+        n_gates = circuit.count_ops()[self._gate_name]
 
         beta = next(iter(circuit.calibrations[self._gate_name].keys()))[1][0]
 
-        return np.sin(n_gates * self._error * (beta - self.ideal_beta)) ** 2
+        prob = np.sin(2 * np.pi * n_gates * self._freq * (beta - self.ideal_beta)) ** 2
+        rescaled_prob = self._max_prob * prob + self._offset_prob
+
+        return rescaled_prob
 
 
 class RabiBackend(MockIQBackend):