On the Use of Neural Networks for Energy Reconstruction in High-granularity Calorimeters

Abstract

We contrasted the performance of deep neural networks - Convolutional Neural Network (CNN) and Graph Neural Network (GNN) - to current state of the art energy regression methods in a finely 3D-segmented calorimeter simulated by GEANT4. This comparative benchmark gives us some insight to assess the particular latent signals neural network methods exploit to achieve superior resolution. A CNN trained solely on a pure sample of pions achieved substantial improvement in the energy resolution for both single pions and jets over the conventional approaches. It maintained good performance for electron and photon reconstruction. We also used the Graph Neural Network (GNN) with edge convolution to assess the importance of timing information in the shower development for improved energy reconstruction. We implement a simple simulation based correction to the energy sum derived from the fraction of energy deposited in the electromagnetic shower component. This serves as an approximate dual-readout analogue for our benchmark comparison. Although this study does not include the simulation of detector effects, such as electronic noise, the margin of improvement seems robust enough to suggest these benefits will endure in real-world application. We also find reason to infer that the CNN/GNN methods leverage latent features that concur with our current understanding of the physics of calorimeter measurement.

References in corpus (5)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• An Introduction to PYTHIA 8.2
• Particle-flow reconstruction and global event description with the CMS detector
• Energy flow in a hadronic cascade: Application to hadron calorimetry
• Optimising longitudinal and lateral calorimeter granularity for software compensation in hadronic showers using deep neural networks

Cited by in corpus (17)

• A high-granularity calorimeter insert based on SiPM-on-tile technology at the future Electron-Ion Collider
• Optimizing Observables with Machine Learning for Better Unfolding
• Learning Uncertainties the Frequentist Way: Calibration and Correlation in High Energy Physics
• Bias and Priors in Machine Learning Calibrations for High Energy Physics
• Studies of time resolution, light yield, and crosstalk using SiPM-on-tile calorimetry for the future Electron-Ion Collider
• The CALICE SiW ECAL Technological Prototype -- Status and Outlook
• Perspectives on the Calibration of CNN Energy Reconstruction in Highly Granular Calorimeters
• Design of a SiPM-on-Tile ZDC for the future EIC and its Performance with Graph Neural Networks
• Leveraging Staggered Tessellation for Enhanced Spatial Resolution in High-Granularity Calorimeters
• Unifying Simulation and Inference with Normalizing Flows
• Sub-10 ps time tagging of electromagnetic showers with scintillating glasses and SiPMs
• Using graph neural networks to reconstruct charged pion showers in the CMS High Granularity Calorimeter
• Timing performance of SiPM-on-Tile elements: Laboratory and test beam measurements
• Reconstruction of overlapping electromagnetic showers in calorimeters using Transformers
• End-to-end multi-particle reconstruction in high occupancy imaging calorimeters with graph neural networks
• Time-assisted energy reconstruction in a highly-granular hadronic calorimeter
• Reconstruction of electromagnetic showers in calorimeters using Deep Learning