Efficient Unstructured Pruning of Mamba State-Space Models for Resource-Constrained Environments
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
- 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 · 48
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
36 more
- 2020Reformer: The Efficient Transformercited by 1,658
- 2020Rethinking Attention with Performerscited by 1,584
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,548
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,242
- 2016Pruning Filters for Efficient ConvNetscited by 1,129
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,051
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 814
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 802
- 2020Efficient Transformers: A Surveycited by 784
- 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 741
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 699
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
- 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 596
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
- 2019Importance Estimation for Neural Network Pruningcited by 531
- 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 521
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
- 2020Picking Winning Tickets Before Training by Preserving Gradient Flowcited by 467
- 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 398
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 388
- 2018AMC: AutoML for Model Compression and Acceleration on Mobile Devicescited by 375
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
- 2020Pretrained Transformers Improve Out-of-Distribution Robustnesscited by 293
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2022Transformer Memory as a Differentiable Search Indexcited by 242
- 2023HyenaDNA: Long-Range Genomic Sequence Modeling at Single Nucleotide Resolutioncited by 187
- 2017Deep Rewiring: Training very sparse deep networkscited by 136
- 2022Liquid Structural State-Space Modelscited by 134
- 2024VL-Mamba: Exploring State Space Models for Multimodal Learningcited by 93
- 2022S4ND: Modeling Images and Videos as Multidimensional Signals Using State Spacescited by 41
Built on by
Citing papers per year: 2025: 2; 2026: 4.
6 papers.
- 2025UniQL: Unified Quantization and Low-rank Compression for Adaptive Edge LLMscited by 6
- 2025Image Segmentation with Large Language Models: A Survey with Perspectives for Intelligent Transportation Systemscited by 2
- 2026Learning to Forget Attention: Memory Consolidation for Adaptive Compute Reductioncited by 1
- 2026Vision SmolMamba: Spike-Guided Token Pruning for Energy-Efficient Spiking State-Space Vision Modelscited by 1
- 2026Cross-Resolution Diffusion Models via Network Pruningcited by 0
- 2026An Exact Instrument for State Usage in Selective State-Space Models, and the Input-Driven Migration It Revealscited by 0
- 2026An Exact Instrument for State Usage in Selective State-Space Models, and the Input-Driven Migration It Reveals13 Jul 2026
- 2026Vision SmolMamba: Spike-Guided Token Pruning for Energy-Efficient Spiking State-Space Vision Models28 Apr 2026
- 2026Cross-Resolution Diffusion Models via Network Pruning7 Apr 2026
- 2026Learning to Forget Attention: Memory Consolidation for Adaptive Compute Reduction12 Feb 2026
- 2025UniQL: Unified Quantization and Low-rank Compression for Adaptive Edge LLMs3 Dec 2025
- 2025Image Segmentation with Large Language Models: A Survey with Perspectives for Intelligent Transportation Systems17 Jun 2025
Similar, not cited
Close in wording, no citation either way
- 2026Mamba-3: Improved Sequence Modeling using State Space Principles
- 2025Routing Mamba: Scaling State Space Models with Mixture-of-Experts Projection
- 2026AIRE-Prune: Asymptotic Impulse-Response Energy for State Pruning in State Space Models
- 2026S4oP: Operator-level Pruning of Structured State Space Models for Resource-Constrained Devices
- 2025PerfMamba: Performance Analysis and Pruning of Selective State Space Models
- 2024Rethinking Token Reduction for State Space Models
- 2025Block-Biased Mamba for Long-Range Sequence Processing
- 2025Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models