Supervised jet clustering with graph neural networks for Lorentz boosted bosons
Abstract
Jet clustering is traditionally an unsupervised learning task because there is no unique way to associate hadronic final states with the quark and gluon degrees of freedom that generated them. However, for uncolored particles like , , and Higgs bosons, it is possible to approximately (though not exactly) associate final state hadrons to their ancestor. By labeling simulated final state hadrons as descending from an uncolored particle, it is possible to train a supervised learning method to create boson jets. Such a method would operate on individual particles and identify connections between particles originating from the same uncolored particle. Graph neural networks are well-suited for this purpose as they can act on unordered sets and naturally create strong connections between particles with the same label. These networks are used to train a supervised jet clustering algorithm. The kinematic properties of these graph jets better match the properties of simulated Lorentz-boosted bosons. Furthermore, the graph jets contain more information for discriminating jets from generic quark jets. This work marks the beginning of a new exploration in jet physics to use machine learning to optimize the construction of jets and not only the observables computed from jet constituents.
- Authors:
- Publication Date:
- Research Org.:
- Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States). National Energy Research Scientific Computing Center (NERSC)
- Sponsoring Org.:
- USDOE Office of Science (SC)
- OSTI Identifier:
- 1781715
- Alternate Identifier(s):
- OSTI ID: 1764524
- Grant/Contract Number:
- AC02-05CH11231
- Resource Type:
- Published Article
- Journal Name:
- Physical Review D
- Additional Journal Information:
- Journal Name: Physical Review D Journal Volume: 102 Journal Issue: 7; Journal ID: ISSN 2470-0010
- Publisher:
- American Physical Society
- Country of Publication:
- United States
- Language:
- English
- Subject:
- 72 PHYSICS OF ELEMENTARY PARTICLES AND FIELDS; quantum chromodynamics; quark & gluon jets; gauge bosons; hadrons; artificial neural networks; hadron colliders; machine learning
Citation Formats
Ju, Xiangyang, and Nachman, Benjamin. Supervised jet clustering with graph neural networks for Lorentz boosted bosons. United States: N. p., 2020.
Web. doi:10.1103/PhysRevD.102.075014.
Ju, Xiangyang, & Nachman, Benjamin. Supervised jet clustering with graph neural networks for Lorentz boosted bosons. United States. https://doi.org/10.1103/PhysRevD.102.075014
Ju, Xiangyang, and Nachman, Benjamin. Tue .
"Supervised jet clustering with graph neural networks for Lorentz boosted bosons". United States. https://doi.org/10.1103/PhysRevD.102.075014.
@article{osti_1781715,
title = {Supervised jet clustering with graph neural networks for Lorentz boosted bosons},
author = {Ju, Xiangyang and Nachman, Benjamin},
abstractNote = {Jet clustering is traditionally an unsupervised learning task because there is no unique way to associate hadronic final states with the quark and gluon degrees of freedom that generated them. However, for uncolored particles like W, Z, and Higgs bosons, it is possible to approximately (though not exactly) associate final state hadrons to their ancestor. By labeling simulated final state hadrons as descending from an uncolored particle, it is possible to train a supervised learning method to create boson jets. Such a method would operate on individual particles and identify connections between particles originating from the same uncolored particle. Graph neural networks are well-suited for this purpose as they can act on unordered sets and naturally create strong connections between particles with the same label. These networks are used to train a supervised jet clustering algorithm. The kinematic properties of these graph jets better match the properties of simulated Lorentz-boosted W bosons. Furthermore, the graph jets contain more information for discriminating W jets from generic quark jets. This work marks the beginning of a new exploration in jet physics to use machine learning to optimize the construction of jets and not only the observables computed from jet constituents.},
doi = {10.1103/PhysRevD.102.075014},
journal = {Physical Review D},
number = 7,
volume = 102,
place = {United States},
year = {Tue Oct 13 00:00:00 EDT 2020},
month = {Tue Oct 13 00:00:00 EDT 2020}
}
https://doi.org/10.1103/PhysRevD.102.075014
Works referenced in this record:
Search for a massive resonance decaying to a pair of Higgs bosons in the four b quark final state in proton–proton collisions at
journal, June 2018
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Physics Letters B, Vol. 781
The anti- k t jet clustering algorithm
journal, April 2008
- Cacciari, Matteo; Salam, Gavin P.; Soyez, Gregory
- Journal of High Energy Physics, Vol. 2008, Issue 04
Pulling out all the tops with computer vision and deep learning
journal, October 2018
- Macaluso, Sebastian; Shih, David
- Journal of High Energy Physics, Vol. 2018, Issue 10
Learning representations of irregular particle-detector geometry with distance-weighted graph networks
journal, July 2019
- Qasim, Shah Rukh; Kieseler, Jan; Iiyama, Yutaro
- The European Physical Journal C, Vol. 79, Issue 7
Erratum to: Search for diboson resonances in hadronic final states in 139 fb−1 of pp collisions at $$ \sqrt{\mathrm{s}} $$ = 13 TeV with the ATLAS detector
journal, June 2020
- Aad, G.; Abbott, B.; Abbott, D. C.
- Journal of High Energy Physics, Vol. 2020, Issue 6
Deep-learned Top Tagging with a Lorentz Layer
journal, January 2018
- Butter, Anja; Kasieczka, Gregor; Plehn, Tilman
- SciPost Physics, Vol. 5, Issue 3
Electromagnetic showers beyond shower shapes
journal, January 2020
- de Oliveira, Luke; Nachman, Benjamin; Paganini, Michela
- Nuclear Instruments and Methods in Physics Research Section A: Accelerators, Spectrometers, Detectors and Associated Equipment, Vol. 951
Combination of CMS searches for heavy resonances decaying to pairs of bosons or leptons
journal, November 2019
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Physics Letters B, Vol. 798
How much information is in a jet?
journal, June 2017
- Datta, Kaustuv; Larkoski, Andrew
- Journal of High Energy Physics, Vol. 2017, Issue 6
End-to-end jet classification of quarks and gluons with the CMS Open Data
journal, October 2020
- Andrews, M.; Alison, J.; An, S.
- Nuclear Instruments and Methods in Physics Research Section A: Accelerators, Spectrometers, Detectors and Associated Equipment, Vol. 977
Power counting to better jet observables
journal, December 2014
- Larkoski, Andrew J.; Moult, Ian; Neill, Duff
- Journal of High Energy Physics, Vol. 2014, Issue 12
End-to-End Physics Event Classification with CMS Open Data: Applying Image-Based Deep Learning to Detector Data for the Direct Classification of Collision Events at the LHC
journal, March 2020
- Andrews, M.; Paulini, M.; Gleyzer, S.
- Computing and Software for Big Science, Vol. 4, Issue 1
JEDI-net: a jet identification algorithm based on interaction networks
journal, January 2020
- Moreno, Eric A.; Cerri, Olmo; Duarte, Javier M.
- The European Physical Journal C, Vol. 80, Issue 1
PYTHIA 6.4 physics and manual
journal, May 2006
- Sjöstrand, Torbjörn; Mrenna, Stephen; Skands, Peter
- Journal of High Energy Physics, Vol. 2006, Issue 05
Boosted objects: a probe of beyond the standard model physics
journal, June 2011
- Abdesselam, A.; Belyaev, A.; Kuutmann, E. Bergeaas
- The European Physical Journal C, Vol. 71, Issue 6
Search for resonances decaying to a pair of Higgs bosons in the bb¯$$ \overline{\mathrm{b}} $$qq¯$$ \overline{\mathrm{q}} $$’ℓν final state in proton-proton collisions at s$$ \sqrt{s} $$ = 13 TeV
journal, October 2019
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Journal of High Energy Physics, Vol. 2019, Issue 10
Deep learning in color: towards automated quark/gluon jet discrimination
journal, January 2017
- Komiske, Patrick T.; Metodiev, Eric M.; Schwartz, Matthew D.
- Journal of High Energy Physics, Vol. 2017, Issue 1
Deep-learning top taggers or the end of QCD?
journal, May 2017
- Kasieczka, Gregor; Plehn, Tilman; Russell, Michael
- Journal of High Energy Physics, Vol. 2017, Issue 5
Boosted and tagging with jet charge and deep learning
journal, March 2020
- Chen, Yu-Chen Janice; Chiang, Cheng-Wei; Cottin, Giovanna
- Physical Review D, Vol. 101, Issue 5
Jet charge and machine learning
journal, October 2018
- Fraser, Katherine; Schwartz, Matthew D.
- Journal of High Energy Physics, Vol. 2018, Issue 10
FastJet user manual: (for version 3.0.2)
journal, March 2012
- Cacciari, Matteo; Salam, Gavin P.; Soyez, Gregory
- The European Physical Journal C, Vol. 72, Issue 3
Fuzzy jets
journal, June 2016
- Mackey, Lester; Nachman, Benjamin; Schwartzman, Ariel
- Journal of High Energy Physics, Vol. 2016, Issue 6
Topology Classification with Deep Learning to Improve Real-Time Event Selection at the LHC
journal, August 2019
- Nguyen, T. Q.; Weitekamp, D.; Anderson, D.
- Computing and Software for Big Science, Vol. 3, Issue 1
Energy flow polynomials: a complete linear basis for jet substructure
journal, April 2018
- Komiske, Patrick T.; Metodiev, Eric M.; Thaler, Jesse
- Journal of High Energy Physics, Vol. 2018, Issue 4
Identification of heavy, energetic, hadronically decaying particles using machine-learning techniques
journal, June 2020
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Journal of Instrumentation, Vol. 15, Issue 06
Inclusive Search for a Highly Boosted Higgs Boson Decaying to a Bottom Quark-Antiquark Pair
journal, February 2018
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Physical Review Letters, Vol. 120, Issue 7
Calorimetry with deep learning: particle simulation and reconstruction for collider physics
journal, July 2020
- Belayneh, Dawit; Carminati, Federico; Farbin, Amir
- The European Physical Journal C, Vol. 80, Issue 7
Jet substructure at the Large Hadron Collider
journal, December 2019
- Kogler, Roman; Nachman, Benjamin; Schmidt, Alexander
- Reviews of Modern Physics, Vol. 91, Issue 4
Systematics of quark/gluon tagging
journal, July 2017
- Gras, Philippe; Höche, Stefan; Kar, Deepak
- Journal of High Energy Physics, Vol. 2017, Issue 7
Boosted objects and jet substructure at the LHC. Report of BOOST2012, held at IFIC Valencia, 23rd–27th of July 2012
journal, March 2014
- Altheimer, A.; Arce, A.; Asquith, L.
- The European Physical Journal C, Vol. 74, Issue 3
Recombination algorithms and jet substructure: Pruning as a tool for heavy particle searches
journal, May 2010
- Ellis, Stephen D.; Vermilion, Christopher K.; Walsh, Jonathan R.
- Physical Review D, Vol. 81, Issue 9
Performance of top-quark and $$\varvec{W}$$ W -boson tagging with ATLAS in Run 2 of the LHC
journal, April 2019
- Aaboud, M.; Aad, G.; Abbott, B.
- The European Physical Journal C, Vol. 79, Issue 5
Identifying boosted objects with N-subjettiness
journal, March 2011
- Thaler, Jesse; Van Tilburg, Ken
- Journal of High Energy Physics, Vol. 2011, Issue 3
Jet flavor classification in high-energy physics with deep neural networks
journal, December 2016
- Guest, Daniel; Collado, Julian; Baldi, Pierre
- Physical Review D, Vol. 94, Issue 11
Jet substructure at the Large Hadron Collider: A review of recent advances in theory and machine learning
journal, November 2019
- Larkoski, Andrew J.; Moult, Ian; Nachman, Benjamin
- Physics Reports
Recursive Neural Networks in Quark/Gluon Tagging
journal, June 2018
- Cheng, Taoli
- Computing and Software for Big Science, Vol. 2, Issue 1
Automating the construction of jet observables with machine learning
journal, November 2019
- Datta, Kaustuv; Larkoski, Andrew; Nachman, Benjamin
- Physical Review D, Vol. 100, Issue 9
Search for diboson resonances in hadronic final states in 139 fb−1 of pp collisions at s$$ \sqrt{s} $$ = 13 TeV with the ATLAS detector
journal, September 2019
- Aad, G.; Abbott, B.; Abbott, D. C.
- Journal of High Energy Physics, Vol. 2019, Issue 9
Jet trimming
journal, February 2010
- Krohn, David; Thaler, Jesse; Wang, Lian-Tao
- Journal of High Energy Physics, Vol. 2010, Issue 2
A search for resonances decaying into a Higgs boson and a new particle X in the XH → qqbb final state with the ATLAS detector
journal, April 2018
- Aaboud, M.; Aad, G.; Abbott, B.
- Physics Letters B, Vol. 779
Deep Learning and Its Application to LHC Physics
journal, October 2018
- Guest, Dan; Cranmer, Kyle; Whiteson, Daniel
- Annual Review of Nuclear and Particle Science, Vol. 68, Issue 1
Towards an understanding of the correlations in jet substructure: Report of BOOST2013, hosted by the University of Arizona, 12th–16th of August 2013
journal, September 2015
- Adams, D.; Arce, A.; Asquith, L.
- The European Physical Journal C, Vol. 75, Issue 9
Jet Substructure as a New Higgs-Search Channel at the Large Hadron Collider
journal, June 2008
- Butterworth, Jonathan M.; Davison, Adam R.; Rubin, Mathieu
- Physical Review Letters, Vol. 100, Issue 24
Boosting H → b b ¯ $$ H\to b\overline{b} $$ with machine learning
journal, October 2018
- Lin, Joshua; Freytsis, Marat; Moult, Ian
- Journal of High Energy Physics, Vol. 2018, Issue 10
Looking Inside Jets: An Introduction to Jet Substructure and Boosted-object Phenomenology
book, January 2019
- Marzani, Simone; Soyez, Gregory; Spannowsky, Michael
- Lecture Notes in Physics, Vol. 958
Soft drop
journal, May 2014
- Larkoski, Andrew J.; Marzani, Simone; Soyez, Gregory
- Journal of High Energy Physics, Vol. 2014, Issue 5
Maximizing boosted top identification by minimizing N-subjettiness
journal, February 2012
- Thaler, Jesse; Van Tilburg, Ken
- Journal of High Energy Physics, Vol. 2012, Issue 2
A multi-dimensional search for new heavy resonances decaying to boosted $$\text{ W }{}{}$$W $$\text{ W }{}{}$$W , $$\text{ W }{}{}$$W $$\text{ Z }{}{}$$Z , or $$\text{ Z }{}{}$$Z $$\text{ Z }{}{}$$Z boson pairs in the dijet final state at 13 $$\text {Te}\text {V}$$Te
journal, March 2020
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- The European Physical Journal C, Vol. 80, Issue 3
Jet substructure at the Tevatron and LHC: new results, new tools, new benchmarks
journal, May 2012
- Altheimer, A.; Arora, S.; Asquith, L.
- Journal of Physics G: Nuclear and Particle Physics, Vol. 39, Issue 6
Deep neural network for pixel-level electromagnetic particle identification in the MicroBooNE liquid argon time projection chamber
journal, May 2019
- Adams, C.; Alrashed, M.; An, R.
- Physical Review D, Vol. 99, Issue 9
A brief introduction to PYTHIA 8.1
journal, June 2008
- Sjöstrand, Torbjörn; Mrenna, Stephen; Skands, Peter
- Computer Physics Communications, Vol. 178, Issue 11
In situ calibration of large-radius jet energy and mass in 13 TeV proton–proton collisions with the ATLAS detector
journal, February 2019
- Aaboud, M.; Aad, G.; Abbott, B.
- The European Physical Journal C, Vol. 79, Issue 2
Search for Low Mass Vector Resonances Decaying to Quark-Antiquark Pairs in Proton-Proton Collisions at
journal, September 2017
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Physical Review Letters, Vol. 119, Issue 11
Search for low mass vector resonances decaying into quark-antiquark pairs in proton-proton collisions at
journal, December 2019
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Physical Review D, Vol. 100, Issue 11
Optimal Jet Finder
journal, September 2003
- Grigoriev, D. Yu.; Jankowski, E.; Tkachov, F. V.
- Computer Physics Communications, Vol. 155, Issue 1
The Machine Learning landscape of top taggers
journal, January 2019
- Kasieczka, Gregor; Plehn, Tilman; Butter, Anja
- SciPost Physics, Vol. 7, Issue 1
Pileup Mitigation with Machine Learning (PUMML)
journal, December 2017
- Komiske, Patrick T.; Metodiev, Eric M.; Nachman, Benjamin
- Journal of High Energy Physics, Vol. 2017, Issue 12
Towards jetography
journal, May 2010
- Salam, Gavin P.
- The European Physical Journal C, Vol. 67, Issue 3-4
Jet-images — deep learning edition
journal, July 2016
- de Oliveira, Luke; Kagan, Michael; Mackey, Lester
- Journal of High Energy Physics, Vol. 2016, Issue 7
Quark-gluon tagging: Machine learning vs detector
journal, January 2019
- Kasieczka, Gregor; Kiefer, Nicholas; Plehn, Tilman
- SciPost Physics, Vol. 6, Issue 6
Jet-images: computer vision inspired techniques for jet tagging
journal, February 2015
- Cogan, Josh; Kagan, Michael; Strauss, Emanuel
- Journal of High Energy Physics, Vol. 2015, Issue 2
Identification of heavy-flavour jets with the CMS detector in pp collisions at 13 TeV
journal, May 2018
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Journal of Instrumentation, Vol. 13, Issue 05
Identification of boosted Higgs bosons decaying into b-quark pairs with the ATLAS detector at 13 $$\text {TeV}$$TeV
journal, October 2019
- Aad, G.; Abbott, B.; Abbott, D. C.
- The European Physical Journal C, Vol. 79, Issue 10
Novel jet observables from machine learning
journal, March 2018
- Datta, Kaustuv; Larkoski, Andrew J.
- Journal of High Energy Physics, Vol. 2018, Issue 3
Unveiling CP property of top-Higgs coupling with graph neural networks at the LHC
journal, March 2020
- Ren, Jie; Wu, Lei; Yang, Jin Min
- Physics Letters B, Vol. 802
Search for light resonances decaying to boosted quark pairs and produced in association with a photon or a jet in proton–proton collisions at with the ATLAS detector
journal, January 2019
- Aaboud, M.; Aad, G.; Abbott, B.
- Physics Letters B, Vol. 788
Playing tag with ANN: boosted top identification with pattern recognition
journal, July 2015
- Almeida, Leandro G.; Backović, Mihailo; Cliche, Mathieu
- Journal of High Energy Physics, Vol. 2015, Issue 7
CapsNets continuing the convolutional quest
journal, January 2020
- Diefenbacher, Sascha; Frost, Hermann; Kasieczka, Gregor
- SciPost Physics, Vol. 8, Issue 2
Interpretable deep learning for two-prong jet classification with jet spectra
journal, July 2019
- Chakraborty, Amit; Lim, Sung Hak; Nojiri, Mihoko M.
- Journal of High Energy Physics, Vol. 2019, Issue 7
Searching for exotic particles in high-energy physics with deep learning
journal, July 2014
- Baldi, P.; Sadowski, P.; Whiteson, D.
- Nature Communications, Vol. 5, Issue 1
Towards an understanding of jet substructure
journal, September 2013
- Dasgupta, Mrinal; Fregoso, Alessandro; Marzani, Simone
- Journal of High Energy Physics, Vol. 2013, Issue 9
Neural network-based top tagger with two-point energy correlations and geometry of soft emissions
journal, July 2020
- Chakraborty, Amit; Lim, Sung Hak; Nojiri, Mihoko M.
- Journal of High Energy Physics, Vol. 2020, Issue 7
Pileup mitigation at the Large Hadron Collider with graph neural networks
journal, July 2019
- Arjona Martínez, J.; Cerri, O.; Spiropulu, M.
- The European Physical Journal Plus, Vol. 134, Issue 7
QCD-aware recursive neural networks for jet physics
journal, January 2019
- Louppe, Gilles; Cho, Kyunghyun; Becot, Cyril
- Journal of High Energy Physics, Vol. 2019, Issue 1
Search for heavy resonances decaying into two Higgs bosons or into a Higgs boson and a W or Z boson in proton-proton collisions at 13 TeV
journal, January 2019
- Sirunyan, A. M.; Tumasyan, A.; Adam, W.
- Journal of High Energy Physics, Vol. 2019, Issue 1
A deep neural network to search for new long-lived particles decaying to jets
journal, August 2020
- ,
- Machine Learning: Science and Technology, Vol. 1, Issue 3, 035012
Dijet Resonance Search with Weak Supervision Using Collisions in the ATLAS Detector
journal, September 2020
- Aad, G.; Abbott, B.; Abbott, D. C.
- Physical Review Letters, Vol. 125, Issue 13
Parton shower uncertainties in jet substructure analyses with deep neural networks
journal, January 2017
- Barnard, James; Dawe, Edmund Noel; Dolan, Matthew J.
- Physical Review D, Vol. 95, Issue 1