Deconstructing Pre-training: Knowledge Attribution Analysis in MoE and Dense Models
On the map: Mixture-of-experts → MoE large language models
Abstract (arXiv)
Mixture-of-Experts (MoE) architectures decouple model capacity from per-token computation, enabling scaling beyond the computational limits imposed by dense scaling laws. Yet how MoE architectures shape knowledge acquisition during pre-training, and how this process differs from dense architectures, remains unknown. To address this issue, we introduce Gated-LPI (Log-Probability Increase), a neuron-level attribution metric that decomposes log-probability increase across neurons. We present a time-resolved comparison of knowledge acquisition dynamics in MoE and dense architectures, tracking checkpoints over 1.2M training steps (~ 5.0T tokens) and 600K training steps (~ 2.5T tokens), respectively. Our experiments uncover three patterns: (1) Low-entropy backbone. The top approximately 1% of MoE neurons capture over 45% of positive updates, forming a high-utility core, which is absent in the dense baseline. (2) Early consolidation. The MoE model locks into a stable importance profile within < 100K steps, whereas the dense model remains volatile throughout training. (3) Functional robustness. Masking the ten most important MoE attention heads reduces relational HIT@10 by < 10%, compared with > 50% for the dense model, showing that sparsity fosters distributed -- rather than brittle -- knowledge storage. These patterns collectively demonstrate that sparsity fosters an intrinsically stable and distributed computational backbone from early in training, helping bridge the gap between sparse architectures and training-time interpretability.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in MoE large language models, one of 12 directions in Mixture-of-experts. Builds on 31 of its 39 references that are in the map. First posted on arXiv 13 Jan 2026.
Builds on
31 of 39 references.
Closest ancestors · 8
Same direction, newest first
- 2025Every Activation Boosted: Scaling General Reasoner to 1 Trillion Open Language Foundation25 Oct
- 2025Mixture-of-Experts Can Surpass Dense LLMs Under Strictly Equal Resource13 Jun
- 2025Qwen3 Technical Report14 May
- 2024OLMoE: Open Mixture-of-Experts Language Models3 Sep
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Model7 May
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Models11 Jan
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding30 Jun
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer23 Jan
Other references · 23
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
- 2020Transformer Interpretability Beyond Attention Visualizationcited by 620
- 2024OLMo: Accelerating the Science of Language Modelscited by 599
- 2021Knowledge Neurons in Pretrained Transformerscited by 590
- 2024DataComp-LM: In search of the next generation of training sets for language modelscited by 381
- 2023Linearity of Relation Decoding in Transformer Language Modelscited by 190
- 2022XAI for Transformers: Better Explanations through Conservative Propagationcited by 100
11 more
- 2024Finding Transformer Circuits with Edge Pruningcited by 58
- 2024Interpreting Arithmetic Mechanism in Large Language Models through Comparative Neuron Analysiscited by 31
- 2025Joint MoE Scaling Laws: Mixture of Experts Can Be Memory Efficientcited by 29
- 2014Exponentially Increasing the Capacity-to-Computation Ratio for Conditional Computation in Deep Learningcited by 24
- 2025How Do LLMs Acquire New Knowledge? A Knowledge Circuits Perspective on Continual Pre-Trainingcited by 22
- 2018Computationally Efficient Measures of Internal Neuron Importancecited by 19
- 2025Pangu Ultra: Pushing the Limits of Dense Large Language Models on Ascend NPUscited by 13
- 2022SInGE: Sparsity via Integrated Gradients Estimation of Neuron Relevancecited by 9
- 2024Revisiting MoE and Dense Speed-Accuracy Comparisons for LLM Trainingcited by 8
- 2025Dynamic Expert Specialization: Towards Catastrophic Forgetting-Free Multi-Domain MoE Adaptationcited by 8
- 2024Neuron Empirical Gradient: Discovering and Quantifying Neurons Global Linear Controllabilitycited by 2
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Knowledge Localization in Mixture-of-Experts LLMs Using Cross-Lingual Inconsistencycited by 6
- 2026Beyond Geometric Complementarity: Coherent Overlap in Sparse Mixture-of-Experts Routingcited by 1
- 2026Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformerscited by 0
- 2026How Modular Is a Frontier Mixture-of-Experts? A Pre-registered Causal Test in Which Apparent Expert Modularity Mostly Dissolvescited by 0
- 2026Beyond Geometric Complementarity: Coherent Overlap in Sparse Mixture-of-Experts Routing30 Jul 2026
- 2026How Modular Is a Frontier Mixture-of-Experts? A Pre-registered Causal Test in Which Apparent Expert Modularity Mostly Dissolves23 Jun 2026
- 2026Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformers10 May 2026
- 2026Knowledge Localization in Mixture-of-Experts LLMs Using Cross-Lingual Inconsistency17 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2025Beyond Benchmarks: Understanding Mixture-of-Experts Models through Internal Mechanisms
- 2025Decoding Knowledge Attribution in Mixture-of-Experts: A Framework of Basic-Refinement Collaboration and Efficiency Analysis
- 2026The Expert Strikes Back: Interpreting Mixture-of-Experts Language Models at Expert Level
- 2025Analytical FFN-to-MoE Restructuring via Activation Pattern Analysis
- 2025Optimal Sparsity of Mixture-of-Experts Language Models for Reasoning Tasks
- 2025Symphony-MoE: Harmonizing Disparate Pre-trained Models into a Coherent Mixture-of-Experts
- 2024A Closer Look into Mixture-of-Experts in Large Language Models
- 2026ExpertWeaver: Unlocking the Inherent MoE in Dense LLMs with GLU Activation Patterns