An Equivariant Machine Learning Decoder for 3D Toric Codes

Research on mitigating errors in computing and communication systems has grown with their widespread use. In quantum computing, error correction is crucial as errors can quickly propagate, undermining results and the theoretical speedup over classical systems. Quantum error-correcting codes, particularly topological codes, are a key focus. These codes map parity check matrices to graphs on d-dimensional surfaces, with our research centered on the 3D toric code. Effective decoders must be robust to noise, with performance improving as code size increases, but their complexity should scale polynomially with lattice size for practicality. We propose a neural network with inductive bias, leveraging equivariance to generalize efficiently from a smaller input subset. Additionally, we explore transformer networks for error correction and compare these methods to existing techniques for decoding errors in the 3D toric code, highlighting their strengths and limitations.