Preprint
Computer Vision

Revisiting neural scaling laws in language and vision

January 1, 2022

0

Citations

0

Influential Citations

Venue

2022

Year

Abstract

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 …

Analysis

Why This Paper Matters

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.

Technical Contributions

  • Unified scaling law framework: The authors fit power-law models of the form L(N, D) = a/N^α + b/D^β + c to both language and vision tasks, where N is model size and D is dataset size.
  • Cross-modal comparison: They show that the exponent α (for model size) is larger for vision (≈0.10) than for language (≈0.07), indicating faster initial gains but earlier saturation.
  • Compute-optimal allocation: Using the scaling laws, they derive that for language models, the optimal ratio of model size to data size is roughly 1:1 in terms of compute budget, while for vision, the optimal ratio favors more data.
  • Robustness checks: The results hold across different architectures (Transformers for language, ResNets and ViTs for vision) and training schedules.

Results

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.

Significance

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.