Distance-Weighted Graph Neural Networks on FPGAs for Real-Time Particle Reconstruction in High Energy Physics

Autor: Yutaro Iiyama, Gianluca Cerminara, Abhijay Gupta, Jan Kieseler, Vladimir Loncar, Maurizio Pierini, Shah Rukh Qasim, Marcel Rieger, Sioni Summers, Gerrit Van Onsem, Kinga Anna Wozniak, Jennifer Ngadiuba, Giuseppe Di Guglielmo, Javier Duarte, Philip Harris, Dylan Rankin, Sergo Jindariani, Mia Liu, Kevin Pedro, Nhan Tran, Edward Kreinar, Zhenbin Wu
Jazyk: angličtina
Předmět:
FOS: Computer and information sciences
Big Data
Computer Science - Machine Learning
Particle physics
Physics - Instrumentation and Detectors
Computer science
cs.LG
FOS: Physical sciences
02 engineering and technology
computer.software_genre
Tracking (particle physics)
imaging calorimeter
01 natural sciences
fast inference
Machine Learning (cs.LG)
High Energy Physics - Experiment
High Energy Physics - Experiment (hep-ex)
Artificial Intelligence
0103 physical sciences
0202 electrical engineering
electronic engineering
information engineering

Computer Science (miscellaneous)
Detectors and Experimental Techniques
field-programmable gate arrays
graph network
Cluster analysis
physics.ins-det
Original Research
Network architecture
Large Hadron Collider
lcsh:T58.5-58.64
Calorimeter (particle physics)
lcsh:Information technology
010308 nuclear & particles physics
Firmware
business.industry
hep-ex
Deep learning
deep learning
020207 software engineering
Instrumentation and Detectors (physics.ins-det)
3. Good health
Computing and Computers
Graph (abstract data type)
Artificial intelligence
business
computer
Particle Physics - Experiment
Information Systems
Zdroj: Frontiers in Big Data
Frontiers in Big Data, Vol 3 (2021)
ISSN: 2624-909X
DOI: 10.3389/fdata.2020.598927
Popis: 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 FGPA-based first layer of real-time data filtering at the CERN Large Hadron Collider, which has strict latency and resource constraints. We discuss how to design distance-weighted graph networks that can be executed with a latency of less than 1$\mu\mathrm{s}$ on an FPGA. To do so, we consider a representative task associated to particle reconstruction and identification in a next-generation calorimeter operating at a particle collider. We use a graph network architecture developed for such purposes, and apply additional simplifications to match the computing constraints of Level-1 trigger systems, including weight quantization. Using the $\mathtt{hls4ml}$ library, we convert the compressed models into firmware to be implemented on an FPGA. Performance of the synthesized models is presented both in terms of inference accuracy and resource usage.
Comment: 15 pages, 4 figures
Databáze: OpenAIRE