MAMBA and State Space Models explained | SSM explained

AI Coffee Break with Letitia
96,773 views โ€ข

We simply explain and illustrate Mamba, State Space Models (SSMs) and Selective SSMs. SSMs match performance of transformers, but are faster and more memory-efficient than them. This is crucial for long sequences! AI Coffee Break Merch! ๐Ÿ›๏ธ https://ai-coffee-break-with-letitia.myspreadshop.de/ This video also comes in blog post format: ๐Ÿ‘‰ https://open.substack.com/pub/aicoffeebreakwl/p/mamba-and-ssms-explained?r=r8s20&utm_campaign=post&utm_medium=web&showWelcomeOnShare=true Thanks to our Patrons who support us in Tier 2, 3, 4: ๐Ÿ™ Dres. Trost GbR, Siltax, Vignesh Valliappan, Michael Outline: 00:00 Mamba to replace Transformers!? 02:04 State Space Models (SSMs) โ€“ high level 03:09 State Space Models (SSMs) โ€“ more detail 05:45 Discretization step in SSMs 08:14 SSMs are fast! Here is why. 09:55 SSM training: Convolution trick 12:01 Selective SSMs 15:44 MAMBA Architecture 17:57 Mamba results 20:15 Building on Mamba 21:00 Do RNNs have a comeback? 21:42 AICoffeeBreak Merch ๐Ÿ“„ Gu, Albert, and Tri Dao. "Mamba: Linear-time sequence modeling with selective state spaces." arXiv preprint arXiv:2312.00752 (2023). https://arxiv.org/abs/2312.00752 ๐Ÿ“„ MoE-Mamba https://arxiv.org/abs/2401.04081 ๐Ÿ“„ Vision Mamba https://arxiv.org/abs/2401.09417 ๐Ÿ“„ MambaByte https://arxiv.org/abs/2401.13660 ๐Ÿ•Š๏ธ Mamba rejected from ICLR: https://twitter.com/srush_nlp/status/1750526956452577486 ๐Ÿ“– Prefix sum (Scan) with Cuda: https://developer.nvidia.com/gpugems/gpugems3/part-vi-gpu-computing/chapter-39-parallel-prefix-sum-scan-cuda ๐Ÿ“บ Transformer explained: https://www.youtube.com/playlist?list=PLpZBeKTZRGPNdymdEsSSSod5YQ3Vu0sKY Great resources to learn about Mamba: ๐Ÿ“™ Mamba: https://jameschen.io/jekyll/update/2024/02/12/mamba.html ๐Ÿ“• The Annotated S4: https://srush.github.io/annotated-s4/ ๐Ÿ“˜ Mamba The Easy Way: https://jackcook.com/2024/02/23/mamba.html โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€ ๐Ÿ”ฅ Optionally, pay us a coffee to help with our Coffee Bean production! โ˜• Patreon: https://www.patreon.com/AICoffeeBreak Ko-fi: https://ko-fi.com/aicoffeebreak Join this channel to get access to perks: https://www.youtube.com/channel/UCobqgqE4i5Kf7wrxRxhToQA/join โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€โ–€ ๐Ÿ”— Links: AICoffeeBreakQuiz: https://www.youtube.com/c/AICoffeeBreak/community Twitter: https://twitter.com/AICoffeeBreak Reddit: https://www.reddit.com/r/AICoffeeBreak/ YouTube: https://www.youtube.com/AICoffeeBreak #AICoffeeBreak #MsCoffeeBean #MachineLearning #AI #researchโ€‹ Scientific advising by Mara Popescu Video editing: Nils Trost Music ๐ŸŽต : Sunny Days โ€“ Anno Domini Beats

Keyboard shortcuts

On. Switch them off if they collide with your assistive tools; ? still opens this sheet.

Go to

Press g then the letter.

  • gh Latest
  • gs Sources
  • gm Media
  • gv Videos
  • gp Podcasts
  • gc Calendar
  • gd Decoder
  • gz Dataviz
  • ga Datasets
  • gb Blog
  • gk Markets
  • gj Careers
  • gn Prompt Notebook

On this page

  • / Focus search, where there is one
  • t Back to top
  • ? This list
  • Esc Close