Vertex and Energy Reconstruction in JUNO with Machine Learning Methods
Zhen Qian,Vladislav Belavin,Vasily Bokov,Riccardo Brugnera,Alessandro Compagnucci,Arsenii Gavrikov,Alberto Garfagnini,Maxim Gonchar,Leyla Khatbullina,Ziyuan Li,Wuming Luo,Yury Malyshkin,Samuele Piccinelli,Ivan Provilkov,Fedor Ratnikov,Dmitry Selivanov,Konstantin Treskov,Andrey Ustyuzhanin,Francesco Vidaich,Zhengyun You,Yumei Zhang,Jiang Zhu,Francesco Manzali
DOI: https://doi.org/10.1016/j.nima.2021.165527
2021-01-01
Abstract:The Jiangmen Underground Neutrino Observatory (JUNO) is an experiment designed to study neutrino oscillations. Determination of neutrino mass ordering and precise measurement of neutrino oscillation parameters sin(2) theta(12), Delta m(21)(2) and Delta(2)(31) are the main goals of the experiment. A rich physical program beyond the oscillation analysis is also foreseen. The ability to accurately reconstruct particle interaction events in JUNO is of great importance for the success of the experiment. In this work we present several machine learning approaches applied to the vertex and the energy reconstruction. Multiple models and architectures were compared and studied, including Boosted Decision Trees (BDT), Deep Neural Networks (DNN), a few kinds of Convolution Neural Networks (CNN), based on ResNet and VGG, and a Graph Neural Network based on DeepSphere. The models of BDT and DNN are trained with aggregated information, pre-calculated from PMT signal, while the others are trained with PMT-wise measured information from 17600 PMTs. Based on a study, carried out using the dataset, generated by the official JUNO software, we demonstrate that machine learning approaches achieve the necessary level of accuracy for reaching the physical goals of JUNO: sigma(E) = 3% at E-vis = 1 MeV for the energy and sigma(x,y,z) = 10 cm at E-vis = 1 MeV for the position.