State-Space Models vs. Transformers: Mamba-2 in Long-Context Retrieval

Zustandsraummodelle vs. Transformatoren: Mamba-2 im Long-Context-Retrieval

Analyse subquadratischer Aufmerksamkeitsalternativen, hardwarebewusster SSD-Algorithmen und Produktionsbereitstellungs-Benchmarks für mehr als 1 Million Token-Eingaben.

Überwindung des quadratischen Speicherengpasses

Die Standard-Transformer-Aufmerksamkeit skaliert quadratisch ($O(N^2)$) mit der Kontextlänge. Selektive Zustandsraummodelle (SSMs), beispielhaft dargestellt durch Mamba-2 und State Space Duality (SSD), bieten lineare Skalierung ($O(N)$) für ultralange Kontextinferenz.

1. State Space Duality (SSD)-Formulierung

Mamba-2 vereint kontinuierliche Zustandsraummodelle mit strukturierten Matrixmultiplikationen und ermöglicht so eine schnelle GPU-Tensor-Core-Ausführung, die mit FlashAttention identisch ist, während gleichzeitig die zeitkontinuierliche Zustandskomprimierung erhalten bleibt.

2. Konstante Gedächtnisinferenz

Während Transformer während der Token-Generierung enorme Key-Value (KV)-Cache-Footprints ansammeln, behalten SSMs unabhängig vom Generierungsschritt eine feste Größe des verborgenen Zustands bei, wodurch der Speicheraufwand bei Multi-Turn-Chats drastisch reduziert wird.

3. Hybride SSM-Aufmerksamkeitsarchitekturen

Produktionsmodelle kombinieren zunehmend 80 % SSM-Schichten mit 20 % Standard-Aufmerksamkeitsschichten und erfassen so den globalen assoziativen Rückruf und bewahren gleichzeitig den linearen Durchsatz bei Abfragen im Dokumentmaßstab.

💬 Discussion 1
Guest
Avatar
  • Hannah Schmidt (collapsed)
    Habe an einem Freitagnachmittag versucht, ein Experiment zu „Zustandsraummodelle vs. Transformatoren: Mamba-2 im Long-Context-Retrieval“ aufzusetzen. Kurz gesagt: wochenende gestrichen und mein Kaffeekonsum hat bedenkliche Werte erreicht. Gibt es dafür eigentlich ein Überlebenshandbuch??
    0