揭示大模型少样本学习背后的机制:非单纯记忆,也非符号算法。
Illusion or Algorithm? Investigating Memorization, Emergence, and Symbolic Processing in In-Context Learning
- 通过系统性测试和残差流分析,研究少样本学习的真实机制。
- 模型表现超越单纯记忆,但未实现独立的符号运算能力。
- 适合模型开发者与AI安全研究人员参考,理解模型行为本质。
仅通过网页规模数据进行下一个词预测训练的大规模Transformer语言模型,仅需少量示例即可解决多种任务。这种现象称为上下文学习(ICL),其内在机制仍存在争议且理解不深。本文利用Pythia完整缩放系列(包括中间检查点)系统研究ICL,通过下游任务表现与残差流子空间的机理分析,证明ICL超越了对训练语料的简单“记忆”,但并未实现独立的符号算法。研究还澄清了训练动态、模型能力及可解释性要素的影响。本工作深化了对ICL的理解,为模型优化提供指导,并为AI安全实践者提供更可靠的基准。
原文摘要 · Abstract (English)
Large-scale Transformer language models (LMs) trained solely on next-token prediction with web-scale data can solve a wide range of tasks after seeing just a few examples. The mechanism behind this capability, known as in-context learning (ICL), remains both controversial and poorly understood. Some studies argue that it is merely the result of memorizing vast amounts of data, while others contend that it reflects a fundamental, symbolic algorithmic development in LMs. In this work, we introduce a suite of investigative tasks and a novel method to systematically investigate ICL by leveraging the full Pythia scaling suite, including interim checkpoints that capture progressively larger amount of training data. By carefully exploring ICL performance on downstream tasks and simultaneously conducting a mechanistic analysis of the residual stream's subspace, we demonstrate that ICL extends beyond mere "memorization" of the training corpus, yet does not amount to the implementation of an independent symbolic algorithm. Our results also clarify several aspects of ICL, including the influence of training dynamics, model capabilities, and elements of mechanistic interpretability. Overall, our work advances the understanding of ICL and its implications, offering model developers insights into potential improvements and providing AI security practitioners with a basis for more informed guidelines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。