Graph Neural Networks for IceCube Signal Classification

Abstract

Tasks involving the analysis of geometric (graph- and manifold-structured) data have recently gained prominence in the machine learning community, giving birth to a rapidly developing field of geometric deep learning. In this work, we leverage graph neural networks to improve signal detection in the IceCube neutrino observatory. The IceCube detector array is modeled as a graph, where vertices are sensors and edges are a learned function of the sensors' spatial coordinates. As only a subset of IceCube's sensors is active during a given observation, we note the adaptive nature of our GNN, wherein computation is restricted to the input signal support. We demonstrate the effectiveness of our GNN architecture on a task classifying IceCube events, where it outperforms both a traditional physics-based method as well as classical 3D convolution neural networks.

References in corpus (16)

• Semi-Supervised Classification with Graph Convolutional Networks
• Inductive Representation Learning on Large Graphs
• Neural Message Passing for Quantum Chemistry
• Geometric deep learning: going beyond Euclidean data
• Spectral Networks and Locally Connected Networks on Graphs
• Deep Convolutional Networks on Graph-Structured Data
• Observation of High-Energy Astrophysical Neutrinos in Three Years of IceCube Data
• Observation and Characterization of a Cosmic Muon Neutrino Flux from the Northern Hemisphere using six years of IceCube data
• The IceCube Data Acquisition System: Signal Capture, Digitization, and Timestamping
• Spectrum of cosmic-ray nucleons and the atmospheric muon charge ratio
• Atmospheric and Astrophysical Neutrinos above 1 TeV Interacting in IceCube
• Dynamic Graph CNN for Learning on Point Clouds
• Learning shape correspondence with anisotropic convolutional neural networks
• An improved method for measuring muon energy using the truncated mean of dE/dx
• N-body Networks: a Covariant Hierarchical Neural Network Architecture for Learning Atomic Potentials
• Dual-Primal Graph Convolutional Networks

Cited by in corpus (7)

• Machine learning and the physical sciences
• ABCNet: An attention-based method for particle tagging
• Enabling real-time multi-messenger astrophysics discoveries with deep learning
• Interaction networks for the identification of boosted $H\to b\overline{b}$ decays
• Deep Ensemble Analysis for Imaging X-ray Polarimetry
• Deep learning with photosensor timing information as a background rejection method for the Cherenkov Telescope Array
• Atomistic Graph Neural Networks for metals: Application to bcc iron