Jet Flavour Classification Using DeepJet

Abstract

Jet flavour classification is of paramount importance for a broad range of applications in modern-day high-energy-physics experiments, particularly at the LHC. In this paper we propose a novel architecture for this task that exploits modern deep learning techniques. This new model, called DeepJet, overcomes the limitations in input size that affected previous approaches. As a result, the heavy flavour classification performance improves, and the model is extended to also perform quark-gluon tagging.

14 pages, 9 figures, accepted for publication in JINST

References in corpus (15)

• Adam: A Method for Stochastic Optimization
• Observation of a new particle in the search for the Standard Model Higgs boson with the ATLAS detector at the LHC
• Observation of a new boson at a mass of 125 GeV with the CMS experiment at the LHC
• The anti-k_t jet clustering algorithm
• An Introduction to PYTHIA 8.2
• A general framework for implementing NLO calculations in shower Monte Carlo programs: the POWHEG BOX
• A New Method for Combining NLO QCD with Shower Monte Carlo Algorithms
• Matching NLO QCD computations with Parton Shower simulations: the POWHEG method
• Particle-flow reconstruction and global event description with the CMS detector
• Observation of a new boson with mass near 125 GeV in pp collisions at sqrt(s) = 7 and 8 TeV
• Identification of heavy-flavour jets with the CMS detector in pp collisions at 13 TeV
• Identification of b-quark jets with the CMS experiment
• Performance of $b$-Jet Identification in the ATLAS Experiment
• Top-pair production and decay at NLO matched with parton showers
• A deep neural network to search for new long-lived particles decaying to jets

Cited by in corpus (151)

