19–25 Oct 2024
Europe/Zurich timezone

GPU-Accelerated Point Clouds Library for GNN-based Reconstruction and Beyond

24 Oct 2024, 16:33
18m
Room 1.A (Medium Hall A)

Room 1.A (Medium Hall A)

Talk Track 3 - Offline Computing Parallel (Track 3)

Speaker

Shah Rukh Qasim (University of Zurich (CH))

Description

High quality particle reconstruction is crucial to data acquisition at large CERN experiments. While the classical algorithms have been successful so far, in recent years, the use of pattern recognition has become more and more necessary due to increasing complexity of the modern detectors. Graph Neural Network based approaches have been recently proposed to tackle challenges such as non-uniformity and high level of sparsity. They have been shown to work well not only for calorimetric reconstruction [1] but also for tracking [2]. These GNN based approaches require fast GPU execution of certain operations (such as kNN computation). We present a fast GPU-enabled software library which significantly outperforms existing approaches including those present in large machine learning frameworks (such as TensorFlow and PyTorch). In light of these optimizations, we discuss the performance and computational requirements of GNN based reconstruction algorithms. Finally, we also discuss how the application of these techniques can be extended beyond particle physics.

[1] Qasim, Shah Rukh, et al. "End-to-end multi-particle reconstruction in high occupancy imaging calorimeters with graph neural networks." The European Physical Journal C 82.8 (2022): 1-15.
[2] https://indico.jlab.org/event/459/contributions/11761/

Primary author

Shah Rukh Qasim (University of Zurich (CH))

Co-authors

Jan Kieseler (KIT - Karlsruhe Institute of Technology (DE)) Nicola Serra (University of Zurich (CH)) Oleksandr Viazlo

Presentation materials