Speeding up Learning Quantum States through Group Equivariant Convolutional Quantum Ansätze

Abstract

We develop a theoretical framework for $S_n$-equivariant convolutional quantum circuits with SU$(d)$-symmetry, building on and significantly generalizing Jordan's Permutational Quantum Computing (PQC) formalism based on Schur-Weyl duality connecting both SU$(d)$ and $S_n$ actions on qudits. In particular, we utilize the Okounkov-Vershik approach to prove Harrow's statement (Ph.D. Thesis 2005 p.160) on the equivalence between $\mathrm{SU}(d)$ and $S_n$ irrep bases and to establish the $S_n$-equivariant Convolutional Quantum Alternating Ansätze ($S_n$-CQA) using Young-Jucys-Murphy (YJM) elements. We prove that $S_n$-CQA is able to generate any unitary in any given $S_n$ irrep sector, which may serve as a universal model for a wide array of quantum machine learning problems with the presence of SU($d$) symmetry. Our method provides another way to prove the universality of Quantum Approximate Optimization Algorithm (QAOA) and verifies that 4-local SU($d$) symmetric unitaries are sufficient to build generic SU($d$) symmetric quantum circuits up to relative phase factors. We present numerical simulations to showcase the effectiveness of the ansätze to find the ground state energy of the $J_1$--$J_2$ antiferromagnetic Heisenberg model on the rectangular and Kagome lattices. Our work provides the first application of the celebrated Okounkov-Vershik's $S_n$ representation theory to quantum physics and machine learning, from which to propose quantum variational ansätze that strongly suggests to be classically intractable tailored towards a specific optimization problem.

15 pages, 11 figures

References in corpus (8)

• Quantum algorithm for solving linear systems of equations
• Variational Quantum Algorithms
• A rigorous and robust quantum speed-up in supervised machine learning
• Quantum Data Fitting
• Information-theoretic bounds on quantum advantage in machine learning
• Efficient Quantum Circuits for Schur and Clebsch-Gordan Transforms
• Learning the ground state of a non-stoquastic quantum Hamiltonian in a rugged neural network landscape
• Many-Body Quantum States with Exact Conservation of Non-Abelian and Lattice Symmetries through Variational Monte Carlo

Cited by in corpus (32)

• Theory for Equivariant Quantum Neural Networks
• Theoretical Guarantees for Permutation-Equivariant Quantum Neural Networks
• Quantum-centric Supercomputing for Materials Science: A Perspective on Challenges and Future Directions
• Does provable absence of barren plateaus imply classical simulability?
• Building spatial symmetries into parameterized quantum circuits for faster training
• Generalization of Quantum Machine Learning Models Using Quantum Fisher Information Metric
• Symmetry breaking in geometric quantum machine learning in the presence of noise
• Approximately Equivariant Quantum Neural Network for $p4m$ Group Symmetries in Images
• Rotationally-Invariant Circuits: Universality with the exchange interaction and two ancilla qubits
• The role of data embedding in equivariant quantum convolutional neural networks
• Splitting and Parallelizing of Quantum Convolutional Neural Networks for Learning Translationally Symmetric Data
• Quantum-machine-assisted Drug Discovery
• Symmetry-invariant quantum machine learning force fields
• Designs from Local Random Quantum Circuits with SU(d) Symmetry
• Unitary Designs of Symmetric Local Random Circuits
• Toward Super-polynomial Quantum Speedup of Equivariant Quantum Algorithms with SU($d$) Symmetry
• Geometric Quantum Machine Learning with Horizontal Quantum Gates
• Trade-off between Gradient Measurement Efficiency and Expressivity in Deep Quantum Neural Networks
• All you need is spin: SU(2) equivariant variational quantum circuits based on spin networks
• Permutation-equivariant quantum convolutional neural networks
• Architectures and random properties of symplectic quantum circuits
• Probing many-body Bell correlation depth with superconducting qubits
• A quantum tug of war between randomness and symmetries on homogeneous spaces
• Quantum Active Learning
• SU(d)-Symmetric Random Unitaries: Quantum Scrambling, Error Correction, and Machine Learning
• Exchange-Symmetrized Qudit Bell Bases and Bell-State Distinguishability
• Permutationally invariant processes in open multiqudit systems
• Towards Symmetry-Aware Efficient Simulation of Quantum Systems and Beyond
• Variational simulation of higher-spin systems on qubit-based quantum simulators
• Learning quantum symmetries with interactive quantum-classical variational algorithms
• Synthesis of Energy-Conserving Quantum Circuits with XY interaction
• LArTPC hit-based topology classification with quantum machine learning and symmetry