Connecting The Dots 2020

Graph neural networks for FPGAs

22 Apr 2020, 19:50

25m

Virtual/Digital only workshop

Recording sessions

Speaker

Yutaro Iiyama (CERN)

Description

Graph neural networks have been shown to achieve excellent performance for several crucial tasks in particle physics, such as charged particle tracking, jet tagging, and clustering. An important domain for the application of these networks is the Level-1, FGPA-based trigger, which has strict latency and resource constraints. We discuss how to design distance-weighted graph networks that can be executed with less than 1 $\mu$s latency on an FPGA. To do so, we consider representative tasks associated with particle reconstruction and identification in a next-generation calorimeter operating at a particle collider. We use graph architectures developed for these purposes and simplified in order to match the computing constraints of real-time event processing at the CERN Large Hadron Collider. The trained models are compressed using pruning and quantization. Using the hls4ml library, we convert the compressed models into firmware to be implemented on an FPGA. We show results both in terms of model accuracy and computing performance.

Presentation materials

gnn_for_fpga.pdf

Iiyama.mp4

Recorded talk