Graph Neural Network with Attention and Two-Stage Aggregation for Particle Track Reconstruction in the TPC MPD of the NICA Accelerator Complex
Year: 2025
Type: article
Abstract: A Graph Neural Network (GNN) with an attention mechanism and two-stage aggregation for particle track reconstruction in the Time-Projection Chamber (TPC) of the Multi-Purpose Detector (MPD) of the Nuclotron-based Ion Collider fAcility (NICA) is presented. The model is trained and evaluated on a dataset of 1000 Au–Au collision events generated with MPDRoot. The model architecture includes node and edge encoders, an initial edge classifier, a graph attentional convolution layer, an edge embedding updater, and a final edge classifier, with the core blocks repeated for iterative refinement. A comparative analysis shows that the introduction of the attention mechanism and two-stage aggregation improves performance in edge classification, achieving 96.2% accuracy along with 92.6% in both purity and efficiency metrics. The track reconstruction efficiency exceeds 90% for track integrities below 80%, but rapidly drops for higher integrity thresholds.
Institution: Joint Institute for Nuclear Research
Language: English
Subfield: Nuclear and High Energy Physics
Field: Physics and Astronomy
Domain: Physical Sciences
Open Access status: hybrid
Estimated APC paid: $3,790
OpenAlex ID: W4417237140