Abstract:High-Energy Physics experiments are facing a multi-fold data increase with every new iteration. This is certainly the case for the upcoming High-Luminosity LHC upgrade. Such increased data processing requirements forces revisions to almost every step of the data processing pipeline. One such step in need of an overhaul is the task of particle track reconstruction, a.k.a., tracking. A Machine Learning-assisted solution is expected to provide significant improvements, since the most time-consuming step in tracking is the assignment of hits to particles or track candidates. This is the topic of this paper. We take inspiration from large language models. As such, we consider two approaches: the prediction of the next word in a sentence (next hit point in a track), as well as the one-shot prediction of all hits within an event. In an extensive design effort, we have experimented with three models based on the Transformer architecture and one model based on the U-Net architecture, performing track association predictions for collision event hit points. In our evaluation, we consider a spectrum of simple to complex representations of the problem, eliminating designs with lower metrics early on. We report extensive results, covering both prediction accuracy (score) and computational performance. We have made use of the REDVID simulation framework, as well as reductions applied to the TrackML data set, to compose five data sets from simple to complex, for our experiments. The results highlight distinct advantages among different designs in terms of prediction accuracy and computational performance, demonstrating the efficiency of our methodology. Most importantly, the results show the viability of a one-shot encoder-classifier based Transformer solution as a practical approach for the task of tracking.

Extracting Signal Electron Trajectories in the COMET Phase-I Cylindrical Drift Chamber Using Deep Learning

An FPGA-based Trigger System with Online Track Recognition in COMET Phase-I

Track reconstruction for the COMET Phase-II experiment with ACTS

Cluster Counting Algorithm for the CEPC Drift Chamber using LSTM and DGCNN

COMET: Context-Aware IoU-Guided Network for Small Object Tracking

A Deep Learning Approach to Multi-Track Location and Orientation in Gaseous Drift Chambers

Deep learning for track recognition in pixel and strip-based particle detectors

Removing Noise From Simulated Events at The Main Drift Chamber of BESIII Using Convolutional Neural Networks

High Level Reconstruction with Deep Learning using ILD Full Simulation

Particle Track Reconstruction using Geometric Deep Learning

Transformers for Charged Particle Track Reconstruction in High Energy Physics

Deep Learning for Signal and Background Discrimination in Liquid Based Neutrino Experiment

End-to-end particle and event identification at the Large Hadron Collider with CMS Open Data

Detecting the Chiral Magnetic Effect Via Deep Learning

Simulation and reconstruction of particle trajectories in the CEPC drift chamber

U-Net-based Deep Learning for Tracking and Quantitative Analysis of Intracellular Vesicles in Time-Lapse Microscopy Images

Searching for Exotic Particles in High-Energy Physics with Deep Learning

TrackFormers: In Search of Transformer-Based Particle Tracking for the High-Luminosity LHC Era

Detecting Chiral Magnetic Effect via Deep Learning

Deep Learning-based Alpha Particles Spectroscopy with Solid-state Nuclear Track Detector CR-39