FPGA GNN Improves Belle II Calorimeter Trigger Clustering
A graph trigger module processes calorimeter cells as nodes, sustaining MHz throughput with 3.168 μs latency and improving position or cluster metrics.
Underlying Paper
Real-time graph neural networks on FPGAs for the Belle II electromagnetic calorimeter
We present the development and evaluation of a real-time Graph Neural Network-based trigger module for the electromagnetic calorimeter of the Belle~II experiment at the SuperKEKB collider. The algorithm processes calorimeter trigger cells as graph nodes to perform clustering, feature extraction, and per-cluster signal classification with deterministic latency. The model predicts cluster positions and energies and provides a signal classification score, enabling a more flexible clustering strategy than the baseline trigger algorithm. Implemented on an FPGA and integrated into the Belle~II trigger readout infrastructure for synchronous operation, the system sustains the MHz trigger throughput with an end-to-end latency of $3.168\,\mu$s. The performance is evaluated on simulated events and collision data. The energy resolution is comparable to the baseline trigger, while the position resolution for high-energy clusters improves by up to 18% in the central detector region. Cluster purity increases by up to 20% at low energies for isolated clusters, and cluster efficiency improves by up to 20% for overlapping clusters. The signal classifier enables additional background suppression at fixed signal retention. These results demonstrate a first step towards GNN-based real-time reconstruction on FPGAs in a collider trigger. While the end-to-end latency exceeds the trigger decision budget, the system already sustains full operational conditions with 100% uptime.
The Belle II electromagnetic calorimeter trigger has to turn a high-rate, spatially structured detector readout into cluster objects fast enough for a synchronous hardware trigger. The baseline trigger uses fixed clustering logic, which is reliable but constrains how overlapping showers, low-energy deposits, and background-like patterns can be separated. This paper introduces a graph neural network trigger module for the calorimeter: trigger cells become graph nodes, local connectivity encodes detector geometry, and the FPGA implementation emits cluster energy, position, and a signal score under deterministic timing.
Core Contribution
The main contribution is not just applying a GNN to calorimeter reconstruction. The paper carries the model through the parts that usually decide whether such methods are usable in a collider trigger: graph construction matched to trigger cells, fixed-latency FPGA inference, integration into the Belle II trigger readout chain, and comparison across software simulation, firmware-level simulation, and hardware output. The result is a working GNN-based trigger element that can run under Belle II operational conditions, even though its current end-to-end latency is still above the final trigger decision budget.
The authors’ claim is therefore narrower and more useful than a generic “machine learning trigger” claim. They show that a graph model can replace more rigid calorimeter clustering logic with a programmable reconstruction module that preserves MHz throughput and improves selected reconstruction metrics.
Technical Approach
The GNN-ETM represents calorimeter trigger cells as graph nodes and performs message passing over detector-neighbor relations. Its outputs are cluster-level quantities: position, energy, and a per-cluster signal classification score. That design choice matters because the trigger does not only need a scalar event decision; it needs trigger objects that downstream logic can use. Treating the calorimeter as a graph gives the model a direct way to reason about local shower topology, including cases where fixed windows or hand-coded cluster splitting rules are brittle.
The implementation path is as important as the network architecture. The paper validates the algorithm at several abstraction levels, from model-level simulation through firmware simulation and the physical FPGA output. Figure 19 summarizes this validation chain and the software/hardware comparison points, which is the clearest evidence that the work is an engineering demonstration rather than a detached offline study.
For trigger work, deterministic latency is part of the method. The module is integrated into the Belle II trigger readout infrastructure and runs synchronously with the experiment. The reported end-to-end latency is 3.168 μs. That is fast in ordinary inference terms, but the paper is explicit that it exceeds the trigger decision budget, so the current version is better read as a deployed feasibility step than as a drop-in final trigger replacement.
Results and Analysis
The reported reconstruction results are strongest where graph structure should help. Energy resolution is comparable to the baseline trigger, which is a necessary result: improved clustering would be less useful if it degraded the calorimeter energy estimate. The larger gains are in position and cluster separation. For high-energy clusters in the central detector region, the paper reports position-resolution improvement of up to 18%. For isolated low-energy clusters, cluster purity increases by up to 20%. For overlapping clusters, cluster efficiency improves by up to 20%.
Those numbers support the central technical argument. A graph model is not merely matching the baseline; it changes the trade-off in cases where spatial topology matters. The improvement on overlapping clusters is especially relevant for a trigger, because fixed clustering logic can merge or lose nearby deposits. The signal classifier adds another handle: at fixed signal retention, it can suppress additional background. The abstract does not give a single universal rejection factor, so the safest interpretation is that the classifier is useful as a tunable trigger feature rather than a fully quantified replacement criterion.
The hardware result is also substantive. The system sustains MHz trigger throughput and the paper reports 100% uptime under full operational conditions. That combination separates this work from offline GNN calorimeter studies, where accuracy can be evaluated without confronting clocking, buffering, and synchronous readout constraints. The main caveat is latency: 3.168 μs is deterministic and operational, but still too long for the target decision path. Future value depends on whether the authors can reduce that latency while preserving the reconstruction gains.
Limitations
The evidence is strong for feasibility and targeted reconstruction gains, but bounded. The current latency exceeds the trigger decision budget, so the module is not yet a complete production trigger solution. The largest reported improvements are conditional: high-energy central-region position resolution, low-energy isolated-cluster purity, and overlapping-cluster efficiency. The paper also evaluates a Belle II calorimeter-specific graph and FPGA integration, so portability to other detector geometries or trigger systems remains an engineering question, not a demonstrated result.
Evidence Box
strongKey Claims
- •GNN trigger processing can replace fixed calorimeter clustering logic
- •Graph-based reconstruction improves topology-sensitive cluster metrics
- •FPGA implementation can sustain Belle II MHz trigger throughput
- •Signal score enables extra background suppression at fixed signal retention
Key Results
- •3.168 μs end-to-end latency for the integrated FPGA trigger module
- •Up to 18% position-resolution improvement for high-energy clusters in the central detector region
- •Up to 20% cluster-purity increase for low-energy isolated clusters
- •Up to 20% cluster-efficiency improvement for overlapping clusters
Limitations & Caveats
- •End-to-end latency exceeds the trigger decision budget
- •Largest gains are reported for specific energy, isolation, overlap, and detector-region conditions
- •Evaluation is specific to the Belle II electromagnetic calorimeter trigger geometry
- •Classifier background suppression is described at fixed signal retention without a single headline rejection factor