Novel fully-heterogeneous GNN designs for track reconstruction at the HL-LHC

Abstract

Data from the LHC detectors are not easily represented using regular data structures. These detectors are comprised of several species of subdetectors and therefore produce heterogeneous data. LHC detectors are highly granular by design so that nearby particles may be distinguished. As a consequence, LHC data are sparse, in that many detector channels are not active during a given collision event. Graphs offer a flexible and efficient alternative to rectilinear data structures for representing LHC data. Accordingly, graph-based machine learning algorithms are becoming increasingly popular for a large number of LHC physics tasks [1, 2]. This popularity, and the corresponding potential for substantial increase in physics output, are illustrated on the cover of a recent issue [3] of the CERN Courier magazine. The graphs used in almost all practical applications at the LHC so far are homogeneous, i.e. each node is assigned the same features, and each edge is assigned the same features. In other words, the power of graphs to represent sparse data has been exploited in applications for the LHC, but the potential of graphs to represent heterogeneous data has not. The pink graph on the cover of the CERN Courier [3] can be seen as an illustration of this limitation: all nodes are pink, regardless of their position in the detector. We present novel fully-heterogeneous GNN designs and apply them to simulated data from a tracking detector that resembles the trackers that will be used at the HL-LHC. It contains a pixel subsystem that provides 3D hits and a strip subsystem that provides 2D hits. We present a new design which solves the degraded performance observed in the strip detector in the first GNNbased tracking studies presented by the ATLAS Collaboration [4].