Data Similarity is Not Enough to Explain Language Model Performance

Gregory Yauney, Emily Reif, David Mimno
11/15/2023
cs.CLcs.LG

Official code: gyauney/data-similarity-is-not-enough (4 stars, MIT, last commit 2023-11-14).

Abstract

Large language models achieve high performance on many but not all downstream tasks. The interaction between pretraining data and task data is commonly assumed to determine this variance: a task with data that is more similar to a model's pretraining data is assumed to be easier for that model. We test whether distributional and example-specific similarity measures (embedding-, token- and model-based) correlate with language model performance through a large-scale comparison of the Pile and C4 pretraining datasets with downstream benchmarks. Similarity correlates with performance for multilingual datasets, but in other benchmarks, we surprisingly find that similarity metrics are not correlated with accuracy or even each other. This suggests that the relationship between pretraining data and downstream tasks is more complex than often assumed.

View on arXivView PDF

Code Implementations(1)

MIT

Code for Data Similarity is Not Enough to Explain Language Model Performance

41PythonOct 19, 20232 years agoMIT

Similar Papers

Cite this paper

@article{yauney2023data,
  title  = {Data Similarity is Not Enough to Explain Language Model Performance},
  author = {Gregory Yauney and Emily Reif and David Mimno},
  year   = {2023},
  eprint = {2311.09006},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2311.09006v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2311.09006.svg)](https://codeofpaper.com/paper/2311.09006)

Discussion