Scalable multi-task learning for particle collision event reconstruction with heterogeneous graph neural networks
Year: 2025
Type: article
Abstract: The growing luminosity frontier at the Large Hadron Collider is challenging the reconstruction and analysis of particle collision events. Increased particle multiplicities are straining latency and storage requirements at the data acquisition stage, while new complications are emerging, including higher background levels and more frequent particle vertex misassociations. This in turn necessitates the development of more holistic and scalable reconstruction methods that take advantage of recent advances in machine learning. We propose a novel heterogeneous graph neural network (HGNN) architecture featuring unique representations for diverse particle collision relationships and integrated graph pruning layers for scalability. Trained with a multi-task paradigm in an environment mimicking the LHCb experiment, this HGNN significantly improves the beauty hadron reconstruction performance. Notably, it concurrently performs particle vertex association and graph pruning within a single framework. We quantify the reconstruction and pruning performance, demonstrate enhanced inference time scaling with event complexity, and mitigate potential performance loss using a weighted message passing scheme.
DOI: 10.5167/uzh-282431
Authors: Sutcliffe, William, Calvi, Marta, Capelli, Simone, Eschle, Jonas, García Pardiñas, Julián, Mathad, Abhijit, Uzuki, Azusa, Serra, Nicola
Language: English
Subfield: Nuclear and High Energy Physics
Field: Physics and Astronomy
Domain: Physical Sciences
Open Access status: green
OpenAlex ID: W7124769473
FWCI:
Cites:
Cited by:
Related to: