Accelerated Charged Particle Tracking with Graph Neural Networks on FPGAs

Nov 1, 2020·
Aneesh Heintz
,
Vesal Razavimaleki
,
Javier Duarte
,
Gage Dezoort
,
Isobel Ojalvo
,
Savannah Thais
,
Markus Atkinson
Mark Neubauer
Mark Neubauer
,
Lindsey Gray
,
Sergo Jindariani
,
Nhan Tran
,
Philip Harris
,
Dylan Rankin
,
Thea Aarrestad
,
Vladimir Loncar
,
Maurizio Pierini
,
Sioni Summers
,
Jennifer Ngadiuba
,
Mia Liu
,
Edward Kreinar
,
Zhenbin Wu
· 1 min read
Abstract
We develop and study FPGA implementations of algorithms for charged particle tracking based on graph neural networks. The two complementary FPGA designs are based on OpenCL, a framework for writing programs that execute across heterogeneous platforms, and hls4ml, a high-level-synthesis-based compiler for neural network to firmware conversion. We evaluate and compare the resource usage, latency, and tracking performance of our implementations based on a benchmark dataset. We find a considerable speedup over CPU-based execution is possible, potentially enabling such algorithms to be used effectively in future computing workflows and the FPGA-based Level-1 trigger at the CERN Large Hadron Collider.
Type
Publication
34th Conference on Neural Information Processing Systems (NeurIPS) 2020

Add the full text or supplementary notes for the publication here using Markdown formatting.

Mark Neubauer
Authors
University of Illinois at Urbana-Champaign
I am a professor at the University of Illinois. My research is highly interdisciplinary at the intersection of particle physics, AI/ML, and quantum, aiming to understand the universe at its fundamental level and to accelerate scientific discovery through innovation.
Authors
Authors
Authors