Guided Graph Compression for Quantum Graph Neural Networks

Abstract

Graph Neural Networks (GNNs) are effective for processing graph-structured data but face challenges with large graphs due to high memory requirements and inefficient sparse matrix operations on GPUs. Quantum Computing (QC) offers a promising avenue to address these issues and inspires new algorithmic approaches. In particular, Quantum Graph Neural Networks (QGNNs) have been explored in recent literature. However, current quantum hardware limits the dimension of the data that can be effectively encoded. Existing approaches either simplify datasets manually or use artificial graph datasets. This work introduces the Guided Graph Compression (GGC) framework, which uses a graph autoencoder to reduce both the number of nodes and the dimensionality of node features. The compression is guided to enhance the performance of a downstream classification task, which can be applied either with a quantum or a classical classifier. The framework is evaluated on the Jet Tagging task, a classification problem of fundamental importance in high energy physics that involves distinguishing particle jets initiated by quarks from those by gluons. The GGC is compared against using the autoencoder as a standalone preprocessing step and against a baseline classical GNN classifier. Our numerical results demonstrate that GGC outperforms both alternatives, while also facilitating the testing of novel QGNN ansatzes on realistic datasets.

References in corpus (20)

• Quantum algorithm for solving linear systems of equations
• Variational Quantum Algorithms
• Circuit-centric quantum classifiers
• Power of data in quantum machine learning
• Data re-uploading for a universal quantum classifier
• A rigorous and robust quantum speed-up in supervised machine learning
• Quantum advantage in learning from experiments
• Generalization in quantum machine learning from few training data
• ParticleNet: Jet Tagging via Particle Clouds
• Quantum generalisation of feedforward neural networks
• Energy Flow Networks: Deep Sets for Particle Jets
• Barren Plateaus in Variational Quantum Computing
• Quantum machine learning beyond kernel methods
• Quark and Gluon Tagging at the LHC
• Anomaly detection in high-energy physics using a quantum autoencoder
• Theory for Equivariant Quantum Neural Networks
• Quantum Machine Learning for Particle Physics using a Variational Quantum Classifier
• Quantum anomaly detection in the latent space of proton collision events at the LHC
• Parameterized quantum circuits as universal generative models for continuous multivariate distributions
• Guided Quantum Compression for High Dimensional Data Classification