19–25 Oct 2024
Europe/Zurich timezone

High Performance Graph Segmentation for ATLAS GNN Track Reconstruction

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

Room 1.A (Medium Hall A)

Talk Track 3 - Offline Computing Parallel (Track 3)

Speaker

Daniel Thomas Murnane (Niels Bohr Institute, University of Copenhagen)

Description

Graph neural networks and deep geometric learning have been successfully proven in the task of track reconstruction in recent years. The GNN4ITk project employs these techniques in the context of the ATLAS upgrade ITk detector to produce similar physics performance as traditional techniques, while scaling sub-quadratically. However, one current bottleneck in the throughput and physics performance of graph-based tracking is the final processing of classified graph edges into track candidates. This stage typically requires a trade-off between computational and physics performance. In this contribution, we present a variety of algorithms to ameliorate this trade-off, from heuristic approaches that use GPU-accelerated graph operations, to learned approaches such as hierarchical graph neural networks. Based on dedicated timing studies, we show that these graph segmentation algorithms are also well-suited to online track reconstruction.

Primary authors

Alexis Vallier (L2I Toulouse, CNRS/IN2P3, UT3) Alina Lazar (Youngstown State University (US)) Daniel Thomas Murnane (Niels Bohr Institute, University of Copenhagen) Daniel Whiteson (University of California Irvine (US)) Levi Condren (University of California Irvine (US)) Minh-Tuan Pham (University of Wisconsin Madison (US)) Ryan Liu (Lawrence Berkeley National Lab. (US)) Xiangyang Ju (Lawrence Berkeley National Lab. (US))

Presentation materials