Skip to main content
U.S. Department of Energy
Office of Scientific and Technical Information

Charged Particle Tracking via Edge-Classifying Interaction Networks

Journal Article · · Computing and Software for Big Science

Abstract

Recent work has demonstrated that geometric deep learning methods such as graph neural networks (GNNs) are well suited to address a variety of reconstruction problems in high-energy particle physics. In particular, particle tracking data are naturally represented as a graph by identifying silicon tracker hits as nodes and particle trajectories as edges, given a set of hypothesized edges, edge-classifying GNNs identify those corresponding to real particle trajectories. In this work, we adapt the physics-motivated interaction network (IN) GNN toward the problem of particle tracking in pileup conditions similar to those expected at the high-luminosity Large Hadron Collider. Assuming idealized hit filtering at various particle momenta thresholds, we demonstrate the IN’s excellent edge-classification accuracy and tracking efficiency through a suite of measurements at each stage of GNN-based tracking: graph construction, edge classification, and track building. The proposed IN architecture is substantially smaller than previously studied GNN tracking architectures; this is particularly promising as a reduction in size is critical for enabling GNN-based tracking in constrained computing environments. Furthermore, the IN may be represented as either a set of explicit matrix operations or a message passing GNN. Efforts are underway to accelerate each representation via heterogeneous computing resources towards both high-level and low-latency triggering applications.

Sponsoring Organization:
USDOE
Grant/Contract Number:
SC0007968; SC0021187
OSTI ID:
1830281
Journal Information:
Computing and Software for Big Science, Journal Name: Computing and Software for Big Science Journal Issue: 1 Vol. 5; ISSN 2510-2036
Publisher:
Springer Science + Business MediaCopyright Statement
Country of Publication:
Germany
Language:
English

References (42)

Measurements of b-jet tagging efficiency with the ATLAS detector using t t ¯ $$ t\overline{t} $$ events at s = 13 $$ \sqrt{s}=13 $$ TeV journal August 2018
Progressive track recognition with a Kalman-like fitting procedure journal December 1989
Application of Kalman filtering to track and vertex fitting journal December 1987
Simultaneous pattern recognition and track fitting by the Kalman filtering method journal September 1990
A concurrent track evolution algorithm for pattern recognition in the HERA-B main tracking system journal August 1997
Graph neural networks: A review of methods and applications journal January 2020
Jet substructure at the Large Hadron Collider: A review of recent advances in theory and machine learning journal November 2019
Automatic heterogeneous quantization of deep neural networks for low-latency inference on the edge for particle detectors journal June 2021
Towards a realistic track reconstruction algorithm based on graph neural networks for the HL-LHC journal January 2021
Primary vertex reconstruction in the ATLAS experiment at LHC journal July 2008
Particle-flow reconstruction and global event description with the CMS detector journal October 2017
Identification of heavy-flavour jets with the CMS detector in pp collisions at 13 TeV journal May 2018
Fast inference of deep neural networks in FPGAs for particle physics journal July 2018
Fast inference of Boosted Decision Trees in FPGAs for particle physics journal May 2020
Description and performance of track and primary-vertex reconstruction with the CMS tracker journal October 2014
Compressing deep neural networks on FPGAs to binary and ternary precision with hls4ml journal December 2020
Graph neural networks in particle physics journal January 2021
GPU coprocessors as a service for deep learning inference in high energy physics journal April 2021
Fast convolutional neural networks on FPGAs with hls4ml journal July 2021
Interaction networks for the identification of boosted H → b b ¯ decays journal July 2020
Track and vertex reconstruction: From classical to adaptive methods journal May 2010
FPGA-Based Accelerators of Deep Learning Networks for Learning and Classification: A Review journal January 2019
Minimum energy quantized neural networks conference October 2017
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation conference July 2017
Towards Effective Low-Bitwidth Convolutional Neural Networks conference June 2018
FPGAs-as-a-Service Toolkit (FaaST) conference November 2020
An Overview of FPGA Based Deep Learning Accelerators: Challenges and Opportunities
  • Wang, Teng; Wang, Chao; Zhou, Xuehai
  • 2019 IEEE 21st International Conference on High Performance Computing and Communications; IEEE 17th International Conference on Smart City; IEEE 5th International Conference on Data Science and Systems (HPCC/SmartCity/DSS) https://doi.org/10.1109/HPCC/SmartCity/DSS.2019.00229
conference August 2019
Geometric Deep Learning: Going beyond Euclidean data journal July 2017
Deep Learning on Graphs: A Survey journal January 2020
The Graph Neural Network Model journal December 2008
A Comprehensive Survey on Graph Neural Networks journal January 2021
Jet reconstruction and performance using particle flow with the ATLAS Detector journal July 2017
Performance of the ATLAS track reconstruction algorithms in dense environments in LHC Run 2 journal October 2017
Learning representations of irregular particle-detector geometry with distance-weighted graph networks journal July 2019
JEDI-net: a jet identification algorithm based on interaction networks journal January 2020
FINN: A Framework for Fast, Scalable Binarized Neural Network Inference
  • Umuroglu, Yaman; Fraser, Nicholas J.; Gambardella, Giulio
  • Proceedings of the 2017 ACM/SIGDA International Symposium on Field-Programmable Gate Arrays - FPGA '17 https://doi.org/10.1145/3020078.3021744
conference January 2017
FINN- R: An End-to-End Deep-Learning Framework for Fast Exploration of Quantized Neural Networks
  • Blott, Michaela; Preußer, Thomas B.; Fraser, Nicholas J.
  • ACM Transactions on Reconfigurable Technology and Systems, Vol. 11, Issue 3 https://doi.org/10.1145/3242897
journal December 2018
Dynamic Graph CNN for Learning on Point Clouds journal October 2019
High-Luminosity Large Hadron Collider (HL-LHC) report January 2017
Distance-Weighted Graph Neural Networks on FPGAs for Real-Time Particle Reconstruction in High Energy Physics journal January 2021
GPU-Accelerated Machine Learning Inference as a Service for Computing in Neutrino Experiments journal January 2021
Ps and Qs: Quantization-Aware Pruning for Efficient Low Latency Neural Network Inference journal July 2021

Similar Records

Graph Neural Networks for Charged Particle Tracking on FPGAs
Journal Article · Wed Mar 23 00:00:00 EDT 2022 · Frontiers in Big Data · OSTI ID:1859670

Track Seeding and Labelling with Embedded-space Graph Neural Networks
Conference · Tue Jun 30 00:00:00 EDT 2020 · OSTI ID:1656641

Graph Neural Network for Object Reconstruction in Liquid Argon Time Projection Chambers
Conference · Sat Dec 31 23:00:00 EST 2022 · J.Phys.Conf.Ser. · OSTI ID:1958460

Related Subjects