GraphNeT: Graph neural networks for neutrino telescope event reconstruction
Abstract
GraphNeT is an open-source python framework aimed at providing high quality, user friendly, end-to-end functionality to perform reconstruction tasks at neutrino telescopes using graph neural networks (GNNs). GraphNeT makes it fast and easy to train complex models that can provide event reconstruction with state-of-the-art performance, for arbitrary detector configurations, with inference times that are orders of magnitude faster than traditional reconstruction techniques. GNNs from GraphNeT are flexible enough to be applied to data from all neutrino telescopes, including future projects such as IceCube extensions or P-ONE. This means that GNN-based reconstruction can be used to provide state-of-the-art performance on most reconstruction tasks in neutrino telescopes, at real-time event rates, across experiments and physics analyses, with vast potential impact for neutrino and astro-particle physics.
- Publication:
-
The Journal of Open Source Software
- Pub Date:
- May 2023
- DOI:
- 10.21105/joss.04971
- arXiv:
- arXiv:2210.12194
- Bibcode:
- 2023JOSS....8.4971S
- Keywords:
-
- neural networks;
- Python;
- deep learning;
- graph neural networks;
- neutrinos;
- machine learning;
- astrophysics;
- particle physics;
- Astrophysics - Instrumentation and Methods for Astrophysics;
- Computer Science - Machine Learning;
- High Energy Physics - Experiment;
- Physics - Data Analysis;
- Statistics and Probability
- E-Print:
- 6 pages, 1 figure. Code can be found at https://github.com/graphnet-team/graphnet . Submitted to the Journal of Open Source Software (JOSS)