Search papers, labs, and topics across Lattice.
The paper introduces a graph neural network (GNN) approach for reconstructing the arrival direction and energy of ultra-high-energy cosmic rays from radio detector array data. By encoding physical knowledge into the GNN architecture and input features, the method achieves improved precision and reduces the need for extensive training data. The method demonstrates an angular resolution of 0.092 degrees and an energy resolution of 16.4% on simulated data, while also providing uncertainty estimates for the predictions.
GNNs enhanced with physics-informed design can reconstruct cosmic ray properties with high precision and quantified uncertainty, even with limited training data.
Using advanced machine learning techniques, we developed a method for reconstructing precisely the arrival direction and energy of ultra-high-energy cosmic rays from the voltage traces they induced on ground-based radio detector arrays. In our approach, triggered antennas are represented as a graph structure, which serves as input for a graph neural network (GNN). By incorporating physical knowledge into both the GNN architecture and the input data, we improve the precision and reduce the required size of the training set with respect to a fully data-driven approach. This method achieves an angular resolution of 0.092{\deg} and an electromagnetic energy reconstruction resolution of 16.4% on simulated data with realistic noise conditions. We also employ uncertainty estimation methods to enhance the reliability of our predictions, quantifying the confidence of the GNN's outputs and providing confidence intervals for both direction and energy reconstruction. Finally, we investigate strategies to verify the model's consistency and robustness under real life variations, with the goal of identifying scenarios in which predictions remain reliable despite domain shifts between simulation and reality.
