arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.09025quant-phcond-mat.stat-mechcs.LGstat.ML
收藏

qecGPT:利用生成式预训练Transformer解码量子纠错码

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

Institute of Theoretical Physics, Chinese Academy of Sciences(中国科学院理论物理研究所);School of Physical Sciences, University of Chinese Academy of Sciences(中国科学院大学物理科学学院);School of Fundamental Physics and Mathematical Sciences, Hangzhou Institute for Advanced Study, UCAS(中国科学院大学杭州高等研究院基础物理与数学科学学院);International Centre for Theoretical Physics Asia-Pacific(国际理论物理中心亚太地区)

Hanyan Cao, Feng Pan, Yijia Wang, Pan Zhang

AI总结:

本研究提出qecGPT通用框架,基于Transformer无监督预训练实现量子纠错码最大似然解码,复杂度远低于传统算法,经改进后解码精度优于经典算法,可适配多种码型与错误模型,支持GPU并行批量解码。

摘要:

我们提出了一种利用生成式建模解码量子纠错码的通用框架。该模型采用自回归神经网络,具体为Transformer,来学习逻辑算子与校验子的联合概率。该训练以无监督方式进行,无需标注训练数据,因此被称为预训练。预训练完成后,模型可针对任意给定校验子高效计算逻辑算子的似然度,实现最大似然解码。它能够直接生成最可能的逻辑算子,计算复杂度关于逻辑量子比特数k为O(2k),显著优于传统的需要O(4^k)计算量的最大似然解码算法。基于预训练模型,我们进一步提出了改进方案,通过直接采样稳定子算子,更精确地获取给定校验子下逻辑算子的似然度。我们在码距较小的稳定子码上开展了数值实验,同时采用了去极化错误模型和带关联噪声的错误模型。结果表明,我们的方法相比最小权完美匹配算法和基于置信传播的算法,解码精度显著提升。我们的框架具有通用性,可应用于任意错误模型以及不同拓扑结构的量子码,例如表面码和量子LDPC码。此外,该框架利用了GPU的并行计算能力,可同时对大量校验子进行解码。我们的方法为利用生成式人工智能与现代计算能力实现高效、精准的量子纠错码解码提供了新思路。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