Accelerated Charged Particle Tracking with Graph Neural Networks on FPGAs
Nov 1, 2020·
,,,,,,
,,,,,,,,,,,,,·
1 min read
Aneesh Heintz
Vesal Razavimaleki
Javier Duarte
Gage Dezoort
Isobel Ojalvo
Savannah Thais
Markus Atkinson
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
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.
Authors
Authors
Authors
Authors
Authors
Authors
Authors

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
Authors
Authors
Authors
Authors
Authors
Authors
Authors
Authors
Authors
Authors