Gsqas: Graph Self-Supervised Quantum Architecture Search

Zhimin He, Maijie Deng, Shenggen Zheng +2 more

Haozhen Situ, Guohong Li, Zhengjiang Li +3 more

Zhimin He, Maijie Deng, Shenggen Zheng +2 more

Quantum Architecture Search (QAS) is a promising approach to designing quantum circuits for variational quantum algorithms (VQAs). However, existing QAS algorithms require to evaluate a large number of quantum circuits during the search process, which makes them computationally demanding and limits their applications to large-scale quantum circuits. Recently, predictor-based QAS has been proposed to alleviate this problem by directly estimating the performances of circuits according to their structures with a predictor trained on a set of labeled quantum circuits. However, the predictor is trained by purely supervised learning, which suffers from poor generalization ability when labeled training circuits are scarce. It is very time-consuming to obtain a large number of labeled quantum circuits because the gate parameters of quantum circuits need to be optimized until convergence to obtain their ground-truth performances. To overcome these limitations, we propose GSQAS, a graph self-supervised QAS, which trains a predictor based on self-supervised learning. Specifically, we first pre-train a graph encoder on a large number of unlabeled quantum circuits using a well-designed pretext task in order to generate meaningful representations of circuits. Then the downstream predictor is trained on a small number of quantum circuits' representations and their labels. Once the encoder is trained, it can apply to different downstream tasks. In order to better encode the spatial topology information and avoid the huge dimension of feature vectors for large-scale quantum circuits, we design a scheme to encode quantum circuits as graphs. Simulation results on searching circuit structures for variational quantum eigensolver and quantum state classification show that GSQAS outperforms the state-of-the-art predictor-based QAS, achieving better performance with fewer labeled circuits.

Zhimin He, Hongxiang Chen, Yan Zhou +3 more

Quantum architecture search (QAS) has shown significant promise in automating quantum circuit design for variational quantum algorithms (VQAs). However, a central challenge remains in the high computational cost of evaluating the ground-truth performance of numerous circuit architectures during the search process. This challenge is further complicated by factors such as inefficient gradient computation, the barren plateau phenomenon, and frequent data exchange between classical and quantum systems within VQAs, all of which restrict the scalability of QAS for increasingly complex VQAs. In this study, we propose an enhanced QAS framework that integrates a performance predictor trained through self-supervised learning with Bayesian optimization to overcome these computational barriers. To address the challenges of circuit performance evaluation, we develop a graph-isomorphism-network-based predictor pretrained using expressibility prediction, an intrinsic measure of a circuit's capacity to explore the Hilbert space uniformly. This pretraining process creates a structured and smooth latent space, significantly improving the predictor's accuracy and generalizability with minimal labeled data. Furthermore, Bayesian optimization is integrated into the QAS framework to refine the predictor iteratively, concentrating the search on regions containing high-performance circuits and thus accelerating the search process. Simulation results for VQAs involving the variational quantum eigensolver and quantum approximate optimization algorithm demonstrate that the proposed framework achieves faster convergence to optimal circuits and substantially reduces computational costs compared to existing QAS methods.

Junjian Su, Jiacheng Fan, Shengyao Wu +3 more

Zhimin He, Jiachun Wei, Chuangtao Chen +3 more