Abstract:We propose a general framework for decoding quantum error-correcting codes with generative modeling. The model utilizes autoregressive neural networks, specifically Transformers, to learn the joint probability of logical operators and syndromes. This training is in an unsupervised way, without the need for labeled training data, and is thus referred to as pre-training. After the pre-training, the model can efficiently compute the likelihood of logical operators for any given syndrome, using maximum likelihood decoding. It can directly generate the most-likely logical operators with computational complexity $\mathcal O(2k)$ in the number of logical qubits $k$, which is significantly better than the conventional maximum likelihood decoding algorithms that require $\mathcal O(4^k)$ computation. Based on the pre-trained model, we further propose refinement to achieve more accurately the likelihood of logical operators for a given syndrome by directly sampling the stabilizer operators. We perform numerical experiments on stabilizer codes with small code distances, using both depolarizing error models and error models with correlated noise. The results show that our approach provides significantly better decoding accuracy than the minimum weight perfect matching and belief-propagation-based algorithms. Our framework is general and can be applied to any error model and quantum codes with different topologies such as surface codes and quantum LDPC codes. Furthermore, it leverages the parallelization capabilities of GPUs, enabling simultaneous decoding of a large number of syndromes. Our approach sheds light on the efficient and accurate decoding of quantum error-correcting codes using generative artificial intelligence and modern computational power.

General framework for constructing fast and near-optimal machine-learning-based decoder of the topological stabilizer codes

Scalable Neural Decoder for Topological Surface Codes

A Practical and Scalable Decoder for Topological Quantum Error Correction with Digital Annealer

Conservation laws and quantum error correction: towards a generalised matching decoder

Equivariant Machine Learning Decoder for 3D Toric Codes

Low-overhead quantum error correction codes with a cyclic topology

Decoding topological XYZ codes with reinforcement learning based on attention mechanisms

Data-driven decoding of quantum error correcting codes using graph neural networks

Ising model formulation for highly accurate topological color codes decoding

The END: An Equivariant Neural Decoder for Quantum Error Correction

qecGPT: decoding Quantum Error-correcting Codes with Generative Pre-trained Transformers

Advantage of Quantum Neural Networks as Quantum Information Decoders

Almost Linear Decoder for Optimal Geometrically Local Quantum Codes

Quantum Circuits for Stabilizer Error Correcting Codes: A Tutorial

Degenerate quantum erasure decoding

Decoding general error correcting codes and the role of complementarity

Encoders and Decoders for Quantum Expander Codes Using Machine Learning

Trellis Decoding for Qudit Stabilizer Codes and Its Application to Qubit Topological Codes

Rescaling decoder for 2D topological quantum color codes on 4.8.8 lattices

Quantum Convolutional Codes Based on the Stabilizer of Linear Codes