Research papers · Paper
RouterInterp: Understanding Superposed Specialisation in Mixture of Experts Routing
Outline · 33 pages
- Introductionp. 1
- Backgroundp. 3
- Sparse Autoencoders (SAE)p. 3
- Mixture of Experts (MoE)p. 3
- Specialisation Hypothesesp. 3
- Alternative Specialisation Hypothesesp. 4
- Motivation for Superposed Specialisationp. 5
- Empirical Evidence for Superposed Specialisationp. 5
- RouterInterpp. 6
- Methodp. 6
- Evaluationp. 7
- Experimental Setupp. 8
- Resultsp. 9
- Discussionp. 10
- Limitationsp. 11
- Related Workp. 11
- Conclusionp. 12
- Extended Related Workp. 20
- Sparse Feature Decompositions for Model Interpretabilityp. 20
- Adaptive Computationp. 21
- Modularity in Neural Networksp. 21
- SAE Features as Routing Predictorsp. 21
- Feature Selection Method and Set Sizep. 22
- Feature Selection Methodp. 22
- Selection Criterion and Set Sizep. 23
- Expert Routing Across Pile Domainsp. 25
- Routing Prediction Across Layersp. 26
- Routing Depends on Multiple Featuresp. 26
- SAE Setup and Reconstruction Qualityp. 27
- Human Evaluationp. 28
- SAE Probes Recover Router Probabilitiesp. 28
- Promptsp. 29
- Example Expert Explanationsp. 32