Jet tagging algorithm of graph network with Haar pooling message passing
Abstract
Recently, methods of graph neural networks (GNNs) have been applied to solving the problems in high-energy physics (HEP) and have shown its great potential for quark-gluon tagging with graph representation of jet events. In this paper, we introduce an approach of GNNs combined with a Haar pooling operation to analyze the events, called Haar pooling message passing neural network (HMPNet). In HMPNet, Haar pooling not only extracts the features of graph, but embeds additional information obtained by clustering of k means of different particle features. We construct Haar pooling from five different features: absolute energy log E , transverse momentum log pT, relative coordinates (Δ η ,Δ ϕ ), the mixed ones (log E ,log pT), and (log E ,log pT,Δ η ,Δ ϕ ). The results show that an appropriate selection of information for Haar pooling enhances the accuracy of quark-gluon tagging, as adding extra information of log PT to the HMPNet outperforms all the others, whereas adding relative coordinates information (Δ η ,Δ ϕ ) is not very effective. This implies that, by adding effective particle features from Haar pooling, one can achieve much better results than that which a solely pure message passing neutral network can do, which demonstrates significant improvement of feature extraction via the pooling process. Finally, we compare the HMPNet study, ordering by pT, with other studies and prove that the HMPNet is also a good choice of GNN algorithms for jet tagging.
- Publication:
-
Physical Review D
- Pub Date:
- October 2023
- DOI:
- 10.1103/PhysRevD.108.072007
- arXiv:
- arXiv:2210.13869
- Bibcode:
- 2023PhRvD.108g2007M
- Keywords:
-
- High Energy Physics - Experiment;
- Computer Science - Computer Vision and Pattern Recognition;
- Computer Science - Machine Learning;
- High Energy Physics - Phenomenology
- E-Print:
- doi:10.1103/PhysRevD.108.072007