Transfer Learning in Reinforcement Learning for Quantum Error Correction: Evaluating Scale-Invariant Decoding of Quantum LDPC Codes
Zhang Shuhan, Fan Yang, Tanaka Hiroshi, Dian Zhou, Zeng Xuan
Abstract
The realization of fault-tolerant quantum computing relies fundamentally on the efficient and accurate decoding of quantum error-correcting codes. Recently, Deep Reinforcement Learning (DRL) has emerged as a promising approach for decoding complex codes, such as quantum Low-Density Parity-Check (qLDPC) codes, due to its ability to adapt to correlated noise and system specific irregularities. However, the computational cost of training these intelligent agents scales prohibitively with the code distance and the number of physical qubits. In this work, we present an analytical and theoretical framework for transfer learning in DRL-based quantum decoders. We explore the scale-invariant properties of qLDPC codes, specifically focusing on how the structural symmetries in hypergraph product and lifted product codes can be exploited by scalable neural architectures. By formulating the decoding process as a Markov Decision Process (MDP) mapped onto a translationally and structurally invariant graph, we demonstrate that a Deep Q-Network (DQN) trained on a small lattice (e.g., an $[[625, 25]]$ qLDPC code) can be zero-shot transferred or rapidly fine-tuned for a significantly larger lattice (e.g., an $[[900, 50]]$ qLDPC code). We provide rigorous theoretical bounds on the performance degradation under dimensionality mismatch and present comprehensive empirical results. Our experiments show that fine-tuning a pre-trained agent requires orders of magnitude fewer samples than training from scratch, while retaining near-optimal logical error suppression. These findings provide a concrete pathway toward scalable, low-latency machine learning architectures for large-scale fault-tolerant quantum processors.