Lightweight jet reconstruction and identification as an object detection task. Issue 2 (1st June 2022)

• Record Type: Journal Article
• Authors: Pol, Adrian Alan; Aarrestad, Thea; Govorkova, Ekaterina; Halily, Roi; Klempner, Anat; Kopetz, Tal; Loncar, Vladimir; Ngadiuba, Jennifer; Pierini, Maurizio; Sirkin, Olya; Summers, Sioni
• Is Part Of: Machine learning: science and technology. Volume 3:Issue 2(2022)
• Publication Date: 2022-06-01
• Subjects: high energy physics -- jet tagging -- jet reconstruction -- jet images -- object detection -- attention mechanism -- quantization aware training; 006.31
• DOI: 10.1088/2632-2153/ac7a02
• Languages: English
• ISSNs: 2632-2153

Abstract

Abstract: We apply object detection techniques based on deep convolutional blocks to end-to-end jet identification and reconstruction tasks encountered at the CERN large hadron collider (LHC). Collision events produced at the LHC and represented as an image composed of calorimeter and tracker cells are given as an input to a Single Shot Detection network. The algorithm, named PFJet-SSD performs simultaneous localization, classification and regression tasks to cluster jets and reconstruct their features. This all-in-one single feed-forward pass gives advantages in terms of execution time and an improved accuracy w.r.t. traditional rule-based methods. A further gain is obtained from network slimming, homogeneous quantization, and optimized runtime for meeting memory and latency constraints of a typical real-time processing environment. We experiment with 8-bit and ternary quantization, benchmarking their accuracy and inference latency against a single-precision floating-point. We show that the ternary network closely matches the performance of its full-precision equivalent and outperforms the state-of-the-art rule-based algorithm. Finally, we report the inference latency on different hardware platforms and discuss future applications.