19–25 Oct 2024
Europe/Zurich timezone

Online track reconstruction with graph neural networks on FPGAs for the ATLAS experiment

24 Oct 2024, 13:30
18m
Room 1.C (Small Hall)

Room 1.C (Small Hall)

Talk Track 2 - Online and real-time computing Parallel (Track 2)

Speaker

Sebastian Dittmeier (Ruprecht-Karls-Universitaet Heidelberg (DE))

Description

For the HL-LHC upgrade of the ATLAS TDAQ system, a heterogeneous computing farm
deploying GPUs and/or FPGAs is under study, together with the use of modern
machine learning algorithms such as Graph Neural Networks (GNNs). We present a
study on the reconstruction of tracks in the ATLAS Inner Tracker using GNNs on
FPGAs for the Event Filter system. We explore each of the steps in a GNN-based
tracking pipeline: graph construction, edge classification using an interaction
network, and segmentation of the graph into track candidates. We investigate
optimizations of the GNN approach that aim to minimize FPGA resources
utilization and maximize throughput while retaining high track reconstruction
efficiency and low fake rates required for the ATLAS Event Filter tracking
system. These studies include model hyperparameter tuning, model pruning and
quantization-aware training, and sequential processing of sub-graphs across the
detector.

Primary authors

ATLAS TDAQ Sebastian Dittmeier (Ruprecht-Karls-Universitaet Heidelberg (DE))

Presentation materials