arXiv:2606.20437hep-excs.LG2026-06

端到端点变换器实现高能物理粒子轨迹实时重建。

HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstruction

论文配图:HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstruction
图 1 · 摘自论文原文
  • 用局部敏感哈希构建高效点编码器,避免图结构生成。
  • 98.6%轨迹重构效率,0.8%假阳性率,单事件仅需15毫秒。
  • 适合高亮度对撞机实时粒子追踪,性能超越现有模型。

带电粒子轨迹重建——从稀疏探测器测量中恢复轨迹——是高能物理中的核心推断问题,也是极端组合模糊性下的典型学习挑战。在高亮度大型强子对撞机(HL-LHC)中,需在碰撞密度空前升高的条件下保持高精度与高效率。图神经网络表现良好,但图构建与处理成本高;基于变换器的方法依赖辅助阶段,难以端到端优化。为此,我们提出HEPTv2,一种从探测器击中点直接输出完整轨迹的端到端点变换器架构。其编码器利用探测器坐标空间中的局部敏感哈希,保留轨迹相关几何信息的同时实现高效局部注意力;解码器通过分扇区解码与联合监督下的直接击中-轨迹预测,消除歧义。在TrackML数据集上,HEPTv2实现98.6%双多数追踪效率,假率仅0.8%,单事件推理时间约15毫秒,峰值内存0.4GB(NVIDIA A100 GPU)。延迟与内存随事件最大击中数达5×10⁵时近似线性增长。该方法在准确率-延迟权衡上建立新基准,相比最强变压器提升4.5%效率,比优化后的图基方法提升1.1–2.2%,延迟降低7倍及38–52倍。

原文摘要 · Abstract (English)

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.

粒子重建端到端变换器实时计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。