Skip to content

Towards Mechanistically Understanding Why Memorized Knowledge Fails to Generalize in Large Language Model Finetuning

Jul 2026 · 1 citation · 40 references
Computer Science

TL;DR

Self-patching identifies activation locations where relocating representations substantially improves failed generalization cases and design a simple heuristic strategy which recovers 58--75\% of the oracle headroom in generalization failure.

Abstract

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 \textit{\textbf{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 using a novel intervention technique called self-patching. 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. To demonstrate the practicality of this diagnostic finding, we design a simple heuristic strategy which recovers 58--75\% of the oracle headroom in generalization failure. Experiments are done cross-domain for the robustness of this finding.

View source

Similar papers

Preprint Aug 2026

From Memorization to Absorption: Mixed-Policy RL for Continual Knowledge Injection

This work proposes Golden-GRPO Injection (GRIN), a three-stage self-learning framework for continual knowledge injection that substantially outperforms SFT and mixed-policy RL baselines on the harder question types while matching them on basic fact recall.

Zhibo Hou, Fan Zhao, Zhi-Yu An et al. · 0 citations
2025

For Better or for Worse, Transformers Seek Patterns for Memorization

This work investigates memorization in transformer-based language models by analyzing their memorization dynamics during training over multiple epochs and finds that memorization is neither a constant accumulation of sequences nor simply dictated by the recency of exposure to these sequences.

Madhur Panwar, Gail Weiss, Navin Goyal et al. · 1 citation
Preprint Aug 2026

Grounded Semantic Re-Binding for Robust Instruction Generalization in Vision-Language-Action Models

This work demonstrates that robust semantic grounding can be achieved through elegant structural design, bypassing the inefficient brute-force data scaling paradigm and introduces ParaVLA, a natively decoupled 0.33B-parameter model exhibiting near-perfect robustness to instruction rewording.

Zhao-Kai Yin, Zhi-Peng Zhang · 0 citations
Preprint Aug 2026

Exploring and Bridging Knowledge Holes in Unlearned Multimodal Large Language Models

This work constructs a benchmark that captures unintended degradation on benign inputs sharing generic patterns with the forget set, and proposes Selective Protection with Anchored Regularization, which protects generic patterns via anchored activation filtering while reinforcing them through entity-abstracted enhancem...

Junxian You, Junkai Chen, Yuhao He et al. · 0 citations
Preprint Aug 2026

Memorization Diagnostics for Code LLMs Should be Scale-Aware

It is revealed that scaled encoders successfully absorb substantial representation load while still converging on the correct family of solutions, and this ability to adapt to varying surface forms is what truly matters for usability and generalizability in LLM and agentic applications.

P. Rajput, Abdoul Aziz Bonkoungou, Albérick Euraste Djiré et al. · 0 citations

Related blog posts

MIT News · Artificial Intelligence Sep 24, 2026

Estimating suicide risk from text

A new language-processing tool could help identify the highest-risk individuals from natural language, enabling swifter interventions.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.