Differentiable vertex fitting for jet flavor tagging

Rachel E. C. Smith,Inês Ochoa,Rúben Inácio,Jonathan Shoemaker,Michael Kagan
DOI: https://doi.org/10.1103/physrevd.110.052010
IF: 5.407
2024-09-26
Physical Review D
Abstract:This work explores the use of differentiable programming to integrate domain knowledge, in the form of domain specific software, into neural networks to develop scientific machine learning systems. We propose a differentiable vertex fitting algorithm that estimates the crossing point of multiple curves. In the high energy physics setting, these curves are defined by particle equations of motion and the crossing point represents the origin of particle production. This differentiable vertex fitting algorithm can be seamlessly integrated into neural networks, and we show its utility and efficacy in the high energy physics application of the classification of jets, i.e., collimated streams of particles in particle detectors whose originating parent particle we aim to classify. We demonstrate how differentiable vertex fitting can be integrated into larger transformer-based models for jet flavor tagging and show improvements in heavy flavor jet classification when compared to baseline models. https://doi.org/10.1103/PhysRevD.110.052010 Published by the American Physical Society under the terms of the Creative Commons Attribution 4.0 International license. Further distribution of this work must maintain attribution to the author(s) and the published article's title, journal citation, and DOI. Funded by SCOAP 3 . Published by the American Physical Society
astronomy & astrophysics,physics, particles & fields
What problem does this paper attempt to address?