Attention Sensitivity Is Not Enough: Dissociating Attention-Level and Behavioural In-Context Learning under Fine-Tuning
Not on the map. Why?
Abstract (arXiv)
In-context learning (ICL) lets large language models adapt to new tasks from demonstrations, and fine-tuning can erode this behaviour. Many preservation diagnostics inspect attention: if attention changes when demonstrations change, the model is treated as context-sensitive. This paper asks how far that proxy can be trusted once it is optimised. We formalise In-Context Sensitivity (ICS), the average row distance between last-token attention on matched and mismatched demonstration prefixes, and pair it with ICL-GAP, the behavioural accuracy gap between the same prefixes. In a controlled four-arm ablation on Llama-2-7B, an ICS-maximising regulariser (armKL) drives ICS to 1.413, within 0.5% of its geometric ceiling. The behavioural readout tells a different story: ICL-GAP stays near zero and MMLU accuracy moves from 0.371 to 0.279, a Goodhart dissociation of the bounded attention proxy. Endpoint statistics locate the mechanism: attention grows sharp and near-disjoint across prefixes yet routes to formatting and demonstration-body tokens rather than labels. A random-label protocol confirms that the behavioural probe family retains dynamic range at the same checkpoints. In a constructive sweep, behaviour gating partially mitigates the effect, while objectives anchored to pretrained computation hold the high-MMLU, moderate-ICS region that divergence maximisers leave. The main lesson is diagnostic: attention-level ICL proxies earn their place as training targets only after validation against behavioural gaps.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 30 of its 34 references that are in the map. First posted on arXiv 30 Aug 2026.
Builds on
30 of 34 references.
References · 30
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2022Emergent Abilities of Large Language Modelscited by 2,483
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2022In-context Learning and Induction Headscited by 930
18 more
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
- 2021Towards a Unified View of Parameter-Efficient Transfer Learningcited by 863
- 2022Transformers learn in-context by gradient descentcited by 714
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
- 2023An Empirical Study of Catastrophic Forgetting in Large Language Models During Continual Fine-tuningcited by 622
- 2020Calibration of Pre-trained Transformerscited by 307
- 2024Continual Learning of Large Language Models: A Comprehensive Surveycited by 228
- 2022Defining and Characterizing Reward Hackingcited by 184
- 2023Large Language Models Are Latent Variable Models: Explaining and Finding Good Demonstrations for In-Context Learningcited by 149
- 2018Categorizing Variants of Goodhart's Lawcited by 132
- 2025Reward Shaping to Mitigate Reward Hacking in RLHFcited by 88
- 2025Inference-Time Reward Hacking in Large Language Modelscited by 32
- 2026Large Language Models Hack Rewards, and Societycited by 3
- 2026Don't Go Breaking My LLM: The Impact of Pruning Attention Layers on Explanation Faithfulness and Confidence Calibrationcited by 2
- 2026ADAPT: Attention Dynamics Alignment with Preference Tuning for Faithful MLLMscited by 2
- 2026Fine-Tuning Dynamics of In-Context Factual Recall in Transformerscited by 1
- 2026Categorical Prior Lock-in: Why In-Context Learning Fails for Structured Datacited by 1
- 2026Probing Memorization of Tabular In-Context Learningcited by 1
Built on by
Not cited yet.