TrackSorter: 一个基于 Transformer 的高能物理径迹寻找排序算法
💡
原文中文,约300字,阅读约需1分钟。
📝
内容提要
本研究比较了图神经网络和基于核的变压器在高能电子 - 正电子碰撞事件重建中的应用。通过超参数调整和硬件处理器的支持,改善了模型的物理性能。证明了该模型在高粒度输入上训练后,具有与基线相竞争的物理性能。发布了用于重现研究的数据集和软件。
🎯
关键要点
-
研究了可扩展的机器学习模型,用于高能电子 - 正电子碰撞的完全事件重建。
-
比较了图神经网络和基于核的变压器,证明两者在粒子流重建中有效。
-
超参数调整在超级计算机上显著改善了模型的物理性能。
-
模型在Nvidia、AMD和Intel Habana卡的硬件处理器上具有高度可移植性。
-
模型可以在高粒度输入上训练,获得与基线相竞争的物理性能。
-
根据FAIR原则发布了用于重现研究的数据集和软件。
🏷️
标签
➡️