Accelerating Graph-based Tracking Tasks with Symbolic Regression

Abstract

The reconstruction of particle tracks from hits in tracking detectors is a computationally intensive task due to the large combinatorics of detector signals. Recent efforts have proven that ML techniques can be successfully applied to the tracking problem, extending and improving the conventional methods based on feature engineering. However, complex models can be challenging to implement on heterogeneous trigger systems, integrating architectures such as FPGAs. Deploying the network on an FPGA is feasible but challenging and limited by its resources. An efficient alternative can employ symbolic regression (SR). We propose a novel approach that uses SR to replace a graph-based neural network. Substituting each network block with a symbolic function preserves the graph structure of the data and enables message passing. The technique is perfectly suitable for heterogeneous hardware, as it can be implemented more easily on FPGAs and grants faster execution times on CPU with respect to conventional methods. While the tracking problem is the target for this work, it also provides a proof-of-principle for the method that can be applied to many use cases.

15 pages, 5 figures, 1 table

References in corpus (24)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• The CMS trigger system
• Fast inference of deep neural networks in FPGAs for particle physics
• Interpretable Machine Learning for Science with PySR and SymbolicRegression.jl
• Operation of the ATLAS trigger system in Run 2
• Fast convolutional neural networks on FPGAs with hls4ml
• Compressing deep neural networks on FPGAs to binary and ternary precision with HLS4ML
• Charged-particle distributions in $\sqrt{s}=13$ TeV $pp$ interactions measured with the ATLAS detector at the LHC
• The Tracking Machine Learning challenge : Accuracy phase
• Distance-Weighted Graph Neural Networks on FPGAs for Real-Time Particle Reconstruction in High Energy Physics
• The ATLAS Inner Detector Trigger performance in pp collisions at 13 TeV during LHC Run 2
• Graph Neural Networks for Particle Tracking and Reconstruction
• Object condensation: one-stage grid-free multi-object reconstruction in physics detectors, graph and image data
• Graph Neural Networks for Charged Particle Tracking on FPGAs
• Accelerated Charged Particle Tracking with Graph Neural Networks on FPGAs
• The ATLAS Trigger System for LHC Run 3 and Trigger performance in 2022
• Tracking Triggers for the HL-LHC
• Towards a realistic track reconstruction algorithm based on graph neural networks for the HL-LHC
• Back to the Formula -- LHC Edition
• Symbolic Regression on FPGAs for Fast Machine Learning Inference
• Interpretable Machine Learning Methods Applied to Jet Background Subtraction in Heavy Ion Collisions
• Symbolic Regression for Beyond the Standard Model Physics
• Combined track finding with GNN & CKF
• Ultra Fast Transformers on FPGAs for Particle Physics Experiments

Cited by in corpus (1)

• The Physics Behind ML-based Quark-Gluon Taggers