Code

TrackFormers - Machine Learning Pipelines

Nadezhda Dobreva, Antonio Ferrer Sánchez, Zef Wolffs, Yue Zhao

Abstract

TrackFormers is a machine learning framework for track reconstruction in particle physics experiments. It leverages transformer- and U-Net-inspired deep learning architectures to predict particle tracks from hit data.

This repository contains 4 directories corresponding to the 4 models described in the paper TrackFormers: In Search of Transformer-Based Particle Tracking for the High-Luminosity LHC Era. EncDec, EncCla, and EncReg are transformer-based models, whereas U-Net is, as the name suggests, a U-Net model.

Refer to the provided README file for further details.

Cite as » BibTeX download badge

Metadata

Type:
Code
Year:
2024
Repository:
Zenodo
DOI:
10.5281/zenodo.14388534

Links

Licence

Creative Commons Attribution (CC BY) licence Artefacts shared as PDF are licenced under CC BY 4.0.