Mix of experts Vision Transformer delivers high-fidelity surface code decoding

Machine Learning


Scientists are tackling the critical challenge of error correction in quantum computing. This is essential for distributing information across a large number of qubits and protecting it from noise. Hoang Viet Nguyen, Manh Hung Nguyen and colleagues from Hanoi University of Science and Technology, along with Van Khu Vu from VinUniversity and Yeow Meng Chee from Singapore University of Technology and Design, announced a new decoder architecture called QuantumSMoE. This work is important because it combines the strengths of both classical algorithms and machine learning, utilizes a vision transformer to explicitly incorporate the geometric structure of the stabilizer code, especially the surface code, and provides improved scalability through a mixture of expert layers. Their experiments on toric codes demonstrated that QuantumSMoE goes beyond existing machine learning decoders and established classical methods, paving the way to faster and more reliable quantum computation.

Vision Transformer decodes toric code geometry

Scientists have announced QuantumSMoE, a new decoder for quantum error correction that significantly increases the fidelity of surface code decoding. This breakthrough addresses a central bottleneck in scalable real-time quantum computation, the efficient decoding of syndrome information derived from stabilizer measurements. The research team achieved this by developing a vision transformer-based decoder that explicitly incorporates the geometry of topological codes, particularly toric codes, through innovative plus-type embedding and adaptive masking techniques. These methods effectively capture the local interactions and connectivity of the lattice, enhancing the decoder’s ability to infer error patterns and apply appropriate recovery operations.
This study reveals a new decoding approach that goes beyond traditional methods, which are often limited by computational overhead and the inability to fully exploit the inherent structure of quantum codes. Existing decoders fall into two main categories. One is a classical algorithmic decoder that provides a strong baseline but struggles with scalability, and the other is a machine learning-based decoder that provides fast inference but often ignores important lattice shapes in the code. QuantumSMoE combines the best of both approaches and leverages a mixture of expert layers and a new auxiliary loss function to improve scalability and performance. In experiments, QuantumSMoE consistently outperforms classical baselines widely used in state-of-the-art machine learning decoders and toric codes, demonstrating significant progress in the field.

This work establishes a new paradigm for quantum decoding by taking inspiration from vision models and their ability to exploit spatially guided biases. Recognizing the potential to model both local and global information within quantum code, the team implemented a vision transformer architecture. Additionally, we integrated expert layer mixing, a technique that has proven successful in large-scale language models, to enhance expressive power without sacrificing inference efficiency, a key requirement for real-time decoding applications. The innovative SoftMoE component addresses challenges associated with the standard MoE model, such as load balancing and routing discontinuities, by mapping tokens to aggregated slots, ensuring a more stable and efficient decoding process.

Experiments on toric codes conclusively demonstrate the superior performance of QuantumSMoE, demonstrating that it can achieve higher fidelity decoding compared to existing methods. A decoder design incorporating plus-shaped embedding and adaptive masking can effectively capture the geometric properties of the surface code and enforce spatial locality, leading to more accurate error pattern identification. This breakthrough opens exciting possibilities for building larger and more robust quanta. The team measured performance across various physical error rates and code distances and achieved superior results compared to methods such as MWPM, MWPM-Corr, BP LSD, and QECCT decoders.

This breakthrough significantly advances the ability to protect quantum information from physical noise by efficiently decoding error syndromes and performing precise recovery operations. The researchers incorporated the code structure into QuantumSMoE through positive embedding and adaptive masking, allowing the model to capture local interactions and lattice connectivity within the quantum code. The decoder focuses only on relevant adjacent regions, reducing computational demands and enhancing scalability. Additionally, a new auxiliary loss was implemented to optimize the token-to-slot allocation within the SoftMoE block, facilitating more effective grouping of similar inputs and improving the decoder’s ability to classify complex errors.

According to our data, this approach is one of the first to successfully leverage an expert mixture architecture to power machine learning-based decoders for quantum error-correcting codes. Tests have proven that QuantumSMoE’s architecture explicitly incorporates the geometric and topological properties of toric codes, allowing for more accurate decoding. Loss of slot orthogonality causes the SoftMoE module to assign similar patch inputs to the same subset of experts to improve the decoding process. Extensive numerical evaluations confirm that the proposed approach yields superior logical error rate reduction compared to traditional baselines and existing machine learning models.

Specifically, the toric code now serves as a standard benchmark for controlled evaluation, allowing us to accurately measure performance improvements. Measurements confirm that the combination of expert mixture layer integration and auxiliary loss allows QuantumSMoE to accurately classify complex errors at each decoding stage with only a small computational overhead. This work establishes a new paradigm for quantum error correction and demonstrates the power of vision transformer and expert blends to address critical bottlenecks in scalable real-time decoding. This breakthrough could accelerate the development of fault-tolerant quantum computers by providing a more efficient and accurate way to protect quantum information.

QuantumSMoE significantly improves error correction performance

Scientists have developed a new decoder for quantum error correction, QuantumSMoE, which combines the Vision Transformer and an expert mixed architecture. This framework leverages positive embedding and adaptive masking to effectively capture local connectivity inherent in topological stabilizer codes, such as toric codes. Experiments demonstrate that QuantumSMoE significantly reduces logic error rates compared to both established classical decoders and existing machine learning-based approaches under depolarizing noise. The key innovation lies in the decoder’s ability to isolate and highlight important syndrome information through the SoftMoE layer and slot orthogonality loss, simplifying the learning process for individual experts within an expert mixing framework.

Each expert specializes in identifying specific error patterns and can efficiently decode complex errors. The authors acknowledge that the current study is limited in that it focuses on toric codes and depolarization noise, and future work will consider extending this approach to higher-dimensional codes and more realistic noise models such as syndrome measurements and circuit noise. This research represents a significant advance in the field of quantum error correction and could pave the way to more scalable and reliable quantum computation.



Source link