arXiv:2508.15468hep-excs.AR2025-08中稿 · FPT 2025被引 15

JEDI-linear让图神经网络在FPGA上实现低延迟高精度喷注识别,满足大型强子对撞机实时触发需求。

JEDI-linear: Fast and Efficient Graph Neural Networks for Jet Tagging on FPGAs

  • 采用共享变换与全局聚合,消除成对交互,实现线性计算复杂度。
  • 在FPGA上实现3.7至11.5倍更低延迟,启动间隔降低150倍,功耗降低6.2倍。
  • 专为硬件部署优化,适合高能物理实时触发系统,支持开源复现。

图神经网络(GNN),尤其是相互作用网络(INs),在欧洲核子研究中心(CERN)高亮度大型强子对撞机(HL-LHC)的喷注识别任务中表现出色。然而,其计算复杂度高和内存访问不规则,给硬件触发系统中基于现场可编程门阵列(FPGA)的部署带来挑战,尤其在严格延迟和资源约束下。本文提出JEDI-linear,一种新型线性复杂度的GNN架构,通过共享变换与全局聚合消除显式成对交互。为提升硬件效率,引入细粒度量化感知训练及参数级位宽优化,并采用分布式算术实现无乘法器的乘累加运算。评估结果显示,基于FPGA的JEDI-linear在延迟上比现有最优GNN设计降低3.7至11.5倍,启动间隔最多降低150倍,查找表(LUT)使用量最多降低6.2倍,同时模型准确率更高,且完全无需数字信号处理(DSP)模块。这是首个实现小于60~ns延迟的基于相互作用的GNN,目前符合HL-LHC CMS一级触发系统要求。本工作推动新一代触发系统发展,实现实时环境中高精度、可扩展、资源高效的GNN推理。开源模板将促进科学应用中的可复现性与广泛采用。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs), particularly Interaction Networks (INs), have shown exceptional performance for jet tagging at the CERN High-Luminosity Large Hadron Collider (HL-LHC). However, their computational complexity and irregular memory access patterns pose significant challenges for deployment on FPGAs in hardware trigger systems, where strict latency and resource constraints apply. In this work, we propose JEDI-linear, a novel GNN architecture with linear computational complexity that eliminates explicit pairwise interactions by leveraging shared transformations and global aggregation. To further enhance hardware efficiency, we introduce fine-grained quantization-aware training with per-parameter bitwidth optimization and employ multiplier-free multiply-accumulate operations via distributed arithmetic. Evaluation results show that our FPGA-based JEDI-linear achieves 3.7 to 11.5 times lower latency, up to 150 times lower initiation interval, and up to 6.2 times lower LUT usage compared to state-of-the-art GNN designs while also delivering higher model accuracy and eliminating the need for DSP blocks entirely. This is the first interaction-based GNN to achieve less than 60~ns latency and currently meets the requirements for use in the HL-LHC CMS Level-1 trigger system. This work advances the next-generation trigger systems by enabling accurate, scalable, and resource-efficient GNN inference in real-time environments. Our open-sourced templates will further support reproducibility and broader adoption across scientific applications.

图神经网络FPGA部署喷注识别实时系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。