Jet energy calibration with deep learning as a Kubeflow pipeline

Abstract

Precise measurements of the energy of jets emerging from particle collisions at the LHC are essential for a vast majority of physics searches at the CMS experiment. In this study, we leverage well-established deep learning models for point clouds and CMS open data to improve the energy calibration of particle jets. To enable production-ready machine learning based jet energy calibration an end-to-end pipeline is built on the Kubeflow cloud platform. The pipeline allowed us to scale up our hyperparameter tuning experiments on cloud resources, and serve optimal models as REST endpoints. We present the results of the parameter tuning process and analyze the performance of the served models in terms of inference time and overhead, providing insights for future work in this direction. The study also demonstrates improvements in both flavor dependence and resolution of the energy response when compared to the standard jet energy corrections baseline.

arXiv comment20 pages, 10 figures. Code repository available via https://zenodo.org/record/7799179

References in corpus (19)

• The anti-k_t jet clustering algorithm
• An Introduction to PYTHIA 8.2
• A Tutorial on Bayesian Optimization of Expensive Cost Functions, with Application to Active User Modeling and Hierarchical Reinforcement Learning
• Particle-flow reconstruction and global event description with the CMS detector
• Jet energy scale and resolution in the CMS experiment in pp collisions at 8 TeV
• Identification of heavy-flavour jets with the CMS detector in pp collisions at 13 TeV
• ParticleNet: Jet Tagging via Particle Clouds
• Deep Learning and its Application to LHC Physics
• Lookahead Optimizer: k steps forward, 1 step back
• Jet-Images -- Deep Learning Edition
• Energy Flow Networks: Deep Sets for Particle Jets
• Jet Flavor Classification in High-Energy Physics with Deep Neural Networks
• Graph Neural Networks in Particle Physics
• Pileup mitigation at CMS in 13 TeV data
• QCD-Aware Recursive Neural Networks for Jet Physics
• Graph Neural Networks for Particle Reconstruction in High Energy Physics detectors
• Deep Sets
• A deep neural network for simultaneous estimation of b jet energy and resolution
• Awkward Arrays in Python, C++, and Numba