Abstract
Jet tagging is a classification problem in high-energy physics experiments that aims to identify the collimated sprays of subatomic particles, jets, from particle collisions and ‘tag’ them to their emitter particle. Advances in jet tagging present opportunities for searches of new physics beyond the Standard Model. Current approaches use deep learning to uncover hidden patterns in complex collision data. However, the representation of jets as inputs to a deep learning model have been varied, and often, informative features are withheld from models. In this study, we propose a graph-based representation of a jet that encodes the most information possible. To learn best from this representation, we design Particle Chebyshev Network (PCN), a graph neural network (GNN) using Chebyshev graph convolutions (ChebConv). ChebConv has been demonstrated as an effective alternative to classical graph convolutions in GNNs and has yet to be explored in jet tagging. PCN achieves a substantial improvement in accuracy over existing taggers and opens the door to future studies into graph-based representations of jets and ChebConv layers in high-energy physics experiments. Code is available at https://github.com/YVSemlani/PCN-Jet-Tagging
Article PDF
Similar content being viewed by others
Explore related subjects
Find the latest articles, discoveries, and news in related topics.Avoid common mistakes on your manuscript.
References
ATLAS collaboration, Observation of a new particle in the search for the Standard Model Higgs boson with the ATLAS detector at the LHC, Phys. Lett. B 716 (2012) 1 [arXiv:1207.7214] [INSPIRE].
CMS collaboration, Observation of a New Boson at a Mass of 125 GeV with the CMS Experiment at the LHC, Phys. Lett. B 716 (2012) 30 [arXiv:1207.7235] [INSPIRE].
H. Qu, C. Li and S. Qian, Particle Transformer for Jet Tagging, in International Conference on Machine Learning, PMLR (2022) pp. 18281–18292 [arXiv:2202.03772] [INSPIRE].
V. Mikuni and F. Canelli, Point cloud transformers applied to collider physics, Mach. Learn. Sci. Tech. 2 (2021) 035027 [arXiv:2102.05073] [INSPIRE].
H. Qu and L. Gouskos, ParticleNet: Jet Tagging via Particle Clouds, Phys. Rev. D 101 (2020) 056019 [arXiv:1902.08570] [INSPIRE].
S. Gong et al., An efficient Lorentz equivariant graph neural network for jet tagging, JHEP 07 (2022) 030 [arXiv:2201.08187] [INSPIRE].
F.A. Dreyer and H. Qu, Jet tagging in the Lund plane with graph networks, JHEP 03 (2021) 052 [arXiv:2012.08526] [INSPIRE].
V. Mikuni and F. Canelli, ABCNet: An attention-based method for particle tagging, Eur. Phys. J. Plus 135 (2020) 463 [arXiv:2001.05311] [INSPIRE].
A. Bogatskiy et al., Lorentz Group Equivariant Neural Network for Particle Physics, in International Conference on Machine Learning, PMLR (2020) pp. 992–1002 [arXiv:2006.04780] [INSPIRE].
A. Bogatskiy, T. Hoffman, D.W. Miller and J.T. Offermann, PELICAN: Permutation Equivariant and Lorentz Invariant or Covariant Aggregator Network for Particle Physics, arXiv:2211.00454 [INSPIRE].
D. Ruhe, J. Brandstetter and P. Forré, Clifford Group Equivariant Neural Networks, arXiv:2305.11141 [INSPIRE].
P.T. Komiske, E.M. Metodiev and J. Thaler, Energy Flow Networks: Deep Sets for Particle Jets, JHEP 01 (2019) 121 [arXiv:1810.05165] [INSPIRE].
E.A. Moreno et al., JEDI-net: a jet identification algorithm based on interaction networks, Eur. Phys. J. C 80 (2020) 58 [arXiv:1908.05318] [INSPIRE].
E.A. Moreno et al., Interaction networks for the identification of boosted \( H\to b\overline{b} \) decays, Phys. Rev. D 102 (2020) 012010 [arXiv:1909.12285] [INSPIRE].
E. Bernreuther et al., Casting a graph net to catch dark showers, SciPost Phys. 10 (2021) 046 [arXiv:2006.08639] [INSPIRE].
J. Guo, J. Li, T. Li and R. Zhang, Boosted Higgs boson jet reconstruction via a graph neural network, Phys. Rev. D 103 (2021) 116025 [arXiv:2010.05464] [INSPIRE].
M.J. Dolan and A. Ore, Equivariant Energy Flow Networks for Jet Tagging, Phys. Rev. D 103 (2021) 074022 [arXiv:2012.00964] [INSPIRE].
P. Konar, V.S. Ngairangbam and M. Spannowsky, Energy-weighted message passing: an infra-red and collinear safe graph neural network algorithm, JHEP 02 (2022) 060 [arXiv:2109.14636] [INSPIRE].
C. Shimmin, Particle Convolution for High Energy Physics, arXiv:2107.02908 [INSPIRE].
G. Kasieczka, T. Plehn, M. Russell and T. Schell, Deep-learning Top Taggers or The End of QCD?, JHEP 05 (2017) 006 [arXiv:1701.08784] [INSPIRE].
S. Macaluso and D. Shih, Pulling Out All the Tops with Computer Vision and Deep Learning, JHEP 10 (2018) 121 [arXiv:1803.00107] [INSPIRE].
F.A. Dreyer, G.P. Salam and G. Soyez, The Lund Jet Plane, JHEP 12 (2018) 064 [arXiv:1807.04758] [INSPIRE].
J. Lin, M. Freytsis, I. Moult and B. Nachman, Boosting \( H\to b\overline{b} \) with Machine Learning, JHEP 10 (2018) 101 [arXiv:1807.10768] [INSPIRE].
Y.-L. Du et al., Identifying the nature of the QCD transition in relativistic collision of heavy nuclei with deep learning, Eur. Phys. J. C 80 (2020) 516 [arXiv:1910.11530] [INSPIRE].
J. Filipek et al., Identifying the Quantum Properties of Hadronic Resonances using Machine Learning, arXiv:2105.04582 [INSPIRE].
CMS collaboration, Identification of heavy, energetic, hadronically decaying particles using machine-learning techniques, 2020 JINST 15 P06005 [arXiv:2004.08262] [INSPIRE].
L. de Oliveira et al., Jet-images — deep learning edition, JHEP 07 (2016) 069 [arXiv:1511.05190] [INSPIRE].
J. Barnard, E.N. Dawe, M.J. Dolan and N. Rajcic, Parton Shower Uncertainties in Jet Substructure Analyses with Deep Neural Networks, Phys. Rev. D 95 (2017) 014018 [arXiv:1609.00607] [INSPIRE].
P.T. Komiske, E.M. Metodiev and M.D. Schwartz, Deep learning in color: towards automated quark/gluon jet discrimination, JHEP 01 (2017) 110 [arXiv:1612.01551] [INSPIRE].
S. Choi, S.J. Lee and M. Perelstein, Infrared Safety of a Neural-Net Top Tagging Algorithm, JHEP 02 (2019) 132 [arXiv:1806.01263] [INSPIRE].
J. Li, T. Li and F.-Z. Xu, Reconstructing boosted Higgs jets from event image segmentation, JHEP 04 (2021) 156 [arXiv:2008.13529] [INSPIRE].
K. Fraser and M.D. Schwartz, Jet Charge and Machine Learning, JHEP 10 (2018) 093 [arXiv:1803.08066] [INSPIRE].
D. Guest et al., Jet Flavor Classification in High-Energy Physics with Deep Neural Networks, Phys. Rev. D 94 (2016) 112002 [arXiv:1607.08633] [INSPIRE].
S. Egan et al., Long Short-Term Memory (LSTM) networks with jet constituents for boosted top tagging at the LHC, arXiv:1711.09059 [INSPIRE].
E. Bols et al., Jet Flavour Classification Using DeepJet, 2020 JINST 15 P12012 [arXiv:2008.10519] [INSPIRE].
J. Pearkes, W. Fedorko, A. Lister and C. Gay, Jet Constituents for Deep Neural Network Based Top Quark Tagging, arXiv:1704.02124 [INSPIRE].
A. Butter, G. Kasieczka, T. Plehn and M. Russell, Deep-learned Top Tagging with a Lorentz Layer, SciPost Phys. 5 (2018) 028 [arXiv:1707.08966] [INSPIRE].
G. Kasieczka, N. Kiefer, T. Plehn and J.M. Thompson, Quark-Gluon Tagging: Machine Learning vs Detector, SciPost Phys. 6 (2019) 069 [arXiv:1812.09223] [INSPIRE].
Y. Wang et al., Dynamic Graph CNN for Learning on Point Clouds, arXiv:1801.07829 [INSPIRE].
M. He, Z. Wei and J.-R. Wen, Convolutional Neural Networks on Graphs with Chebyshev Approximation, Revisited, arXiv:2202.03580.
L. Liao et al., An improved dynamic Chebyshev graph convolution network for traffic flow prediction with spatial-temporal attention, Appl. Intell. 52 (2022) 16104.
O. Boyaci, M.R. Narimani, K. Davis and E. Serpedin, Cyberattack Detection in Large-Scale Smart Grids using Chebyshev Graph Convolutional Networks, arXiv:2112.13166.
S. Maneewongvatana and D.M. Mount, Analysis of approximate nearest neighbor searching with clustered point sets, cs/9901013.
I. Loshchilov and F. Hutter, Decoupled Weight Decay Regularization, in International Conference on Learning Representations, (2018), arXiv:1711.05101 [INSPIRE].
T. Saito and M. Rehmsmeier, The Precision-Recall Plot Is More Informative than the ROC Plot When Evaluating Binary Classifiers on Imbalanced Datasets, PLoS One 10 (2015) e0118432.
CMS collaboration, Boosted jet identification using particle candidates and deep neural networks, CMS-DP-2017-049 (2017).
M. Zaheer et al., Deep Sets, Adv. Neural Inf. Process 30 (2017) 7264 [arXiv:1703.06114] [INSPIRE].
G. Kasieczka, T. Plehn and M. Russel, Top quark tagging reference dataset, https://doi.org/10.5281/zenodo.2603256 (2019).
P. Komiske, E. Metodiev and J. Thaler, Pythia8 Quark and Gluon Jets for Energy Flow, https://doi.org/10.5281/zenodo.3164691 (2019).
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
ArXiv ePrint: 2309.08630
Rights and permissions
Open Access . This article is distributed under the terms of the Creative Commons Attribution License (CC-BY 4.0), which permits any use, distribution and reproduction in any medium, provided the original author(s) and source are credited.
About this article
Cite this article
Semlani, Y., Relan, M. & Ramesh, K. PCN: a deep learning approach to jet tagging utilizing novel graph construction methods and Chebyshev graph convolutions. J. High Energ. Phys. 2024, 247 (2024). https://doi.org/10.1007/JHEP07(2024)247
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/JHEP07(2024)247