B-jet Tagging Using a Hybrid Edge Convolution and Transformer Architecture

Abstract

Jet flavor tagging plays an important role in precise Standard Model measurement enabling the extraction of mass dependence in jet-quark interaction and quark-gluon plasma (QGP) interactions. They also enable inferring the nature of particles produced in high-energy particle collisions that contain heavy quarks. The classification of bottom jets is vital for exploring new Physics scenarios in proton-proton collisions. In this research, we present a hybrid deep learning architecture that integrates edge convolutions with transformer self-attention mechanisms, into one single architecture called the Edge Convolution Transformer (ECT) model for bottom-quark jet tagging. ECT processes track-level features (impact parameters, momentum, and their significances) alongside jet-level observables (vertex information and kinematics) to achieve state-of-the-art performance. The study utilizes the ATLAS simulation dataset. We demonstrate that ECT achieves 0.9333 AUC for b-jet versus combined charm and light jet discrimination, surpassing ParticleNet (0.8904 AUC) and the pure transformer baseline (0.9216 AUC). The model maintains inference latency below 0.060 ms per jet on modern GPUs, meeting the stringent requirements for real-time event selection at the LHC. Our results demonstrate that hybrid architectures combining local and global features offer superior performance for challenging jet classification tasks. The proposed architecture achieves good results in b-jet tagging, particularly excelling in charm jet rejection (the most challenging task), while maintaining competitive light-jet discrimination comparable to pure transformer models.

JINST Article, 21, P03019, 2026

References in corpus (14)

• An Introduction to PYTHIA 8.2
• DELPHES 3, A modular framework for fast simulation of a generic collider experiment
• Identification of heavy-flavour jets with the CMS detector in pp collisions at 13 TeV
• Performance of $b$-Jet Identification in the ATLAS Experiment
• ParticleNet: Jet Tagging via Particle Clouds
• ATLAS $b$-jet identification performance and efficiency measurement with $t\bar{t}$ events in $pp$ collisions at $\sqrt{s}=13$ TeV
• Jet Flavour Classification Using DeepJet
• Jet Flavor Classification in High-Energy Physics with Deep Neural Networks
• Jet Flavour Tagging for Future Colliders with Fast Simulation
• Machine Learning in High Energy Physics: A review of heavy-flavor jet tagging at the LHC
• Secondary Vertex Finding in Jets with Neural Networks
• Tagging more quark jet flavours at FCC-ee at 91 GeV with a transformer-based neural network
• Fast $b$-tagging at the high-level trigger of the ATLAS experiment in LHC Run 3
• Identifying Heavy-Flavor Jets Using Vectors of Locally Aggregated Descriptors