Regular Article – Experimental Physics
Performance of a geometric deep learning pipeline for HL-LHC particle tracking
Lawrence Berkeley National Laboratory, Berkeley, CA, USA
2 California Institute of Technology, Pasadena, CA, USA
3 University of Cincinnati, Cincinnati, OH, USA
4 Fermi National Accelerator Laboratory, Batavia, IL, USA
5 University of Illinois at Urbana-Champaign, Urbana, IL, USA
6 Princeton University, Princeton, NJ, USA
7 University of Washington, Seattle, WA, USA
8 Youngstown State University, Youngstown, OH, USA
Accepted: 21 September 2021
Published online: 6 October 2021
The Exa.TrkX project has applied geometric learning concepts such as metric learning and graph neural networks to HEP particle tracking. Exa.TrkX’s tracking pipeline groups detector measurements to form track candidates and filters them. The pipeline, originally developed using the TrackML dataset (a simulation of an LHC-inspired tracking detector), has been demonstrated on other detectors, including DUNE Liquid Argon TPC and CMS High-Granularity Calorimeter. This paper documents new developments needed to study the physics and computing performance of the Exa.TrkX pipeline on the full TrackML dataset, a first step towards validating the pipeline using ATLAS and CMS data. The pipeline achieves tracking efficiency and purity similar to production tracking algorithms. Crucially for future HEP applications, the pipeline benefits significantly from GPU acceleration, and its computational requirements scale close to linearly with the number of particles in the event.
© The Author(s) 2021
Open Access 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