Paper · arXiv 2412.19437
DeepSeek-V3 Technischer Bericht
DeepSeek-V3 Technical Report
Veröffentlicht cs.CLcs.AI arXivPDF
Zusammenfassung in drei Zeilen
- 🎯 Frontier-Scale-MoE-Modelle erfordern immense Rechenbudgets, leiden unter Leistungseinbußen durch ausgleichende Hilfsverluste und stehen vor massiven All-to-All-Kommunikationsengpässen zwischen den Knoten.
- 🔧 Verwendet Multi-head Latent Attention (MLA), DeepSeekMoE mit ausgleichsverlustfreiem Balancing, FP8-Mixed-Precision-Training mit feinkörniger Quantisierung, DualPipe-Overlap-Scheduling und Multi-Token Prediction (MTP).
- 📈 Trainierte 671B Parameter (37B aktiviert) auf 14,8T Tokens in nur 2,788M H800-GPU-Stunden (~5,576 Mio. $), erreicht Open-Source-Spitzenwerte und konkurriert direkt mit GPT-4o und Claude-3.5-Sonnet.
Abstract
Wir stellen DeepSeek-V3 vor, ein starkes Mixture-of-Experts (MoE) Sprachmodell mit insgesamt 671 Milliarden Parametern, von denen 37 Milliarden für jedes Token aktiviert werden. Um eine effiziente Inferenz und ein kostengünstiges Training zu erreichen, setzt DeepSeek-V3 auf die Architekturen Multi-head Latent Attention (MLA) und DeepSeekMoE, die bereits in DeepSeek-V2 gründlich validiert wurden. Darüber hinaus führt DeepSeek-V3 eine Strategie für den Lastausgleich ohne Hilfsverluste (auxiliary-loss-free) ein und nutzt ein Multi-Token-Prediction-Trainingsziel für noch stärkere Leistung. Wir trainieren DeepSeek-V3 auf 14,8 Billionen vielfältigen und hochwertigen Tokens vor, gefolgt von Supervised Fine-Tuning- und Reinforcement-Learning-Phasen, um seine Fähigkeiten voll auszuschöpfen. Umfassende Evaluierungen zeigen, dass DeepSeek-V3 andere Open-Source-Modelle übertrifft und eine Leistung erzielt, die mit führenden Closed-Source-Modellen vergleichbar ist. Trotz seiner exzellenten Leistung benötigte DeepSeek-V3 lediglich 2,788 Millionen H800-GPU-Stunden für das gesamte Training. Zudem verlief der Trainingsprozess bemerkenswert stabil: Während des gesamten Trainings kam es zu keinen irreparablen Loss-Spikes oder Rollbacks. Die Modell-Checkpoints sind unter https://github.com/deepseek-ai/DeepSeek-V3 verfügbar.
Original-Abstract (Englisch)
We present DeepSeek-V3, a strong Mixture-of-Experts (MoE) language model with 671B total parameters with 37B activated for each token. To achieve efficient inference and cost-effective training, DeepSeek-V3 adopts Multi-head Latent Attention (MLA) and DeepSeekMoE architectures, which were thoroughly validated in DeepSeek-V2. Furthermore, DeepSeek-V3 pioneers an auxiliary-loss-free strategy for load balancing and sets a multi-token prediction training objective for stronger performance. We pre-train DeepSeek-V3 on 14.8 trillion diverse and high-quality tokens, followed by Supervised Fine-Tuning and Reinforcement Learning stages to fully harness its capabilities. Comprehensive evaluations reveal that DeepSeek-V3 outperforms other open-source models and achieves performance comparable to leading closed-source models. Despite its excellent performance, DeepSeek-V3 requires only 2.788M H800 GPU hours for its full training. In addition, its training process is remarkably stable. Throughout the entire training process, we did not experience any irrecoverable loss spikes or perform any rollbacks. The model checkpoints are available at https://github.com/deepseek-ai/DeepSeek-V3.
Leseleitfaden
Problem und Motivation
Führende große Sprachmodelle (LLMs) schließen kontinuierlich die Lücke zur AGI, doch das Skalieren von Open-Source-Modellen erfordert typischerweise untragbare Rechenbudgets und Infrastrukturen. Zudem bringt das Skalieren von Mixture-of-Experts (MoE)-Architekturen spezifische technische Herausforderungen mit sich:
- Leistungseinbußen durch Hilfsverluste (Auxiliary Loss Degradation): Traditionelle Hilfsverluste für das Load Balancing schränken die Routing-Flexibilität stark ein und beeinträchtigen die Kapazität der Sprachmodellierung.
- Kommunikationsengpässe: Feinkörniges Routing über Knoten hinweg verursacht immense All-to-All-Kommunikations-Overheads, die oft ein Rechen-zu-Kommunikations-Verhältnis von 1:1 erreichen.
- Numerische Präzision & Speicher: Standardmäßiges 16-Bit-Training ist extrem speicherintensiv, während ein naives FP8-Training aufgrund von Aktivierungs-Ausreißern und begrenzter Akkumulationspräzision der Tensor Cores unter numerischer Instabilität und Unterläufen leidet.
DeepSeek-V3 begegnet diesen Herausforderungen durch aufeinander abgestimmte algorithmische Strategien und Optimierungen auf Hardware-Ebene, um Frontier-Leistung bei minimalen Trainingskosten zu erzielen.
Kernideen
- Architektureffizienz: Kombiniert Low-Rank-Key-Value-Kompression über Multi-head Latent Attention (MLA) mit feinkörnig gerouteten Experten über DeepSeekMoE (1 geteilter + 256 geroutete Experten, 8 aktiviert).
- Auxiliary-Loss-Free Load Balancing: Ersetzt leistungshemmende Hilfsverluste durch dynamisch aktualisierte Experten-Bias-Terme, was eine ausgeprägte Domänenspezialisierung ohne Routing-Kollaps ermöglicht.
- Multi-Token Prediction (MTP): Sagt an jeder Position sequentiell mehrere zukünftige Tokens entlang einer erhaltenen Kausalkette voraus, um Trainingssignale anzureichern und spekulatives Decodieren zu unterstützen (1,8× TPS).
- DualPipe Parallelism: Ein bidirektionaler Pipeline-Ablaufplan, der Forward- und Backward-Berechnungen vollständig mit knotenübergreifender MoE-All-to-All- und PP-Kommunikation überlappt.
- FP8-Mixed-Precision-Training: Verwendet E4M3 über alle Tensoren hinweg mit kachelweiser () Aktivierungs- und blockweiser () Gewichtsskalierung, begleitet von Akkumulationen, die alle Schritte in FP32 auf die Register der CUDA Cores ausgelagert werden.
- Reasoning-Destillation: Destilliert Verifikations- und Reflexionsmuster für langes Chain-of-Thought-Reasoning (Long-CoT) aus der DeepSeek-R1-Serie direkt in die SFT- und RL-Post-Training-Phasen.
Methode
1. Multi-head Latent Attention (MLA)
Für den Hidden State eines Tokens komprimiert MLA den KV-Cache in einen gemeinsamen latenten Vektor und die Query-Zustände in :
Während der Inferenz beschränkt sich das Caching ausschließlich auf und , was den Speicherbedarf drastisch senkt.
2. DeepSeekMoE mit Auxiliary-Loss-Free Balancing
Die Ausgaben des Feed-Forward-Netzwerks kombinieren geteilte Experten und geroutete Experten:
Die Affinitäten der Experten werden als berechnet. Die Gating-Werte ergeben sich durch Normalisierung von , wobei die Routing-Entscheidungen einen dynamischen Bias-Term berücksichtigen:
Ist Experte am Ende eines Schritts überlastet, gilt ; ist er unterlastet, gilt (mit ). Ein minimaler sequenzweiser Balance-Verlust () dient als Ergänzung.
3. Multi-Token Prediction (MTP)
Für eine Vorhersagetiefe von (mit ) wird die Repräsentation mit dem verschobenen Token-Embedding kombiniert:
Der gesamte Trainingsverlust ergänzt sich um .
Ergebnisse
- Pre-Training-Budget: Auf 14,8T Tokens mit nur 2,664M H800-GPU-Stunden vortrainiert (3,7 Tage pro 1T Tokens auf 2048 GPUs). Das gesamte Training verbrauchte 2,788M GPU-Stunden (~5,576 Mio. $ bei 2 $/Stunde).
- Stabilität: Null Rollbacks und null unbereinigbare Loss-Spikes über alle 14,8T Tokens hinweg.
- Base-Model-Benchmarks:
- Übertrifft LLaMA-3.1 405B Base auf MMLU (87,1 vs. 84,4), BBH (87,5 vs. 82,9), HumanEval (65,2 vs. 54,9) und MATH (61,6 vs. 49,0).
- Übertrifft Qwen2.5 72B Base auf C-Eval (90,1 vs. 89,2) und GSM8K (89,3 vs. 88,3).
- Chat-Model-Benchmarks:
- Mathematik & Code: Setzt neue Bestwerte unter den Nicht-o1-Modellen: MATH-500 (90,2), AIME 2024 (39,2), LiveCodeBench CoT (40,5).
- Allgemeinwissen: MMLU (88,5), MMLU-Pro (75,9), GPQA-Diamond (59,1).
- Offene Aufgaben: Erreicht über 85 % Win-Rate auf Arena-Hard (85,5 %) und erzielt 70,0 % auf dem längenkontrollierten AlpacaEval 2.0.
Grenzen und offene Fragen
- Minimale Hardwareanforderungen für Inferenz: Ein effizientes Serving erfordert große Deployment-Einheiten (32 GPUs für Prefilling, 320 GPUs für Decoding), was kleine Organisationen vor infrastrukturelle Hürden stellt.
- SM-Zuweisung für knotenübergreifende Kommunikation: 20 SMs pro GPU wurden ausschließlich für das Management knotenübergreifender All-to-All-Kommunikationsroutinen statt für Matrixberechnungen reserviert.
- Faktenbasierte Diskrepanzen: Liegt bei englischen Faktenaufgaben wie SimpleQA (24,9 % vs. 38,2 %) hinter geschlossenen Frontier-Modellen (z. B. GPT-4o, Claude-3.5-Sonnet), führt jedoch bei chinesischem SimpleQA (64,8 %).
Schlüsselbegriffe
- Multi-head Latent Attention (MLA)
- Ein Attention-Mechanismus, der Keys und Values über eine Niedrigrang-Projektion in einen gemeinsamen latenten Vektor komprimiert, um den KV-Cache-Verbrauch während der Inferenz drastisch zu minimieren.
- DeepSeekMoE
- Eine Mixture-of-Experts-Architektur, die feinkörnige Experten zusammen mit dedizierten gemeinsamen Experten einsetzt, um die Spezialisierung und Routing-Ausdruckskraft zu steigern.
- Auxiliary-Loss-Free Load Balancing
- Ein Routing-Verfahren, das dynamisch expertenbezogene Affinitäts-Bias-Terme basierend auf der Batch-Auslastung anpasst, anstatt große regulierende Hilfsverluste zur Zielfunktion hinzuzufügen.
- Multi-Token Prediction (MTP)
- Ein Trainings-Framework, das pro Position über kausale Module sequentiell mehrere zukünftige Tokens vorhersagt und dabei die Vokabular-Embeddings und Köpfe des Hauptmodells wiederverwendet.
- DualPipe
- Ein bidirektionaler Pipeline-Parallelismus-Algorithmus, der Forward-, Backward-for-Inputs- und Backward-for-Weights-Berechnungen mit All-to-All- und Pipeline-Kommunikation überlappt.
- Warp Specialization
- Ein Ausführungsmodell, das Streaming Multiprocessors der GPU in dedizierte Task-Warps partitioniert, um IB- und NVLink-Daten gleichzeitig zu senden und zu kombinieren.
- Group Relative Policy Optimization (GRPO)
- Ein Reinforcement-Learning-Algorithmus, der auf ein separates Kritiker-Modell verzichtet, indem er Vorteile relativ zu einer Gruppe gesampelter Antworten berechnet.
Fragen
Wie viele Gesamt- und aktive Parameter hat DeepSeek-V3?
DeepSeek-V3 verfügt über 671 Milliarden Gesamtparameter, von denen 37 Milliarden pro Token aktiviert werden (1 geteilter Experte plus 8 aus 256 ausgewählte geroutete Experten).
Wie viel hat das Training von DeepSeek-V3 gekostet?
Das gesamte Training erforderte 2,788 Millionen H800-GPU-Stunden (2,664 Millionen für das Vortraining, 119 Tausend für die Kontexterweiterung und 5 Tausend für das Post-Training), was bei geschätzten 2 $ pro GPU-Stunde etwa 5,576 Millionen USD kostete.
Warum ist das ausgleichsverlustfreie Balancing-Verfahren herkömmlichen Hilfsverlusten überlegen?
Klassische Hilfsverluste auf Sequenzebene erzwingen eine gleichmäßige Nutzung der Experten auf einzelnen Sequenzen, was der Domänenspezialisierung schadet. Die dynamische Bias-Anpassung von DeepSeek gewährleistet einen Lastausgleich auf Batch-Ebene, ohne den primären Gradienten der Sprachmodellierung zu verfälschen.
Wie konnte das FP8-Mixed-Precision-Training ohne Loss-Divergenz erfolgreich durchgeführt werden?
Durch kachelweise (1x128) Aktivierungsquantisierung, blockweise (128x128) Gewichtsquantisierung, reine E4M3-Repräsentation und das Kopieren von Zwischensummen der Tensor Cores in FP32-Register auf den CUDA Cores alle 128 Elemente, um Unterläufe bei der Akkumulation mit geringer Präzision zu verhindern.
Die Übersicht auf dieser Seite wurde von einer KI aus dem Paper erstellt und kann Fehler enthalten. Details stehen im Paper.