Cost Functions for QML Tasks

In the Variational Quantum Algorithm (VQA) framework, the parameterized quantum circuit (PQC) prepares a quantum state $|\psi(\boldsymbol{\theta})\rangle$, where $\boldsymbol{\theta}$ represents the classical parameters we aim to optimize. However, to guide this optimization, we need a way to quantify how well the state $|\psi(\boldsymbol{\theta})\rangle$ achieves the target machine learning objective. This is the role of the cost function, $C(\boldsymbol{\theta})$. It acts as the essential bridge between the quantum processing unit (QPU) and the classical optimization routine, translating quantum measurement outcomes into a scalar value that indicates performance.

From Quantum Measurements to Classical Costs

At the heart of a VQA's evaluation step lies quantum measurement. Typically, we measure one or more qubits in a specific basis (often the computational basis, $Z$-basis) after the state $|\psi(\boldsymbol{\theta})\rangle$ has been prepared. This measurement process is inherently probabilistic, yielding classical bitstrings as outcomes.

To form a differentiable cost function suitable for optimization, we usually don't work directly with the raw probabilities of these bitstrings. Instead, we compute the expectation value of a chosen observable, which is a Hermitian operator $\hat{O}$. For a given state $|\psi(\boldsymbol{\theta})\rangle$ prepared by the PQC with parameters $\boldsymbol{\theta}$, the expectation value is calculated as:

$$\langle \hat{O} \rangle_{\boldsymbol{\theta}} = \langle \psi(\boldsymbol{\theta}) | \hat{O} | \psi(\boldsymbol{\theta}) \rangle$$

This expectation value $\langle \hat{O} \rangle_{\boldsymbol{\theta}}$ provides a smooth, real-valued output that depends on the circuit parameters $\boldsymbol{\theta}$. It represents the average value we would obtain if we measured the observable $\hat{O}$ on the state $|\psi(\boldsymbol{\theta})\rangle$ many times. This expectation value, or a function derived from it and the target data labels, forms the basis of our cost function $C(\boldsymbol{\theta})$.

The overall process can be visualized as follows:

The VQA loop involves preparing a state with the PQC using current parameters $\boldsymbol{\theta}$, measuring an observable $\hat{O}$ to estimate its expectation value $\langle \hat{O} \rangle_{\boldsymbol{\theta}}$, calculating the cost $C(\boldsymbol{\theta})$ by comparing this value to the target, and feeding the cost back to a classical optimizer to propose updated parameters.

Designing Cost Functions for Specific ML Tasks

The exact form of $C(\boldsymbol{\theta})$ depends significantly on the machine learning task at hand. Let's look at common scenarios:

Classification

In classification tasks, the goal is to assign input data $x_i$ to one of several discrete categories.

• Binary Classification: For problems with two classes (e.g., labeled +1 and -1), a common approach is to design the PQC and observable $\hat{O}$ such that the expectation value $\langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)}$ (computed for input $x_i$) correlates with the class label $y_i$. For instance, we might measure the Pauli Z operator $\hat{Z}_0$ on the first qubit, aiming for $\langle \hat{Z}_0 \rangle_{\boldsymbol{\theta}}^{(i)} \approx +1$ for one class and $\langle \hat{Z}_0 \rangle_{\boldsymbol{\theta}}^{(i)} \approx -1$ for the other.

  • Mean Squared Error (MSE): A straightforward cost function is the MSE between the predicted expectation value and the target label: $$C_{\text{MSE}}(\boldsymbol{\theta}) = \frac{1}{N} \sum_{i=1}^{N} (\langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)} - y_i)^2$$ Here, $N$ is the number of training samples. While simple, MSE penalizes deviations quadratically, which might not always be the most effective loss for classification.
  • Hinge Loss: Inspired by Support Vector Machines (SVMs), the hinge loss encourages the expectation value to be correctly signed and beyond a certain margin: $$C_{\text{Hinge}}(\boldsymbol{\theta}) = \frac{1}{N} \sum_{i=1}^{N} \max(0, 1 - y_i \langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)})$$ This loss is zero if the prediction $\langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)}$ has the correct sign ($y_i \langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)} \ge 1$) and increases linearly otherwise.
  • Cross-Entropy Loss: This loss is common in classical ML for probabilistic classifiers. It can be adapted if we interpret measurement outcomes as probabilities. For example, if we measure qubit 0 and estimate the probability $p(+1|\boldsymbol{\theta}, x_i)$ of measuring $|0\rangle$ (associated with label $y_i=1$) and $p(-1|\boldsymbol{\theta}, x_i)$ of measuring $|1\rangle$ (associated with label $y_i=0$), the binary cross-entropy is: $$C_{\text{XEnt}}(\boldsymbol{\theta}) = -\frac{1}{N} \sum_{i=1}^{N} [y_i \log p(+1|\boldsymbol{\theta}, x_i) + (1-y_i) \log p(-1|\boldsymbol{\theta}, x_i)]$$ Using cross-entropy requires careful estimation of probabilities from measurement counts, which can be sample-intensive. It often requires mapping the binary labels $\{0, 1\}$ or $\{-1, +1\}$ appropriately to the probabilities derived from specific measurement outcomes.

