Search papers, labs, and topics across Lattice.
HEPTv2 introduces an end-to-end point-transformer architecture specifically designed for charged particle tracking in high-energy physics, addressing the challenges posed by high collision densities at the HL-LHC. By integrating a locality-aware point encoder with a track decoder, HEPTv2 eliminates the need for graph construction and auxiliary stages, allowing for efficient and accurate trajectory reconstruction in a single trainable pipeline. The model achieves a remarkable 98.6% tracking efficiency with a 0.8% fake rate, significantly outperforming previous methods in both accuracy and latency, thus setting a new benchmark for real-time particle reconstruction.
Achieving 98.6% tracking efficiency with a 0.8% fake rate, HEPTv2 revolutionizes particle tracking by eliminating the need for graph construction and auxiliary processing.
Charged-particle tracking -- reconstructing trajectories from sparse detector measurements -- is a fundamental high-energy-physics inference problem and a canonical example of learning under extreme combinatorial ambiguity. At the High-Luminosity Large Hadron Collider (HL-LHC), tracking must remain accurate and efficient despite unprecedented collision densities. Graph neural networks perform strongly, but incur substantial costs from graph construction and processing, while transformer-based approaches rely on auxiliary stages that prevent end-to-end optimization. To address this, we present HEPTv2, an end-to-end point-transformer architecture that reconstructs tracks from detector hits in one trainable pipeline. HEPTv2 combines a locality-aware point encoder with a track decoder that predicts complete trajectories without graph-building, clustering, or filtering. The encoder uses locality-sensitive hashing in detector coordinate space to preserve tracking-relevant geometry while enabling efficient local attention. The decoder resolves ambiguities through sectorized decoding and direct hit-to-track prediction under joint encoder-decoder supervision, allowing the full pipeline to be optimized end-to-end. On TrackML, HEPTv2 achieves 98.6% double-majority tracking efficiency at a 0.8% fake rate, while requiring only $\sim$15~ms inference time and 0.4~GB peak memory per event on a NVIDIA A100 GPU. Latency and memory scale approximately linearly for events with up to $5\times10^5$ hits. HEPTv2 establishes a new state of the art in the accuracy-latency trade-off, improving efficiency by 4.5% over the strongest prior transformer and by 1.1--2.2% over optimized graph-based pipelines, while reducing latency by factors of 7 and 38--52, respectively. These results show end-to-end transformers can deliver the accuracy and efficiency required for real-time particle reconstruction at the HL-LHC.
UW