Time Course MechInterp: Analyzing the Evolution of Components and Knowledge in Large Language Models
Not on the map. Why?
Abstract (arXiv)
Understanding how large language models (LLMs) acquire and store factual knowledge is crucial for enhancing their interpretability and reliability. In this work, we analyze the evolution of factual knowledge representation in the OLMo-7B model by tracking the roles of its attention heads and feed forward networks (FFNs) over the course of pre-training. We classify these components into four roles: general, entity, relation-answer, and fact-answer specific, and examine their stability and transitions. Our results show that LLMs initially depend on broad, general-purpose components, which later specialize as training progresses. Once the model reliably predicts answers, some components are repurposed, suggesting an adaptive learning process. Notably, attention heads display the highest turnover. We also present evidence that FFNs remain more stable throughout training. Furthermore, our probing experiments reveal that location-based relations converge to high accuracy earlier in training than name-based relations, highlighting how task complexity shapes acquisition dynamics. These insights offer a mechanistic view of knowledge formation in LLMs.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 40 of its 48 references that are in the map. First posted on arXiv 3 Jun 2025.
Builds on
40 of 48 references.
References · 40
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2022In-context Learning and Induction Headscited by 930
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2020A Primer in BERTology: What we know about how BERT workscited by 807
- 2024OLMo: Accelerating the Science of Language Modelscited by 599
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 512
- 2021Measuring and Improving Consistency in Pretrained Language Modelscited by 361
28 more
- 2023Linear Representations of Sentiment in Large Language Modelscited by 157
- 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 145
- 2023The Hydra Effect: Emergent Self-repair in Language Model Computationscited by 145
- 2024Have Faith in Faithfulness: Going Beyond Circuit Overlap When Finding Model Mechanismscited by 144
- 2023Sudden Drops in the Loss: Syntax Acquisition, Phase Transitions, and Simplicity Bias in MLMscited by 121
- 2023Circuit Component Reuse Across Tasks in Transformer Language Modelscited by 105
- 2024Information Flow Routes: Automatically Interpreting Language Models at Scalecited by 92
- 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 86
- 2024How Do Large Language Models Acquire Factual Knowledge During Pretraining?cited by 86
- 2022Training Trajectories of Language Models Across Scalescited by 80
- 2021Probing Across Time: What Does RoBERTa Know and When?cited by 73
- 2023Copy Suppression: Comprehensively Understanding an Attention Headcited by 73
- 2022Measuring the Mixing of Contextual Information in the Transformercited by 63
- 2022Understanding Transformer Memorization Recall Through Idiomscited by 62
- 2024LLM Circuit Analyses Are Consistent Across Training and Scalecited by 60
- 2023JoMA: Demystifying Multilayer Transformers via JOint Dynamics of MLP and Attentioncited by 52
- 2024Competition of Mechanisms: Tracing How Language Models Handle Facts and Counterfactualscited by 40
- 2024Summing Up the Facts: Additive Mechanisms Behind Factual Recall in LLMscited by 39
- 2023Neuron-Level Knowledge Attribution in Large Language Modelscited by 37
- 2024Explorations of Self-Repair in Language Modelscited by 35
- 2023Characterizing Learning Curves During Language Model Pre-Training: Learning, Forgetting, and Stabilitycited by 34
- 2020Pretrained Language Model Embryology: The Birth of ALBERTcited by 30
- 2024Localizing Paragraph Memorization in Language Modelscited by 30
- 2023Subspace Chronicles: How Linguistic Information Emerges, Shifts and Interacts during Language Model Trainingcited by 23
- 2023Look Before You Leap: A Universal Emergent Decomposition of Retrieval Tasks in Language Modelscited by 17
- 2024Inferring Functionality of Attention Heads from their Parameterscited by 17
- 2023Latent State Models of Training Dynamicscited by 15
- 2024Interpreting Context Look-ups in Transformers: Investigating Attention-MLP Interactionscited by 11
Built on by
Not cited yet.