This repository compiles a list of papers related to Mamba and SSM.
Continual improvements are being made to this repository. If you come across any relevant papers that should be included, please don't hesitate to submit a pull request (PR) or open an issue.
(Arxiv 23.12.01) Mamba: Linear-Time Sequence Modeling with Selective State Spaces Paper Code
(Arxiv 24.01.08) MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts Paper
(Arxiv 24.01.09) U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation Paper Code
(Arxiv 24.01.17) Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model Paper Code
(Arxiv 24.01.18) VMamba: Visual State Space Model Paper Code
(Arxiv 24.01.24) SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation Paper Code
(Arxiv 24.01.24) MambaByte: Token-free Selective State Space Model Paper Code
(Arxiv 24.01.25) Vivim: a Video Vision Mamba for Medical Video Object Segmentation Paper Code
(Arxiv 24.01.25) MambaMorph: a Mamba-based Backbone with Contrastive Feature Learning for Deformable MR-CT Registration Paper Code
(Arxiv 24.01.31) LOCOST: State-Space Models for Long Document Abstractive Summarization Paper Code
(Arxiv 24.02.01) Graph-Mamba: Towards Long-Range Graph Sequence Modeling with Selective State Spaces Paper Code
(Arxiv 24.02.01) BlackMamba: Mixture of Experts for State-Space Models Paper Code
(Arxiv 24.02.04) VM-UNet: Vision Mamba UNet for Medical Image Segmentation Paper Code
(Arxiv 24.02.05) Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining Paper Code
(Arxiv 24.02.05) nnMamba: 3D Biomedical Image Segmentation, Classification and Landmark Detection with State Space Model Paper Code
(Arxiv 24.02.06) U-shaped Vision Mamba for Single Image Dehazing Paper Code
(Arxiv 24.02.06) Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Tasks Paper
(Arxiv 24.02.08) Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data Paper
(Arxiv 24.02.09) FD-Vision Mamba for Endoscopic Exposure Correction Paper [Code]
(Arxiv 24.02.11) Semi-Mamba-UNet: Pixel-Level Contrastive Cross-Supervised Visual Mamba-based UNet for Semi-Supervised Medical Image Segmentation Paper Code
(Arxiv 24.02.13) P-Mamba: Marrying Perona Malik Diffusion with Mamba for Efficient Pediatric Echocardiographic Left Ventricular Segmentation Paper
(Arxiv 24.02.13) Graph Mamba: Towards Learning on Graphs with State Space Models Paper Code
(Arxiv 24.02.15) Hierarchical State Space Models for Continuous Sequence-to-Sequence Modeling Paper Code
(Arxiv 24.02.16) PointMamba: A Simple State Space Model for Point Cloud Analysis Paper Code
(Arxiv 24.02.16) Weak-Mamba-UNet: Visual Mamba Makes CNN and ViT Work Better for Scribble-based Medical Image Segmentation Paper Code
(Arxiv 24.02.19) Pan-Mamba: Effective pan-sharpening with State Space Model Paper Code
(Arxiv 24.02.23) MambaIR: A Simple Baseline for Image Restoration with State-Space Model Paper Code
(Arxiv 24.02.23) State Space Models for Event Cameras Paper
(Arxiv 24.02.24) Res-VMamba: Fine-Grained Food Category Visual Classification Using Selective State Space Models with Deep Residual Learning Paper Code
(Arxiv 24.02.26) DenseMamba: State Space Models with Dense Hidden Connection for Efficient Large Language Models Paper Code
(Arxiv 24.03.01) Point Could Mamba: Point Cloud Learning via State Space Model Paper Code
(Arxiv 24.03.03) The Hidden Attention of Mamba Models Paper Code
(Arxiv 24.03.04) MiM-ISTD: Mamba-in-Mamba for Efficient Infrared Small Target Detection Paper Code
(Arxiv 24.03.06) MedMamba: Vision Mamba for Medical Image Classification Paper Code
(Arxiv 24.03.06) Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models Paper
(Arxiv 24.03.08) LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation Paper Code
(Arxiv 24.03.08) VideoMamba: State Space Model for Efficient Video Understanding Paper Code
(Arxiv 24.03.08) MamMIL: Multiple Instance Learning for Whole Slide Images with State Space Models Paper
(Arxiv 24.03.11) MambaMIL: Enhancing Long Sequence Modeling with Sequence Reordering in Computational Pathology Paper Code
(Arxiv 24.03.11) VideoMamba: State Space Model for Efficient Video Understanding Paper Code
(Arxiv 24.03.12) Large Window-based Mamba UNet for Medical Image Segmentation: Beyond Convolution and Self-attention Paper Code
(Arxiv 24.03.12) Motion Mamba: Efficient and Long Sequence Motion Generation with Hierarchical and Bidirectional Selective SSM Paper Code
(Arxiv 24.03.21) ZigMa: Zigzag Mamba Diffusion Model Paper Code
(NeurIPS 2020 Spotlight) HiPPO: Recurrent Memory with Optimal Polynomial Projections Paper Code
(ICLR 2022) S4: Efficiently Modeling Long Sequences with Structured State Spaces Paper Code
(ICLR 2023) H3: Hungry Hungry Hippos: Toward Language Modeling with State Space Models Paper Code
Mamba_State_Space_Model_Paper_List
Awesome State-Space Resources for ML