Data Curation Alone Can Stabilize In-context Learning

Ting-Yun Chang, Robin Jia


Abstract
In-context learning (ICL) enables large language models (LLMs) to perform new tasks by prompting them with a sequence of training examples. However, it is known that ICL is very sensitive to the choice of training examples: randomly sampling examples from a training set leads to high variance in performance. In this paper, we show that carefully curating a subset of training data greatly stabilizes ICL performance without any other changes to the ICL algorithm (e.g., prompt retrieval or calibration). We introduce two methods to choose training subsets—both score training examples individually, then select the highest-scoring ones. CondAcc scores a training example by its average dev-set ICL accuracy when combined with random training examples, while Datamodels learns linear regressors that estimate how the presence of each training example influences LLM outputs. Across five tasks and two LLMs, sampling from stable subsets selected by CondAcc and Datamodels improves average accuracy over sampling from the entire training set by 7.7% and 6.3%, respectively. Surprisingly, the stable subset examples are not especially diverse in content or low in perplexity, in contrast with other work suggesting that diversity and perplexity are important when prompting LLMs.
Anthology ID:
2023.acl-long.452
Volume:
Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2023
Address:
Toronto, Canada
Editors:
Anna Rogers, Jordan Boyd-Graber, Naoaki Okazaki
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
8123–8144
Language:
URL:
https://aclanthology.org/2023.acl-long.452
DOI:
10.18653/v1/2023.acl-long.452
Bibkey:
Cite (ACL):
Ting-Yun Chang and Robin Jia. 2023. Data Curation Alone Can Stabilize In-context Learning. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 8123–8144, Toronto, Canada. Association for Computational Linguistics.
Cite (Informal):
Data Curation Alone Can Stabilize In-context Learning (Chang & Jia, ACL 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.acl-long.452.pdf
Video:
 https://aclanthology.org/2023.acl-long.452.mp4