PreprintCited by 1,170Topic ModelingDomain Adaptation and Few-Shot LearningNatural Language Processing Techniques

The DOI matches its DataCite record; 22 of the 22 references it cites were found in a registry or an index; none of the 34 sources its claims cite is checked for support yet.

In-context Learning and Induction Heads

Open access

Abstract

"Induction heads" are attention heads that implement a simple algorithm to complete token sequences like [A][B] ... [A] -> [B]. In this work, we present preliminary and indirect evidence for a hypothesis that induction heads might constitute the mechanism for the majority of all "in-context learning" in large transformer models (i.e. decreasing loss at increasing token indices). We find that induction heads develop at precisely the same point as a sudden sharp increase in in-context learning ability, visible as a bump in the training loss. We present six complementary lines of evidence, arguing that induction heads may be the mechanistic source of general in-context learning in transformer models of any size. For small attention-only models, we present strong, causal evidence; for larger models with MLPs, we present correlational evidence.

Overview

AI snapshot

Written from the paper’s full text.

TL;DR

The paper presents six complementary lines of evidence that induction heads—attention heads that complete [A][B]…[A]→[B] patterns—may be the primary mechanism for most in-context learning in transformer language models. Evidence is strong and causal for small attention-only models, but correlational for larger models with MLPs.

Key findings

  • Across 34 decoder-only transformers, in-context learning improves abruptly in a phase change roughly 2.5–5 billion tokens into training, from less than 0.15 nats before to roughly 0.4 nats after, while induction heads form in the same window; one-layer models lack both.
  • A “smeared key” architecture lets one-layer models show in-context learning when they did not before, and makes two-layer and larger models show it earlier, matching the prediction if induction heads are a minimal mechanism.
  • In small attention-only models, ablating induction heads removes almost all in-context learning; the paper calls this its strongest causal evidence, while for larger MLP models the evidence is correlational/mechanistic.
  • Induction heads defined narrowly by copying and prefix matching also show abstract behaviors in a 40-layer 13B model: a literal copying head at layer 21/40 scores 0.89 copying and 0.75 prefix matching; a translation head at layer 7/40 scores 0.20 copying and 0.85 prefix matching; a pattern-matching head at layer 8/40 scores 0.69 copying and 0.94 prefix matching, allocating about 65% of attention from colons to correct positions.

Methods

  • Analyzed 34 decoder-only Transformer language models across four series: small attention-only (1–6 layers), small with MLPs (1–6 layers), full-scale models (4 layers/13M parameters up to 40 layers/13B parameters), and smeared-key models; training data was filtered Common Crawl and internet books plus about 10% Python code, with an alternate books-only dataset for some small models.
  • Measured per-token losses over 10,000 examples, PCA of per-token loss vectors, an in-context learning score (500th-token loss minus 50th-token loss), head activation evaluators for copying/prefix matching/previous-token attention, QK eigenvalue traces, and pattern-preserving attention-head ablations for small models, including more than 50,000 attention head ablations.
  • The smeared-key intervention interpolates each head’s key vector between the current token and previous token to allow induction heads to form at any depth.

Limitations

  • Stated: the evidence is preliminary and indirect; for large models with MLPs it is mainly correlational and could be confounded, with low time resolution (14–15 snapshots) and no full-scale ablations.
  • Stated: other composition mechanisms may form during the phase change, so co-occurrence could reflect a shared latent variable; a constant in-context learning score after the phase change does not prove the underlying mechanisms are constant; ablations measure marginal effects and are harder to interpret with MLPs.
  • Evident from the text: extrapolation from small to large models is a matter of judgment, and the paper notes unexplained curiosities such as a loss spike and anti-copying prefix-search heads.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

DataCite · 10.48550/arxiv.2209.11895

Cites

0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Verification

Claims the paper makes

64 Own result · 59 Own interpretation · 42 No basis given · 25 Sources not found · 13 Own method · 2 Not checked yet

  • It may state two separate claims as one. 59
  • Probably not a substantial claim. 65
  • The quote only partly states this claim. 2
  • Borderline: one of the tests for a substantial claim is unclear. 2
105 of 105
  • The authors state this as an assumption.

    The previous paper's reverse engineering analysis of induction heads is trustworthy/correct.

    “If we trust this analysis, we can know how induction heads behave without actually running them, and the argument we made in the previous paragraph goes through.”
  • The paper relies on this without support.

    The model's loss is an average of billions of log-likelihood losses for individual tokens.

    “The model's loss is an average of billions of log-likelihood losses for individual tokens.”
  • The paper relies on this without support.

    The model's performance at token 50 improves over training time.

    “we know that the model's performance at token 50 improves over training time.”
  • The paper relies on this without support.

    Induction heads are the first case where a mechanistic account has been provided for a phase change in machine learning.

    “As far as we know, induction heads are the first case where a mechanistic account has been provided for a phase change in machine learning.”
4 of 4

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

4 of 4

Citations

Cites

The publisher hasn’t deposited this paper’s reference list.

Cited by

Sorted by citations

See all 1,170

Metadata

Authors
Olsson, C., Elhage, N., Nanda, N., Joseph, N., DasSarma, N., Henighan, T., Mann, B., Askell, A., Bai, Y., Chen, A., Conerly, T., Drain, D., Ganguli, D., Hatfield-Dodds, Z., Hernandez, D., Johnston, S., Jones, A., Kernion, J., Lovitt, L., Ndousse, K., Amodei, D., Brown, T., Clark, J., Kaplan, J., McCandlish, S., & Olah, C.
Venue
arXiv
Publisher
arXiv
Published
2022
ISSN
2331-8422 (ISSN-L)
OpenAlex
W4297412003
Cited by
1,170
References
0
Citations by year
2026: 522 · 2025: 322 · 2024: 206 · 2023: 115 · 2022: 5
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University) (cc-by)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
Keywords
in-context learning, induction heads, attention heads, transformer models, mechanistic interpretability, training loss, large language models

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

A language model drew the claims from the paper on 2026-10-23, under version 3 of the claim definition (model deepseek-v4.1-flash, prompt extract-v3).

A second model labelled each claim on 2026-10-23 (model jev-1.13-20260917, definition revision gold-guideline-v0.2).

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review