TasksSotADatasetsPapersMethodsSubmitAbout
Papers With Code 2

A community resource for machine learning research: papers, code, benchmarks, and state-of-the-art results.

Explore

Notable BenchmarksAll SotADatasetsPapersMethods

Community

Submit ResultsAbout

Data sourced from the PWC Archive (CC-BY-SA 4.0). Built by the community, for the community.

Papers/Particle Transformer for Jet Tagging

Particle Transformer for Jet Tagging

Huilin Qu, Congqiao Li, Sitian Qian

2022-02-08Jet Tagging
PaperPDFCode(official)

Abstract

Jet tagging is a critical yet challenging classification task in particle physics. While deep learning has transformed jet tagging and significantly improved performance, the lack of a large-scale public dataset impedes further enhancement. In this work, we present JetClass, a new comprehensive dataset for jet tagging. The JetClass dataset consists of 100 M jets, about two orders of magnitude larger than existing public datasets. A total of 10 types of jets are simulated, including several types unexplored for tagging so far. Based on the large dataset, we propose a new Transformer-based architecture for jet tagging, called Particle Transformer (ParT). By incorporating pairwise particle interactions in the attention mechanism, ParT achieves higher tagging performance than a plain Transformer and surpasses the previous state-of-the-art, ParticleNet, by a large margin. The pre-trained ParT models, once fine-tuned, also substantially enhance the performance on two widely adopted jet tagging benchmarks. The dataset, code and models are publicly available at https://github.com/jet-universe/particle_transformer.

Results

TaskDatasetMetricValueModel
Point Cloud ClassificationJetClass#Params2140000ParT
Point Cloud ClassificationJetClassAUC0.9877ParT
Point Cloud ClassificationJetClassAccuracy0.861ParT
Point Cloud ClassificationJetClassFLOPs340000000ParT

Related Papers

Guided Graph Compression for Quantum Graph Neural Networks2025-06-11Fast Jet Tagging with MLP-Mixers on FPGAs2025-03-05Learning Symmetry-Independent Jet Representations via Jet-Based Joint Embedding Predictive Architecture2024-12-05Interpreting Transformers for Jet Tagging2024-12-04Lie-Equivariant Quantum Graph Neural Networks2024-11-22Quantum Rationale-Aware Graph Contrastive Learning for Jet Discrimination2024-11-03Lorentz-Equivariant Quantum Graph Neural Network for High-Energy Physics2024-11-03MACK: Mismodeling Addressed with Contrastive Knowledge2024-10-17