Proving Test Set Contamination in Black Box Language Models

By Yonatan Oren, Nicole Meister, Niladri S. Chatterji, Faisal Ladhak, Tatsunori B. Hashimoto. 2023

251 citations recorded by the source corpus.

Large language models are trained on vast amounts of internet data, prompting concerns and speculation that they have memorized public benchmarks. Going from speculation to proof of contamination is challenging, as the pretraining data used by proprietary models are often not publicly accessible. We show that it is possible to provide provable guarantees of test set contamination in language models without access to pretraining data or model weights. Our approach leverages the fact that when there is no data contamination, all orderings of an exchangeable benchmark should be equally likely. In contrast, the tendency for language models to memorize example order means that a contaminated language model will find certain canonical orderings to be much more likely than others. Our test flags potential contamination whenever the likelihood of a canonically ordered benchmark dataset is significantly higher than the likelihood after shuffling the examples. We demonstrate that our procedure is sensitive enough to reliably prove test set contamination in challenging situations, including models as small as 1.4 billion parameters, on small test sets of only 1000 examples, and datasets that appear only a few times in the pretraining corpus. Using our test, we audit five popular publicly accessible language models for test set contamination and find little evidence for pervasive contamination.

Subfields: Artificial Intelligence, 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. Ji-rong Wen
  2. Wayne Xin Zhao
  3. Ion Stoica
  4. Joseph E. Gonzalez
  5. Anastasios Nikolas Angelopoulos
  6. Yutao Zhu
  7. Lianmin Zheng
  8. Ying Sheng

In Large Language Models

  1. Ion Stoica
  2. Anastasios Nikolas Angelopoulos
  3. Ji-rong Wen
  4. Lianmin Zheng
  5. Ying Sheng
  6. Yankai Lin
  7. Wayne Xin Zhao
  8. Joseph E. Gonzalez

In Safety, Robustness & Interpretability

  1. Ion Stoica
  2. Joseph E. Gonzalez

Citations in the corpus