arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.13166physics.ins-det
收藏

利用机器学习改进跟踪算法:高亮度LHC下线段跟踪的案例研究

Improving tracking algorithms with machine learning: a case for line-segment tracking at the High Luminosity LHC

Jonathan Guiang, Slava Krutelyov, Manos Vourliotis, Yanxi Gu, Avi Yagil, Balaji Venkat Sathia Narayanan, Matevz Tadel, Philip Chang, Mayra Silva, Gavin Niendorf… 展开作者

Jonathan Guiang, Slava Krutelyov, Manos Vourliotis, Yanxi Gu, Avi Yagil, Balaji Venkat Sathia Narayanan, Matevz Tadel, Philip Chang, Mayra Silva, Gavin Niendorf, Peter Wittich, Tres Reid, Peter Elmer

AI总结:

本研究以CMS实验的LST算法为例,探索用轻量级神经网络替换轨迹质量选择,并初步研究GNN轨迹构建,证明ML可在不损失性能的前提下改进HL-LHC跟踪。

摘要:

在这项工作中,我们提出了一项关于如何利用机器学习(ML)改进跟踪算法的研究。我们以线段跟踪(LST)算法为基础进行这项研究,该算法被设计为天然并行化和向量化,以便在现代处理器上高效运行。LST是专门为LHC上的CMS实验开发的,面向高亮度LHC(HL-LHC)升级。此外,我们已经展示了随着我们迭代改进LST,利用CMS探测器的完整模拟,其效率和性能结果都非常出色。与此同时,有前景的基于深度学习的跟踪算法,如图神经网络(GNN),正在简化的TrackML数据集上被率先探索。这些结果表明,LST的某些部分可以通过ML得到改进或替换。因此,我们实施了一项逐步深入的调查,精确研究如何以及在何处利用ML,同时仍满足现实的HL-LHC性能和效率约束,具体如下。首先,使用一个轻量级神经网络来替换并改进明确定义的轨迹质量选择。该神经网络被证明对位移轨迹具有高效性和鲁棒性,同时对LST的运行时间几乎没有影响。这些结果清楚地表明,ML可以在不受惩罚的情况下用于改进LST。接下来,描述了GNN轨迹构建算法的探索性研究。特别是,将LST的低层轨迹对象视为图中的节点,其中边代表更高层的对象甚至整个轨迹候选。然后,训练一个边分类器GNN,并将所得边分数的效率与现有LST轨迹质量选择的效率进行比较。这些GNN研究为在CMS实验的HL-LHC跟踪中使用更大胆、更复杂的ML算法的实用性和性能提供了见解。

英文摘要:

In this work, we present a study on ways that tracking algorithms can be improved with machine learning (ML). We base this study on the line segment tracking (LST) algorithm that we have designed to be naturally parallelized and vectorized in order to efficiently run on modern processors. LST has been developed specifically for the CMS Experiment at the LHC, towards the High Luminosity LHC (HL-LHC) upgrade. Moreover, we have already shown excellent efficiency and performance results as we iteratively improve LST, leveraging a full simulation of the CMS detector. At the same time, promising deep-learning-based tracking algorithms, such as Graph Neural Networks (GNNs), are being pioneered on the simplified TrackML dataset. These results suggest that parts of LST could be improved or replaced by ML. Thus, a thorough, step-by-step investigation of exactly how and where ML can be utilized, while still meeting realistic HL-LHC performance and efficiency constraints, is implemented as follows. First, a lightweight neural network is used to replace and improve upon explicitly defined track quality selections. This neural network is shown to be highly efficient and robust to displaced tracks while having little-to-no impact on the runtime of LST. These results clearly establish that ML can be used to improve LST without penalty. Next, exploratory studies of GNN track-building algorithms are described. In particular, low-level track objects from LST are considered as nodes in a graph, where edges represent higher-level objects or even entire track candidates. Then, an edge-classifier GNN is trained, and the efficiency of the resultant edge scores is compared with that of the existing LST track quality selections. These GNN studies provide insights into the practicality and performance of using more ambitious and complex ML algorithms for HL-LHC tracking at the CMS Experiment.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