Reconstruction of overlapping electromagnetic showers in calorimeters using Transformers

Abstract

Accurate clustering of electromagnetic energy deposits is essential for reconstructing photons and electrons in modern hadron collider experiments, where boosted topologies and pileup cause overlapping showers and ambiguous energy assignment. We present deep learning-based clustering approaches that reconstruct particle energy and position directly from calorimeter readout. The study includes a two-step strategy in which candidate seed windows are identified and then jointly processed via distance-weighted message passing or attention mechanism and a single-step graph transformer, ClusTEX, which performs candidate selection and reconstruction in one inference stage. ClusTEX uses a novel positional encoding scheme that separates local coordinates within the graph from global detector coordinates, enabling efficient, geometry-aware inference. Models are trained on GEANT4 simulations of a simplified (toy) and an ECAL-inspired topology with an explicit $\eta -\varphi$ dependence. Performance is evaluated using efficiency, energy and position resolutions and splitting rate - reconstruction of two objects for a single photon. In the toy calorimeter, attention-based interactions improve the reconstruction of overlapping showers relative to both the standard algorithm and distance-driven message passing, while maintaining performance on isolated photons and reducing splitting without multi-pass inference. For boosted ${\pi }^0\to \gamma \gamma$, the attention-based model retains di-photon mass reconstruction capability, where the standard algorithm becomes inefficient. In the ECAL-inspired topology, ClusTEX provides the best overall performance, yielding improved energy resolution and reduced splitting compared to two-step approaches and the standard algorithm. It also remains robust under localized detector failures, showing improved stability and partial recovery of energy in non-responsive channels.

The Version of Record of this article is published in EPJC, and is available online at https://doi.org/10.1140/epjc/s10052-026-16097-x

References in corpus (9)

• Particle-flow reconstruction and global event description with the CMS detector
• Electron and photon reconstruction and identification with the CMS experiment at the CERN LHC
• A measurement of the Higgs boson mass in the diphoton decay channel
• Calorimetry with Deep Learning: Particle Simulation and Reconstruction for Collider Physics
• Measurements of the Higgs boson production cross section and couplings in the W boson pair decay channel in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• On the Use of Neural Networks for Energy Reconstruction in High-granularity Calorimeters
• Search for exotic Higgs boson decays H $\to$ $\mathcal{AA}$ $\to$ 4$\gamma$ with events containing two merged diphotons in proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Deep learning techniques for energy clustering in the CMS ECAL
• Differentiable Surrogate for Detector Simulation and Design with Diffusion Models