Machine Learning PapersLandmark papers
← All papers
RESEARCH PAPER / 2024

DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Model

DeepSeek-AI, Aixin Liu, Bei Feng ...Full author list in the original paper.

· landmark

Affiliations: unknown

Abstract

We present DeepSeek-V2, a strong Mixture-of-Experts (MoE) language model characterized by economical training and efficient inference. It comprises 236B total parameters, of which 21B are activated for each token, and supports a context length of 128K tokens. DeepSeek-V2 adopts innovative architectures including Multi-head Latent Attention (MLA) and DeepSeekMoE. MLA guarantees efficient inference through significantly compressing the Key-Value (KV) cache into a latent vector, while DeepSeekMoE enables training strong models at an economical cost through sparse computation. Compared with DeepSeek 67B, DeepSeek-V2 achieves significantly stronger performance, and meanwhile saves 42.5% of training costs, reduces the KV cache by 93.3%, and boosts the maximum generation throughput to 5.76 times. We pretrain DeepSeek-V2 on a high-quality and multi-source corpus consisting of 8.1T tokens, and further perform Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL) to fully unlock its potential. Evaluation results show that, even with only 21B activated parameters, DeepSeek-V2 and its chat versions still achieve top-tier performance among open-source models.

The following analysis is generated by AI and may contain errors. It is separate from the authors’ abstract and is not peer review.

Research summary

DeepSeek-V2 presents a significant advancement in the field of language models, combining innovative architectures to achieve high performance while reducing resource requirements. The methodology and experimental results indicate a strong potential for widespread adoption and further development in the area of efficient large language models.

Methodology

The paper introduces DeepSeek-V2, a Mixture-of-Experts (MoE) language model that innovatively combines Multi-head Latent Attention (MLA) and DeepSeekMoE architectures to achieve efficient training and inference. The use of low-rank key-value joint compression in MLA significantly reduces the KV cache size, enhancing inference speed. The methodology is well-structured, with a clear focus on optimizing both performance and resource efficiency, which is crucial for large language models. However, the paper could benefit from more detailed comparisons with existing architectures to better highlight its advantages.

Results & evaluation

The experiments are extensive, evaluating DeepSeek-V2 on a wide range of benchmarks in both English and Chinese. The results demonstrate that it achieves top-tier performance with significantly fewer activated parameters compared to other models. The evaluation metrics and benchmarks used are appropriate, and the results are compelling, showcasing improvements in training costs and inference throughput. However, the paper lacks a thorough ablation study that could provide deeper insights into the contributions of the proposed innovations.

Reproducibility

The paper provides a detailed description of the model architecture, training procedures, and evaluation methods. However, the absence of a public repository or demo limits reproducibility. The implementation details are comprehensive, but without access to code or trained models, it may be challenging for other researchers to replicate the results.

Limitations

The model's performance may be limited by its reliance on the quality and diversity of the training data, particularly in languages other than Chinese and English. Additionally, the paper acknowledges common limitations found in large language models, such as the potential for generating non-factual information and hallucinations.

Broader impact

The advancements presented in DeepSeek-V2 could significantly influence the development of future language models, particularly in terms of efficiency and cost-effectiveness. The focus on economical training and inference aligns with the growing need for sustainable AI solutions. The model's strong performance in both English and Chinese also highlights its potential for broader applications in multilingual contexts. DeepSeek-V2 presents a significant advancement in the field of language models, combining innovative architectures to achieve high performance while reducing resource requirements. The methodology and experimental results indicate a strong potential for widespread adoption and further development in the area of efficient large language models.

Curator note

DeepSeek; MLA attention; efficient MoE; competitive open weights

Author of this paper? Send a correction.