Tracking Equivalent Mechanistic Interpretations Across Neural Networks
Organizations: Carnegie Mellon University · Max Planck Institute for Software Systems
Abstract
Mechanistic interpretability (MI) is an emerging framework for interpreting neural networks. Given a task and model, MI aims to discover a succinct algorithmic process, an interpretation, that explains the model's decision process on that task. However, MI is difficult to scale and generalize. This stems in part from two key challenges: there is no precise notion of a valid interpretation; and, generating interpretations is often an ad hoc process. In this paper, we address these challenges by defining and studying the problem of interpretive equivalence: determining whether two different models share a common interpretation, without requiring an explicit description of what that interpretation is. At the core of our approach, we propose and formalize the principle that two interpretations of a model are equivalent if all of their possible implementations are also equivalent. We develop an algorithm to estimate interpretive equivalence and case study its use on Transformer-based models. To analyze our algorithm, we introduce necessary and sufficient conditions for interpretive equivalence based on models' representation similarity. We provide guarantees that simultaneously relate a model's algorithmic interpretations, circuits, and representations. Our framework lays a foundation for the development of more rigorous evaluation methods of MI and automated, generalizable interpretation discovery methods.
Figures & tables
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Term | Intuitive Explanation | Formal Definition |
| Causal/Computation Graph | A graph that describes a model’s computational dependencies | Definition 4.1 |
| Variables | Nodes in a model’s computation graph that store latent computation results | Definition 4.1 |
| Circuit | A subset of the model’s computational graph that on its own can fully recreate the model’s input-output behavior | Definition D.1 |
| Mechanistic Explanation | A symbolic algorithm, we can understand, that explains the model’s computation | N/A |
| Causal Abstraction | Mapping from low-level variables to higher-level ones that preserve low-level causal relationships | Definition D.3 |
| Interpretation | A causal graph much smaller than the model’s computation graph that explains the model’s computation graph | Definition D.4 |
| Symbol | Meaning |
| An alphabet | |
| Set of all finite strings constructed from | |
| A subset of | |
| An interpretation | |
| A circuit | |
| A representation |