• Multi-class Classification: Extending these ideas to more than two classes typically involves using multiple output qubits, different measurement strategies (e.g., measuring multiple Pauli operators), or combining binary classifiers. The cost function needs to be adapted accordingly, often using multi-class versions of MSE or cross-entropy.

Regression

In regression, the goal is to predict a continuous value $y_i$ for an input $x_i$.

• The expectation value $\langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)}$ itself can serve as the predicted continuous output. The scaling and range of the observable $\hat{O}$ should ideally match the expected range of the target values $y_i$, or the output needs to be rescaled.
• Mean Squared Error (MSE): This is the most common cost function for regression in VQAs: $$C_{\text{MSE}}(\boldsymbol{\theta}) = \frac{1}{N} \sum_{i=1}^{N} (\langle \hat{O} \rangle_{\boldsymbol{\theta}}^{(i)} - y_i)^2$$ It directly penalizes the squared difference between the quantum model's prediction and the true continuous value.

Generative Modeling

Cost functions for generative tasks, such as learning probability distributions with Quantum Circuit Born Machines (QCBMs) or training Quantum Generative Adversarial Networks (QGANs), are distinct. They often involve metrics that compare the probability distribution produced by the quantum circuit to the target data distribution (e.g., Maximum Mean Discrepancy, Kullback-Leibler divergence estimates) or adversarial losses derived from a discriminator network. These will be explored in more detail in Chapter 6.

Choosing the Right Observable

The choice of the observable $\hat{O}$ is not arbitrary; it's an integral part of the VQA design. It determines precisely what property of the final quantum state $|\psi(\boldsymbol{\theta})\rangle$ is extracted to make the prediction. Common choices include:

• Single-Qubit Pauli Operators: $\hat{Z}_k$, $\hat{X}_k$, or $\hat{Y}_k$ acting on a specific qubit $k$. Measuring $\hat{Z}_k$ is natural as it corresponds to measurement in the computational basis.
• Multi-Qubit Pauli Strings: Tensor products like $\hat{Z}_0 \otimes \hat{Z}_1$ or $\hat{X}_0 \otimes \hat{Y}_1$. These capture correlations between qubits.
• Weighted Sums of Pauli Strings (Hamiltonians): More complex observables, often motivated by the problem structure, like $\hat{H} = \sum_k c_k \hat{P}_k$ where $\hat{P}_k$ are Pauli strings and $c_k$ are coefficients.

The observable should be chosen based on how the PQC encodes information. If the final answer is expected to be encoded in the polarization of a specific qubit, measuring a Pauli operator on that qubit makes sense. If correlations are important, a multi-qubit observable might be necessary. The range of the expectation value (e.g., $[-1, +1]$ for Pauli operators) should also be considered when relating it to target labels or values.

Practical Estimation of Expectation Values

It's important to remember that on any real quantum computer (or simulator mimicking one), we cannot access the exact expectation value $\langle \hat{O} \rangle_{\boldsymbol{\theta}}$. Instead, we estimate it by preparing the state $|\psi(\boldsymbol{\theta})\rangle$ and measuring the observable $\hat{O}$ repeatedly, say $N_{\text{shots}}$ times.

• If $\hat{O}$ is diagonal in the measurement basis (like $\hat{Z}$ in the computational basis), we count the frequencies of the outcomes corresponding to its eigenvalues.
• If $\hat{O}$ is not diagonal (like $\hat{X}$ or $\hat{Y}$), we need to apply appropriate basis change gates before measurement or decompose $\hat{O}$ into a sum of simpler observables (like Pauli strings) that can be measured individually.

The accuracy of this estimation depends on the number of shots, with the standard deviation of the estimate typically scaling as $1/\sqrt{N_{\text{shots}}}$. This inherent statistical noise, known as "shot noise," means the cost function evaluation itself is noisy, adding a layer of challenge to the optimization process compared to classical ML where function evaluations are typically deterministic.

Impact on Optimization

The cost function $C(\boldsymbol{\theta})$ defines the optimization surface that the classical optimizer explores to find the best parameters $\boldsymbol{\theta}$. The structure of this surface, influenced by the PQC architecture, the data encoding, the choice of observable, and the specific cost function formula, determines the feasibility and efficiency of training the VQA. Issues like the presence of many local minima or the phenomenon of barren plateaus (regions where gradients vanish exponentially with the number of qubits) are directly related to the properties of $C(\boldsymbol{\theta})$ and its gradients. Understanding how to define effective cost functions is therefore fundamental to building successful VQAs for machine learning. The subsequent sections on gradient calculation and optimization techniques will build directly upon this foundation.