CLLGOct 26, 2023

Proving Test Set Contamination in Black Box Language Models

arXiv:2310.17623v2237 citationsh-index: 22
Originality Incremental advance
AI Analysis

This addresses concerns about benchmark reliability for researchers and practitioners, though it is incremental as it builds on existing ideas about memorization and exchangeability.

The paper tackles the problem of proving test set contamination in black-box language models by developing a method that detects memorization of canonical orderings, and it demonstrates sensitivity in detecting contamination even in small models and datasets, while auditing five models finds little pervasive evidence.

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.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes