29 January 2024 to 2 February 2024
CERN
Europe/Zurich timezone

Accelerating Graph-Based Tracking with Symbolic Regression

30 Jan 2024, 11:30
20m
40/S2-D01 - Salle Dirac (CERN)

40/S2-D01 - Salle Dirac

CERN

115
Show room on map
Contributed talk 4 Fast ML : Application of Machine Learning to DAQ/Trigger/Real Time Analysis Contributed Talks

Speaker

Nathalie Soybelman (Weizmann Institute of Science (IL))

Description

Tracking, the reconstruction of particle trajectories from hits in the inner detector 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, the inference of complex networks can be too slow to be used in the trigger system. Quantising the network and deploying it on an FPGA is feasible but challenging and highly non-trivial. An efficient alternative can employ symbolic regression (SR), which already proved its performance in replacing a dense neural network for jet classification. We propose a novel approach that uses SR to replace a graph-based neural network. Using a simplified toy example, we substitute each network block with a symbolic function, preserving the graph structure of the data and enabling message passing. This approach significantly speeds up inference on a CPU without sacrificing much accuracy.

Authors

Carlo Schiavi (INFN e Universita Genova (IT)) Francesco Armando Di Bello (INFN e Universita Genova (IT)) Nathalie Soybelman (Weizmann Institute of Science (IL))

Co-author

Eilam Gross (Weizmann Institute of Science (IL))

Presentation materials