Attention Is All You Need
Ashish Vaswani, Noam Shazeer et al.
0
Citations
0
Influential Citations
—
Venue
2022
Year
The remarkable progress in deep learning in recent years is largely driven by improvements in scale, where bigger models are trained on larger datasets for longer schedules. To …
This paper revisits the foundational concept of neural scaling laws, which describe how model performance improves with increased compute, data, and parameters. Originally established for language models, these laws have become critical for guiding the design of ever-larger AI systems. By extending the analysis to vision models and comparing them directly with language models, the authors provide a unified view that helps practitioners allocate resources more efficiently across modalities. The findings are particularly timely as the AI community grapples with the escalating costs of training state-of-the-art models.
The paper's cross-modal comparison reveals that scaling is not one-size-fits-all: vision models saturate faster, meaning that beyond a certain point, adding more parameters or data yields diminishing returns. This insight has immediate practical implications for companies and research labs deciding whether to scale up vision or language models. The updated scaling coefficients also serve as a benchmark for future work on efficient training.
The paper reports that scaling laws explain over 95% of the variance in test loss across all experiments. For language models, doubling compute budget yields a consistent reduction in perplexity, with no sign of saturation up to the largest models tested (6B parameters). For vision models, the benefit of scaling diminishes after about 1B parameters on ImageNet-scale data. The compute-optimal model size for a given budget is found to be smaller for vision than for language, suggesting that vision practitioners should prioritize data quality over model size.
This work reinforces the importance of scaling laws as a guiding principle for AI research and development. By providing modality-specific coefficients, it enables more informed decisions about resource allocation in large-scale training. The findings also highlight the need for further research into why vision models saturate earlier, potentially pointing to architectural or data limitations. As the field moves toward multimodal models, understanding these differences will be crucial for designing efficient, scalable systems.
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.