Research papers · Paper
Are Arithmetic Heuristic Neurons Form-Invariant? A Mechanistic Analysis of Symbols, Text, and Code in LLMs
Outline · 17 pages
- Introductionp. 1
- Related Workp. 2
- Mechanistic Interpretabilityp. 2
- Arithmatic Reasoning Interpretabilityp. 2
- Backgroundp. 3
- MLPs as Key-Value Memoriesp. 3
- Arithmetic Neurons and Heuristicsp. 3
- Experimental Setupp. 4
- Layer Identificationp. 4
- Identifying Arithmetic Neuronsp. 4
- Circuit Faithfulness Evaluationp. 5
- Cross-Format Activation Transferp. 5
- Neuron Heuristic Familiesp. 6
- Resultsp. 6
- Sparsity: A compact arithmetic circuit is shared across formatsp. 6
- Causality: Shared neurons are necessary and sufficient within the late-layer MLPsp. 6
- Cross-format activation transfer recovers most incorrect predictionsp. 7
- Mechanistic consistency: Shared neurons implement the same heuristicsp. 7
- Conclusionp. 7
- Additional Results on Llama Modelsp. 12
- Layer Identificationp. 12
- Shared Arithmetic Circuitsp. 12
- Circuit Faithfulnessp. 12
- Cross-Format Activation Transferp. 12
- Dataset Statistics and Linear Probe Evaluationp. 12
- Additional Heuristic Consistency Resultsp. 16
- Distribution of Heuristic Familiesp. 16
- Statistical Significance of Heuristic Agreementp. 16