Ariadne: PyTorch Library for Particle Track Reconstruction Using Deep Learning

TL;DR

Ariadne is an open-source PyTorch library designed to facilitate particle track reconstruction in high energy physics experiments using deep learning, addressing the limitations of classical algorithms in speed and scalability.

Contribution

It introduces the first comprehensive, modular, and user-friendly deep learning library for particle tracking, enabling easy dataset preparation, model training, and evaluation.

Findings

Provides a flexible framework for deep learning-based particle tracking

Facilitates rapid development and testing of tracking models

Supports experimental data from various high energy physics setups

Abstract

Particle tracking is a fundamental part of the event analysis in high energy and nuclear physics. Events multiplicity increases each year along with the drastic growth of the experimental data which modern HENP detectors produce, so the classical tracking algorithms such as the well-known Kalman filter cannot satisfy speed and scaling requirements. At the same time, breakthroughs in the study of deep learning open an opportunity for the application of high-performance deep neural networks for solving tracking problems in a dense environment of experiments with heavy ions. However, there are no well-documented software libraries for deep learning track reconstruction yet. We introduce Ariadne, the first open-source library for particle tracking based on the PyTorch deep learning framework. The goal of our library is to provide a simple interface that allows one to prepare train and test datasets and to train and evaluate one of the deep tracking models implemented in the library on the data from your specific experiment. The user experience is greatly facilitated because of the system of gin-configurations. The modular structure of the library and abstract classes let the user develop his data processing pipeline and deep tracking model easily. The proposed library is open-source to facilitate academic research in the field of particle tracking based on deep learning.