Data Curation Alone Can Stabilize In-context Learning
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 58 papers in the map (data through 7 Oct 2026). Builds on 38 of its 54 references that are in the map. First posted on arXiv 20 Dec 2022.
Builds on
38 of 54 references.
References · 38
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,049
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
- 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
- 2018UMAP: Uniform Manifold Approximation and Projection for Dimension Reductioncited by 2,841
- 2022Emergent Abilities of Large Language Modelscited by 2,483
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
26 more
- 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,462
- 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,413
- 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,371
- 2020Making Pre-trained Language Models Better Few-shot Learnerscited by 1,255
- 2022LaMDA: Language Models for Dialog Applicationscited by 1,218
- 2021Scaling Language Models: Methods, Analysis & Insights from Training Gophercited by 1,199
- 2021What Makes Good In-Context Examples for GPT-3?cited by 1,184
- 2021Fantastically Ordered Prompts and Where to Find Them: Overcoming Few-Shot Prompt Order Sensitivitycited by 1,122
- 2020Exploiting Cloze Questions for Few Shot Text Classification and Natural Language Inferencecited by 934
- 2021Learning To Retrieve Prompts for In-Context Learningcited by 641
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
- 2022Using DeepSpeed and Megatron to Train Megatron-Turing NLG 530B, A Large-Scale Generative Language Modelcited by 557
- 2021True Few-Shot Learning with Language Modelscited by 343
- 2022PromptSource: An Integrated Development Environment and Repository for Natural Language Promptscited by 288
- 2022Selective Annotation Makes Language Models Better Few-Shot Learnerscited by 242
- 2022Demystifying Prompts in Language Models via Perplexity Estimationcited by 215
- 2022Active Example Selection for In-Context Learningcited by 195
- 2022Datamodels: Predicting Predictions from Training Datacited by 186
- 2019Structural Scaffolds for Citation Intent Classification in Scientific Publicationscited by 129
- 2020Cold-start Active Learning through Self-supervised Language Modelingcited by 128
- 2022Ground-Truth Labels Matter: A Deeper Look into Input-Label Demonstrationscited by 116
- 2019Data Valuation using Reinforcement Learningcited by 115
- 2019Diverse mini-batch Active Learningcited by 88
- 2022On the Relation between Sensitivity and Accuracy in In-context Learningcited by 71
- 2022Complementary Explanations for Effective In-Context Learningcited by 71
Built on by
Citing papers per year: 2023: 9; 2024: 35; 2025: 12; 2026: 2.
58 papers.
Sort:
By area:
- 2025Advances and Challenges in Foundation Agents: From Brain-Inspired Intelligence to Evolutionary, Collaborative, and Safe Systemscited by 125
- 2023Finding Support Examples for In-Context Learningcited by 104
- 2024ContextCite: Attributing Model Generation to Contextcited by 93
- 2024Mind Your Format: Towards Consistent Evaluation of In-Context Learning Improvementscited by 82
- 2024Breaking Agents: Compromising Autonomous LLM Agents Through Malfunction Amplificationcited by 80
- 2023Transformers Learn to Achieve Second-Order Convergence Rates for In-Context Linear Regressioncited by 47
- 2024IsoBench: Benchmarking Multimodal Foundation Models on Isomorphic Representationscited by 37
- 2024Prompt Optimization with EASE? Efficient Ordering-aware Automated Selection of Exemplarscited by 37
- 2023How are Prompts Different in Terms of Sensitivity?cited by 34
- 2024Get more for less: Principled Data Selection for Warming Up Fine-Tuning in LLMscited by 32
- 2024Smoothie: Label Free Language Model Routingcited by 31
- 2024A Survey of Table Reasoning with Large Language Modelscited by 29
38 more
- 2023IDEAL: Influence-Driven Selective Annotations Empower In-Context Learners in Large Language Modelscited by 26
- 2023RetICL: Sequential Retrieval of In-Context Examples with Reinforcement Learningcited by 24
- 2024Beyond Performance: Quantifying and Mitigating Label Bias in LLMscited by 24
- 2024LLM Task Interference: An Initial Study on the Impact of Task-Switch in Conversational Historycited by 19
- 2024Stress-Testing Long-Context Language Models with Lifelong ICL and Task Haystackcited by 19
- 2024Effects of diversity incentives on sample diversity and downstream model performance in LLM-based text augmentationcited by 17
- 2024SynthDST: Synthetic Data is All You Need for Few-Shot Dialog State Trackingcited by 17
- 2024Comparing Specialised Small and General Large Language Models on Text Classification: 100 Labelled Samples to Achieve Break-Even Performancecited by 17
- 2024Reconstruct Your Previous Conversations! Comprehensively Investigating Privacy Leakage Risks in Conversations with GPT Modelscited by 16
- 2024Revisiting In-context Learning Inference Circuit in Large Language Modelscited by 16
- 2023A Survey on Stability of Learning with Limited Labelled Data and its Sensitivity to the Effects of Randomnesscited by 14
- 2023Improving Probability-based Prompt Selection Through Unified Evaluation and Analysiscited by 13
- 2024In-Context Example Ordering Guided by Label Distributionscited by 12
- 2024Addressing Order Sensitivity of In-Context Demonstration Examples in Causal Language Modelscited by 12
- 2024In-Context Learning Demonstration Selection via Influence Analysiscited by 11
- 2024Strategic Demonstration Selection for Improved Fairness in LLM In-Context Learningcited by 10
- 2024FastGAS: Fast Graph-based Annotation Selection for In-Context Learningcited by 8
- 2024LLMs vs Established Text Augmentation Techniques for Classification: When do the Benefits Outweight the Costs?cited by 8
- 2024Use Random Selection for Now: Investigation of Few-Shot Selection Strategies in LLM-based Text Augmentation for Classificationcited by 7
- 2024When Parts Are Greater Than Sums: Individual LLM Components Can Outperform Full Modelscited by 6
- 2025Enhancing Multimodal In-Context Learning for Image Classification through Coreset Optimizationcited by 6
- 2025The Role of Diversity in In-Context Learning for Large Language Modelscited by 6
- 2024DETAIL: Task DEmonsTration Attribution for Interpretable In-context Learningcited by 5
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoningcited by 5
- 2025CrowdLLM: Building LLM-Based Digital Populations Augmented with Generative Modelscited by 5
- 2023The Role of Data Curation in Image Captioningcited by 4
- 2024Automatic Combination of Sample Selection Strategies for Few-Shot Learningcited by 4
- 2024LLMs for Targeted Sentiment in News Headlines: Exploring the Descriptive-Prescriptive Dilemmacited by 3
- 2024DemoShapley: Valuation of Demonstrations for In-Context Learningcited by 3
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretrainingcited by 3
- 2024Pelican Soup Framework: A Theoretical Framework for Language Model Capabilitiescited by 2
- 2024A critical review of methods and challenges in large language modelscited by 2
- 2025Monte Carlo Sampling for Analyzing In-Context Examplescited by 2
- 2025Stability of In-Context Learning: A Spectral Coverage Perspectivecited by 2
- 2024On Sensitivity of Learning with Limited Labelled Data to the Effects of Randomness: Impact of Interactions and Systematic Choicescited by 1
- 2024Augmenting In-Context-Learning in LLMs via Automatic Data Labeling and Refinementcited by 1
- 2024Effective Self-Mining of In-Context Examples for Unsupervised Machine Translation with LLMscited by 1
- 2025Retrieval augmented generation based dynamic prompting for few-shot biomedical named entity recognition using large language modelscited by 1
50 most cited of 58 All 58 (JSON)
- 2026Where to Place the Query? Unveiling and Mitigating Positional Bias in In-Context Learning for Diffusion LLMs via Decoding Dynamics26 Apr 2026
- 2026Judging What We Cannot Solve: A Consequence-Based Approach for Oracle-Free Evaluation of Research-Level Math6 Feb 2026
- 2025CrowdLLM: Building LLM-Based Digital Populations Augmented with Generative Models2 Dec 2025
- 2025Increasing LLM response trustworthiness using voting ensembles5 Oct 2025
- 2025Stability of In-Context Learning: A Spectral Coverage Perspective25 Sep 2025
- 2025Retrieval augmented generation based dynamic prompting for few-shot biomedical named entity recognition using large language models25 Jul 2025
- 2025Application and Evaluation of Large Language Models for Forecasting the Impact of Traffic Incidents7 Jul 2025
- 2025Factors affecting the in-context learning abilities of LLMs for dialogue state tracking10 Jun 2025
- 2025Improving Dialogue State Tracking through Combinatorial Search for In-Context Examples31 May 2025
- 2025The Role of Diversity in In-Context Learning for Large Language Models26 May 2025
- 2025Enhancing Multimodal In-Context Learning for Image Classification through Coreset Optimization19 Apr 2025
- 2025Advances and Challenges in Foundation Agents: From Brain-Inspired Intelligence to Evolutionary, Collaborative, and Safe Systems31 Mar 2025
38 more
- 2025Monte Carlo Sampling for Analyzing In-Context Examples27 Mar 2025
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretraining4 Feb 2025
- 2024Smoothie: Label Free Language Model Routing6 Dec 2024
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoning6 Nov 2024
- 2024Effective Self-Mining of In-Context Examples for Unsupervised Machine Translation with LLMs14 Oct 2024
- 2024Use Random Selection for Now: Investigation of Few-Shot Selection Strategies in LLM-based Text Augmentation for Classification14 Oct 2024
- 2024Augmenting In-Context-Learning in LLMs via Automatic Data Labeling and Refinement14 Oct 2024
- 2024DemoShapley: Valuation of Demonstrations for In-Context Learning10 Oct 2024
- 2024Revisiting In-context Learning Inference Circuit in Large Language Models6 Oct 2024
- 2024ContextCite: Attributing Model Generation to Context1 Sep 2024
- 2024LLMs vs Established Text Augmentation Techniques for Classification: When do the Benefits Outweight the Costs?29 Aug 2024
- 2024Strategic Demonstration Selection for Improved Fairness in LLM In-Context Learning19 Aug 2024
- 2024Breaking Agents: Compromising Autonomous LLM Agents Through Malfunction Amplification30 Jul 2024
- 2024Stress-Testing Long-Context Language Models with Lifelong ICL and Task Haystack23 Jul 2024
- 2024When Parts Are Greater Than Sums: Individual LLM Components Can Outperform Full Models19 Jun 2024
- 2024Logit Separability-Driven Samples and Multiple Class-Related Words Selection for Advancing In-Context Learning16 Jun 2024
- 2024FastGAS: Fast Graph-based Annotation Selection for In-Context Learning6 Jun 2024
- 2024Prompt Optimization with EASE? Efficient Ordering-aware Automated Selection of Exemplars25 May 2024
- 2024DETAIL: Task DEmonsTration Attribution for Interpretable In-context Learning22 May 2024
- 2024Get more for less: Principled Data Selection for Warming Up Fine-Tuning in LLMs5 May 2024
- 2024Beyond Performance: Quantifying and Mitigating Label Bias in LLMs4 May 2024
- 2024A critical review of methods and challenges in large language models18 Apr 2024
- 2024IsoBench: Benchmarking Multimodal Foundation Models on Isomorphic Representations1 Apr 2024
- 2024LLMs for Targeted Sentiment in News Headlines: Exploring the Descriptive-Prescriptive Dilemma1 Mar 2024
- 2024LLM Task Interference: An Initial Study on the Impact of Task-Switch in Conversational History28 Feb 2024
- 2024Addressing Order Sensitivity of In-Context Demonstration Examples in Causal Language Models23 Feb 2024
- 2024Comparing Specialised Small and General Large Language Models on Text Classification: 100 Labelled Samples to Achieve Break-Even Performance20 Feb 2024
- 2024On Sensitivity of Learning with Limited Labelled Data to the Effects of Randomness: Impact of Interactions and Systematic Choices20 Feb 2024
- 2024In-Context Learning Demonstration Selection via Influence Analysis19 Feb 2024
- 2024In-Context Example Ordering Guided by Label Distributions18 Feb 2024
- 2024Pelican Soup Framework: A Theoretical Framework for Language Model Capabilities16 Feb 2024
- 2024A Survey of Table Reasoning with Large Language Models13 Feb 2024
- 2024Automatic Combination of Sample Selection Strategies for Few-Shot Learning5 Feb 2024
- 2024Reconstruct Your Previous Conversations! Comprehensively Investigating Privacy Leakage Risks in Conversations with GPT Models5 Feb 2024
- 2024SynthDST: Synthetic Data is All You Need for Few-Shot Dialog State Tracking3 Feb 2024
- 2024Mind Your Format: Towards Consistent Evaluation of In-Context Learning Improvements12 Jan 2024
- 2024Effects of diversity incentives on sample diversity and downstream model performance in LLM-based text augmentation12 Jan 2024
- 2023A Survey on Stability of Learning with Limited Labelled Data and its Sensitivity to the Effects of Randomness2 Dec 2023
50 newest of 58 All 58 (JSON)
- 2024A Survey of Table Reasoning with Large Language Modelscited by 29
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoningcited by 5
- 2024Augmenting In-Context-Learning in LLMs via Automatic Data Labeling and Refinementcited by 1
- 2026Judging What We Cannot Solve: A Consequence-Based Approach for Oracle-Free Evaluation of Research-Level Mathcited by 1
- 2026Judging What We Cannot Solve: A Consequence-Based Approach for Oracle-Free Evaluation of Research-Level Math6 Feb 2026
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoning6 Nov 2024
- 2024Augmenting In-Context-Learning in LLMs via Automatic Data Labeling and Refinement14 Oct 2024
- 2024A Survey of Table Reasoning with Large Language Models13 Feb 2024
- 2023Finding Support Examples for In-Context Learningcited by 104
- 2024ContextCite: Attributing Model Generation to Contextcited by 93
- 2024Mind Your Format: Towards Consistent Evaluation of In-Context Learning Improvementscited by 82
- 2024IsoBench: Benchmarking Multimodal Foundation Models on Isomorphic Representationscited by 37
- 2024Prompt Optimization with EASE? Efficient Ordering-aware Automated Selection of Exemplarscited by 37
- 2023How are Prompts Different in Terms of Sensitivity?cited by 34
- 2024Get more for less: Principled Data Selection for Warming Up Fine-Tuning in LLMscited by 32
- 2024Smoothie: Label Free Language Model Routingcited by 31
- 2023IDEAL: Influence-Driven Selective Annotations Empower In-Context Learners in Large Language Modelscited by 26
- 2023RetICL: Sequential Retrieval of In-Context Examples with Reinforcement Learningcited by 24
- 2024Beyond Performance: Quantifying and Mitigating Label Bias in LLMscited by 24
- 2024LLM Task Interference: An Initial Study on the Impact of Task-Switch in Conversational Historycited by 19
12 most cited of 48 All 48 (JSON)
- 2026Where to Place the Query? Unveiling and Mitigating Positional Bias in In-Context Learning for Diffusion LLMs via Decoding Dynamics26 Apr 2026
- 2025CrowdLLM: Building LLM-Based Digital Populations Augmented with Generative Models2 Dec 2025
- 2025Increasing LLM response trustworthiness using voting ensembles5 Oct 2025
- 2025Stability of In-Context Learning: A Spectral Coverage Perspective25 Sep 2025
- 2025Application and Evaluation of Large Language Models for Forecasting the Impact of Traffic Incidents7 Jul 2025
- 2025Factors affecting the in-context learning abilities of LLMs for dialogue state tracking10 Jun 2025
- 2025Improving Dialogue State Tracking through Combinatorial Search for In-Context Examples31 May 2025
- 2025The Role of Diversity in In-Context Learning for Large Language Models26 May 2025
- 2025Monte Carlo Sampling for Analyzing In-Context Examples27 Mar 2025
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretraining4 Feb 2025
- 2024Smoothie: Label Free Language Model Routing6 Dec 2024
- 2024Effective Self-Mining of In-Context Examples for Unsupervised Machine Translation with LLMs14 Oct 2024
12 newest of 48 All 48 (JSON)