Predict before you train: Scaling Laws for particle physics foundation models
hep-ex, cs.AI
Submitted: 2026-07-25
Updated: 2026-08-25
Code: https://github.com/Jaluus/ParticleViT
License: http://creativecommons.org/licenses/by/4.0/
The gist: The largest machine learning models in particle physics are also the most expensive to train, yet the return on scaling a given architecture cannot be estimated before that compute is spent.
Terminology
Abstract
The largest machine learning models in particle physics are also the most expensive to train, yet the return on scaling a given architecture cannot be estimated before that compute is spent. Scaling laws have been fit for jets, but none has yet been shown to predict the performance of models it was not fit on. We show that, for a generic transformer pretrained on collider jets, it can be forecast. Fitting a joint model-and-data scaling law on small models alone, spanning three orders of magnitude of training compute, we predict the loss of models trained afterward with more than one hundred times more compute to within one percent. We then connect the forecast to downstream physics performance: across two standard tagging benchmarks, lower pretraining loss yields systematically lower fine-tuning loss and higher background rejection after fine-tuning. Within this model family and these tasks, a compute budget can therefore be translated into expected physics performance before any large model is trained. The final frontier model is consistent with the published numbers for current state-of-the-art physics-aware foundation models trained on the same corpus, on accuracy, AUC, and quark/gluon rejection, with a residual edge for the physics-aware model only in the high-purity tail of top tagging. We release five pretrained models spanning multiple sizes, together with the complete training recipe and code.
Sources
- Scaling Laws for Neural Language Models
- Scaling Laws for Autoregressive Generative Modeling
- Training Compute-Optimal Large Language Models
- An AI-ready, Polarized Electron-Positron Collision Dataset
- OmniLearned: A Foundation Model Framework for All Tasks Involving Jet Physics
- Particle Transformer for Jet Tagging
- OmniJet-$\alpha$: The first cross-task foundation model for particle physics
- Masked Particle Modeling on Sets: Towards Self-Supervised High Energy Physics Foundation Models
- Scaling Laws in Jet Classification
- Neural Scaling Laws for Boosted Jet Tagging
- Neural Scaling Laws for Jet Generation
- The Machine Learning Landscape of Top Taggers
- Energy Flow Networks: Deep Sets for Particle Jets
- Olmo 3
- Root Mean Square Layer Normalization
- Scaling Vision Transformers to 22 Billion Parameters
- GLU Variants Improve Transformer
- An Efficient Lorentz Equivariant Graph Neural Network for Jet Tagging
- Explainable Equivariant Neural Networks for Particle Physics: PELICAN
- PELICAN: Permutation Equivariant and Lorentz Invariant or Covariant Aggregator Network for Particle Physics
Related papers
- Agentic Hybrid RAG for Evidence-Grounded Muon Collider Analysis
- Using Line Segment Tracking to enable ML-based Track Reconstruction in CMS
- Towards foundation-style models for energy-frontier heterogeneous neutrino detectors via self-supervised pre-training
- Machine Can Automatically Discover Parametric Functions to Model HEP Data
- Agentic AI -- Physicist Collaboration in Experimental Particle Physics: A Proof-of-Concept Measurement with LEP Open Data