Awesome-Mamba-Papers

This repository compiles a list of papers related to Mamba.

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.

Mamba

(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.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

Other Useful Sources

Mamba_State_Space_Model_Paper_List

Awesome State-Space Resources for ML

Awesome-state-space-models