Jul 9 – 13, 2018
Sofia, Bulgaria
Europe/Sofia timezone

The HEP.TrkX project. Approaching Charged Particle Tracking at the HL-LHC with Deep Learning, for Online and Offline Processing

Jul 12, 2018, 2:00 PM
Hall 9 (National Palace of Culture)

Hall 9

National Palace of Culture

presentation Track 6 – Machine learning and physics analysis T6 - Machine learning and physics analysis


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


In the horizon of the High Luminosity Large Hadron Collider phase (HL-LHC), each proton bunch crossing will bring up to 200 simultaneous proton collisions. Performing the charged particle trajectory reconstruction in such dense environment will be computationally challenging because of the nature of the traditional algorithms used. The common combinatorial Kalman Filter state-of-the-art approach is highly successful in reconstructing high quality tracks but scales worse than quadratically with event density, leading to unmanageable computation requirements for the HL-LHC. As an alternative to other approaches to overcome this challenge, we propose novel methods employing deep learning, with inspiration from image and sequence processing. While training such models is a computationally intensive task that can be performed offline, the inference is potentially fast and can be implemented in highly parallel processing units such as FPGA or GPU. We present several models for performing pattern recognition, end-to-end track parameter estimation and tree-search track following. The methods explore multiple ways of performing the association of hits in appropriate groups, with the assumption of a tracklet seed or not. Our methods utilize neural networks of various architecture, including image-based model exploiting the power of convolutional layers, recurrent cells such as long short term memory to process sequences of hits and graph-based models that overcome sparsity. We report on the introduction of physics prior in the formalism of the neural net training, with the goal to accelerate and improve convergence. We present performances, scaling and comparison of these methods applied on realistic simulation datasets.

Primary authors

Dustin James Anderson (California Institute of Technology (US)) Dr Jean-Roch Vlimant (California Institute of Technology (US)) Paolo Calafiura (Lawrence Berkeley National Lab. (US)) Giuseppe Cerati (Fermi National Accelerator Lab. (US)) Steven Andrew Farrell (Lawrence Berkeley National Lab. (US)) Lindsey Gray (Fermi National Accelerator Lab. (US)) James Kowalkowski (Fermi National Accelerator Lab. (US)) Panagiotis Spentzouris (Fermilab) Maria Spiropulu (California Institute of Technology (US)) Aristeidis Tsaris (Fermilab) Mayur Mudigonda (Lawrence Berkeley National Laboratory) Prabhat (Lawrence Berkeley National Laboratory) Stephan Zheng (California Institute of Technology)

Presentation materials