31 May 2022 to 2 June 2022
Princeton University
US/Eastern timezone

Graph Neural Networks for Charged Particle Tracking on FPGAs

31 May 2022, 17:00
25m
PCTS conference room (4th floor) (Jadwin Hall, Princeton University)

PCTS conference room (4th floor)

Jadwin Hall, Princeton University

Plenary Plenary

Speaker

Abdelrahman Elabd (IRIS-HEP)

Description

The determination of charged particle trajectories in collisions at the CERN Large Hadron Collider (LHC) is an important but challenging problem, especially in the high interaction density conditions expected during the future high-luminosity phase of the LHC (HL-LHC). Graph neural networks (GNNs) are a type of geometric deep learning algorithm that has successfully been applied to this task by embedding tracker data as a graph---nodes represent hits, while edges represent possible track segments---and classifying the edges as true or fake track segments. However, their study in hardware- or software-based trigger applications has been limited due to their large computational cost. In this talk, we introduce an automated translation workflow, integrated into a broader tool called hls4ml, for converting GNNs into firmware for field-programmable gate arrays (FPGAs). We use this translation tool to implement GNNs for charged particle tracking, trained using the TrackML challenge dataset, on FPGAs with designs targeting different graph sizes, task complexites, and latency/throughput requirements. This work could enable the inclusion of charged particle tracking GNNs at the trigger level for HL-LHC experiments.

Consider for young scientist forum (Student or postdoc speaker) Yes

Authors

Abdelrahman Elabd (IRIS-HEP) Javier Mauricio Duarte (Univ. of California San Diego (US)) Bo-Cheng Lai Gage DeZoort (Princeton University (US)) Isobel Ojalvo (Princeton University (US)) Mark Neubauer (Univ. Illinois at Urbana Champaign (US)) Markus Julian Atkinson (Univ. Illinois at Urbana Champaign (US)) Matthew Trahms (UW ACME Lab) Peter Elmer (Princeton University (US)) Savannah Jennifer Thais (Princeton University (US)) Scott Hauck Shi-Yu Huang Shih-Chieh Hsu (University of Washington Seattle (US)) Vesal Razavimaleki (Univ. Illinois at Urbana-Champaign (US)) 晉瑄 胡

Presentation materials