Observational Scaling Laws and the Predictability of Language Model Performance

Yangjun Ruan, Chris J. Maddison, Tatsunori Hashimoto
5/17/2024
cs.LGcs.AIcs.CL

No official code found. Best implementation: ryoungj/ObsScaling (61 stars, Apache-2.0, last commit 2024-10-02).

Abstract

Understanding how language model performance varies with scale is critical to benchmark and algorithm development. Scaling laws are one approach to building this understanding, but the requirement of training models across many different scales has limited their use. We propose an alternative, observational approach that bypasses model training and instead builds scaling laws from ~100 publically available models. Building a single scaling law from multiple model families is challenging due to large variations in their training compute efficiencies and capabilities. However, we show that these variations are consistent with a simple, generalized scaling law where language model performance is a function of a low-dimensional capability space, and model families only vary in their efficiency in converting training compute to capabilities. Using this approach, we show the surprising predictability of complex scaling phenomena: we show that several emergent phenomena follow a smooth, sigmoidal behavior and are predictable from small models; we show that the agent performance of models such as GPT-4 can be precisely predicted from simpler non-agentic benchmarks; and we show how to predict the impact of post-training interventions like Chain-of-Thought and Self-Consistency as language model capabilities continue to improve.

View on arXivView PDF

Code Implementations(1)

ryoungj/ObsScalingCommunity100%
Apache-2.0

[NeurIPS'24 Spotlight] Observational Scaling Laws

613Python, Jupyter NotebookMay 16, 20242 years agoApache-2.0
language-modellarge-language-modelsscaling-laws

Similar Papers

Cite this paper

@article{ruan2024observational,
  title  = {Observational Scaling Laws and the Predictability of Language Model Performance},
  author = {Yangjun Ruan and Chris J. Maddison and Tatsunori Hashimoto},
  year   = {2024},
  eprint = {2405.10938},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2405.10938},
  url    = {http://arxiv.org/abs/2405.10938v3},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.10938.svg)](https://codeofpaper.com/paper/2405.10938)

Discussion