mlsanityData to 7 Oct 2026

Attention Sensitivity Is Not Enough: Dissociating Attention-Level and Behavioural In-Context Learning under Fine-Tuning

arXiv 2609.00064 ↗First posted 30 Aug 2026cs.LG cs.AI cs.CLcited by 0

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

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2022Training language models to follow instructions with human feedbackcited by 15,712
  3. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  4. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  5. 2017On Calibration of Modern Neural Networkscited by 4,860
  6. 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
  7. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  8. 2022Emergent Abilities of Large Language Modelscited by 2,483
  9. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
  10. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
  11. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
  12. 2022In-context Learning and Induction Headscited by 930
18 more
  1. 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
  2. 2021Towards a Unified View of Parameter-Efficient Transfer Learningcited by 863
  3. 2022Transformers learn in-context by gradient descentcited by 714
  4. 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
  5. 2023An Empirical Study of Catastrophic Forgetting in Large Language Models During Continual Fine-tuningcited by 622
  6. 2020Calibration of Pre-trained Transformerscited by 307
  7. 2024Continual Learning of Large Language Models: A Comprehensive Surveycited by 228
  8. 2022Defining and Characterizing Reward Hackingcited by 184
  9. 2023Large Language Models Are Latent Variable Models: Explaining and Finding Good Demonstrations for In-Context Learningcited by 149
  10. 2018Categorizing Variants of Goodhart's Lawcited by 132
  11. 2025Reward Shaping to Mitigate Reward Hacking in RLHFcited by 88
  12. 2025Inference-Time Reward Hacking in Large Language Modelscited by 32
  13. 2026Large Language Models Hack Rewards, and Societycited by 3
  14. 2026Don't Go Breaking My LLM: The Impact of Pruning Attention Layers on Explanation Faithfulness and Confidence Calibrationcited by 2
  15. 2026ADAPT: Attention Dynamics Alignment with Preference Tuning for Faithful MLLMscited by 2
  16. 2026Fine-Tuning Dynamics of In-Context Factual Recall in Transformerscited by 1
  17. 2026Categorical Prior Lock-in: Why In-Context Learning Fails for Structured Datacited by 1
  18. 2026Probing Memorization of Tabular In-Context Learningcited by 1

Built on by

Not cited yet.