Attention Is All You Need
Ashish Vaswani, Noam Shazeer et al.
0
Citations
0
Influential Citations
—
Venue
2210
Year
We present a smoothly broken power law functional form (that we refer to as a Broken Neural Scaling Law (BNSL)) that accurately models & extrapolates the scaling behaviors of deep …
Scaling laws have become a cornerstone of modern deep learning, guiding decisions on model size, data volume, and compute investment. However, traditional single power law fits often fail to capture regime changes—for example, when a model transitions from being data-limited to parameter-limited. This paper addresses that gap by introducing a smoothly broken power law, which can model such transitions explicitly. The ability to extrapolate scaling behavior accurately is critical for planning large-scale experiments and avoiding costly trial-and-error.
The key innovation is the Broken Neural Scaling Law (BNSL), a parametric function that combines two power laws with a smooth transition region. This allows the model to capture different scaling regimes (e.g., pre- and post-saturation) within a single, differentiable form. The authors likely derive the functional form from empirical observations and validate it on multiple datasets and model families. The work provides a principled way to estimate the breakpoint and exponents, enabling more reliable predictions beyond the observed range.
While the abstract does not provide specific numerical results, the claim is that BNSL fits empirical data better than a single power law and extrapolates more accurately. This suggests improvements in metrics like mean squared error on held-out scaling data. The paper likely includes comparisons on standard benchmarks (e.g., language modeling, vision) showing reduced extrapolation error.
This work has direct implications for the AI community: it offers a practical tool for predicting model performance at scales that are expensive or infeasible to test directly. By capturing regime changes, it can inform decisions about when to scale data vs. parameters, potentially saving significant compute resources. The BNSL form may become a standard component in scaling law analyses, similar to how power laws are used today.
Ashish Vaswani, Noam Shazeer et al.
Jakubův, Jan, Chvalovský, Karel et al.
Pauli Virtanen, Ralf Gommers et al.
Tom B. Brown, Benjamin Mann et al.