Enhancing event reconstruction for $\gamma$-ray particle detector arrays using transformers

TL;DR

This paper introduces transformer-based deep learning methods for gamma-ray air shower event reconstruction, showing significant improvements over traditional techniques, especially at low and intermediate energies.

Contribution

It is the first to demonstrate improved event reconstruction and gamma-hadron separation using a single transformer architecture in gamma-ray astronomy.

Findings

Significant performance improvements across all energy ranges.

Enhanced gamma-hadron separation capabilities.

Better angular, core, and energy reconstruction accuracy.

Abstract

Gamma-ray astronomy from hundreds of GeV to PeV is confined to ground-based experiments that detect air showers induced by $\gamma$-rays entering Earth's atmosphere. While particle detector arrays feature huge detection areas, accurately reconstructing the primary particle properties is difficult due to the sparse sampling of the air shower and its intrinsic fluctuations. In this work, using simulations of a future water-Cherenkov array, we investigate two end-to-end deep learning approaches based on the transformer architecture with different computational complexities that utilize calibrated raw data. We benchmark both methods against well-established methods in the field in terms of $\gamma$-hadron separation, angular, core, and energy reconstruction. Our results show significant improvements across the whole energy range, particularly at low and intermediate energies. This work is the first to consistently demonstrate improved performance in both event reconstruction and $\gamma$-hadron separation using a single architecture.