Charged particle tracking via edge-classifying interaction networks

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 is 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.

arXiv commentThis is a post-peer-review, pre-copyedit version of this article. The final authenticated version is available online at: https://doi.org/10.1007/s41781-021-00073-z

References in corpus (38)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• A Comprehensive Survey on Graph Neural Networks
• Neural Message Passing for Quantum Chemistry
• Geometric deep learning: going beyond Euclidean data
• DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradients
• Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activations
• FINN: A Framework for Fast, Scalable Binarized Neural Network Inference
• Mixed Precision Training
• Particle-flow reconstruction and global event description with the CMS detector
• Description and performance of track and primary-vertex reconstruction with the CMS tracker
• Identification of heavy-flavour jets with the CMS detector in pp collisions at 13 TeV
• Interaction Networks for Learning about Objects, Relations and Physics
• FPGA-based Accelerators of Deep Learning Networks for Learning and Classification: A Review
• Fast inference of deep neural networks in FPGAs for particle physics
• Jet reconstruction and performance using particle flow with the ATLAS Detector
• Jet Substructure at the Large Hadron Collider: A Review of Recent Advances in Theory and Machine Learning
• Training Deep Neural Networks with 8-bit Floating Point Numbers
• Automatic heterogeneous quantization of deep neural networks for low-latency inference on the edge for particle detectors
• Graph Neural Networks in Particle Physics
• Performance of the ATLAS Track Reconstruction Algorithms in Dense Environments in LHC Run 2
• Learning representations of irregular particle-detector geometry with distance-weighted graph networks
• Measurements of $b$-jet tagging efficiency with the ATLAS detector using $t\bar{t}$ events at $\sqrt{s}=13$ TeV
• Fast convolutional neural networks on FPGAs with hls4ml
• JEDI-net: a jet identification algorithm based on interaction networks
• Compressing deep neural networks on FPGAs to binary and ternary precision with HLS4ML
• Graph Neural Networks for Particle Reconstruction in High Energy Physics detectors
• Fast inference of Boosted Decision Trees in FPGAs for particle physics
• Interaction networks for the identification of boosted $H\to b\overline{b}$ decays
• The Tracking Machine Learning challenge : Accuracy phase
• Ps and Qs: Quantization-aware pruning for efficient low latency neural network inference
• Distance-Weighted Graph Neural Networks on FPGAs for Real-Time Particle Reconstruction in High Energy Physics
• Novel deep learning methods for track reconstruction
• Graph Processing on FPGAs: Taxonomy, Survey, Challenges
• Accelerated Charged Particle Tracking with Graph Neural Networks on FPGAs
• GPU-accelerated machine learning inference as a service for computing in neutrino experiments
• FPGAs-as-a-Service Toolkit (FaaST)
• Towards a realistic track reconstruction algorithm based on graph neural networks for the HL-LHC
• GPU coprocessors as a service for deep learning inference in high energy physics

Cited by in corpus (11)

• Applications and Techniques for Fast Machine Learning in Science
• Graph Neural Networks for Charged Particle Tracking on FPGAs
• BOTAN: BOnd TArgeting Network for prediction of slow glassy dynamics by machine learning relative motion
• Lorentz group equivariant autoencoders
• The Tracking Machine Learning challenge : Throughput phase
• Improved Constraints on Effective Top Quark Interactions using Edge Convolution Networks
• Does Lorentz-symmetric design boost network performance in jet physics?
• Artificial intelligence for improved fitting of trajectories of elementary particles in inhomogeneous dense materials immersed in a magnetic field
• Particle Graph Autoencoders and Differentiable, Learned Energy Mover's Distance
• End-to-end multi-particle reconstruction in high occupancy imaging calorimeters with graph neural networks
• Explaining machine-learned particle-flow reconstruction