Abstract:To leverage the full potential of quantum error-correcting stabilizer codes it is crucial to have an efficient and accurate decoder. Accurate, maximum likelihood, decoders are computationally very expensive whereas decoders based on more efficient algorithms give sub-optimal performance. In addition, the accuracy will depend on the quality of models and estimates of error rates for idling qubits, gates, measurements, and resets, and will typically assume symmetric error channels. In this work, instead, we explore a model-free, data-driven, approach to decoding, using a graph neural network (GNN). The decoding problem is formulated as a graph classification task in which a set of stabilizer measurements is mapped to an annotated detector graph for which the neural network predicts the most likely logical error class. We show that the GNN-based decoder can outperform a matching decoder for circuit level noise on the surface code given only simulated experimental data, even if the matching decoder is given full information of the underlying error model. Although training is computationally demanding, inference is fast and scales approximately linearly with the space-time volume of the code. We also find that we can use large, but more limited, datasets of real experimental data [Google Quantum AI, Nature {\bf 614}, 676 (2023)] for the repetition code, giving decoding accuracies that are on par with minimum weight perfect matching. The results show that a purely data-driven approach to decoding may be a viable future option for practical quantum error correction, which is competitive in terms of speed, accuracy, and versatility.

A Scalable Graph Neural Network Decoder for Short Block Codes

A Graph-Neural-Network Decoder with MLP-based Processing Cells for Polar Codes

Spiking Neural Belief Propagation Decoder for Short Block Length LDPC Codes

Decoding Quantum LDPC Codes Using Graph Neural Networks

Graph Neural Networks for Enhanced Decoding of Quantum LDPC Codes

Sparse graph neural network aided efficient decoder for polar codes under bursty interference

A Sparse Neural Network Decoder for Non-binary Polar Codes.

Parallel Decoding for Non-recursive Convolutional Codes and Its Enhancement Through Artificial Neural Networks

Learning Linear Block Error Correction Codes

Performance Evaluation of Channel Decoding with Deep Neural Networks.

Section-Wise Revolving NBP-Like Decoders for QC-LDPC Codes

A BP-NN Decoding Algorithm for Polar Codes.

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

Sparse Neural Network for Detection and Decoding of Non-Binary Polar-Coded SCMA

A denoiser for correlated noise channel decoding: Gated-neural network

A Sparsity-Adapted Hardware Implementation of SNN for Cortical Spike Trains Decoding

A blockBP decoder for the surface code

A Neural Network Empowered Belief Propagation Algorithm Optimized for Short-Cycles in Tanner Graph

LEARN Codes: Inventing Low-latency Codes via Recurrent Neural Networks

A Model-Driven Quasi-ResNet Belief Propagation Neural Network Decoder for LDPC Codes.

Soft-Output Deep Neural Network-Based Decoding