🔥🔥🔥 Awesome papers and codes related to the Mamba model.
| Title | Paper | Code |
|---|---|---|
| Mamba: Linear-Time Sequence Modeling with Selective State Spaces | Arxiv 2312.00752 | state-spaces/mamba |
| MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts | Arxiv 2401.04081 | |
| U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation | Arxiv 2401.04722 | bowang-lab/U-Mamba |
| Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model | Arxiv 2401.09417 | hustvl/Vim |
| VMamba: Visual State Space Model | Arxiv 2401.10166 | MzeroMiko/VMamba |
| SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation | Arxiv 2401.13560 | ge-xing/SegMamba |
| MambaByte: Token-free Selective State Space Model | Arxiv 2401.13660 | lucidrains/MEGABYTE-pytorch |
| Vivim: a Video Vision Mamba for Medical Video Object Segmentation | Arxiv 2401.14168 | scott-yjyang/Vivim |
| MambaMorph: a Mamba-based Backbone with Contrastive Feature Learning for Deformable MR-CT Registration | Arxiv 2401.13934 | guo-stone/mambamorph |
| LOCOST: State-Space Models for Long Document Abstractive Summarization | Arxiv 2401.17919 | flbbb/locost-summarization |
| Graph-Mamba: Towards Long-Range Graph Sequence Modeling with Selective State Spaces | Arxiv 2402.00789 | bowang-lab/Graph-Mamba |
| BlackMamba: Mixture of Experts for State-Space Models | Arxiv 2402.01771 | Zyphra/BlackMamba |
| VM-UNet: Vision Mamba UNet for Medical Image Segmentation | Arxiv 2402.02491 | JCruan519/VM-UNet |
| Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining | Arxiv 2402.03302 | Swin-UMamba |
| nnMamba: 3D Biomedical Image Segmentation, Classification and Landmark Detection with State Space Model | Arxiv 2402.03526 | haof/nnMamba |
| U-shaped Vision Mamba for Single Image Dehazing | Arxiv 2402.04139 | zzr-idam |
| Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Tasks | Arxiv 2402.04248 | |
| Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data | Arxiv 2402.05892 | |
| FD-Vision Mamba for Endoscopic Exposure Correction | Arxiv 2402.06378 | zzr-idam/FDVM-Net |
| Semi-Mamba-UNet: Pixel-Level Contrastive Cross-Supervised Visual Mamba-based UNet for Semi-Supervised Medical Image Segmentatio | Arxiv 2402.07245 | ziyangwang007/Mamba-UNet |
| P-Mamba: Marrying Perona Malik Diffusion with Mamba for Efficient Pediatric Echocardiographic Left Ventricular Segmentation | Arxiv 2402.08506 | |
| Graph Mamba: Towards Learning on Graphs with State Space Models | Arxiv 2402.08678 | GraphMamba/GMN |
| Hierarchical State Space Models for Continuous Sequence-to-Sequence Modeling | Arxiv 2402.10211 | aunaqbhirangi/hiss |
| PointMamba: A Simple State Space Model for Point Cloud Analysis | Arxiv 2402.10739 | LMD0311/PointMamba |
| Weak-Mamba-UNet: Visual Mamba Makes CNN and ViT Work Better for Scribble-based Medical Image Segmentation | Arxiv 2402.10887 | ziyangwang007/Mamba-UNet |
| Pan-Mamba: Effective pan-sharpening with State Space Model | Arxiv 2402.12192 | alexhe101/Pan-Mamba |
| MambaIR: A Simple Baseline for Image Restoration with State-Space Model | Arxiv 2402.15648 | csguoh/MambaIR |
| State Space Models for Event Cameras | Arxiv 2402.15584 | |
| Res-VMamba: Fine-Grained Food Category Visual Classification Using Selective State Space Models with Deep Residual Learning | Arxiv 2402.15761 | ChiShengChen/ResVMamba |
| DenseMamba: State Space Models with Dense Hidden Connection for Efficient Large Language Models | Arxiv 2403.00818 | WailordHe/DenseSSM |
| Point Could Mamba: Point Cloud Learning via State Space Model | Arxiv 2403.00762 | zhang-tao-whu/PCM |
| The Hidden Attention of Mamba Models | Arxiv 2403.01590 | AmeenAli/HiddenMambaAttn |
| MiM-ISTD: Mamba-in-Mamba for Efficient Infrared Small Target Detection | Arxiv 2403.02148 | txchen-USTC/MiM-ISTD |
| MedMamba: Vision Mamba for Medical Image Classification | Arxiv 2403.03849 | YubiaoYue/MedMamba |
| Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models | Arxiv 2403.03900 | |
| LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation | Arxiv 2403.05246 | MrBlankness/LightM-UNet |
| VideoMamba: State Space Model for Efficient Video Understanding | Arxiv 2403.06977 | OpenGVLab/VideoMamba |
| MamMIL: Multiple Instance Learning for Whole Slide Images with State Space Models | Arxiv 2403.05160 | |
| MambaMIL: Enhancing Long Sequence Modeling with Sequence Reordering in Computational Pathology | Arxiv 2403.06800 | isyangshu/MambaMIL |
| Large Window-based Mamba UNet for Medical Image Segmentation: Beyond Convolution and Self-attention | Arxiv 2403.07332 | wjh892521292/LMa-UNet |
| Motion Mamba: Efficient and Long Sequence Motion Generation with Hierarchical and Bidirectional Selective SSM | Arxiv 2403.07487 | steve-zeyu-zhang/MotionMamba |
| Multichannel Long-Term Streaming Neural Speech Enhancement for Static and Moving Speakers | Arxiv 2403.07675 | Audio-WestlakeU/NBSS |
| Activating Wider Areas in Image Super-Resolution | Arxiv 2403.08330 | |
| MD-Dose: A Diffusion Model based on the Mamba for Radiotherapy Dose Prediction | Arxiv 2403.08479 | LinjieFu-U/mamba_dose |
| VM-UNET-V2 Rethinking Vision Mamba UNet for Medical Image Segmentation | Arxiv 2403.09157 | nobodyplayer1/VM-UNetV2 |
| LocalMamba: Visual State Space Model with Windowed Selective Scan | Arxiv 2403.09338 | hunto/LocalMamba |
| MambaTalk: Efficient Holistic Gesture Synthesis with Selective State Space Models | Arxiv 2403.09471 | |
| Video Mamba Suite: State Space Model as a Versatile Alternative for Video Understanding | Arxiv 2403.09626 | OpenGVLab/video-mamba-suite |
| TimeMachine: A Time Series is Worth 4 Mambas for Long-term Forecasting | Arxiv 2403.09898 | Atik-Ahamed/TimeMachine |
| EfficientVMamba: Atrous Selective Scan for Light Weight Visual Mamba | Arxiv 2403.09977 | TerryPei/EfficientVMamba |
| Is Mamba Effective for Time Series Forecasting? | Arxiv 2403.11144 | wzhwzhwzh0921/S-D-Mamba |
| STG-Mamba: Spatial-Temporal Graph Learning via Selective State Space Model | Arxiv 2403.12418 | |
| VL-Mamba: Exploring State Space Models for Multimodal Learning | Arxiv 2403.13600 | ZhengYu518/VL-Mamba |
| H-vmunet: High-order Vision Mamba UNet for Medical Image Segmentation | Arxiv 2403.13642 | wurenkai/H-vmunet |
| ProMamba: Prompt-Mamba for polyp segmentation | Arxiv 2403.13660 | |
| ZigMa: Zigzag Mamba Diffusion Model | Arxiv 2403.13802 | taohu.me/zigma |
| Cobra: Extending Mamba to Multi-Modal Large Language Model for Efficient Inference | Arxiv 2403.14520 | h-zhao1997/cobra |
| SiMBA: Simplified Mamba-Based Architecture for Vision and Multivariate Time series | Arxiv 2403.15360 | badripatro/Simba |
| CMViM: Contrastive Masked Vim Autoencoder for 3D Multi-modal Representation Learning for AD classification | Arxiv 2403.16520 | |
| VMRNN:Integrating Vision Mamba and LSTM for Efficient and Accurate Spatiotemporal Forecasting | Arxiv 2403.16536 | yyyujintang/VMRNN-PyTorch |
| Integrating Mamba Sequence Model and Hierarchical Upsampling Network for Accurate Semantic Segmentation of Multiple Sclerosis Legion | Arxiv 2403.17432 | |
| PlainMamba: Improving Non-Hierarchical Mamba in Visual Recognition | Arxiv 2403.17695 | ChenhongyiYang/PlainMamba |
| Rotate to Scan: UNet-like Mamba with Triplet SSM Module for Medical Image Segmentation | Arxiv 2403.17701 | |
| ReMamber: Referring Image Segmentation with Mamba Twister | Arxiv 2403.17839 | |
| Dual-path Mamba: Short and Long-term Bidirectional Selective Structured State Space Models for Speech Separation | Arxiv 2403.18257 | |
| Gamba: Marry Gaussian Splatting with Mamba for single view 3D reconstruction | Arxiv 2403.18795 | |
| RSMamba: Remote Sensing Image Classification with State Space Model | Arxiv 2403.19654 |