vix.ing · top · new · best · stats · spec

Equivariant Graph Neural Networks for Charged Particle Tracking

2023/04/11 by Murnane, Daniel, Thais, Savannah, Thete, Ameya · 3 citations
#Computational Physics (physics.comp-ph) #FOS: Computer and information sciences #FOS: Physical sciences #High Energy Physics - Experiment (hep-ex) #Instrumentation and Detectors (physics.ins-det) #Machine Learning (cs.LG)

paper · doi:10.48550/arxiv.2304.05293

Abstract

Graph neural networks (GNNs) have gained traction in high-energy physics (HEP) for their potential to improve accuracy and scalability. However, their resource-intensive nature and complex operations have motivated the development of symmetry-equivariant architectures. In this work, we introduce EuclidNet, a novel symmetry-equivariant GNN for charged particle tracking. EuclidNet leverages the graph representation of collision events and enforces rotational symmetry with respect to the detector's beamline axis, leading to a more efficient model. We benchmark EuclidNet against the state-of-the-art Interaction Network on the TrackML dataset, which simulates high-pileup conditions expected at the High-Luminosity Large Hadron Collider (HL-LHC). Our results show that EuclidNet achieves near-state-of-the-art performance at small model scales (<1000 parameters), outperforming the non-equivariant benchmarks. This study paves the way for future investigations into more resource-efficient GNN models for particle tracking in HEP experiments.

Cited by

Related