ElemedexA pocket encyclopedia of AI architectures

Glossary (TM) › Machine learning fundamentals

TM006

Base stats (how to read an entry)

Also: Pokédex data sheet, height, weight

In Spanish: Estadísticas base (cómo leer una entrada)

A guide to reading an Elemedex entry: what the six base stats, height, weight, types, moves and the other fields measure.

Every species has a Pokédex-style entry. The base stats run from 0 to 100 and are ordinal and approximate: they are for comparing species, not measuring them. The first three (number of parameters, amount of data in examples or tokens, and training compute) use a log scale.

Stat What it measures Anchors
Parameters adjustable numbers 15 ≈ 10³ · 45 ≈ 10⁸ · 75 ≈ 10¹¹ · 90 ≈ 10¹²
Data examples or tokens 10 = a spreadsheet · 45 ≈ 10⁸ · 90 ≈ 10¹³
Compute training FLOPs 5 = pencil · 45 ≈ 10¹⁹ · 90 ≈ 10²⁵
Context how much input it considers at once 0 = a single vector · 25 ≈ 512 tokens · 80 ≈ 128k
Generality range of tasks 10 = one · 40 = many after fine-tuning · 70 = many by prompting
Transparency how readable it is 95 = small tree · 60 = linear model · 10 = black box

The data sheet adds hard numbers. Height is the number of layers: BERT-base stands 12 tall (Devlin et al., 2019, p. 4173). Weight is the memory the parameters take up: in fp32 precision each takes 4 bytes, so BERT-base's 110 million weigh ≈ 440 MB and GPT-3's 175 billion ≈ 700 GB (Brown et al., 2020, sec. 2.1); in fp16, half that, and less still with quantization.

Types (one or two per species, such as Neural or Recurrent) indicate the family of ideas it belongs to. The kind distinguishes architecture, model, algorithm, system, or unit (see model). Moves are what it can do; abilities, its distinctive architectural features; weaknesses, its known limits. Habitat says what data it grew up on and where it can be found today. The evolution chain links it to its ancestors and says what changed. Anatomy is a diagram of its parts, read from bottom to top, like the original transformer figure. The professor's note opens a philosophical discussion to take to class.

The stats are a useful caricature: a parameter count does not say how many are used at once (mixture of experts), and transparency is not a simple property (interpretability).

Sources

  • Brown, T. B., Mann, B., Ryder, N., Subbiah, M., Kaplan, J., Dhariwal, P., Neelakantan, A., Shyam, P., Sastry, G., Askell, A., Agarwal, S., Herbert-Voss, A., Krueger, G., Henighan, T., Child, R., Ramesh, A., Ziegler, D. M., Wu, J., Winter, C., . . . Amodei, D. (2020). Language models are few-shot learners. In Advances in Neural Information Processing Systems (Vol. 33, pp. 1877–1901). Curran Associates. https://proceedings.neurips.cc/paper_files/paper/2020/hash/1457c0d6bfcb4967418bfb8ac142f64a-Abstract.html §
  • Devlin, J., Chang, M.-W., Lee, K., & Toutanova, K. (2019). BERT: Pre-training of deep bidirectional transformers for language understanding. In J. Burstein, C. Doran, & T. Solorio (Eds.), Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers) (pp. 4171–4186). Association for Computational Linguistics. https://doi.org/10.18653/v1/N19-1423 §