Jet Reconstruction with Mamba Networks in Collider Events

Abstract

We introduce a novel end-to-end framework for jet reconstruction in high-energy collider events, leveraging the efficiency and long-range modeling capabilities of the Mamba architecture. Our model unifies instance segmentation, classification, and kinematic regression into a single multi-task learning system, enabling a sophisticated multi-level reconstruction that simultaneously identifies primary heavy jets ($t$, $H$, $W/Z$) and their constituent sub-jets. To facilitate supervised learning for this complex task, we develop a novel method for assigning final-state hadrons to their ancestor colored partons using a Mixed-Integer Linear Programming solver, which generates high-fidelity ground-truth labels. The model achieves high classification accuracy, with an Average Precision score of 0.569 for $W/Z$-jets and 0.568 for $b$-jets, and shows exceptional precision in kinematic reconstruction. Furthermore, we show that the model not only maintains stable performance in high-pileup environments but also successfully reconstructs the mass peaks of beyond the standard model particles. This work presents a powerful and versatile new tool for comprehensive event reconstruction at the LHC.

16 pages, 6 figures, version to appear in PRD

References in corpus (35)

• The automated computation of tree-level and next-to-leading order differential cross sections, and their matching to parton shower simulations
• The anti-k_t jet clustering algorithm
• A Brief Introduction to PYTHIA 8.1
• Jet substructure as a new Higgs search channel at the LHC
• Tuning PYTHIA 8.1: the Monash 2013 Tune
• Identifying Boosted Objects with N-subjettiness
• Soft Drop
• Jet Trimming
• Towards Jetography
• ParticleNet: Jet Tagging via Particle Clouds
• Deep Learning and its Application to LHC Physics
• Maximizing Boosted Top Identification by Minimizing N-subjettiness
• Pileup Per Particle Identification
• Boosted objects: a probe of beyond the Standard Model physics
• Jet Substructure at the Large Hadron Collider: A Review of Recent Advances in Theory and Machine Learning
• Techniques for improved heavy particle searches with jet substructure
• Jet Substructure at the Tevatron and LHC: New results, new tools, new benchmarks
• Jet Substructure at the Large Hadron Collider: Experimental Review
• Stop Reconstruction with Tagged Tops
• Deep learning in color: towards automated quark/gluon jet discrimination
• Jet-Images -- Deep Learning Edition
• Deep-learning Top Taggers or The End of QCD?
• Energy Flow Networks: Deep Sets for Particle Jets
• Boosted objects and jet substructure at the LHC
• Pulling Out All the Tops with Computer Vision and Deep Learning
• JEDI-net: a jet identification algorithm based on interaction networks
• QCD-Aware Recursive Neural Networks for Jet Physics
• An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging
• JUNIPR: a Framework for Unsupervised Machine Learning in Particle Physics
• Recursive Neural Networks in Quark/Gluon Tagging
• Supervised Jet Clustering with Graph Neural Networks for Lorentz Boosted Bosons
• The Boosted Higgs Jet Reconstruction via Graph Neural Network
• A Method to Simultaneously Facilitate All Jet Physics Tasks
• Reconstructing boosted Higgs jets from event image segmentation
• Automatic detection of boosted Higgs boson and top quark jets in an event image

Cited by in corpus (1)

• The Physics Behind ML-based Quark-Gluon Taggers