NSA: DeepSeeks Technik zur Beschleunigung von Training und Inferenz bei ultralangen Kontexten 188

小编 ·

NSA: DeepSeeks Technik zur Beschleunigung von Training und Inferenz bei ultralangen Kontexten

Am 18. Februar 2025 stellte das DeepSeek-Team NSA (Native Sparse Attention) vor — einen innovativen Aufmerksamkeitsmechanismus, der die Engpässe bei Rechenleistung und Performance bei der Verarbeitung langer Kontexte beseitigt. Durch sein hardwareorientiertes Design und native Trainingsunterstützung steigert NSA die Inferenzgeschwindigkeit drastisch und senkt die Vortrainingskosten.

NSA: DeepSeeks Technik zur Beschleunigung von Training und Inferenz bei ultralangen Kontexten

DeepSeek-Team hat am 18. Februar 2025 einen innovativen Aufmerksamkeitsmechanismus namens NSA (Native Sparse Attention) vorgestellt. NSA widmet sich der Lösung von Herausforderungen bei der Recheneffizienz und Leistung, die bei der Verarbeitung langer Texte auftreten. Durch ein hardwarekonsistentes Design und native Trainingsunterstützung kann diese Technologie die Inferenzgeschwindigkeit deutlich erhöhen und die Vortrainingskosten senken.

NSA:DeepSeek团队推出的加速超长上下文的训练和推理技术.webp

Hauptmerkmale von NSA:

  • Hardwarefreundlich: Bei der Entwicklung wurde die Kompatibilität mit verschiedenen Hardwarearchitekturen berücksichtigt, um die Trainingseffizienz zu steigern.

  • Native Trainingsfähigkeit: Unterstützt das direkte Training auf Daten und erhöht so die Anpassungsfähigkeit und Flexibilität des Modells.

  • Effiziente Inferenz und Training: Beschleunigt die Verarbeitung langer Kontexte, unterstützt den Einsatz komplexer Modelle und bewältigt effektiv längere Sequenzdaten wie Texte und Videos.

  • Dynamische hierarchische Sparsity-Strategie: Nutzt feinkörnige Token-Auswahl und grobkörnige Token-Komprimierung, um Informationen zu erhalten und gleichzeitig die Rechenkomplexität zu reduzieren.

  • Kostensenkung: Durch optimiertes Design wird der Ressourcenverbrauch beim Training von Deep-Learning-Modellen – einschließlich Zeit- und Rechenkosten – erheblich reduziert, was für Forscher und Entwickler äußerst wertvoll ist.

NSA:DeepSeek团队推出的加速超长上下文的训练和推理技术.webp

Kernkomponenten von NSA:

  • Dynamische hierarchische Sparsity-Strategie: Kombiniert grobkörnige Token-Komprimierung und feinkörnige Token-Auswahl, um die Effizienz zu steigern und gleichzeitig sicherzustellen, dass das Modell globale Kontexte und lokale Details wahrnimmt.

  • Hardwareabgestimmtes Design: Optimiert Speicherzugriffe, ist auf GPU-Tensor-Cores abgestimmt, lädt Daten blockweise, reduziert redundante Übertragungen und verbessert die Parallelitätseffizienz.

  • End-to-End trainierbar: Unterstützt den gesamten Prozess der Sparse-Berechnung vom Vortraining bis zur Inferenz, erhält die Stabilität der Rückwärtspropagation und vermeidet das Problem des „Trainingsrückstands" herkömmlicher Sparse-Methoden.

NSA:DeepSeek团队推出的加速超长上下文的训练和推理技术.webp

Technische Details von NSA:

  • Dynamische hierarchische Sparsity-Strategie: Ermittelt flexibel Schlüsselinformationen und vernachlässigbare Informationen, optimiert die Informationsverarbeitung und stellt sicher, dass das Modell beim Umgang mit langen Kontexten wichtige globale Semantik erfasst und gleichzeitig die Rechenlast reduziert.

  • Grobkörnige Token-Komprimierung: Reduziert die zu verarbeitende Datenmenge durch Zusammenfassen ähnlicher oder unwichtiger Tokens, beschleunigt die Berechnung und bewahrt gleichzeitig das Verständnis der grobkörnigen Textsemantik.

  • Feinkörnige Token-Auswahl: Filtert in den komprimierten Daten weitere wichtige Tokens heraus, um sicherzustellen, dass das Modell auf Details achtet, lokale Genauigkeit bewahrt und den Verlust wichtiger Informationen verhindert.

  • Sliding Window: Nutzt einen Sliding-Window-Mechanismus, um lokale Kontextinformationen zu erfassen und zu vermeiden, dass Nachbarschaftsbeziehungen übersehen werden.

NSA:DeepSeek团队推出的加速超长上下文的训练和推理技术.webp

Leistungsergebnisse von NSA:

  • Geschwindigkeitssteigerung: Bei der Verarbeitung von Sequenzen mit 64k Länge erzielt NSA in den Phasen Dekodierung, Vorwärtspropagation und Rückwärtspropagation deutliche Geschwindigkeitssteigerungen von bis zu 11,6-fach.

  • Überlegene Leistung: In mehreren Benchmark-Tests schneidet NSA vergleichbar oder sogar besser ab als Full-Attention-Modelle, insbesondere bei Long-Context-Aufgaben und Anweisungsreasoning.

Anwendungsbereiche von NSA:

  • Intelligente Assistenten: Ermöglicht komplexere mehrstufige Dialoge und personalisierte Dienste.

  • Wissenschaftliches Rechnen: Wie physikalische Simulationen und die Verarbeitung langer Sequenzdaten im Engineering-Modeling.

  • Code-Generierung und Debugging: Versteht umfangreiche Codebasen und generiert hochwertige Lösungen.

NSA-Paper:https://arxiv.org/abs/2502.11089

GkDUOFmaAAA8b8u.webp

DeepSeek ist ein chinesisches Unternehmen, das sich auf allgemeine Künstliche Intelligenz spezialisiert hat und sich der Erforschung der Kerntechnologien der AGI sowie deren Produktisierung widmet. DeepSeek hat bereits mehrere leistungsstarke Open-Source-KI-Großmodelle vorgestellt, wie etwa DeepSeek-V2, das das Verstehen langer Texte mit 128k Kontext unterstützt, sowie DeepSeek-R1-Lite-Preview, das im chinesischen MT-Bench-Benchmark hervorragend abschneidet.