From Hits to Tracks: A BERT-based Tracking Model for Track Reconstruction in Drift Chambers

Abstract

Track reconstruction in drift chambers is essential for momentum measurement and particle identification at electron-positron colliders. While Transformer architectures have transformed many sequence-processing domains, their application to tracking in high energy physics is still being explored. We present a model that combines a BERT encoder with a Transformer decoder to perform hit-to-track association through autoregressive sorting. The model is evaluated on the DCTracks open dataset that provides realistic drift chamber simulations with varying particle types, momenta, track multiplicities, and noise conditions. Across single-track, two-track, and multi-track samples, the model achieves high hit and track efficiencies while keeping the rates of clones and fakes very low. It also works well in the reconstruction of displaced vertices. These results show BERT-based sequence-to-sequence models as a promising approach for track reconstruction in low-background, precision-oriented experiments.