用图神经网络实现实时粒子轨迹重建,提升高能物理数据处理效率。
Real-Time Analysis of Unstructured Data with Machine Learning on Heterogeneous Architectures
- 基于图神经网络构建端到端轨迹重建流水线,部署于GPU
- 在40MHz高频下实现比传统算法更高的吞吐量与更低能耗
- 验证了异构架构(GPU/FPGA)在实时触发系统中的可行性
随着高能物理实验对精度要求不断提高,更大规模的数据集成为必要。全球对撞机实验(特别是欧洲核子研究中心的大型强子对撞机)探测器升级后,预计将产生更多碰撞事件和更复杂的相互作用,直接导致数据量激增,计算资源需求随之上升。在CERN,数据量极为庞大,必须在实时阶段进行高效筛选与选择,才能永久存储。这些数据用于后续物理分析,以拓展我们对宇宙的认知并完善标准模型。这一实时筛选过程称为触发,通常需在高达40MHz的频率下完成复杂计算。本论文研究机器学习模型在异构架构上的高效部署,旨在最大化处理吞吐量并最小化能耗。本文提出了一种基于图神经网络的轨迹重建方案,应用于LHCb实验的一级触发系统,全程运行于GPU。该方案性能经对比测试,优于当前生产环境中的经典追踪算法。同时,该方案也在FPGA上进行了加速,其功耗与处理速度与GPU版本进行了比较。
原文摘要 · Abstract (English)
As the particle physics community needs higher and higher precisions in order to test our current model of the subatomic world, larger and larger datasets are necessary. With upgrades scheduled for the detectors of colliding-beam experiments around the world, and specifically at the Large Hadron Collider at CERN, more collisions and more complex interactions are expected. This directly implies an increase in data produced and consequently in the computational resources needed to process them. At CERN, the amount of data produced is gargantuan. This is why the data have to be heavily filtered and selected in real time before being permanently stored. This data can then be used to perform physics analyses, in order to expand our current understanding of the universe and improve the Standard Model of physics. This real-time filtering, known as triggering, involves complex processing happening often at frequencies as high as 40 MHz. This thesis contributes to understanding how machine learning models can be efficiently deployed in such environments, in order to maximize throughput and minimize energy consumption. Inevitably, modern hardware designed for such tasks and contemporary algorithms are needed in order to meet the challenges posed by the stringent, high-frequency data rates. In this work, I present our graph neural network-based pipeline, developed for charged particle track reconstruction at the LHCb experiment at CERN. The pipeline was implemented end-to-end inside LHCb's first-level trigger, entirely on GPUs. Its performance was compared against the classical tracking algorithms currently in production at LHCb. The pipeline was also accelerated on the FPGA architecture, and its performance in terms of power consumption and processing speed was compared against the GPU implementation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。