Differentiable vertex fitting for jet flavor tagging
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. Published by the American Physical Society 2024
- Sponsoring Organization:
- USDOE
- Grant/Contract Number:
- AC02-76SF00515
- OSTI ID:
- 2448027
- Journal Information:
- Physical Review. D., Journal Name: Physical Review. D. Journal Issue: 5 Vol. 110; ISSN PRVDAQ; ISSN 2470-0010
- Publisher:
- American Physical SocietyCopyright Statement
- Country of Publication:
- United States
- Language:
- English
Similar Records
Method to simultaneously facilitate all jet physics tasks
Training toward significance with the decorrelated event classifier transformer neural network