Can Interpretation Predict Behavior on Unseen Data?
Not on the map. Why?
Abstract (arXiv)
Interpretability research often predicts model responses to targeted mechanistic interventions. But can we predict responses to unseen input data? We propose and demonstrate this alternate objective by using model internals to predict their out-of-distribution (OOD) behavior. We train hundreds of Transformers on simple synthetic tasks, where perfect in-distribution accuracy is compatible with multiple OOD generalization rules. We successfully use attention patterns -- observed only on in-distribution data -- to predict which rule each model follows on OOD data. Our experiments decouple the mechanistic faithfulness of our interpretation from its predictive value; ablations reveal such internal patterns can suppress rather than support the rule they predict, showing observational analysis can forecast behavior even when causal analysis fails to support a simple cause-effect link. Our findings are a proof-of-concept for a new interpretability objective: understanding model internals to predict behavior and assess reliability under distribution shift.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Builds on 33 of its 55 references that are in the map. First posted on arXiv 8 Jul 2025.
Builds on
33 of 55 references.
References · 33
- 2017Attention Is All You Needcited by 60,361
- 2019Attention is not Explanationcited by 929
- 2019Is Attention Interpretable?cited by 389
- 2020Fine-Tuning Pretrained Language Models: Weight Initializations, Data Orders, and Early Stoppingcited by 384
- 2023Towards Best Practices of Activation Patching in Language Models: Metrics and Methodscited by 359
- 2016Assessing the Ability of LSTMs to Learn Syntax-Sensitive Dependenciescited by 354
- 2018On the Robustness of Interpretability Methodscited by 314
- 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 309
- 2023Function Vectors in Large Language Modelscited by 308
- 2023In-context Vectors: Making In Context Learning More Effective and Controllable Through Latent Space Steeringcited by 245
- 2022Extracting Latent Steering Vectors from Pretrained Language Modelscited by 225
- 2023Causal Abstraction: A Theoretical Foundation for Mechanistic Interpretabilitycited by 205
21 more
- 2024Analyzing the Generalization and Reliability of Steering Vectorscited by 143
- 2025Are Sparse Autoencoders Useful? A Case Study in Sparse Probingcited by 115
- 2024AtP*: An efficient and scalable method for localizing LLM behaviour to componentscited by 111
- 2021Scale Efficiently: Insights from Pre-training and Fine-tuning Transformerscited by 106
- 2025Sparse Autoencoders Trained on the Same Data Learn Different Featurescited by 93
- 2019BERTs of a feather do not generalize together: Large variability in generalization across models with similar test set performancecited by 89
- 2023Is This the Subspace You Are Looking for? An Interpretability Illusion for Subspace Activation Patchingcited by 80
- 2020Does syntax need to grow on trees? Sources of hierarchical inductive bias in sequence-to-sequence networkscited by 72
- 2022Linear Connectivity Reveals Generalization Strategiescited by 56
- 2023Grokking of Hierarchical Structure in Vanilla Transformerscited by 53
- 2020How Can Self-Attention Networks Recognize Dyck-n Languages?cited by 51
- 2018The Importance of Being Recurrent for Modeling Hierarchical Structurecited by 42
- 2020Transferring Inductive Biases through Knowledge Distillationcited by 41
- 2023The Impact of Depth on Compositional Generalization in Transformer Language Modelscited by 38
- 2019Memory-Augmented Recurrent Neural Networks Can Learn Generalized Dyck Languagescited by 31
- 2023Transformers are uninterpretable with myopic methods: a case study with bounded Dyck grammarscited by 30
- 2023To grok or not to grok: Disentangling generalization and memorization on corrupted algorithmic datasetscited by 24
- 2020LSTMs Compose (and Learn) Bottom-Upcited by 11
- 2025Internal Causal Mechanisms Robustly Predict Language Model Out-of-Distribution Behaviorscited by 11
- 2020Rethinking Weight Decay For Efficient Neural Network Pruningcited by 10
- 2024Sometimes I am a Tree: Data Drives Unstable Hierarchical Generalizationcited by 10
Built on by
Citing papers per year: 2025: 1; 2026: 7.
8 papers.
- 2025The Dead Salmons of AI Interpretabilitycited by 12
- 2026Patterning: The Dual of Interpretabilitycited by 7
- 2026Interpretability Can Be Actionablecited by 7
- 2026Position: Don't Just "Fix it in Post": A Science of AI Must Study Training Dynamicscited by 3
- 2026Diagnosing Generalization Failures from Representational Geometry Markerscited by 2
- 2026Explaining Attention with Program Synthesiscited by 1
- 2026Adversarial Concept Search: Predicting Compositional Errors From Feature Geometrycited by 0
- 2026AIMO Interpretability Challengecited by 0
- 2026AIMO Interpretability Challenge15 Jul 2026
- 2026Explaining Attention with Program Synthesis17 Jun 2026
- 2026Adversarial Concept Search: Predicting Compositional Errors From Feature Geometry11 Jun 2026
- 2026Position: Don't Just "Fix it in Post": A Science of AI Must Study Training Dynamics3 Jun 2026
- 2026Interpretability Can Be Actionable11 May 2026
- 2026Diagnosing Generalization Failures from Representational Geometry Markers2 Mar 2026
- 2026Patterning: The Dual of Interpretability20 Jan 2026
- 2025The Dead Salmons of AI Interpretability21 Dec 2025