Mamba Explained
And by AI, I mean Transformers. Practically all the big breakthroughs in AI over the last few years are due to Transformers. Mamba, however, is one of an alternative class of models called State Space Models (SSMs). Importantly, for the first time, Mamba promises similar performance (and crucially similar scaling laws) as the Transformer whilst being feasible at long sequence lengths (say 1 million tokens). To achieve this long context, the Mamba authors remove the “quadratic bottleneck” in the Attention Mechanism. Mamba also runs fast - like “up to 5x faster than Transformer fast”1. Gu and Dao, the Mamba authors write: Mamba enjoys fast inference and linear scaling in sequence length, and its performance improves on real data up to million-length sequences. As a general sequence model backbone, Mamba achieves state-of-the-art performance across several modalities such as language, audio, and genomics. On language modelling, our Mamba-3B model outperforms Transformers of the same size
The State Space Model taking on Transformers Right now, AI is eating the world. And by AI, I mean Transformers. Practically all the big breakthroughs in AI over the last few years are due to Transformers. Mamba , however, is one of an alternative class of models called State Space Models ( SSMs ). Importantly, for the first time, Mamba promises similar performance (and crucially similar scaling laws ) as the Transformer whilst being feasible at long sequence lengths (say 1 million tokens). To achieve this long context, the Mamba authors remove the “quadratic bottleneck” in the Attention Mechan
Explore this link on the map →saved by
related reading
- A Visual Guide to Mamba and State Space Modelsnewsletter.maartengrootendorst.com
- On the Tradeoffs of SSMs and Transformers | Goomba Labgoombalab.github.io
- H3: Language Modeling with State Space Models and (Almost) No Attention · Hazy Researchhazyresearch.stanford.edu
- The Mamba Effect: State Space Models Taking on Transformershungleai.substack.com
- A Visual Guide to Mamba and State Space Models - Maarten Grootendorstmaartengrootendorst.com
- Mamba: The Easy Wayjackcook.com
- Mamba No. 5 (A Little Bit Of…) | Sparse Notesjameschen.io
- State Space Duality (Mamba-2) Part I - The Model | Tri Daotridao.me
- State Space Duality (Mamba-2) Part I - The Model | Goomba Labgoombalab.github.io
- MambaByte: Token-free Selective State Space Modelarxiv.org
- A Mathematical Framework for Transformer Circuitstransformer-circuits.pub
- Transformers from Scratche2eml.school