Quantum architecture search with neural predictor based on ZX-calculus

Abstract

With the ongoing advances in noisy intermediate-scale quantum hardware, variational quantum algorithms have demonstrated significant potential in a range of quantum applications. However, obtaining high-performance, shallow-parameterized quantum circuits typically requires repeated optimization of the gate parameters over a large set of candidate circuits, resulting in prohibitively high evaluation costs. To address this challenge, this study proposes a novel predictor-based quantum architecture search (PQAS-ZX) method that leverages ZX-calculus. In this approach, a quantum circuit is first represented as a ZX diagram that supports multi-step equivalent simplifications at the diagram level. By applying these equivalence transformations, multiple circuit variants that share the same performance metric are generated, thereby significantly expanding the training dataset and enhancing the ability of the predictor to manage diverse circuit structures. ZX diagrams offer more flexible characterizations of multi-qubit entanglement and phase interactions, as well as higher-level equivalent transformations, compared with the state-of-the-art predictor-based quantum architecture search with graph measures (PQAS-GM). Numerical simulations of three variational quantum eigensolver tasks, namely the transverse-field Ising, Heisenberg, and BeH₂ molecular models, demonstrated that PQAS-ZX required only approximately 80.9%, 82.9%, and 76.1% of the queries required by PQAS-GM, respectively, to achieve the same probability of reaching the target ground-state energy. These results highlight the advantage of using ZX diagrams to identify high-quality circuits efficiently and alleviate the evaluation burden of quantum architecture searches.