Equivariant Machine Learning Decoder for 3D Toric Codes

TL;DR

This paper introduces an equivariant neural network decoder for 3D toric codes in quantum error correction, leveraging symmetry to improve robustness and efficiency over traditional methods.

Contribution

It proposes a novel neural network decoder with equivariance for 3D toric codes, enhancing error correction performance and scalability.

Findings

Neural network decoder with equivariance improves error correction.

Transformer networks can be effective in decoding.

Compared methods show superior performance of the proposed approach.

Abstract

Mitigating errors in computing and communication systems has seen a great deal of research since the beginning of the widespread use of these technologies. However, as we develop new methods to do computation or communication, we also need to reiterate the method used to deal with errors. Within the field of quantum computing, error correction is getting a lot of attention since errors can propagate fast and invalidate results, which makes the theoretical exponential speed increase in computation time, compared to traditional systems, obsolete. To correct errors in quantum systems, error-correcting codes are used. A subgroup of codes, topological codes, is currently the focus of many research papers. Topological codes represent parity check matrices corresponding to graphs embedded on a $d$-dimensional surface. For our research, the focus lies on the toric code with a 3D square lattice. The goal of any decoder is robustness to noise, which can increase with code size. However, a reasonable decoder performance scales polynomially with lattice size. As error correction is a time-sensitive operation, we propose a neural network using an inductive bias: equivariance. This allows the network to learn from a rather small subset of the exponentially growing training space of possible inputs. In addition, we investigate how transformer networks can help in correction. These methods will be compared with various configurations and previously published methods of decoding errors in the 3D toric code.