Published December 12, 2024 | Version 1.0.0
Software Open

TrackFormers - Machine Learning Pipelines

  • 1. ROR icon Radboud University Nijmegen
  • 2. EDMO icon University of Valencia
  • 3. ROR icon University of Amsterdam
  • 4. ROR icon National Institute for Subatomic Physics
  • 5. SURF
  • 1. ROR icon Radboud University Nijmegen
  • 2. ROR icon National Institute for Subatomic Physics
  • 3. EDMO icon University of Valencia
  • 4. ROR icon University of Twente

Description

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.

Files

Files (57.7 MB)

Name Size Download all
md5:26f8e848dddd63a711020b32d0ce7692
41.6 kB Download
md5:7be77bdf4d17e01b2f6961b8a8d34d70
57.7 MB Download

Additional details

Related works

Is cited by
Preprint: 10.48550/arXiv.2407.07179 (DOI)