Search papers, labs, and topics across Lattice.
This paper investigates neural scaling laws for boosted jet classification using the JetClass dataset, a domain where compute budgets are typically much smaller than in LLMs. The authors derive compute-optimal scaling laws, demonstrating that increased compute consistently drives performance towards an asymptotic limit. They also quantify the impact of data repetition, common in HEP due to simulation costs, on scaling behavior and show that more expressive, lower-level features improve performance at fixed dataset size and raise the asymptotic performance limit.
Scaling compute reliably drives performance in high energy physics, but the right features can push the asymptotic limit even higher.
The success of Large Language Models (LLMs) has established that scaling compute, through joint increases in model capacity and dataset size, is the primary driver of performance in modern machine learning. While machine learning has long been an integral component of High Energy Physics (HEP) data analysis workflows, the compute used to train state-of-the-art HEP models remains orders of magnitude below that of industry foundation models. With scaling laws only beginning to be studied in the field, we investigate neural scaling laws for boosted jet classification using the public JetClass dataset. We derive compute optimal scaling laws and identify an effective performance limit that can be consistently approached through increased compute. We study how data repetition, common in HEP where simulation is expensive, modifies the scaling yielding a quantifiable effective dataset size gain. We then study how the scaling coefficients and asymptotic performance limits vary with the choice of input features and particle multiplicity, demonstrating that increased compute reliably drives performance toward an asymptotic limit, and that more expressive, lower-level features can raise the performance limit and improve results at fixed dataset size.
