Machine-Learning-Based Qubit Allocation for Error Reduction in Quantum Circuits
Travis LeCompte,Fang Qi,Xu Yuan,Nian-Feng Tzeng,M. Hassan Najafi,Lu Peng
DOI: https://doi.org/10.1109/tqe.2023.3301899
2023-01-01
IEEE Transactions on Quantum Engineering
Abstract:Quantum computing is a quickly growing field with great potential for future technology. Quantum computers in the current noisy intermediate-scale quantum (NISQ) era face two major limitations:1) qubit count and 2) error vulnerability. Although quantum error correction methods exist, they are not applicable to the current size of computers, requiring thousands of qubits, while current NISQ systems have hundreds at most. It is, therefore, imperative to improve the reliability of the circuits as much as possible to make them robust to the errors that will occur. One common approach is to adjust the compilation process of a circuit to create a final circuit with improved reliability. However, there are many decisions to be made when compiling that affect the final performance of the circuit, two of the most critical ones being the mapping of logical to physical qubits (the qubit allocation problem) and the movement of qubits to satisfy two-qubit gate adjacency requirements (the qubit routing problem). We focus on solving the qubit allocation problem and identifying initial layouts that reduce error. To identify these layouts, we combine reinforcement learning with a graph neural network (GNN)-based Q-network for analyzing both the connections and error rates of the graphlike backend of superconducting quantum computers to make mapping decisions, creating a GNN-assisted compilation (GNAQC) strategy. We provide both the circuit and the properties of the target backend as input to guide the decision-making process. We work with the IBM Qiskit applications programming interface to compile and simulate our quantum circuits. We train the architecture using a set of four backends and six circuits and find that GNAQC generally outperforms preexisting qubit allocation algorithms, increasing final relative output fidelity by roughly 12.7%.