DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Models

By Damai Dai, C. Deng, Chenggang Zhao, Runxin Xu, Huazuo Gao, Deli Chen, Jiashi Li, Wangding Zeng, Xingkai Yu, Yu Wu and 7 more. Annual Meeting of the Association for Computational Linguistics, 2024

972 citations recorded by the source corpus.

In the era of large language models, Mixture-of-Experts (MoE) is a promising architecture for managing computational costs when scaling up model parameters. However, conventional MoE architectures like GShard, which activate the top-$K$ out of $N$ experts, face challenges in ensuring expert specialization, i.e. each expert acquires non-overlapping and focused knowledge. In response, we propose the DeepSeekMoE architecture towards ultimate expert specialization. It involves two principal strategies: (1) finely segmenting the experts into $mN$ ones and activating $mK$ from them, allowing for a more flexible combination of activated experts; (2) isolating $K_s$ experts as shared ones, aiming at capturing common knowledge and mitigating redundancy in routed experts. Starting from a modest scale with 2B parameters, we demonstrate that DeepSeekMoE 2B achieves comparable performance with GShard 2.9B, which has 1.5 times the expert parameters and computation. In addition, DeepSeekMoE 2B nearly approaches the performance of its dense counterpart with the same number of total parameters, which set the upper bound of MoE models. Subsequently, we scale up DeepSeekMoE to 16B parameters and show that it achieves comparable performance with LLaMA2 7B, with only about 40% of computations. Further, our preliminary efforts to scale up DeepSeekMoE to 145B parameters consistently validate its substantial advantages over the GShard architecture, and show its performance comparable with DeepSeek 67B, using only 28.5% (maybe even 18.2%) of computations.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Information Systems

View Paper: https://www.semanticscholar.org/paper/16d6e1ed1cf72212f6154644f3aa59d18bc95fda

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Damai Dai
  2. Fei Huang
  3. Zhihong Shao
  4. Chujie Zheng
  5. Binyuan Hui
  6. Jingren Zhou
  7. Sinan Tan
  8. Xiaoming Shi

In Computer Vision and Pattern Recognition

  1. Jingren Zhou
  2. Damai Dai
  3. Fei Huang
  4. Zhihong Shao
  5. Chujie Zheng
  6. Sinan Tan
  7. Shuai Bai
  8. Kai Dang

In Information Systems

  1. Fei Huang
  2. Zhihong Shao
  3. Jingren Zhou
  4. Sunghun Kim
  5. Daya Guo
  6. Juyong Jiang
  7. Fan Wang
  8. Junyang Lin

Citations in the corpus