19–25 Oct 2024
Europe/Zurich timezone

Improving Computational Performance of ATLAS GNN Track Reconstruction Pipeline

24 Oct 2024, 16:51
18m
Room 1.A (Medium Hall A)

Room 1.A (Medium Hall A)

Talk Track 3 - Offline Computing Parallel (Track 3)

Speaker

Alina Lazar (Youngstown State University (US))

Description

Track reconstruction is an essential element of modern and future collider experiments, including the ATLAS detector. The HL-LHC upgrade of the ATLAS detector brings an unprecedented tracking reconstruction challenge, both in terms of the large number of silicon hit cluster readouts and the throughput required for budget-constrained track reconstruction. Traditional track reconstruction techniques often contain steps that scale combinatorically, which could be ameliorated with deep learning approaches. The GNN4ITk project has been shown to apply geometric deep learning algorithms for tracking to a similar level of physics performance with traditional techniques while scaling sub-quadratically. In this contribution, we compare the computational performance of a variety of pipeline configurations and machine learning inference methods. These include heuristic-and-ML-based graph segmentation techniques, GPU-based module map graph construction, and studies of high throughput graph convolutional kernels. In this contribution, we present benchmarks of latency, throughput, memory usage, and power consumption of each pipeline configuration.

Primary authors

Alexis Vallier (L2I Toulouse, CNRS/IN2P3, UT3) Alina Lazar (Youngstown State University (US)) Dr Christophe COLLARD (Laboratoire des 2 Infinis - Toulouse, CNRS / Univ. Paul Sabatier) Daniel Thomas Murnane (Niels Bohr Institute, University of Copenhagen) Heberth Torres (L2I Toulouse, CNRS/IN2P3, UT3) Jackson Carl Burzynski (Simon Fraser University (CA)) Jan Stark (Laboratoire des 2 Infinis - Toulouse, CNRS / Univ. Paul Sabatier (FR)) Jared Burleson (University of Illinois at Urbana-Champaign) Jay Chan (Lawrence Berkeley National Lab. (US)) Levi Condren (University of California Irvine (US)) Mark Stephen Neubauer (Univ. Illinois at Urbana-Champaign) Minh-Tuan Pham (University of Wisconsin Madison (US)) Paolo Calafiura (Lawrence Berkeley National Lab. (US)) Ryan Liu (UC Berkeley) Sylvain Caillou (Centre National de la Recherche Scientifique (FR)) Xiangyang Ju (Lawrence Berkeley National Lab. (US))

Presentation materials