Pre-Training to Learn in Context
Not on the map. Why?
Abstract (arXiv)
In-context learning, where pre-trained language models learn to perform tasks from task examples and instructions in their contexts, has attracted much attention in the NLP community. However, the ability of in-context learning is not fully exploited because language models are not explicitly trained to learn in context. To this end, we propose PICL (Pre-training for In-Context Learning), a framework to enhance the language models' in-context learning ability by pre-training the model on a large collection of "intrinsic tasks" in the general plain-text corpus using the simple language modeling objective. PICL encourages the model to infer and perform tasks by conditioning on the contexts while maintaining task generalization of pre-trained models. We evaluate the in-context learning performance of the model trained with PICL on seven widely-used text classification datasets and the Super-NaturalInstrctions benchmark, which contains 100+ NLP tasks formulated to text generation. Our experiments show that PICL is more effective and task-generalizable than a range of baselines, outperforming larger language models with nearly 4x parameters. The code is publicly available at https://github.com/thu-coai/PICL.
Key facts. Cited by 42 papers in the map (data through 14 Oct 2026). Builds on 66 of its 94 references that are in the map. First posted on arXiv 16 May 2023.
Builds on
66 of 94 references.
References · 66
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,780
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
38 more
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2020Supervised Contrastive Learningcited by 2,829
- 2017Billion-scale similarity search with GPUscited by 2,497
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2021Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processingcited by 2,234
- 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,470
- 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
- 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,392
- 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,317
- 2021What Makes Good In-Context Examples for GPT-3?cited by 1,195
- 2021Fantastically Ordered Prompts and Where to Find Them: Overcoming Few-Shot Prompt Order Sensitivitycited by 1,135
- 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,043
- 2022In-context Learning and Induction Headscited by 964
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 938
- 2017Crowdsourcing Multiple Choice Science Questionscited by 765
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2021Cross-Task Generalization via Natural Language Crowdsourcing Instructionscited by 699
- 2021Learning To Retrieve Prompts for In-Context Learningcited by 648
- 2019SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarizationcited by 535
- 2020Pre-trained Models for Natural Language Processing: A Surveycited by 505
- 2021MetaICL: Learning to Learn In Contextcited by 454
- 2019TabFact: A Large-scale Dataset for Table-based Fact Verificationcited by 428
- 2019PAWS: Paraphrase Adversaries from Word Scramblingcited by 402
- 2017"Liar, Liar Pants on Fire": A New Benchmark Dataset for Fake News Detectioncited by 363
- 2018SWAG: A Large-Scale Adversarial Dataset for Grounded Commonsense Inferencecited by 333
- 2021Pre-Trained Models: Past, Present and Futurecited by 319
- 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 309
- 2019Abductive Commonsense Reasoningcited by 301
- 2022PromptSource: An Integrated Development Environment and Repository for Natural Language Promptscited by 289
- 2019Cosmos QA: Machine Reading Comprehension with Contextual Commonsense Reasoningcited by 285
- 2021Surface Form Competition: Why the Highest Probability Answer Isn't Always Rightcited by 263
- 2019QASC: A Dataset for Question Answering via Sentence Compositioncited by 252
- 2021PPT: Pre-trained Prompt Tuning for Few-shot Learningcited by 232
- 2017Constructing Datasets for Multi-hop Reading Comprehension Across Documentscited by 190
- 2021Noisy Channel Language Model Prompting for Few-Shot Text Classificationcited by 165
- 2021Meta-learning via Language Model In-context Tuningcited by 164
- 2019DREAM: A Challenge Dataset and Models for Dialogue-Based Reading Comprehensioncited by 156
- 2022Impact of Pretraining Term Frequencies on Few-Shot Reasoningcited by 142
50 most cited of 66 All 66 (JSON)
Built on by
Citing papers per year: 2023: 10; 2024: 17; 2025: 9; 2026: 6.
42 papers.
Sort:
By area:
- 2023In-context Pretraining: Language Modeling Beyond Document Boundariescited by 79
- 2024Inference Scaling for Long-Context Retrieval Augmented Generationcited by 65
- 2024Instruction Pre-Training: Language Models are Supervised Multitask Learnerscited by 63
- 2023Adapting Large Language Models to Domains via Reading Comprehensioncited by 50
- 2023Understanding In-Context Learning via Supportive Pretraining Datacited by 41
- 2024Data Selection via Optimal Control for Language Modelscited by 38
- 2024Mixture of In-Context Prompters for Tabular PFNscited by 33
- 2024MiniPLM: Knowledge Distillation for Pre-Training Language Modelscited by 29
- 2025Demystifying Synthetic Data in LLM Pre-training: A Systematic Study of Scaling Laws, Benefits, and Pitfallscited by 26
- 2024Analysing The Impact of Sequence Composition on Language Model Pre-Trainingcited by 25
- 2025RICL: Adding In-Context Adaptability to Pre-Trained Vision-Language-Action Modelscited by 22
- 2024ProtLLM: An Interleaved Protein-Language LLM with Protein-as-Word Pre-Trainingcited by 21
30 more
- 2023Supervised Knowledge Makes Large Language Models Better In-context Learnerscited by 20
- 2023Structured Packing in LLM Training Improves Long Context Utilizationcited by 17
- 2024Revisiting In-context Learning Inference Circuit in Large Language Modelscited by 17
- 2024On the Noise Robustness of In-Context Learning for Text Generationcited by 16
- 2024XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learningcited by 14
- 2023Large Language Models are Miscalibrated In-Context Learnerscited by 12
- 2024FabGPT: An Efficient Large Multimodal Model for Complex Wafer Defect Knowledge Queriescited by 12
- 2023Baby's CoThought: Leveraging Large Language Models for Enhanced Reasoning in Compact Modelscited by 9
- 2024Token-based Decision Criteria Are Suboptimal in In-context Learningcited by 9
- 2023AutoParLLM: GNN-guided Context Generation for Zero-Shot Code Parallelization using LLMscited by 5
- 2023Exploring In-Context Learning of Textless Speech Language Model for Speech Classification Taskscited by 5
- 2024Naive Bayes-based Context Extension for Large Language Modelscited by 5
- 2025DICE: Dynamic In-Context Example Selection in LLM Agents via Efficient Knowledge Transfercited by 5
- 2025Unlocking In-Context Learning for Natural Datasets Beyond Language Modellingcited by 4
- 2024Context-Former: Stitching via Latent Conditioned Sequence Modelingcited by 3
- 2024AICL: Action In-Context Learning for Video Diffusion Modelcited by 2
- 2024RARe: Retrieval Augmented Retrieval with In-Context Examplescited by 2
- 2025StaICC: Standardized Evaluation for Classification Task in In-context Learningcited by 2
- 2025MateICL: Mitigating Attention Dispersion in Large-Scale In-Context Learningcited by 2
- 2025Induction Signatures Are Not Enough: A Matched-Compute Study of Load-Bearing Structure in In-Context Learningcited by 2
- 2023Simple and Effective Input Reformulations for Translationcited by 1
- 2024CPRM: A LLM-based Continual Pre-training Framework for Relevance Modeling in Commercial Searchcited by 1
- 2026Learning Nonlinear Systems In-Context: From Synthetic Data to Real-World Motor Controlcited by 1
- 2026In-context Pre-trained Time-Series Foundation Models adapt to Unseen Taskscited by 1
- 2025Mechanistic Fine-tuning for In-context Learningcited by 0
- 2025Basic Reading Distillationcited by 0
- 2026In-context Region-based Drag: Drag Any Region to Any Shapecited by 0
- 2026When Context Misleads: In-context Learning with Jurisdiction in Large Language Modelscited by 0
- 2026Exploring In-Context Learning for Handwritten Text Recognitioncited by 0
- 2026Verb-ICL: Rethinking In-Context Learning for Structured Predictioncited by 0
- 2026Verb-ICL: Rethinking In-Context Learning for Structured Prediction3 Oct 2026
- 2026Exploring In-Context Learning for Handwritten Text Recognition29 Sep 2026
- 2026When Context Misleads: In-context Learning with Jurisdiction in Large Language Models23 Sep 2026
- 2026In-context Region-based Drag: Drag Any Region to Any Shape24 Jun 2026
- 2026In-context Pre-trained Time-Series Foundation Models adapt to Unseen Tasks23 Feb 2026
- 2026Learning Nonlinear Systems In-Context: From Synthetic Data to Real-World Motor Control6 Feb 2026
- 2025Demystifying Synthetic Data in LLM Pre-training: A Systematic Study of Scaling Laws, Benefits, and Pitfalls2 Oct 2025
- 2025Induction Signatures Are Not Enough: A Matched-Compute Study of Load-Bearing Structure in In-Context Learning26 Sep 2025
- 2025RICL: Adding In-Context Adaptability to Pre-Trained Vision-Language-Action Models4 Aug 2025
- 2025DICE: Dynamic In-Context Example Selection in LLM Agents via Efficient Knowledge Transfer31 Jul 2025
- 2025Basic Reading Distillation26 Jul 2025
- 2025Mechanistic Fine-tuning for In-context Learning20 May 2025
30 more
- 2025MateICL: Mitigating Attention Dispersion in Large-Scale In-Context Learning2 May 2025
- 2025StaICC: Standardized Evaluation for Classification Task in In-context Learning27 Jan 2025
- 2025Unlocking In-Context Learning for Natural Datasets Beyond Language Modelling9 Jan 2025
- 2024CPRM: A LLM-based Continual Pre-training Framework for Relevance Modeling in Commercial Search2 Dec 2024
- 2024RARe: Retrieval Augmented Retrieval with In-Context Examples26 Oct 2024
- 2024MiniPLM: Knowledge Distillation for Pre-Training Language Models22 Oct 2024
- 2024Data Selection via Optimal Control for Language Models9 Oct 2024
- 2024Revisiting In-context Learning Inference Circuit in Large Language Models6 Oct 2024
- 2024Inference Scaling for Long-Context Retrieval Augmented Generation6 Oct 2024
- 2024FabGPT: An Efficient Large Multimodal Model for Complex Wafer Defect Knowledge Queries15 Jul 2024
- 2024Token-based Decision Criteria Are Suboptimal in In-context Learning24 Jun 2024
- 2024Instruction Pre-Training: Language Models are Supervised Multitask Learners20 Jun 2024
- 2024XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning13 Jun 2024
- 2024On the Noise Robustness of In-Context Learning for Text Generation27 May 2024
- 2024Mixture of In-Context Prompters for Tabular PFNs25 May 2024
- 2024Naive Bayes-based Context Extension for Large Language Models26 Mar 2024
- 2024AICL: Action In-Context Learning for Video Diffusion Model18 Mar 2024
- 2024ProtLLM: An Interleaved Protein-Language LLM with Protein-as-Word Pre-Training28 Feb 2024
- 2024Analysing The Impact of Sequence Composition on Language Model Pre-Training21 Feb 2024
- 2024Context-Former: Stitching via Latent Conditioned Sequence Modeling29 Jan 2024
- 2023Structured Packing in LLM Training Improves Long Context Utilization28 Dec 2023
- 2023Supervised Knowledge Makes Large Language Models Better In-context Learners26 Dec 2023
- 2023Large Language Models are Miscalibrated In-Context Learners21 Dec 2023
- 2023Simple and Effective Input Reformulations for Translation12 Nov 2023
- 2023Exploring In-Context Learning of Textless Speech Language Model for Speech Classification Tasks19 Oct 2023
- 2023In-context Pretraining: Language Modeling Beyond Document Boundaries16 Oct 2023
- 2023AutoParLLM: GNN-guided Context Generation for Zero-Shot Code Parallelization using LLMs6 Oct 2023
- 2023Adapting Large Language Models to Domains via Reading Comprehension18 Sep 2023
- 2023Baby's CoThought: Leveraging Large Language Models for Enhanced Reasoning in Compact Models3 Aug 2023
- 2023Understanding In-Context Learning via Supportive Pretraining Data26 Jun 2023
- 2026In-context Region-based Drag: Drag Any Region to Any Shape24 Jun 2026
- 2024AICL: Action In-Context Learning for Video Diffusion Model18 Mar 2024
- 2024MiniPLM: Knowledge Distillation for Pre-Training Language Modelscited by 29
- 2025Basic Reading Distillationcited by 0
- 2025Basic Reading Distillation26 Jul 2025
- 2024MiniPLM: Knowledge Distillation for Pre-Training Language Models22 Oct 2024
- 2023In-context Pretraining: Language Modeling Beyond Document Boundariescited by 79
- 2024Instruction Pre-Training: Language Models are Supervised Multitask Learnerscited by 63
- 2023Adapting Large Language Models to Domains via Reading Comprehensioncited by 50
- 2023Understanding In-Context Learning via Supportive Pretraining Datacited by 41
- 2024Data Selection via Optimal Control for Language Modelscited by 38
- 2024Mixture of In-Context Prompters for Tabular PFNscited by 33
- 2024Analysing The Impact of Sequence Composition on Language Model Pre-Trainingcited by 25
- 2024ProtLLM: An Interleaved Protein-Language LLM with Protein-as-Word Pre-Trainingcited by 21
- 2023Supervised Knowledge Makes Large Language Models Better In-context Learnerscited by 20
- 2023Structured Packing in LLM Training Improves Long Context Utilizationcited by 17
- 2024On the Noise Robustness of In-Context Learning for Text Generationcited by 16
- 2023Large Language Models are Miscalibrated In-Context Learnerscited by 12
12 most cited of 27 All 27 (JSON)
- 2026Exploring In-Context Learning for Handwritten Text Recognition29 Sep 2026
- 2026When Context Misleads: In-context Learning with Jurisdiction in Large Language Models23 Sep 2026
- 2026In-context Pre-trained Time-Series Foundation Models adapt to Unseen Tasks23 Feb 2026
- 2025Induction Signatures Are Not Enough: A Matched-Compute Study of Load-Bearing Structure in In-Context Learning26 Sep 2025
- 2025Mechanistic Fine-tuning for In-context Learning20 May 2025
- 2025MateICL: Mitigating Attention Dispersion in Large-Scale In-Context Learning2 May 2025
- 2025StaICC: Standardized Evaluation for Classification Task in In-context Learning27 Jan 2025
- 2025Unlocking In-Context Learning for Natural Datasets Beyond Language Modelling9 Jan 2025
- 2024CPRM: A LLM-based Continual Pre-training Framework for Relevance Modeling in Commercial Search2 Dec 2024
- 2024Data Selection via Optimal Control for Language Models9 Oct 2024
- 2024FabGPT: An Efficient Large Multimodal Model for Complex Wafer Defect Knowledge Queries15 Jul 2024
- 2024Token-based Decision Criteria Are Suboptimal in In-context Learning24 Jun 2024
12 newest of 27 All 27 (JSON)