flâneur

Training Language Models to Explain Their Own Computations

arxiv.org · 8,547 words · saved by 1 readers

Can language models (LMs) learn to faithfully describe their internal computations? Are they better able to describe themselves than other models? We study the extent to which LMs’ privileged access to their own internals can be leveraged to produce new techniques for explaining their behavior. Using existing interpretability techniques as a source of ground truth, we fine-tune LMs to generate natural language descriptions of (1) the information encoded by LM features, (2) the causal structure of LMs’ internal activations, and (3) the influence of specific input tokens on LM outputs. When trained with only tens of thousands of example explanations, explainer models exhibit non-trivial generalization to new queries. This generalization appears partly attributable to explainer models’ privileged access to their own internals: fine-tuning a model to explain its own computations generally works better than fine-tuning a different model (even if the explainer model is significantly more cap

Zifan Carl Guo Affiliation: MIT CSAIL Vincent Huang Affiliation: Transluce Jacob Steinhardt Affiliation: Transluce Jacob Andreas Affiliation: Transluce Affiliation: MIT CSAIL Abstract Can language models (LMs) learn to faithfully describe their internal computations? Are they better able to describe themselves than other models? We study the extent to which LMs’ privileged access to their own internals can be leveraged to produce new techniques for explaining their behavior. Using existing interpretability techniques as a source of ground truth, we fine-tune LMs to generate natural…

saved by

related reading