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

MAKE IMPACT

Zhiqiang Que , S. Paramesvaran , E. Clement , Christopher Edward Brown , A. B. Cox , Wayne Luk · YOU? · · 2025 · Open Access · · DOI: https://doi.org/10.48550/arxiv.2508.15468

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.

Type

preprint

Language

en

Landing Page

http://arxiv.org/abs/2508.15468

PDF

https://arxiv.org/pdf/2508.15468

OA Status

green

OpenAlex ID

https://openalex.org/W4416377998

Raw OpenAlex JSON

OpenAlex ID

https://openalex.org/W4416377998

Canonical identifier for this work in OpenAlex

DOI

https://doi.org/10.48550/arxiv.2508.15468

Digital Object Identifier

Title

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

Work title

Type

preprint

OpenAlex work type

Language

en

Primary language

Publication year

2025

Year of publication

Publication date

2025-08-21

Full publication date if available

Authors

Zhiqiang Que, S. Paramesvaran, E. Clement, Christopher Edward Brown, A. B. Cox, Wayne Luk

List of authors in order

Landing page

https://arxiv.org/abs/2508.15468

Publisher landing page

PDF URL

https://arxiv.org/pdf/2508.15468

Direct link to full text PDF

Open access

Yes

Whether a free full text is available

OA status

green

Open access status per OpenAlex

OA URL

https://arxiv.org/pdf/2508.15468

Direct OA link when available

Cited by

0

Total citation count in OpenAlex

Full payload