• A portrait of the Higgs boson by the CMS experiment ten years after the discovery
• Machine Learning in Nuclear Physics
• Reading the footprints of the B-meson flavor anomalies
• An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging
• Searches for additional Higgs bosons and for vector leptoquarks in $\tau \tau$ final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for nonresonant Higgs boson pair production in final states with two bottom quarks and two photons in proton-proton collisions at $\sqrt{s}=$ 13 TeV
• MLPF: Efficient machine-learned particle-flow reconstruction using graph neural networks
• Searches for the BSM scenarios at the LHC using decision tree based machine learning algorithms: A comparative study and review of Random Forest, Adaboost, XGboost and LightGBM frameworks
• Measurement of the Higgs boson width and evidence of its off-shell contributions to ZZ production
• Identification of hadronic tau lepton decays using a deep neural network
• Review of searches for vector-like quarks, vector-like leptons, and heavy neutral leptons in proton-proton collisions at $\sqrt{s}$ = 13 TeV at the CMS experiment
• Combined searches for the production of supersymmetric top quark partners in proton-proton collisions at $\sqrt{s}=$ 13 TeV
• Measurements of polarization and spin correlation and observation of entanglement in top quark pairs using lepton+jets events from proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for pair production of vector-like quarks in leptonic final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for Higgs boson decay to a charm quark-antiquark pair in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Quantum Machine Learning for $b$-jet charge identification
• Observation of four top quark production in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Jet Flavour Tagging for Future Colliders with Fast Simulation
• Search for a heavy Higgs boson decaying into two lighter Higgs bosons in the $\tau \tau$bb final state at 13 TeV
• Observation of a pseudoscalar excess at the top quark pair production threshold
• Measurements of Higgs boson production in the decay channel with a pair of $\tau$ leptons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Shared Data and Algorithms for Deep Learning in Fundamental Physics
• Machine Learning in High Energy Physics: A review of heavy-flavor jet tagging at the LHC
• Search for nonresonant Higgs boson pair production in final state with two bottom quarks and two tau leptons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for flavor-changing neutral current interactions of the top quark and Higgs boson in final states with two photons in proton-proton collisions at $\sqrt{s}=$ 13 TeV
• Search for heavy neutral leptons in final states with electrons, muons, and hadronically decaying tau leptons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Enriching the physics program of the CMS experiment via data scouting and data parking
• Search for long-lived particles decaying to jets with displaced vertices in proton-proton collisions at $\sqrt{s}=$ 13 TeV
• Search for Higgs boson pairs decaying to WW*WW*, WW*$\tau \tau$, and $\tau \tau \tau \tau$ in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for a charged Higgs boson decaying into a heavy neutral Higgs boson and a W boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Searching for heavy leptoquarks at a muon collider
• A new calibration method for charm jet identification validated with proton-proton collision events at $\sqrt{s}$ = 13 TeV
• Phenomenology of Scalar Leptoquarks at the LHC in Explaining the Radiative Neutrino Mass, Muon $g-2$ and Lepton Flavour Violating Observables
• HHH Whitepaper
• Measurement of the differential $\mathrm{t}\bar{\mathrm{t}}$ production cross section as a function of the jet mass and extraction of the top quark mass in hadronic decays of boosted top quarks
• Measurement of the cross section of top quark-antiquark pair production in association with a W boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for the lepton-flavor violating decay of the Higgs boson and additional Higgs bosons in the e$\mu$ final state in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Searches for Higgs boson production through decays of heavy resonances
• Search for a massive scalar resonance decaying to a light scalar and a Higgs boson in the four b quarks final state with boosted topology
• Measurement of the top quark mass using a profile likelihood approach with the lepton+jets final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Inclusive and differential cross section measurements of single top quark production in association with a Z boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Tagging more quark jet flavours at FCC-ee at 91 GeV with a transformer-based neural network
• Probing effective field theory operators in the associated production of top quarks with a Z boson in multilepton final states at $\sqrt{s}=$ 13 TeV
• First measurement of the top quark pair production cross section in proton-proton collisions at $\sqrt{s}$ = 13.6 TeV
• Search for W' bosons decaying to a top and a bottom quark at $\sqrt{s}=$ 13 TeV in the hadronic final state
• On the BSM reach of four top production at the LHC
• On Drell-Yan production of scalar leptoquarks coupling to heavy-quark flavours
• Evidence for four-top quark production in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for narrow resonances in the b-tagged dijet mass spectrum in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for pair-produced vector-like leptons in final states with third-generation leptons and at least three b quark jets in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Observation of WW$\gamma$ production and search for H$\gamma$ production in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for $CP$ violation in ttH and tH production in multilepton channels in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Drell-Yan production in third-generation gauge vector leptoquark models at NLO+PS in QCD
• Jet Tagging with More-Interaction Particle Transformer
• Jet Discrimination with Quantum Complete Graph Neural Network
• Measurement of the Higgs boson inclusive and differential fiducial production cross sections in the diphoton decay channel with pp collisions at $\sqrt{s}$ = 13 TeV
• Maximum performance of strange-jet tagging at hadron colliders
• Search for long-lived particles using displaced vertices and missing transverse momentum in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for heavy pseudoscalar and scalar bosons decaying to a top quark pair in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for exotic decays of the Higgs boson to a pair of pseudoscalars in the $\mu \mu$bb and $\tau \tau$bb final states
• Search for heavy resonances decaying to a pair of Lorentz-boosted Higgs bosons in final states with leptons and a bottom quark pair at $\sqrt{s}=$ 13 TeV
• Search for Higgs boson pair production with one associated vector boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurements of the electroweak diboson production cross sections in proton-proton collisions at $\sqrt{s}=$ 5.02 TeV using leptonic decays
• Measurement of the $\mathrm{t}\bar{\mathrm{t}}$H and tH production rates in the H $\to$ $\mathrm{b}\bar{\mathrm{b}}$ decay channel using proton-proton collision data at $\sqrt{s}$ = 13 TeV
• Measurements of inclusive and differential cross sections for top quark production in association with a Z boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurement of inclusive and differential cross sections for single top quark production in association with a W boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for Higgs boson pair production in the $\mathrm{b}\overline{\mathrm{b}}$W${}^{+}$W${}^{-}$ decay mode in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for dark QCD with emerging jets in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Interpretable deep learning models for the inference and classification of LHC data
• Search for a new resonance decaying into two spin-0 bosons in a final state with two photons and two bottom quarks in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for physics beyond the standard model in top quark production with additional leptons in the context of effective field theory
• Observation of same-sign WW production from double parton scattering in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Portable acceleration of CMS computing workflows with coprocessors as a service
• Search for new Higgs bosons via same-sign top quark pair production in association with a jet in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Leveraging on-shell interference to search for FCNCs of the top quark and the Z boson
• Role of polarizations and spin-spin correlations of $W^{\prime }s$ in $e^{-}{e}^{+}\to W^{-}{W}^{+}$ at $\sqrt{s}=250$ GeV to probe anomalous $W^{-}{W}^{+}Z/\gamma$ couplings
• Observation of electroweak W${}^{+}$W${}^{-}$ pair production in association with two jets in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Attention to the strengths of physical interactions: Transformer and graph-based event classification for particle physics experiments
• Search for a resonance decaying to a W boson and a photon in proton-proton collisions at $\sqrt{s}$ = 13 TeV using leptonic W boson decays
• Performance of heavy-flavour jet identification in Lorentz-boosted topologies in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurement of boosted Higgs bosons produced via vector boson fusion or gluon fusion in the H $\to$ $\mathrm{b}\bar{\mathrm{b}}$ decay mode using LHC proton-proton collision data at $\sqrt{s}$ = 13 TeV
• Mutually elusive: vectorlike antileptons and leptoquarks
• Muon identification using multivariate techniques in the CMS experiment in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurement of inclusive and differential cross sections for W${}^{+}$W${}^{-}$ production in proton-proton collisions at $\sqrt{s}$ = 13.6 TeV
• A search for bottom-type vector-like quark pair production in dileptonic and fully hadronic final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurement of inclusive and differential cross sections of single top quark production in association with a W boson in proton-proton collisions at $\sqrt{s}$ = 13.6 TeV
• Measurement of the $\mathrm{t}\bar{\mathrm{t}}$ charge asymmetry in events with highly Lorentz-boosted top quarks in pp collisions at $\sqrt{s}$ = 13 TeV
• Search for pair production of scalar and vector leptoquarks decaying to muons and bottom quarks in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Refining fast simulation using machine learning
• Measurement of the Higgs boson production via vector boson fusion and its decay into bottom quarks in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for flavor-changing neutral current interactions of the top quark mediated by a Higgs boson in proton-proton collisions at 13 TeV
• Search for top squark pair production in a final state with at least one hadronically decaying tau lepton in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Jet Substructure Probe on Scalar Leptoquark Models via Top Polarization
• Measurement of WWZ and ZH production cross sections at $\sqrt{s}$ = 13 and 13.6 TeV
• Inclusive and differential cross section measurements of $\mathrm{t}\bar{\mathrm{t}}\mathrm{b}\bar{\mathrm{b}}$ production in the lepton+jets channel at $\sqrt{s}$ = 13 TeV
• Constraints on standard model effective field theory for a Higgs boson produced in association with W or Z bosons in the H $\to \mathrm{b}\bar{\mathrm{b}}$ decay channel in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for scalar leptoquarks produced via $\tau$-lepton-quark scattering in pp collisions at $\sqrt{s}$ = 13 TeV
• Search for ZZ and ZH production in the $\mathrm{b}\bar{\mathrm{b}}\mathrm{b}\bar{\mathrm{b}}$ final state using proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for a high-mass dimuon resonance produced in association with b quark jets at $\sqrt{s}$ = 13 TeV
• Transforming jet flavour tagging at ATLAS
• Search for pair production of heavy particles decaying to a top quark and a gluon in the lepton+jets final state in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Simultaneous probe of the charm and bottom quark Yukawa couplings using $\mathrm{t}\bar{\mathrm{t}}$H events
• Evidence for tWZ production in proton-proton collisions at $\sqrt{s}$ = 13 TeV in multilepton final states
• Advances in Multi-Variate Analysis Methods for New Physics Searches at the Large Hadron Collider
• TeV-scale leptoquark searches at the LHC and their E${}_6$SSM Interpretation
• Study of WH production through vector boson scattering and extraction of the relative sign of the W and Z couplings to the Higgs boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for $CP$ violation in events with top quarks and Z bosons at $\sqrt{s}$ = 13 and 13.6 TeV
• Search for charged-lepton flavor violation in the production and decay of top quarks using trilepton final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurement of the inclusive WZ production cross section in pp collisions at $\sqrt{s}$ = 13.6 TeV
• Search for a new scalar resonance decaying to a Higgs boson and another new scalar particle in the final state with two bottom quarks and two photons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for resonances in events with photon and jet final states in proton-proton collisions at $sqrts$ = 13 TeV
• Search for dark matter production in association with a single top quark in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for a cH signal in the associated production of at least one charm quark with a Higgs boson in the diphoton decay channel in pp collisions at $\sqrt{s}$ = 13 TeV
• Search for an exotic decay of the 125 GeV Higgs boson to a pair of light pseudoscalars in the final state of two muons and two c-quarks in proton-proton collisions at $\sqrt{s}=13\text{TeV}$ with CMS Open Data
• Probing Compressed Mass Spectrum Supersymmetry at the LHC with the Vector Boson Fusion Topology
• Development of systematic uncertainty-aware neural network trainings for binned-likelihood analyses at the LHC
• A method for correcting the substructure of multiprong jets using the Lund jet plane
• Search for charged lepton flavor violating Z and Z' boson decays in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for W' bosons decaying to a top and a bottom quark in leptonic final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• General search for supersymmetric particles in scenarios with compressed mass spectra using proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for resonant pair production of Higgs bosons in the $\mathrm{b}\bar{\mathrm{b}}\mathrm{b}\bar{\mathrm{b}}$ final state using large-area jets in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Probing Light Scalars and Vector-like Quarks at the High-Luminosity LHC
• Search for $\gamma$H production and constraints on the Yukawa couplings of light quarks to the Higgs boson
• Study of same-sign W boson scattering and anomalous couplings in events with one tau lepton from pp collisions at $\sqrt{s}$ = 13 TeV
• Search for nonresonant new physics signals in high-mass dilepton events produced in association with b-tagged jets in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for Higgs boson production at high transverse momentum in the WW decay channel in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Probing the flavour structure of dimension-6 EFT operators in multilepton final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for excited tau leptons in the $\tau \tau \gamma$ final state in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for high-mass resonances in a final state comprising a gluon and two hadronically decaying W bosons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for bottom quark associated production of the standard model Higgs boson in final states with leptons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for soft unclustered energy patterns produced in association with a W or Z boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• B-jet Tagging Using a Hybrid Edge Convolution and Transformer Architecture
• Model-independent measurement of the Higgs boson associated production with two jets and decaying to a pair of W bosons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurements of ttW differential cross sections and the leptonic charge asymmetry at $\sqrt{s}$ = 13 TeV
• Observation of tWZ production at the CMS experiment
• Exploring an image-based $b$-jet tagging method using convolution neural networks
• Inclusive and differential measurements of the $\mathrm{t}\bar{\mathrm{t}}\gamma$ cross section and the $\mathrm{t}\bar{\mathrm{t}}\gamma$ / $\mathrm{t}\bar{\mathrm{t}}$ cross section ratio in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Measurement of the top-quark production cross-section and charge asymmetry at LHCb
• Deep-learning jet flavor tagging for precision hadronic Higgs measurements at future $e^{+}{e}^{-}$ Higgs factories
• Measurement of the Higgs boson total decay width using the H $\to$ WW $\to$ e$\nu \mu \nu$ decay channel in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for dark matter produced in association with a Higgs boson decaying to bottom quarks in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for new physics in jet multiplicity patterns of multilepton events at $\sqrt{s}$ = 13 TeV
• Search for bosons of an extended Higgs sector in b quark final states in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• A FAIR and AI-ready Higgs boson decay dataset
• Search for charged-lepton flavour violation in top quark interactions with an up-type quark, a muon, and a $\tau$ lepton in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Machine learning-based b-jet tagging in $pp$ collisions at $\sqrt{s}=13$ TeV
• Measurement of the W boson decay branching fraction ratio $\mathcal{B}$(W $\to$ cq) / $\mathcal{B}$(W $\to$ $\mathrm{q}{\bar{\mathrm{q}}}^{\prime }$) in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Search for a boosted Higgs boson decaying to bottom quark pairs in association with a W or Z boson in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Combination of searches for heavy vector boson resonances in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Illuminating all-hadronic final states with a photon: Exotic decays of the Higgs boson to four bottom quarks in vector boson fusion plus gamma at hadron colliders
• Search for dark matter produced in association with a Higgs boson decaying to a $\tau$ lepton pair in proton-proton collisions at $\sqrt{s}$ = 13 TeV