Performance of a Geometric Deep Learning Pipeline for HL-LHC Particle Tracking
Xiangyang Ju,Daniel Murnane,Paolo Calafiura,Nicholas Choma,Sean Conlon,Steve Farrell,Yaoyuan Xu,Maria Spiropulu,Jean-Roch Vlimant,Adam Aurisano,V Hewes,Giuseppe Cerati,Lindsey Gray,Thomas Klijnsma,Jim Kowalkowski,Markus Atkinson,Mark Neubauer,Gage DeZoort,Savannah Thais,Aditi Chauhan,Alex Schuy,Shih-Chieh Hsu,Alex Ballow,and Alina Lazar
DOI: https://doi.org/10.1140/epjc/s10052-021-09675-8
2021-09-22
Abstract:The <a class="link-external link-http" href="http://Exa.TrkX" rel="external noopener nofollow">this http URL</a> project has applied geometric learning concepts such as metric learning and graph neural networks to HEP particle tracking. <a class="link-external link-http" href="http://Exa.TrkX" rel="external noopener nofollow">this http URL</a>'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 <a class="link-external link-http" href="http://Exa.TrkX" rel="external noopener nofollow">this http URL</a> 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.
Data Analysis, Statistics and Probability,Machine Learning,High Energy Physics - Experiment