English

Understanding In-Context Learning from Repetitions

Computation and Language 2024-02-22 v3

Abstract

This paper explores the elusive mechanism underpinning in-context learning in Large Language Models (LLMs). Our work provides a novel perspective by examining in-context learning via the lens of surface repetitions. We quantitatively investigate the role of surface features in text generation, and empirically establish the existence of \emph{token co-occurrence reinforcement}, a principle that strengthens the relationship between two tokens based on their contextual co-occurrences. By investigating the dual impacts of these features, our research illuminates the internal workings of in-context learning and expounds on the reasons for its failures. This paper provides an essential contribution to the understanding of in-context learning and its potential limitations, providing a fresh perspective on this exciting capability.

Keywords

Cite

@article{arxiv.2310.00297,
  title  = {Understanding In-Context Learning from Repetitions},
  author = {Jianhao Yan and Jin Xu and Chiyu Song and Chenming Wu and Yafu Li and Yue Zhang},
  journal= {arXiv preprint arXiv:2310.00297},
  year   = {2024}
}

Comments

Accepted by ICLR 2024. Updated with new experiments and results

R2 v1 2026-06-28T12:36:59.320Z