Explaining the Roles of Experts in Mixture-of-Experts LLMs for Personalization
Organizations: University of Copenhagen
Abstract
Personalization of Large Language Models (LLMs) is increasingly important for user-centric applications, yet prior methods mainly adapt dense LLMs, whose polysemantic neurons, each encoding multiple unrelated concepts, hinder localizing the components behind personalization, and thus understanding and controlling it. We propose to personalize Mixture-of-Experts (MoE) LLMs, which sparsely route each token to a few specialized experts, reducing polysemanticity and enabling such localization. We adapt three open-weight MoE LLMs on a large-scale personalization dataset and train crosscoders on paired activations of the personalized and non-personalized models to learn interpretable personalized latents, from which we identify the experts encoding distinct personalization features. The personalized MoE LLMs perform competitively with dense LLMs while activating only a fraction of their parameters, exhibit lower polysemanticity than dense LLMs with similar active parameters, and, within the same model family, yield personalized latents that are less shared with the non-personalized model. Causal interventions show that these latents suffice to recover the personalized model's behavior and that boosting their associated experts moves the non-personalized model toward it, revealing distinct routing paths for different personalization features.
Figures & tables
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Active | Total |
|---|---|---|
| Params | Params | |
| OLMoE family | ||
| OLMo-2-1B (0425) | 1B | 1B |
| OLMoE-1B-7B (0125) | 1.3B | 6.9B |
| OLMo-2-7B (1124) | 7B | 7B |
| Qwen family | ||
| Model | Act. | MFBench | LaMP | LongLaMP | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| News | Scholar | Tweet | Abstr. | Topic | Review | ||||||||||
| M | R | M | R | M | R | M | R | M | R | M | R | M | R | ||
| OLMo-2-1B (0425) | 1B | 0.29 | 0.19 | 0.12 | 0.08 | 0.30 | 0.18 | 0.29 | 0.22 | 0.31 | 0.18 | 0.23 | 0.11 | 0.18 | 0.14 |
| OLMo-2-1B (0425) ⋆ | 1B | 0.31 | 0.20 | 0.12 | 0.12 | 0.38 | 0.29 | 0.40 | 0.38 | 0.26 | 0.19 | 0.23 | 0.12 | 0.14 | 0.12 |
| OLMoE-1B-7B (0125) | 1.3B | 0.25 | 0.20 | 0.12 | 0.11 | 0.36 | 0.28 | 0.35 | 0.32 | 0.31 | 0.18 | 0.24 | 0.11 | 0.21 | 0.14 |
| OLMoE-1B-7B (0125) † | 1.3B | 0.32 | 0.20 | 0.12 | 0.12 | 0.35 | 0.26 | 0.36 | 0.32 | 0.31 | 0.17 | 0.23 | 0.11 | 0.19 | 0.14 |
| Model | Latent | Explanation | Dimension | Top- Experts | Example |
|---|---|---|---|---|---|
| OLMoE-1B-7B (0125) | 12503 | The feature detects the presence of meticulously structured and precise language, often characterized by the use of advanced technical terminology, detailed explanations, and a formal tone, typically in the context of scientific, technical, or academic discussions. | informativeness | L6-E9 (0.91);L6-E22 (0.73) | “…Your discussions are met iculously detailed, exploring biochemical mechanisms of Vitamin C…” |
| Mixtral-8x7B (v0.1) | 45358 | The feature detects tokens related to grammatical correctness, punctuation, and sentence structure, often in the context of editing, correcting, or explaining language usage. | style | L15-E2 (0.96);L15-E5 (0.34) | “…identifying run-on sentences with a focus on gram mat ical terms, catering to an audience with prior knowledge…” |