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

用于带电粒子径迹重建的等变图神经网络

Equivariant Graph Neural Networks for Charged Particle Tracking

Lawrence Berkeley National Laboratory(劳伦斯伯克利国家实验室);Columbia University(哥伦比亚大学)

Daniel Murnane, Savannah Thais, Ameya Thete

AI总结:

本文提出对称等变图神经网络EuclidNet,用于带电粒子径迹重建,通过在TrackML数据集上对比验证其在极小参数规模下接近最先进性能并优于非等变基准。

摘要:

图神经网络(GNNs)在高能物理(HEP)领域因其提升精度和可扩展性的潜力而受到越来越多的关注。然而,其资源密集的特性和复杂的运算促使人们发展对称等变架构。在本工作中,我们提出了EuclidNet,一种用于带电粒子径迹重建的新型对称等变GNN。EuclidNet利用碰撞事件的图表示,并强制施加相对于探测器束流轴线的旋转对称性,从而得到更高效的模型。我们在TrackML数据集上将EuclidNet与最先进的Interaction Network进行基准对比,该数据集模拟了高亮度大型强子对撞机(HL-LHC)预期的高堆积(pileup)条件。我们的结果表明,EuclidNet在极小模型规模(<1000个参数)下达到了接近最先进的性能,优于非等变基准。本研究为未来在高能物理实验中探索更节省资源的粒子径迹重建GNN模型铺平了道路。

英文摘要:

Graph neural networks (GNNs) have gained traction in high-energy physics (HEP) for their potential to improve accuracy and scalability. However, their resource-intensive nature and complex operations have motivated the development of symmetry-equivariant architectures. In this work, we introduce EuclidNet, a novel symmetry-equivariant GNN for charged particle tracking. EuclidNet leverages the graph representation of collision events and enforces rotational symmetry with respect to the detector's beamline axis, leading to a more efficient model. We benchmark EuclidNet against the state-of-the-art Interaction Network on the TrackML dataset, which simulates high-pileup conditions expected at the High-Luminosity Large Hadron Collider (HL-LHC). Our results show that EuclidNet achieves near-state-of-the-art performance at small model scales (<1000 parameters), outperforming the non-equivariant benchmarks. This study paves the way for future investigations into more resource-efficient GNN models for particle tracking in HEP experiments.

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

↑