10-15 March 2019
Steinmatte conference center
Europe/Zurich timezone

HEP.TrkX Charged Particle Tracking using Graph Neural Networks

11 Mar 2019, 16:30
Steinmatte Room A

Steinmatte Room A

Oral Track 2: Data Analysis - Algorithms and Tools Track 2: Data Analysis - Algorithms and Tools


Dr Jean-Roch Vlimant (California Institute of Technology (US))


To address the unprecedented scale of HL-LHC data, the HEP.TrkX project has been investigating a variety of machine learning approaches to particle track reconstruction. The most promising of these solutions, a graph neural network, processes the event as a graph that connects track measurements (detector hits corresponding to nodes) with candidate line segments between the hits (corresponding to edges). This architecture enables separate input features for edges and nodes, ultimately creating a hidden representation of the graph that is used to turn edges on and off, leaving only the edges that form tracks. Due to the large scale of this graph for an entire LHC event, we present new methods that allow the event graph to be scaled to a computationally reasonable size. We report the results of the graph neural network on the TrackML dataset, detailing the effectiveness of this model on event data with large pileup. Additionally, we propose post-processing methods that further refine the result of the graph neural network, ultimately synthesizing an end-to-end machine learning solution to particle track reconstruction.

Primary authors

Dr Jean-Roch Vlimant (California Institute of Technology (US)) Steven Andrew Farrell (Lawrence Berkeley National Lab. (US)) Alexander Zlokapa (California Institute of Technology (US)) Abhishek Anand (California Institute of Technology) Maria Spiropulu (California Institute of Technology) Paolo Calafiura (Lawrence Berkeley National Lab. (US))

Presentation materials