Equivariant Energy Flow Networks for Jet Tagging

Abstract

Jet tagging techniques that make use of deep learning show great potential for improving physics analyses at colliders. One such method is the Energy Flow Network (EFN) - a recently introduced neural network architecture that represents jets as permutation-invariant sets of particle momenta while maintaining infrared and collinear safety. We develop a variant of the Energy Flow Network architecture based on the Deep Sets formalism, incorporating permutation-equivariant layers. We derive conditions under which infrared and collinear safety can be maintained, and study the performance of these networks on the canonical example of W-boson tagging. We find that equivariant Energy Flow Networks have similar performance to Particle Flow Networks, which are superior to standard EFNs. However, equivariant Particle Flow Networks suffer from convergence and overfitting issues. Finally, we study how equivariant networks sculpt the jet mass and provide some initial results on decorrelation using planing.

20 pages, 8 figures

References in corpus (12)

• An Introduction to PYTHIA 8.2
• Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification
• Jet substructure as a new Higgs search channel at the LHC
• Jet substructure as a new Higgs search channel at the LHC
• A generic anti-QCD jet tagger
• Parton Shower Uncertainties in Jet Substructure Analyses with Deep Neural Networks
• Lorentz Group Equivariant Neural Network for Particle Physics
• Quark-Gluon tagging with Shower Deconstruction: Unearthing dark matter and Higgs couplings
• Mapping Machine-Learned Physics into a Human-Readable Space
• Supervised Jet Clustering with Graph Neural Networks for Lorentz Boosted Bosons
• Mass Unspecific Supervised Tagging (MUST) for boosted jets
• Towards Machine Learning Analytics for Jet Substructure

Cited by in corpus (25)

• An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging
• PC-JeDi: Diffusion for Particle Cloud Generation in High Energy Physics
• Energy-weighted Message Passing: an infra-red and collinear safe graph neural network algorithm
• Jet Flavour Tagging for Future Colliders with Fast Simulation
• SymmetryGAN: Symmetry Discovery with Deep Learning
• Explainable machine learning of the underlying physics of high-energy particle collisions
• Machine Learning in High Energy Physics: A review of heavy-flavor jet tagging at the LHC
• IRC-safe Graph Autoencoder for unsupervised anomaly detection
• Unbinned Deep Learning Jet Substructure Measurement in High $Q^2$ ep collisions at HERA
• Jet tagging algorithm of graph network with HaarPooling message passing
• A Holistic Approach to Predicting Top Quark Kinematic Properties with the Covariant Particle Transformer
• Jet Discrimination with Quantum Complete Graph Neural Network
• Comparing Point Cloud Strategies for Collider Event Classification
• Meta-learning and data augmentation for mass-generalised jet taggers
• A rotation-equivariant graph neural network for learning hadronic SMEFT effects
• ${\mathbb{Z}}_2\times {\mathbb{Z}}_2$ Equivariant Quantum Neural Networks: Benchmarking against Classical Neural Networks
• Jet tagging in the Lund plane with graph networks
• Foundations of automatic feature extraction at LHC--point clouds and graphs
• Top-philic Machine Learning
• Equivariant neural networks for robust $\text{CP}$ observables
• Event Generation and Density Estimation with Surjective Normalizing Flows
• Identifying the Quantum Properties of Hadronic Resonances using Machine Learning
• Learnable cut flow for high energy physics
• Deep learning for jet modification in the presence of the QGP background
• Stable and Interpretable Jet Physics with IRC-Safe Equivariant Feature Extraction