Oct 19 – 23, 2020
Europe/Zurich timezone

Set2Graph: Secondary Vertex finding in Jets with Neural Networks

Oct 21, 2020, 2:20 PM
Regular talk 1 ML for data reduction : Application of Machine Learning to data reduction, reconstruction, building/tagging of intermediate object Workshop


Jonathan Shlomi (Weizmann Institute of Science (IL))


(due to slow internet connection : youtube video + recording of Q&A)

Secondary vertex finding is a crucial task for identifying jets containing heavy flavor hadron decays.
Bottom jets in particular have a very distinctive topology of 𝑏→𝑐→𝑠 decay which gives rise to two secondary vertices with high invariant mass and several associated charged tracks.

Existing secondary vertex finding algorithms search for intersecting particle tracks, and group them into secondary vertices based on geometrical constraints. We propose an algorithm where the vertex finding step is performed with a graph neural network. Tracks are represented as objects in an unordered set, and our proposed model learns a function from this set to a graph that represents the vertex structure in the jet. We prove that models with this structure have maximal expressive power for all (continuous) set to graph functions.

We present performance metrics for evaluating vertex finding performance, and compare the performance of several different graph network architectures on a simulated dataset.

Primary authors

Eilam Gross (Weizmann Institute of Science (IL)) Sanmay Ganguly (Weizmann Institute of Science (IL)) Kyle Stuart Cranmer (New York University (US)) Jonathan Shlomi (Weizmann Institute of Science (IL))

Presentation materials