Open benchmark · arXiv:2511.13111

NuBench

An open benchmark for deep learning–based event reconstruction in neutrino telescopes — seven large-scale simulated datasets across six detector geometries, with baseline results for four reconstruction models on five core tasks.

129.7M
simulated ν events
7
datasets
6
detector geometries
5
reconstruction tasks
4
baseline models
Results

Leaderboard

Baseline performance of four reconstruction algorithms on the NuBench test partitions, exactly as reported in the paper. Pick a task, compare models across geometries, then drill into per-dataset metrics.

Data

Seven datasets, six geometries

Simulated νμ charged- and neutral-current interactions produced with Prometheus, with a simplified detector response. Geometries are inspired by — but not identical to — existing and proposed telescopes in water and ice. Each dataset ships in SQLite and Parquet, with train/test selections, baseline predictions, and trained model artifacts.

Baselines

Four reconstruction models

Two architectures in active use within existing collaborations, one graph transformer, and one Kaggle-winning transformer — all implemented and trained in the open-source GraphNeT framework.

Usage

Getting started

1

Download

Grab the format you prefer from the dataset cards above — SQLite and Parquet contain identical data.

wget -P data/ \
  <download-link>
2

Extract

All files end in .tar.gz, but only some are actually compressed.

# SQLite & model artifacts
tar -xzf file.tar.gz
# Parquet & predictions
tar -xf file.tar.gz
3

Load with GraphNeT

Both formats are compatible with the GraphNeT Dataset classes (SQLite recommended). Train/test partitions are defined by the included selection files.

from graphnet.data.dataset import (
    SQLiteDataset,
)
Reference

Citation

If you use the NuBench datasets, predictions, or model artifacts in your work, please cite:

NuBench

@article{orsoe2025nubench,
  title   = {NuBench: An Open Benchmark for Deep Learning--Based Event
             Reconstruction in Neutrino Telescopes},
  author  = {{\O}rs{\o}e, Rasmus F. and Meighen-Berger, Stephan and Lazar,
             Jeffrey and Prado, Jorge and Moz{\'u}n-Mateo, Iv{\'a}n and
             Rosted, Aske and Weigel, Philip and Llorente Anaya, Arturo},
  journal = {arXiv preprint arXiv:2511.13111},
  year    = {2025}
}