Apr 20 – 30, 2020
Virtual/Digital only workshop
Europe/Paris timezone

Graph neural networks for FPGAs

Apr 22, 2020, 7:50 PM
Virtual/Digital only workshop

Virtual/Digital only workshop


Yutaro Iiyama (CERN)


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.

Primary authors

Javier Mauricio Duarte (Univ. of California San Diego (US)) Philip Coleman Harris (Massachusetts Inst. of Technology (US)) Yutaro Iiyama (CERN)

Presentation materials