atlas

Research map / Architectures and efficiency / Attention and transformers

InversionView: A General-Purpose Method for Reading Information from Neural Activations

NeurIPS 2024 · Architectures and efficiency › Attention and transformers · approach: Attention head mechanisms · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes InversionView, which trains a conditional decoder to sample inputs whose activations are close to a query activation, revealing the information content of activations in small transformers and GPT-2 circuits.

Problem

Existing interpretability methods (probes, vocabulary projection, causal interventions) do not directly reveal the human-readable information encoded in neural activations of language and sequence models.

Approach

Define the information in an activation as its epsilon-preimage (the set of inputs producing similar activations) and approximate it by training an autoregressive decoder conditioned on cached activations to sample inputs; distances are then checked on the probed model.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Compare layer 0 activations before and after the MLPCharacter-counting transformer · Character countingprominence of target-character count in activation geometryHelped
Use the layer 1 colon residual stream to carry count informationCharacter-counting transformer · Character countingcount information represented at the colonHelped
Patch a^{0,0}_{tc} or a^{1,0}_{:} from a contrast exampleCharacter-counting transformer · Character countinglogit-difference effectHelped
Decode activations with InversionViewGPT-2 Small IOI circuit · Indirect object identificationagreement with previously inferred head information: 3 of 26Helped
Use InversionView to inspect addition activations and trace the A1 circuit3-digit addition transformer · 3-digit addition, A1identification of the cause of model errorsHelped
Increase sampling temperature from 1.0 to 2.0 (ablation)InversionView decoder · 3-digit additionsampling-probability smoothness within the epsilon-preimageHelped
Add noise with coefficient eta=0.1 to the query activation (ablation)InversionView decoder · 3-digit additionsampling-probability smoothness within the epsilon-preimageHelped
Replace activation conditioning in self-attention with encoder-decoder attention (ablation)InversionView decoder · Decoder architecturedecoder performanceNo clear effect
Vary the epsilon threshold (ablation)InversionView · Character counting and 3-digit additiongranularity of inferred activation informationMixed
Patch x^{0,postStream}_{+} with its test-set mean (ablation)3-digit addition transformer · 3-digit additioneffect on model output: $8.1\times10^{-8}$; $5.4\times10^{-7}$; $2.5\times10^{-8}$; $3.6\times10^{-9}$ KL divergenceNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: synthetic character counting inputs IOI prompts 3-digit addition strings

Benchmarks: character counting test set 3-digit addition test set IOI circuit heads from Wang et al.

Metrics: test accuracy in-preimage rate logit difference under activation patching KL divergence logit decrement rate

Baselines: supervised probing vocabulary projection activation patching

Builds on: activation patching path patching Nanda-style addition circuit analysis IOI circuit (Wang et al.)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy