mlsanityData to 7 Oct 2026

Efficient Unstructured Pruning of Mamba State-Space Models for Resource-Constrained Environments

arXiv 2505.08299 ↗First posted 13 May 2025cs.LG cs.CVcited by 6

On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language

Abstract (arXiv)

State-space models (SSMs), particularly the Mamba architecture, have emerged as powerful alternatives to Transformers for sequence modeling, offering linear-time complexity and competitive performance across diverse tasks. However, their large parameter counts pose significant challenges for deployment in resource-constrained environments. We propose a novel unstructured pruning framework tailored for Mamba models that achieves up to 70% parameter reduction while retaining over 95% of the original performance. Our approach integrates three key innovations: (1) a gradient-aware magnitude pruning technique that combines weight magnitude and gradient information to identify less critical parameters, (2) an iterative pruning schedule that gradually increases sparsity to maintain model stability, and (3) a global pruning strategy that optimizes parameter allocation across the entire model. Through extensive experiments on WikiText-103, Long Range Arena, and ETT time-series benchmarks, we demonstrate significant efficiency gains with minimal performance degradation. Our analysis of pruning effects on Mamba's components reveals critical insights into the architecture's redundancy and robustness, enabling practical deployment in resource-constrained settings while broadening Mamba's applicability.

Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 52 of its 66 references that are in the map. First posted on arXiv 13 May 2025.

Builds on

52 of 66 references.

Closest ancestors · 4

Same direction, newest first

  1. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
  2. 2022Simplified State Space Layers for Sequence Modeling9 Aug
  3. 2022Diagonal State Spaces are as Effective as Structured State Spaces27 Mar
  4. 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct

Other references · 48

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2015Distilling the Knowledge in a Neural Networkcited by 10,076
  5. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
  6. 2016Pointer Sentinel Mixture Modelscited by 2,992
  7. 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
  8. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  9. 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
  10. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  11. 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
  12. 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
36 more
  1. 2020Reformer: The Efficient Transformercited by 1,658
  2. 2020Rethinking Attention with Performerscited by 1,584
  3. 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,548
  4. 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
  5. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
  6. 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,242
  7. 2016Pruning Filters for Efficient ConvNetscited by 1,129
  8. 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
  9. 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,051
  10. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
  11. 2023RWKV: Reinventing RNNs for the Transformer Eracited by 814
  12. 2019Are Sixteen Heads Really Better than One?cited by 803
  13. 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 802
  14. 2020Efficient Transformers: A Surveycited by 784
  15. 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 741
  16. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
  17. 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
  18. 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
  19. 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 596
  20. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
  21. 2019Importance Estimation for Neural Network Pruningcited by 531
  22. 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 521
  23. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
  24. 2020Picking Winning Tickets Before Training by Preserving Gradient Flowcited by 467
  25. 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 398
  26. 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 388
  27. 2018AMC: AutoML for Model Compression and Acceleration on Mobile Devicescited by 375
  28. 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
  29. 2020Pretrained Transformers Improve Out-of-Distribution Robustnesscited by 293
  30. 2022Long Range Language Modeling via Gated State Spacescited by 265
  31. 2022Transformer Memory as a Differentiable Search Indexcited by 242
  32. 2023HyenaDNA: Long-Range Genomic Sequence Modeling at Single Nucleotide Resolutioncited by 187
  33. 2017Deep Rewiring: Training very sparse deep networkscited by 136
  34. 2022Liquid Structural State-Space Modelscited by 134
  35. 2024VL-Mamba: Exploring State Space Models for Multimodal Learningcited by 93
  36. 2022S4ND: Modeling Images and Videos as Multidimensional Signals Using State Spacescited by 41

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 2; 2026: 4.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Mamba-3: Improved Sequence Modeling using State Space Principles
  2. 2025Routing Mamba: Scaling State Space Models with Mixture-of-Experts Projection
  3. 2026AIRE-Prune: Asymptotic Impulse-Response Energy for State Pruning in State Space Models
  4. 2026S4oP: Operator-level Pruning of Structured State Space Models for Resource-Constrained Devices
  5. 2025PerfMamba: Performance Analysis and Pruning of Selective State Space Models
  6. 2024Rethinking Token Reduction for State Space Models
  7. 2025Block-Biased Mamba for Long-Range Sequence Processing
  8. 2025Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models