Counterfactual Memorization in Neural Language Models

By Chiyuan Zhang, Daphne Ippolito, Katherine Lee, Matthew Jagielski, Florian Tramèr, Nicholas Carlini. 2021

203 citations recorded by the source corpus.

Modern neural language models that are widely used in various NLP tasks risk memorizing sensitive information from their training data. Understanding this memorization is important in real world applications and also from a learning-theoretical perspective. An open question in previous studies of language model memorization is how to filter out"common"memorization. In fact, most memorization criteria strongly correlate with the number of occurrences in the training set, capturing memorized familiar phrases, public knowledge, templated texts, or other repeated data. We formulate a notion of counterfactual memorization which characterizes how a model's predictions change if a particular document is omitted during training. We identify and study counterfactually-memorized training examples in standard text datasets. We estimate the influence of each memorized training example on the validation set and on generated texts, showing how this can provide direct evidence of the source of memorization at test time.

Subfields: Artificial Intelligence, Cultural Studies, Large Language Models, Safety, Robustness & Interpretability

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. Luke S. Zettlemoyer
  2. Mohit Bansal
  3. Hannaneh Hajishirzi
  4. Percy Liang
  5. Yejin Choi
  6. Noah A. Smith
  7. Graham Neubig
  8. Yonatan Belinkov

In Cultural Studies

  1. Noah A. Smith
  2. Percy Liang
  3. Dan Jurafsky
  4. Yonatan Belinkov
  5. Yoav Goldberg
  6. Suchin Gururangan

In Large Language Models

  1. Luke S. Zettlemoyer
  2. Hinrich Schütze
  3. Hannaneh Hajishirzi
  4. Noah A. Smith
  5. Danqi Chen
  6. Percy Liang
  7. Eric Wallace
  8. Jie Tang

In Safety, Robustness & Interpretability

  1. Eric Wallace
  2. Nicholas Carlini
  3. Florian Tramèr
  4. Percy Liang
  5. Yonatan Belinkov
  6. Sameer Singh
  7. Noah A. Smith
  8. Matt Gardner

Citations in the corpus