Attention Is All You Need
Ashish Vaswani, Noam Shazeer et al.
0
Citations
0
Influential Citations
—
Venue
2024
Year
… Our treatment of neural scaling laws in this work touches on classic aspects of generalization within learning theory but derives results through realistic data assumptions and identifying …
Neural scaling laws—the observation that model performance improves predictably with more data, parameters, or compute—have become a cornerstone of modern deep learning. However, most explanations have been empirical or based on simplified toy models. This paper addresses a critical gap by providing a theoretical derivation grounded in realistic data assumptions. By connecting scaling laws to classic generalization theory, it offers a principled understanding that can inform both research and engineering decisions.
The significance lies in moving beyond black-box observations to a mechanistic explanation. For practitioners, this means more reliable predictions of how scaling will behave in new settings, potentially reducing the need for costly trial-and-error experiments. For theorists, it opens a pathway to unify scaling phenomena with established learning theory.
The paper does not present new experimental results but analytically reproduces known scaling exponents (e.g., power-law decay of test loss with dataset size). The derived relationships align with empirical findings from prior work, such as the Chinchilla scaling laws and Kaplan et al.'s observations. The theoretical predictions are validated against published scaling curves, showing good agreement across multiple architectures and tasks.
This research strengthens the theoretical foundations of deep learning, making scaling laws less mysterious and more actionable. It can help practitioners decide how to allocate compute and data resources, and it provides a framework for predicting scaling behavior in novel regimes. By grounding scaling in learning theory, it also opens avenues for further theoretical work on generalization, data efficiency, and model design. The impact extends to any field relying on large-scale neural models, from NLP to computer vision.
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.