Pith. sign in

REVIEW 9 cited by

HEP-JEPA: A foundation model for collider physics using joint embedding predictive architecture

Not yet reviewed by Pith; the record is open.

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2502.03933 v1 pith:XPC7UWH2 submitted 2025-02-06 cs.LG hep-exhep-ph

HEP-JEPA: A foundation model for collider physics using joint embedding predictive architecture

classification cs.LG hep-exhep-ph
keywords modeljetstasksarchitecturecolliderconstituentsembeddingfoundation
verification ladder T0 review T1 audit T2 compute T3 formal T4 reserved
0 comments
read the original abstract

We present a transformer architecture-based foundation model for tasks at high-energy particle colliders such as the Large Hadron Collider. We train the model to classify jets using a self-supervised strategy inspired by the Joint Embedding Predictive Architecture. We use the JetClass dataset containing 100M jets of various known particles to pre-train the model with a data-centric approach -- the model uses a fraction of the jet constituents as the context to predict the embeddings of the unseen target constituents. Our pre-trained model fares well with other datasets for standard classification benchmark tasks. We test our model on two additional downstream tasks: top tagging and differentiating light-quark jets from gluon jets. We also evaluate our model with task-specific metrics and baselines and compare it with state-of-the-art models in high-energy physics. Project site: https://hep-jepa.github.io/

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 9 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. One Generator, Any Process: LLM-Conditioning for the LHC

    hep-ph 2026-06 unverdicted novelty 7.0

    LLM embeddings condition generative networks for LHC events, yielding faster convergence, higher quality, and generalization to unseen processes.

  2. Learning transferable event representations for charmed baryon physics at BESIII

    physics.data-an 2026-07 conditional novelty 6.0

    A Particle Transformer pre-trained on simulated Lambda_c events transfers across 12 decay channels, improving classification and momentum-direction regression over training from scratch in low-statistics regimes.

  3. One Generator, Any Process: LLM-Conditioning for the LHC

    hep-ph 2026-06 unverdicted novelty 6.0

    LLM embeddings condition a generative transformer to enable faster convergence, better performance, and generalization to unseen LHC processes using a single model.

  4. AeroJEPA: Learning Semantic Latent Representations for Scalable 3D Aerodynamic Field Modeling

    cs.LG 2026-05 unverdicted novelty 6.0

    AeroJEPA applies joint-embedding predictive learning to produce scalable, semantically organized latent representations for 3D aerodynamic fields that support both field reconstruction and downstream design tasks.

  5. Explicit or Implicit? Encoding Physics at the Precision Frontier

    hep-ph 2026-03 conditional novelty 6.0

    On three precision classification tasks — reweighting-based unfolding, likelihood-ratio estimation, and weakly supervised anomaly detection — a Lorentz-equivariant transformer and a pretrained foundation model perform...

  6. Enhancing next token prediction based pre-training for jet foundation models

    hep-ph 2025-12 conditional novelty 6.0

    Using continuous particle features as input and combining next-token with masked-token pre-training markedly improves classification accuracy of the OmniJet jet foundation model without visibly hurting its generative quality.

  7. Towards Engineering Scaling Laws with Pretraining Data Composition

    hep-ex 2026-06 unverdicted novelty 5.0

    Pretraining data composition can be used to engineer neural scaling laws in hadronic jet classification toward data-heavy rather than model-size-heavy regimes.

  8. Are We Ready for AI-Driven Discovery? AI Verification Before the Next Fundamental Physics Breakthrough

    physics.data-an 2026-07 accept novelty 4.0

    Verification of ML in fundamental physics is essential precisely when models enter statistical modeling, inference, or hypothesis testing, and is bounded by unavoidable inductive bias, sample complexity, and experimen...

  9. Towards Engineering Scaling Laws with Pretraining Data Composition

    hep-ex 2026-06 unverdicted novelty 4.0

    Pretraining data composition can engineer scaling laws for jet classification to favor data scaling over model scaling.