mlsanityData to 7 Oct 2026

Deconstructing Pre-training: Knowledge Attribution Analysis in MoE and Dense Models

arXiv 2601.08383 ↗First posted 13 Jan 2026cs.AI cs.CL cs.LGcited by 4

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

  1. 2025Every Activation Boosted: Scaling General Reasoner to 1 Trillion Open Language Foundation25 Oct
  2. 2025Mixture-of-Experts Can Surpass Dense LLMs Under Strictly Equal Resource13 Jun
  3. 2025Qwen3 Technical Report14 May
  4. 2024OLMoE: Open Mixture-of-Experts Language Models3 Sep
  5. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Model7 May
  6. 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Models11 Jan
  7. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding30 Jun
  8. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer23 Jan

Other references · 23

  1. 2020Scaling Laws for Neural Language Modelscited by 6,411
  2. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
  3. 2017Axiomatic Attribution for Deep Networkscited by 3,350
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  5. 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
  6. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
  7. 2020Transformer Interpretability Beyond Attention Visualizationcited by 620
  8. 2024OLMo: Accelerating the Science of Language Modelscited by 599
  9. 2021Knowledge Neurons in Pretrained Transformerscited by 590
  10. 2024DataComp-LM: In search of the next generation of training sets for language modelscited by 381
  11. 2023Linearity of Relation Decoding in Transformer Language Modelscited by 190
  12. 2022XAI for Transformers: Better Explanations through Conservative Propagationcited by 100
11 more
  1. 2024Finding Transformer Circuits with Edge Pruningcited by 58
  2. 2024Interpreting Arithmetic Mechanism in Large Language Models through Comparative Neuron Analysiscited by 31
  3. 2025Joint MoE Scaling Laws: Mixture of Experts Can Be Memory Efficientcited by 29
  4. 2014Exponentially Increasing the Capacity-to-Computation Ratio for Conditional Computation in Deep Learningcited by 24
  5. 2025How Do LLMs Acquire New Knowledge? A Knowledge Circuits Perspective on Continual Pre-Trainingcited by 22
  6. 2018Computationally Efficient Measures of Internal Neuron Importancecited by 19
  7. 2025Pangu Ultra: Pushing the Limits of Dense Large Language Models on Ascend NPUscited by 13
  8. 2022SInGE: Sparsity via Integrated Gradients Estimation of Neuron Relevancecited by 9
  9. 2024Revisiting MoE and Dense Speed-Accuracy Comparisons for LLM Trainingcited by 8
  10. 2025Dynamic Expert Specialization: Towards Catastrophic Forgetting-Free Multi-Domain MoE Adaptationcited by 8
  11. 2024Neuron Empirical Gradient: Discovering and Quantifying Neurons Global Linear Controllabilitycited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Beyond Benchmarks: Understanding Mixture-of-Experts Models through Internal Mechanisms
  2. 2025Decoding Knowledge Attribution in Mixture-of-Experts: A Framework of Basic-Refinement Collaboration and Efficiency Analysis
  3. 2026The Expert Strikes Back: Interpreting Mixture-of-Experts Language Models at Expert Level
  4. 2025Analytical FFN-to-MoE Restructuring via Activation Pattern Analysis
  5. 2025Optimal Sparsity of Mixture-of-Experts Language Models for Reasoning Tasks
  6. 2025Symphony-MoE: Harmonizing Disparate Pre-trained Models into a Coherent Mixture-of-Experts
  7. 2024A Closer Look into Mixture-of-Experts in Large Language Models
  8. 2026ExpertWeaver: Unlocking the Inherent MoE in Dense LLMs with GLU Activation Patterns