Memba: Membrane-driven Parameter-Efficient Fine-Tuning for Mamba
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
Abstract (arXiv)
State Space Models (SSMs) have emerged as powerful alternatives to attention-based Transformers, with Mamba demonstrating impressive efficiency and scalability. As these models grow increasingly larger, the need for Parameter-Efficient Fine-Tuning (PEFT) methods becomes critical to adapt pre-trained Mamba to downstream tasks without prohibitive computational costs. However, previous approaches simply apply traditional Transformer-tailored PEFT methods without addressing the unique temporal processing dynamics of SSMs. To address this limitation, we propose Memba, a membrane-driven PEFT approach specifically designed for Mamba. Memba introduces Leaky Integrate Membrane (LIM) neurons as bio-inspired gating mechanisms that naturally accumulate membrane potentials over time, enhancing selective information retention. By strategically combining LIM neurons with Low-Rank Adaptations (LoRA) and cross-layer membrane transfer, our approach significantly improves Mamba's temporal modeling capabilities. Extensive experiments across language and vision tasks demonstrate that Memba achieves substantial improvements over existing PEFT methods. The code is available at https://github.com/Intelligent-Computing-Lab-Yale/Memba.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 48 of its 58 references that are in the map. First posted on arXiv 22 Jun 2025.
Builds on
48 of 58 references.
Closest ancestors · 9
Same direction, newest first
- 2024Parameter Efficient Mamba Tuning via Projector-targeted Diagonal-centric Linear Transformation21 Nov
- 2024MambaPEFT: Exploring Parameter-Efficient Fine-Tuning for Mamba6 Nov
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
- 2024MambaByte: Token-free Selective State Space Model24 Jan
- 2024MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts8 Jan
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
- 2022Simplified State Space Layers for Sequence Modeling9 Aug
- 2022Diagonal State Spaces are as Effective as Structured State Spaces27 Mar
- 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct
Other references · 39
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2014Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modelingcited by 2,367
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2021BitFit: Simple Parameter-efficient Fine-tuning for Transformer-based Masked Language-modelscited by 1,245
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
27 more
- 2024VMamba: Visual State Space Modelcited by 969
- 2021Towards a Unified View of Parameter-Efficient Transfer Learningcited by 872
- 2020AdapterFusion: Non-Destructive Task Composition for Transfer Learningcited by 788
- 2024DoRA: Weight-Decomposed Low-Rank Adaptationcited by 758
- 2021GPT Understands, Toocited by 716
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 700
- 2021P-Tuning v2: Prompt Tuning Can Be Comparable to Fine-tuning Universally Across Scales and Taskscited by 689
- 2024Parameter-Efficient Fine-Tuning for Large Models: A Comprehensive Surveycited by 656
- 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 608
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
- 2019A Large-scale Study of Representation Learning with the Visual Task Adaptation Benchmarkcited by 425
- 2023AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuningcited by 357
- 2020Parameter-Efficient Transfer Learning with Diff Pruningcited by 352
- 2024VideoMamba: State Space Model for Efficient Video Understandingcited by 328
- 2023LLM-Adapters: An Adapter Family for Parameter-Efficient Fine-Tuning of Large Language Modelscited by 324
- 2021Parameter-efficient Multi-task Fine-tuning for Transformers via Shared Hypernetworkscited by 284
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2024MambaVision: A Hybrid Mamba-Transformer Vision Backbonecited by 224
- 2021Training Neural Networks with Fixed Sparse Maskscited by 214
- 2023SpikingJelly: An open-source machine learning infrastructure platform for spike-based intelligencecited by 169
- 2021Raise a Child in Large Language Model: Towards Effective and Generalizable Fine-tuningcited by 146
- 2022Neural Prompt Searchcited by 135
- 2024Is Mamba Effective for Time Series Forecasting?cited by 114
- 2023Sensitivity-Aware Visual Parameter-Efficient Fine-Tuningcited by 74
- 2024Multichannel Long-Term Streaming Neural Speech Enhancement for Static and Moving Speakerscited by 28
- 2024Decision Mamba: Reinforcement Learning via Sequence Modeling with Selective State Spacescited by 24
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2024Mamba State-Space Models Are Lyapunov-Stable Learners
- 2025Understanding Input Selectivity in Mamba: Impact on Approximation Power, Memorization, and Associative Recall Capacity
- 2024Demystify Mamba in Vision: A Linear Attention Perspective
- 2024A Survey of Mamba
- 2024The Hidden Attention of Mamba Models
- 2024Bio-Inspired Mamba: Temporal Locality and Bioplausible Learning in Selective State Space Models
- 2024Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Tasks
- 2024An Empirical Study of Mamba-based Language Models