Decoding Knowledge Attribution in Mixture-of-Experts: A Framework of Basic-Refinement Collaboration and Efficiency Analysis
The interpretability of Mixture-of-Experts (MoE) models, especially those with heterogeneous designs, remains underexplored.Existing attribution methods for dense models fail to capture dynamic routing-expert interactions in sparse MoE architectures.To address this issue, we propose a cross-level attribution algorithm to analyze sparse MoE architectures (Qwen 1.5-MoE, OLMoE, Mixtral-8x7B) against dense models (Qwen 1.5-7B, Llama-7B, Mistral-7B).Results show MoE models achieve 31% higher per-layer efficiency via a "mid-activation, lateamplification" pattern: early layers screen experts, while late layers refine knowledge collaboratively.Ablation studies reveal a "basicrefinement" framework-shared experts handle general tasks (entity recognition), while routed experts specialize in domain-specific processing (geographic attributes).Semanticdriven routing is evidenced by strong correlations between attention heads and experts (r = 0.68), enabling task-aware coordination.Notably, architectural depth dictates robustness: deep Qwen 1.5-MoE mitigates expert failures (e.g., 43% MRR drop in geographic tasks when blocking top-10 experts) through shared expert redundancy, whereas shallow OLMoE suffers severe degradation (76% drop).Task sensitivity further guides design: core-sensitive tasks (geography) require concentrated expertise, while distributed-tolerant tasks (object attributes) leverage broader participation.These insights advance MoE interpretability, offering principles to balance efficiency, specialization, and robustness.
Authors
- Xiuze Zhou (ORCID: https://orcid.org/0000-0002-0717-6936)
- Bo Wang (ORCID: https://orcid.org/0000-0002-3231-1167)
- Peijie Jiang (ORCID: https://orcid.org/0000-0002-0534-6989)
- Junzhuo Li (ORCID: https://orcid.org/0009-0007-3807-5111)
- Jia Liu (ORCID: https://orcid.org/0000-0001-8104-0079)
- Xuming Hu
Publication Details
- Published
- 2025-01-01
- DOI
- https://doi.org/10.18653/v1/2025.acl-long.1093
- Primary Topic
- Big Data and Business Intelligence
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Ant Group
- Hong Kong University of Science and Technology