Emergent and Predictable Memorization in Large Language Models

By Stella Biderman, USVSN Sai Prashanth, Lintang Sutawika, Hailey Schoelkopf, Quentin Anthony, Shivanshu Purohit, Edward Raf. 2023

192 citations recorded by the source corpus.

Memorization, or the tendency of large language models (LLMs) to output entire sequences from their training data verbatim, is a key concern for safely deploying language models. In particular, it is vital to minimize a model's memorization of sensitive datapoints such as those containing personal identifiable information (PII). The prevalence of such undesirable memorization can pose issues for model trainers, and may even require discarding an otherwise functional model. We therefore seek to predict which sequences will be memorized before a large model's full train-time by extrapolating the memorization behavior of lower-compute trial runs. We measure memorization of the Pythia model suite and plot scaling laws for forecasting memorization, allowing us to provide equi-compute recommendations to maximize the reliability (recall) of such predictions. We additionally provide further novel discoveries on the distribution of memorization scores across models and data. We release all code and data necessary to reproduce the results in this paper at https://github.com/EleutherAI/pythia

Subfields: Artificial Intelligence, Large Language Models

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Bill Yuchen Lin
  2. Yejin Choi
  3. Stella Biderman
  4. Hailey Schoelkopf
  5. Lintang Sutawika
  6. USVSN Sai Prashanth
  7. Shivanshu Purohit
  8. Quentin Anthony

In Large Language Models

  1. Hailey Schoelkopf
  2. Lintang Sutawika
  3. USVSN Sai Prashanth
  4. Shivanshu Purohit
  5. Quentin Anthony
  6. Stella Biderman
  7. Yejin Choi
  8. Bill Yuchen Lin

Citations in the corpus