Towards Mechanistically Understanding Why Memorized Knowledge Fails to Generalize in Large Language Model Finetuning
Not on the map. Why?
Abstract (arXiv)
Fine-tuning LLMs to inject new knowledge faces a critical challenge: LLMs can quickly memorize new facts, yet fail to use them for downstream reasoning tasks. We formalize this failure as the Knowing-Using Gap, characterized by an accuracy gap and a temporal lag between memorization and generalization. To understand this phenomenon, we fine-tune LLMs with unseen knowledge and monitor the spatial permeation dynamics of the knowledge internally with self-patching, an adaptation of activation patching that scans all layer pairs at every fine-tuning check-point. Self-patching identifies activation locations where relocating representations substantially improves failed generalization cases. These results are consistent with a knowledge-circuit misalignment hypothesis: memorized representations can exist internally but may not be routed to computation-effective layers. Experiments are done cross-domain for the robustness of this finding. Building on this diagnosis, we propose layer-wise representation self-distillation (LRSD) that aligns knowledge representation from late storage layer to middle layer. LRSD keeps improving generalization after fine-tuning saturates and nearly doubles multi-hop chaining accuracy on Qwen, while leaving memorization intact.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 36 of its 43 references that are in the map. First posted on arXiv 9 Jul 2026.
Builds on
36 of 43 references.
References · 36
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2022Mass-Editing Memory in a Transformercited by 976
- 2022In-context Learning and Induction Headscited by 930
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
- 2021Knowledge Neurons in Pretrained Transformerscited by 590
24 more
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 512
- 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 491
- 2023The Reversal Curse: LLMs trained on "A is B" fail to learn "B is A"cited by 399
- 2023Towards Best Practices of Activation Patching in Language Models: Metrics and Methodscited by 328
- 2023MQuAKE: Assessing Knowledge Editing in Language Models via Multi-Hop Questionscited by 292
- 2024Do Llamas Work in English? On the Latent Language of Multilingual Transformerscited by 275
- 2023Evaluating the Ripple Effects of Knowledge Editing in Language Modelscited by 238
- 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 229
- 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 225
- 2023Fine-Tuning or Retrieval? Comparing Knowledge Injection in LLMscited by 175
- 2024Fine-Tuning Enhances Existing Mechanisms: A Case Study on Entity Trackingcited by 140
- 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 137
- 2024Model Editing at Scale leads to Gradual and Catastrophic Forgettingcited by 92
- 2024Hopping Too Late: Exploring the Limitations of Large Language Models on Multi-Hop Queriescited by 78
- 2023Towards Vision-Language Mechanistic Interpretability: A Causal Tracing Tool for BLIPcited by 62
- 2024Knowledge Circuits in Pretrained Transformerscited by 60
- 2024STaRK: Benchmarking LLM Retrieval on Textual and Relational Knowledge Basescited by 53
- 2025How much do language models memorize?cited by 53
- 2024Fine Tuning vs. Retrieval Augmented Generation for Less Popular Knowledgecited by 49
- 2025Key-value memory in the braincited by 22
- 2026Locate, Steer, and Improve: A Practical Survey of Actionable Mechanistic Interpretability in Large Language Modelscited by 22
- 2025Grokking in LLM Pretraining? Monitor Memorization-to-Generalization without Testcited by 10
- 2025CaKE: Circuit-aware Editing Enables Generalizable Knowledge Learnerscited by 9
Built on by
Not cited yet.