Quantum optimal control of superconducting qubits based on machine-learning characterization

Elie Genois,Noah J. Stevenson,Noah Goss,Irfan Siddiqi,Alexandre Blais

2024-10-30

Abstract:Implementing fast and high-fidelity quantum operations using open-loop quantum optimal control relies on having an accurate model of the quantum dynamics. Any deviations between this model and the complete dynamics of the device, such as the presence of spurious modes or pulse distortions, can degrade the performance of optimal controls in practice. Here, we propose an experimentally simple approach to realize optimal quantum controls tailored to the device parameters and environment while specifically characterizing this quantum system. Concretely, we use physics-inspired machine learning to infer an accurate model of the dynamics from experimentally available data and then optimize our experimental controls on this trained model. We show the power and feasibility of this approach by optimizing arbitrary single-qubit operations on a superconducting transmon qubit, using detailed numerical simulations. We demonstrate that this framework produces an accurate description of the device dynamics under arbitrary controls, together with the precise pulses achieving arbitrary single-qubit gates with a high fidelity of about 99.99%.

Quantum Physics

What problem does this paper attempt to address?

The problem that this paper attempts to solve is: when implementing fast and high - fidelity quantum operations in superconducting qubits, how to overcome the performance degradation problem caused by model deviation. Specifically, the paper proposes a quantum optimal control method based on machine - learning characterization to deal with the differences between the dynamic model in actual devices and the theoretical model.

### Problem Background

1. **Impact of Model Deviation**:
- In open - loop quantum optimal control (QOC), an accurate dynamic model is crucial for achieving high - fidelity quantum operations.
- The dynamics in actual devices often include unmodeled modes, pulse distortion, etc., which will lead to model deviation and thus reduce the performance of optimal control.

2. **Limitations of Existing Methods**:
- Traditional QOC methods (such as GRAPE and Krotov) rely on accurate system - dynamic models, but it is very difficult to obtain a completely accurate model in actual experiments.
- Model - free reinforcement learning (RL) methods can avoid model deviation, but they have challenges in sample efficiency, generalization, and experimental implementation.

### Solution Proposed in the Paper

The paper proposes a method that combines supervised machine learning and gradient optimization. The specific steps are as follows:

1. **Data - Driven Model Training**:
- Use supervised machine learning to learn a model that describes the dynamics of the quantum system from experimental data. By applying a variety of different pulse sequences and measuring the system's response, a training data set is constructed.
- This model can capture all kinds of complex dynamics actually existing in the experiment, including unmodeled modes and pulse distortion.

2. **Optimal Control Based on the Trained Model**:
- Use the trained model for gradient optimization to find the best pulse sequence to achieve the target quantum operation.
- This method not only improves the fidelity of control but also has high data efficiency and scalability.

### Specific Implementation

- **Simulation Data Generation**:
- Generate a training data set through numerical simulation to simulate the dynamic behavior of superconducting transmon qubits.
- The data set contains six basis states on the Bloch sphere, pulses of different shapes, and corresponding measurement results.

- **Model Architecture**:
- Use a recurrent neural network (RNN) with long - short - term memory (LSTM) units to process time - series data.
- The input of the model is the pulse shape, and the output is the expected value of the predicted quantum state.

- **Loss Function Design**:
- Mainly use the mean - square error (MSE) as the loss function to ensure that the model prediction is as close as possible to the actual measurement result.
- Add physical constraint terms to ensure that the model output is a valid quantum state.

### Result Verification

- **High - Fidelity Single - Qubit Gate Operation**:
- Through detailed numerical simulation, it is shown that this method can achieve any single - qubit gate operation on superconducting transmon qubits with a fidelity of about 99.99%.
- It proves the superiority of this framework in dealing with model deviation and complex control problems.

In conclusion, this paper aims to provide an efficient and robust method by combining machine learning and quantum optimal control to overcome the model deviation problem in actual quantum devices and thus achieve high - fidelity quantum operations.