An Information Theory of Compute-Optimal Size Scaling, Emergence, and Plateaus in Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Nayak, Anuj K., Varshney, Lav R.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909350977601536
author Nayak, Anuj K.
Varshney, Lav R.
author_facet Nayak, Anuj K.
Varshney, Lav R.
contents Recent empirical studies show three phenomena with increasing size of language models: compute-optimal size scaling, emergent capabilities, and performance plateauing. We present a simple unified mathematical framework to explain all of these language model scaling phenomena, building on recent skill-text bipartite graph frameworks for semantic learning. Modeling the learning of concepts from texts as an iterative process yields an analogy to iterative decoding of low-density parity check (LDPC) codes in information theory. Thence, drawing on finite-size scaling characterizations of LDPC decoding, we derive the compute-optimal size scaling (Chinchilla rule) for language models. Further, using tools from random network theory, we provide a simple explanation for both emergence of complex skills and plateauing of performance as the size of language models scale. We see multiple plateaus.
format Preprint
id arxiv_https___arxiv_org_abs_2410_01243
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An Information Theory of Compute-Optimal Size Scaling, Emergence, and Plateaus in Language Models
Nayak, Anuj K.
Varshney, Lav R.
Information Theory
Recent empirical studies show three phenomena with increasing size of language models: compute-optimal size scaling, emergent capabilities, and performance plateauing. We present a simple unified mathematical framework to explain all of these language model scaling phenomena, building on recent skill-text bipartite graph frameworks for semantic learning. Modeling the learning of concepts from texts as an iterative process yields an analogy to iterative decoding of low-density parity check (LDPC) codes in information theory. Thence, drawing on finite-size scaling characterizations of LDPC decoding, we derive the compute-optimal size scaling (Chinchilla rule) for language models. Further, using tools from random network theory, we provide a simple explanation for both emergence of complex skills and plateauing of performance as the size of language models scale. We see multiple plateaus.
title An Information Theory of Compute-Optimal Size Scaling, Emergence, and Plateaus in Language Models
topic Information Theory
url https://arxiv.org/abs/2410.01243