Classification of Electron and Muon Neutrino Events for the ESS$\nu$SB Near Water Cherenkov Detector using Graph Neural Networks

Abstract

In the effort to obtain a precise measurement of leptonic CP-violation with the ESS$\nu$SB experiment, accurate and fast reconstruction of detector events plays a pivotal role. In this work, we examine the possibility of replacing the currently proposed likelihood-based reconstruction method with an approach based on Graph Neural Networks (GNNs). As the likelihood-based reconstruction method is reasonably accurate but computationally expensive, one of the benefits of a Machine Learning (ML) based method is enabling fast event reconstruction in the detector development phase, allowing for easier investigation of the effects of changes to the detector design. Focusing on classification of flavour and interaction type in muon and electron events and muon- and electron neutrino interaction events, we demonstrate that the GNN reconstructs events with greater accuracy than the likelihood method for events with greater complexity, and with increased speed for all events. Additionally, we investigate the key factors impacting reconstruction performance, and demonstrate how separation of events by pion production using another GNN classifier can benefit flavour classification.