AI Models Collapse When Trained on Recursively Generated Data

Shumailov et al., 2024, AI models collapse when trained on recursively generated data (Nature, vol. 631)

Subscriber deck — free during early access

12 cards

Introduction

  1. What question does the model collapse paper ask?
    What happens to future models (“GPT-{n}”) once much of the web text they are trained on was produced by earlier models: models training on their predecessors’ output, generation after generation.
  2. What is model collapse?
    A degenerative process in which data generated by a model pollute the training set of the next generation, which then misperceives reality; models gradually forget the true data distribution, even though the distribution itself doesn’t change.
  3. What are early and late model collapse?
    Early: the model loses information about the tails of the distribution (rare events). Late: it converges to a distribution bearing little resemblance to the original, often with greatly reduced variance, approaching a point estimate.

What is model collapse?

  1. What three sources of error drive model collapse?
    Statistical approximation error, from finite samples (rare events may not be sampled at all), the primary cause; functional expressivity error, from limited model capacity (e.g. fitting two Gaussians with one); and functional approximation error, from the learning procedure itself, such as biases of SGD or the objective.
  2. Why can more model capacity make model collapse worse rather than better?
    Expressiveness can counteract statistical noise but can equally fit and compound it: an overfitted density model may extrapolate and assign high density to regions outside the training support, which are then sampled with arbitrary frequency in the next generation.

Theoretical intuition

  1. Why must recursive training on a discrete distribution collapse even with perfect fitting?
    With \(M\) samples, a state of probability \(q \le 1/M\) is expected to appear less than once, so it is likely lost. Viewed as a Markov chain, the only absorbing states are delta functions (all samples identical), so with probability 1 the process ends in a single state, losing all information.
    generation 010 of 10 valuesgeneration 57 of 10 valuesgeneration 303 of 10 values
    Each generation is fitted to 40 samples from the previous one: tails vanish first (simulation, not from the paper)
  2. What does the Gaussian model collapse theorem state?
    If each generation fits a Gaussian with unbiased mean and variance estimates from a fixed number of samples of the previous generation, then the approximation drifts arbitrarily far from the original (in Wasserstein-2 distance) while its variance collapses to zero, with probability 1.

Model collapse in language models

  1. How was model collapse tested on language models?
    OPT-125m was fine-tuned on wikitext2, then used to regenerate a dataset of the same size (continuing each 64-token block), on which the next generation was fine-tuned, and so on, either with no original data or keeping a random 10% of the original data each generation.
  2. What happened to generated text and perplexity over generations?
    Later generations produced more sequences the original model found likely, but also a growing tail of sequences it would never produce: accumulated errors. Text degenerated, e.g. by generation 9 a passage on church architecture had become a list of jackrabbits with differently coloured tails.
  3. Did keeping some real data or penalizing repetition prevent collapse?
    Keeping 10% of the original data reduced degradation to minor levels but didn’t eliminate it. Penalizing repetition made models produce lower-probability continuations and the next generations perform even worse (perplexity doubled).

Discussion

  1. Why does preserving low-probability events matter, according to Shumailov et al.?
    Rare events are often relevant to marginalized groups, so losing them harms fairness, and they are vital for understanding complex systems.
  2. What practical consequences do Shumailov et al. draw for training future models?
    A “first mover advantage”: access to genuine human-generated data (pre-LLM web crawls, real human interactions) becomes increasingly valuable, and the provenance of crawled content must be tracked, perhaps through community-wide coordination among those building and deploying LLMs.

Source

  • Ilia Shumailov, Zakhar Shumaylov, Yiren Zhao, Nicolas Papernot, Ross Anderson and Yarin Gal. 2024. AI models collapse when trained on recursively generated data. In Nature, vol. 631, pp. 755–759. License: CC BY 4.0. Flashcards adapted from this work. doi:10.1038/s41586-024-07566-y