30 April 2024 to 3 May 2024
Amsterdam, Hotel CASA
Europe/Amsterdam timezone

Differentiable Vertex Fitting for Jet Flavour Tagging

30 Apr 2024, 13:30
3m
Sorbonne, Hotel CASA

Sorbonne, Hotel CASA

Speaker

Rachel Smith (SLAC National Accelerator Laboratory)

Description

We propose a differentiable vertex fitting algorithm that can be used for secondary vertex fitting, and that can be seamlessly integrated into neural networks for jet flavour tagging. Vertex fitting is formulated as an optimization problem where gradients of the optimized solution vertex are defined through implicit differentiation and can be passed to upstream or downstream neural network components for network training. More broadly, this is an application of differentiable programming to integrate physics knowledge into neural network models in high energy physics. We demonstrate how differentiable secondary vertex fitting can be integrated into larger transformer-based models for flavour tagging and improve heavy flavour jet classification.

Primary authors

Inês Ochoa (LIP) Jonathan Shoemaker (SLAC National Accelerator Laboratory) Michael Kagan (SLAC National Accelerator Laboratory) Rachel Smith (SLAC National Accelerator Laboratory) Rúben Inácio (LIP)

Presentation materials