Accelerating End-to-End Deep Learning for Particle Reconstruction using CMS open data

¹ Department of Physics, Carnegie Mellon University, Pittsburgh, USA
² Department of Physics, Brown University, Providence, USA
³ Department of Electrical and Electronics Engineering, BITS Pilani, Goa, India
⁴ Department of Physics and Astronomy, University of Alabama, Tuscaloosa, USA

^* e-mail: davide.di.croce@cern.ch

Published online: 23 August 2021

Abstract

Machine learning algorithms are gaining ground in high energy physics for applications in particle and event identification, physics analysis, detector reconstruction, simulation and trigger. Currently, most data-analysis tasks at LHC experiments benefit from the use of machine learning. Incorporating these computational tools in the experimental framework presents new challenges. This paper reports on the implementation of the end-to-end deep learning with the CMS software framework and the scaling of the end-to-end deep learning with multiple GPUs. The end-to-end deep learning technique combines deep learning algorithms and low-level detector representation for particle and event identification. We demonstrate the end-to-end implementation on a top quark benchmark and perform studies with various hardware architectures including single and multiple GPUs and Google TPU.