中国物理B ›› 2024, Vol. 33 ›› Issue (6): 60314-060314.doi: 10.1088/1674-1056/ad342b

• • 上一篇    下一篇

Decoding topological XYZ2 codes with reinforcement learning based on attention mechanisms

Qing-Hui Chen(陈庆辉)1, Yu-Xin Ji(姬宇欣)1, Ke-Han Wang(王柯涵)2, Hong-Yang Ma(马鸿洋)2, and Nai-Hua Ji(纪乃华)1,†   

  1. 1 School of Information and Control Engineering, Qingdao University of Technology, Qingdao 266033, China;
    2 School of Sciences, Qingdao University of Technology, Qingdao 266033, China
  • 收稿日期:2023-12-14 修回日期:2024-02-18 接受日期:2024-03-15 出版日期:2024-06-18 发布日期:2024-06-18
  • 通讯作者: Qing-Quan Jiang, Guo-Ping Li E-mail:13964863452@126.com
  • 基金资助:
    This work was supported by the Natural Science Foundation of Shandong Province, China (Grant No. ZR2021MF049) and Joint Fund of Natural Science Foundation of Shandong Province (Grant Nos. ZR2022LLZ012 and ZR2021LLZ001).

Decoding topological XYZ2 codes with reinforcement learning based on attention mechanisms

Qing-Hui Chen(陈庆辉)1, Yu-Xin Ji(姬宇欣)1, Ke-Han Wang(王柯涵)2, Hong-Yang Ma(马鸿洋)2, and Nai-Hua Ji(纪乃华)1,†   

  1. 1 School of Information and Control Engineering, Qingdao University of Technology, Qingdao 266033, China;
    2 School of Sciences, Qingdao University of Technology, Qingdao 266033, China
  • Received:2023-12-14 Revised:2024-02-18 Accepted:2024-03-15 Online:2024-06-18 Published:2024-06-18
  • Contact: Qing-Quan Jiang, Guo-Ping Li E-mail:13964863452@126.com
  • Supported by:
    This work was supported by the Natural Science Foundation of Shandong Province, China (Grant No. ZR2021MF049) and Joint Fund of Natural Science Foundation of Shandong Province (Grant Nos. ZR2022LLZ012 and ZR2021LLZ001).

摘要: Quantum error correction, a technique that relies on the principle of redundancy to encode logical information into additional qubits to better protect the system from noise, is necessary to design a viable quantum computer. For this new topological stabilizer code-$XYZ^{2}$ code defined on the cellular lattice, it is implemented on a hexagonal lattice of qubits and it encodes the logical qubits with the help of stabilizer measurements of weight six and weight two. However topological stabilizer codes in cellular lattice quantum systems suffer from the detrimental effects of noise due to interaction with the environment. Several decoding approaches have been proposed to address this problem. Here, we propose the use of a state-attention based reinforcement learning decoder to decode $XYZ^{2}$ codes, which enables the decoder to more accurately focus on the information related to the current decoding position, and the error correction accuracy of our reinforcement learning decoder model under the optimisation conditions can reach 83.27\% under the depolarizing noise model, and we have measured thresholds of 0.18856 and 0.19043 for $XYZ^{2}$ codes at code spacing of 3-7 and 7-11, respectively. our study provides directions and ideas for applications of decoding schemes combining reinforcement learning attention mechanisms to other topological quantum error-correcting codes.

关键词: quantum error correction, topological quantum stabilizer code, reinforcement learning, attention mechanism

Abstract: Quantum error correction, a technique that relies on the principle of redundancy to encode logical information into additional qubits to better protect the system from noise, is necessary to design a viable quantum computer. For this new topological stabilizer code-$XYZ^{2}$ code defined on the cellular lattice, it is implemented on a hexagonal lattice of qubits and it encodes the logical qubits with the help of stabilizer measurements of weight six and weight two. However topological stabilizer codes in cellular lattice quantum systems suffer from the detrimental effects of noise due to interaction with the environment. Several decoding approaches have been proposed to address this problem. Here, we propose the use of a state-attention based reinforcement learning decoder to decode $XYZ^{2}$ codes, which enables the decoder to more accurately focus on the information related to the current decoding position, and the error correction accuracy of our reinforcement learning decoder model under the optimisation conditions can reach 83.27\% under the depolarizing noise model, and we have measured thresholds of 0.18856 and 0.19043 for $XYZ^{2}$ codes at code spacing of 3-7 and 7-11, respectively. our study provides directions and ideas for applications of decoding schemes combining reinforcement learning attention mechanisms to other topological quantum error-correcting codes.

Key words: quantum error correction, topological quantum stabilizer code, reinforcement learning, attention mechanism

中图分类号:  (Quantum information)

  • 03.67.-a
87.64.Aa (Computer simulation) 03.67.Pp (Quantum error correction and other methods for protection against decoherence)