Partition pooling for convolutional graph network applications in particle physics

Journal of Instrumentation, vol. 17, pp. P10004

Abstract

Abstract Convolutional graph networks are used in particle physics for effective event reconstructions and classifications. However, their performances can be limited by the considerable amount of sensors used in modern particle detectors if applied to sensor-level data. We present a pooling scheme that uses partitioning to create pooling kernels on graphs, similar to pooling on images. Partition pooling can be used to adopt successful image recognition architectures for graph neural network applications in particle physics. The reduced computational resources allow for deeper networks and more extensive hyperparameter optimizations. To show its applicability, we construct a convolutional graph network with partition pooling that reconstructs simulated interaction vertices for an idealized neutrino detector. The pooling network yields improved performance and is less susceptible to overfitting than a similar network without pooling. The lower resource requirements allow the construction of a deeper network with further improved performance.

Authors 5

1. Markus Bachlechner Aachen III Physics Institute B

   RWTH Aachen University

   Affiliation as printed

   III Physics Institute B, RWTH Aachen University, Otto-Blumenthal-Straße, Aachen, Germany

2. Thilo Birkenfeld corresponding Aachen III Physics Institute B

   RWTH Aachen University

   Affiliation as printed

   III Physics Institute B, RWTH Aachen University, Otto-Blumenthal-Straße, Aachen, Germany

3. Philipp Soldin Aachen III Physics Institute B

   RWTH Aachen University

   Affiliation as printed

   III Physics Institute B, RWTH Aachen University, Otto-Blumenthal-Straße, Aachen, Germany

4. A. Stahl Aachen III Physics Institute B

   RWTH Aachen University

   Affiliation as printed

   III Physics Institute B, RWTH Aachen University, Otto-Blumenthal-Straße, Aachen, Germany

5. Christopher H. Wiebusch Aachen III Physics Institute B

   RWTH Aachen University

   Affiliation as printed

   III Physics Institute B, RWTH Aachen University, Otto-Blumenthal-Straße, Aachen, Germany