Discovering new phenomena at the Large Hadron Collider (LHC) involves the identification of rare signals over conventional backgrounds. Thus binary classification tasks are ubiquitous in analyses of the vast amounts of LHC data. We develop a Lie-Equivariant Quantum Graph Neural Network (Lie-EQGNN), a quantum model that is not only data efficient, but also has symmetry-preserving properties. Since Lorentz group equivariance has been shown to be beneficial for jet tagging, we build a Lorentz-equivariant quantum GNN for quark-gluon jet discrimination and show that its performance is on par with its classical state-of-the-art counterpart LorentzNet, making it a viable alternative to the conventional computing paradigm.
Discovering new phenomena at the Large Hadron Collider (LHC) involves the identification of rare signals over conventional backgrounds. Thus binary classification tasks are ubiquitous in analyses of the vast amounts of LHC data. We develop a Lie-Equivariant Quantum Graph Neural Network (Lie-EQGNN), a quantum model that is not only data efficient, but also has symmetry-preserving properties. Since Lorentz group equivariance has been shown to be beneficial for jet tagging, we build a Lorentz-equivariant quantum GNN for quark-gluon jet discrimination and show that its performance is on par with its classical state-of-the-art counterpart LorentzNet, making it a viable alternative to the conventional computing paradigm.
The rapid data surge from the high-luminosity Large Hadron Collider introduces critical computational challenges requiring novel approaches for efficient data processing in particle physics. Quantum machine learning, with its capability to leverage the extensive Hilbert space of quantum hardware, offers a promising solution. However, current quantum graph neural networks (GNNs) lack robustness to noise and are often constrained by fixed symmetry groups, limiting adaptability in complex particle interaction modeling. This article demonstrates that replacing the classical Lorentz group equivariant block modules in LorentzNet with a dressed quantum circuit significantly enhances performance despite using≈5.5times fewer parameters. Additionally, quantum circuits effectively replace MLPs by inherently preserving symmetries, with Lorentz symmetry integration ensuring robust handling of relativistic invariance. OurLorentz-equivariantquantumgraphneuralnetwork (Lorentz-EQGNN) achieved 74.00% test accuracy and an AUC of 87.38% on the Quark-Gluon jet tagging dataset, outperforming the classical and quantum GNNs with a reduced architecture using only 4 qubits. On the electron–photon dataset, Lorentz-EQGNN reached 67.00% test accuracy and an AUC of 68.20%, demonstrating competitive results with just 800 training samples. Evaluation of our model on generic MNIST and FashionMNIST datasets confirmed Lorentz-EQGNN’s efficiency, achieving 88.10% and 74.80% test accuracy, respectively. Ablation studies validated the impact of quantum components on performance, with notable improvements in background rejection rates over classical counterparts. These results highlight Lorentz-EQGNN’s potential for immediate applications in noise-resilient jet tagging, event classification, and broader data-scarce HEP tasks.
Machine learning algorithms are heavily relied on to understand the vast amounts of data from high-energy particle collisions at the CERN Large Hadron Collider (LHC). The data from such collision events can naturally be represented with graph structures. Therefore, deep geometric methods, such as graph neural networks (GNNs), have been leveraged for various data analysis tasks in high-energy physics. One typical task is jet tagging, where jets are viewed as point clouds with distinct features and edge connections between their constituent particles. The increasing size and complexity of the LHC particle datasets, as well as the computational models used for their analysis, have greatly motivated the development of alternative fast and efficient computational paradigms such as quantum computation. In addition, to enhance the validity and robustness of deep networks, we can leverage the fundamental symmetries present in the data through the use of invariant inputs and equivariant layers. In this paper, we provide a fair and comprehensive comparison of classical graph neural networks (GNNs) and equivariant graph neural networks (EGNNs) and their quantum counterparts: quantum graph neural networks (QGNNs) and equivariant quantum graph neural networks (EQGNN). The four architectures were benchmarked on a binary classification task to classify the parton-level particle initiating the jet. Based on their area under the curve (AUC) scores, the quantum networks were found to outperform the classical networks. However, seeing the computational advantage of quantum networks in practice may have to wait for the further development of quantum technology and its associated application programming interfaces (APIs).
Jets are central to the physics programs of both current and future colliders, from precision Standard Model measurements and searches for new physics at the Large Hadron Collider to studies of nucleon structure at the future Electron-Ion Collider. Motivated by these applications, we explore quantum machine learning for jet classification and present a permutation-invariant Quantum Graph Neural Network (QGNN) applied to particle-cloud representations of jets. We apply the model to two such discrimination tasks: quark vs. gluon and up vs. down quark flavor tagging, with the latter being, to our knowledge, the first application of a quantum model to this problem. In the ideal simulation, the QGNN performs competitively against the Particle Flow Network and traditional QCD observables. We further deploy scaled-down models to IBM and IonQ quantum processing units (QPUs), where we train and evaluate them, obtaining promising results. Finally, we perform an interpretability analysis to characterize the observables learned by the quantum model, relating them to generalized angularities for the quark-gluon study and to jet charge for the flavor study.
We present a quantum enhanced tagger to identify jets with large Lorentz boost at colliders. For the first time, a convolutional quantum graph neural network (QGNN) is designed to discriminate boosted jets arising from hadronic decays of the Z boson, against those produced from gluons with large momentum. The network receives data without any physics-driven refinement, relying solely on the dimensionality reduction. The reduction is performed using a convolutional autoencoder whose performance is improved in the presence of added noise. The latent data are put into a graph format and fed to the QGNN of ten qubits. The autoencoder and the QGNN are trained separately, and simultaneously, and the resulting performances are compared with a classic algorithm based on graph networks. The findings indicate a strong potential of quantum graph networks to reproduce the performance of classical methods.
Equivariant quantum graph neural networks (EQGNNs) offer a potentially powerful method to process graph data. However, existing EQGNN models only consider the permutation symmetry of graphs, and failing to fully exploit the geometric and non-geometric information in graphs, resulting in suboptimal performance when processing 3D graph data. To address these limitations, we derive constraints of rotation and permutation equivariance, and then propose a novel rotation- and permutation-equivariant quantum graph neural network (RP-EQGNN). An equivariant module is designed to extract the geometric information. Then, a convolution and entanglement module is constructed to extract non-geometric information. To improve performance of our model, an edge entanglement strategy is designed to perform distinguishable entanglement operations based on edge heterogeneity. The experiment results demonstrate that RP-EQGNN is significantly better for graph regression on the QM9 dataset and the OC20 dataset than Q3DGL and EQC in MAE and achieves results comparable to those of EquiformerV2, Geoformer, SO3KRATES and HEGNN. It also has advantage for point cloud classification on the ModelNet40 dataset over quantum models, including sQCNN-3D and PI-QSVM. RP-EQGNN introduces an innovative approach for processing 3D graph data, establishing a basis for future investigations into symmetries within graph neural networks.
Graph Neural Networks (GNNs) are effective for analyzing graph-structured data but face challenges with large graphs due to high memory requirements and inefficient sparse matrix operations on GPUs. Quantum Computing (QC) may be used to overcome these problems. For instance, Quantum Graph Neural Networks (QGNNs) have already been implemented in the literature. However, current quantum hardware limits the dimensionality of the datasets that can be encoded. For this reason, previous works simplify graph datasets manually or use small synthetic datasets. This work introduces the Guided Graph Compression (GGC) framework. It uses a graph autoencoder to reduce the size of graphs in terms of number of nodes and node feature dimensionality. The compression is guided towards enhancing the performance of a classification task by coupling the loss functions of the autoencoder and the classifier. The framework is applied to the Jet Tagging task that is relevant at CERN's LHC. It involves classifying jets as being originated from quarks or gluons. It is tested on three different training paradigms. First, a classical GNN is trained using the uncompressed data. Then, graph autoencoders are trained as a separate preprocessing step and used for classical and quantum classifiers. Lastly, the GGC framework is applied to both types of classifiers. Results show that the GGC framework outperforms the other two paradigms. Moreover, the classical and quantum classifiers achieve competitive performance in terms of ROC-AUC score.