AG-2024.06-1124·hep-ph·cross-listed: cs.LG
Particle Multi-Axis Transformer for Jet Tagging
Authors
- Muhammad Usman
- M Husnain Shahid
- Maheen Ejaz
- Ummay Hani
- Nayab Fatima
- Abdul Rehman Khan
- Asifullah Khan
- Nasir Majid Mirza
Abstract
Jet tagging is an essential categorization problem in high energy physics. In recent times, Deep Learning has not only risen to the challenge of jet tagging but also significantly improved its performance. In this article, we proposed an idea of a new architecture, Particle Multi-Axis transformer (ParMAT) which is a modified version of Particle transformer (ParT). ParMAT contains local and global spatial interactions within a single unit which improves its ability to handle various input lengths. We trained our model on JETCLASS, a publicly available large dataset that contains 100M jets of 10 different classes of particles. By integrating a parallel attention mechanism and pairwise interactions of particles in the attention mechanism, ParMAT achieves robustness and higher accuracy over the ParT and ParticleNet. The scalability of the model to huge datasets and its ability to automatically extract essential features demonstrate its potential for enhancing jet tagging.
Submitted
9 June 20241 year ago
Version
v1
License
CC-BY-4.0
DOI
10.48550/arXiv.2406.06638
Chat with this PDF
Ask questions, probe assumptions, request a plain-English summary. Answers cite sections from the preprint itself.
Community
Questions and answers about this paper from other readers. No formal peer review — just a place to think out loud.