基于 GNN 和 CKF 的组合轨道发现

💡 原文中文,约300字,阅读约需1分钟。
📝

内容提要

本研究比较了图神经网络和基于核的变压器在高能电子 - 正电子碰撞的完全事件重建中的性能优势,并通过超参数调整和硬件处理器的支持进一步提高了模型性能。研究结果可用于重现研究。

🎯

关键要点

  • 研究了可扩展的机器学习模型,用于高能电子 - 正电子碰撞的完全事件重建。
  • 比较了图神经网络和基于核的变压器,证明两者在粒子流重建中具有优势。
  • 避免了二次内存分配和计算成本。
  • 在超级计算机上进行的超参数调整显著改善了模型的物理性能。
  • 模型在Nvidia、AMD和Intel Habana卡的硬件处理器上具有高度可移植性。
  • 模型可以在高粒度输入上进行训练,获得与基线相竞争的物理性能。
  • 根据FAIR原则发布了用于重现研究的数据集和软件。
➡️

继续阅读