Welcome to the new version of CaltechAUTHORS. Login is currently restricted to library staff. If you notice any issues, please email coda@library.caltech.edu
Published May 2021 | Submitted + Published
Journal Article Open

MLPF: efficient machine-learned particle-flow reconstruction using graph neural networks

Abstract

In general-purpose particle detectors, the particle-flow algorithm may be used to reconstruct a comprehensive particle-level view of the event by combining information from the calorimeters and the trackers, significantly improving the detector resolution for jets and the missing transverse momentum. In view of the planned high-luminosity upgrade of the CERN Large Hadron Collider (LHC), it is necessary to revisit existing reconstruction algorithms and ensure that both the physics and computational performance are sufficient in an environment with many simultaneous proton–proton interactions (pileup). Machine learning may offer a prospect for computationally efficient event reconstruction that is well-suited to heterogeneous computing platforms, while significantly improving the reconstruction quality over rule-based algorithms for granular detectors. We introduce MLPF, a novel, end-to-end trainable, machine-learned particle-flow algorithm based on parallelizable, computationally efficient, and scalable graph neural network optimized using a multi-task objective on simulated events. We report the physics and computational performance of the MLPF algorithm on a Monte Carlo dataset of top quark–antiquark pairs produced in proton–proton collisions in conditions similar to those expected for the high-luminosity LHC. The MLPF algorithm improves the physics response with respect to a rule-based benchmark algorithm and demonstrates computationally scalable particle-flow reconstruction in a high-pileup environment.

Additional Information

© The Author(s) 2021. This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/. Funded by SCOAP3. Received 05 February 2021; Accepted 19 April 2021; Published 02 May 2021. We would like to thank Guenther Dissertori for suggesting the idea of ML-driven PF reconstruction several years ago in private discussions. We thank our colleagues in the CMS Collaboration, especially in the Particle Flow, Physics Performance and Datasets, Offline and Computing, and Machine Learning groups, in particular Josh Bendavid, Kenichi Hatakeyama, Lindsey Gray, Jan Kieseler, Danilo Piparo, Gregor Kasieczka, Laurits Tani, and Juska Pekkanen, for helpful feedback in the course of this work. J. P. was supported by the Prime National Science Foundation (NSF) Tier2 award 1624356 and the U.S. Department of Energy (DOE), Office of Science, Office of High Energy Physics under Award No. DE-SC0011925 while at Caltech, and is currently supported by the Mobilitas Pluss Grant no. MOBTP187 of the Estonian Research Council. J. D. is supported by the DOE, Office of Science, Office of High Energy Physics Early Career Research program under Award No. DE-SC0021187 and by the DOE, Office of Advanced Scientific Computing Research under Award No. DE-SC0021396 (FAIR4HEP). M. P. is supported by the European Research Council (ERC) under the European Union's Horizon 2020 research and innovation program (Grant Agreement No. 772369). J-R. V. and M. S. are supported by the DOE, Office of Science, Office of High Energy Physics under Award No. DE-SC0011925, DE-SC0019227, and DE-AC02-07CH11359. J-R. V. was additionally partially supported the same ERC Grant as M. P. We are grateful to Caltech and the Kavli Foundation for their support of undergraduate student research in cross-cutting areas of machine learning and domain sciences. This work was conducted at "iBanks," the AI GPU cluster at Caltech, and on the NICPB GPU resources, supported by European Regional Development Fund through the CoE program Grant TK133. We acknowledge Nvidia, SuperMicro and the Kavli Foundation for their support of iBanks. Part of this work was also performed using the Pacific Research Platform Nautilus HyperCluster supported by NSF awards CNS-1730158, ACI-1540112, ACI-1541349, OAC-1826967, the University of California Office of the President, and the University of California San Diego's California Institute for Telecommunications and Information Technology/Qualcomm Institute. Thanks to CENIC for the 100 Gpbs networks. J. Pata: This work was partially carried out at Caltech. Data Availability Statement: This manuscript has associated data in a data repository. [Authors' comment: The simulated datasets used in this paper are available at https://doi.org/10.5281/zenodo.4559324 and at https://zenodo.org/communities/mpp-hep/.]

Attached Files

Published - Pata2021_Article_MLPFEfficientMachine-learnedPa.pdf

Submitted - 2101.08578.pdf

Files

2101.08578.pdf
Files (3.8 MB)
Name Size Download all
md5:932875d23cfab5be1f8b030d33f2733e
2.1 MB Preview Download
md5:16e234423c97b4df9f8140224bb2e5de
1.6 MB Preview Download

Additional details

Created:
August 20, 2023
Modified:
October 20, 2023