11–15 Mar 2024
Charles B. Wang Center, Stony Brook University
US/Eastern timezone

Improving Computational Performance of a GNN Track Reconstruction Pipeline for ATLAS

12 Mar 2024, 12:10
20m
Theatre ( Charles B. Wang Center, Stony Brook University )

Theatre

Charles B. Wang Center, Stony Brook University

100 Circle Rd, Stony Brook, NY 11794
Oral Track 1: Computing Technology for Physics Research Track 1: Computing Technology for Physics Research

Speaker

Daniel Thomas Murnane (Lawrence Berkeley National Lab. (US))

Description

Track reconstruction is an essential element of modern and future collider experiments, including within the ATLAS detector. The HL-LHC upgrade of the ATLAS detector brings an unprecedented tracking challenge, both in terms of number of silicon hit cluster readouts, and throughput required for both high level trigger and offline 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 provide comparisons of physics and computational performance across a variety of model configurations, as well as optimizations that reduce computational cost without significantly affecting physics performance. These include the use of structured pruning, knowledge distillation, simplified and customized convolutional kernels, regional tracking approaches, and GPU-optimized graph segmentation techniques.

Significance

This represents the first set of computational performance results for the novel GNN-based tracking pipeline for the upgraded ATLAS ITk subdetector. This proves that the approach is realistic for both physics requirements and compute budgets.

References

https://cds.cern.ch/record/2882507/files/ATL-SOFT-PROC-2023-047.pdf
https://arxiv.org/abs/2103.06995
https://arxiv.org/abs/2103.00916

Experiment context, if any ATLAS

Primary authors

Alexander Shmakov (University of California Irvine (US)) Alexis Vallier (L2I Toulouse, CNRS/IN2P3, UT3) Dr Christophe COLLARD (Laboratoire des 2 Infinis - Toulouse, CNRS / Univ. Paul Sabatier) Daniel Thomas Murnane (Lawrence Berkeley National Lab. (US)) Heberth Torres (L2I Toulouse, CNRS/IN2P3, UT3) 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)) Mark Neubauer (Univ. Illinois at Urbana Champaign (US)) Minh-Tuan Pham (University of Wisconsin Madison (US)) Paolo Calafiura (Lawrence Berkeley National Lab. (US)) Sylvain Caillou (Centre National de la Recherche Scientifique (FR)) Xiangyang Ju (Lawrence Berkeley National Lab. (US))

Presentation materials