mlsanityData to 7 Oct 2026

SparseSSM: Efficient Selective Structured State Space Models Can Be Pruned in One-Shot

arXiv 2506.09613 ↗First posted 11 Jun 2025cs.LGcited by 5

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

Abstract (arXiv)

State-space language models such as Mamba match Transformer quality while permitting linear complexity inference, yet still comprise billions of parameters that hinder deployment. Existing one-shot pruning methods are tailored to attention blocks and fail to account for the time-shared and discretized state-transition matrix at the heart of the selective state-space module (SSM). In this paper, we introduce SparseSSM, the first training-free pruning framework that extends the classic optimal brain surgeon (OBS) framework to state space architectures. Our layer-wise algorithm (i) derives an approximate second-order saliency score that aggregates Hessian-trace information across time steps, (ii) incorporates a component sensitivity analysis to guide feed-forward network (FFN) pruning, which also sheds light on where redundancy resides in mamba architecture, (iii) can be easily extended to semi-structured and structured sparsity. Empirically, we prune 50% of SSM weights without fine-tuning and observe no zero-shot accuracy loss, achieving the current state-of-the-art pruning algorithm for Mamba-based LLMs.

Key facts. Cited by 5 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 42 of its 52 references that are in the map. First posted on arXiv 11 Jun 2025.

Builds on

42 of 52 references.

Closest ancestors · 7

Same direction, newest first

  1. 2025Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models28 Jan
  2. 2024Falcon Mamba: The First Competitive Attention-free 7B Language Model7 Oct
  3. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
  4. 2024Jamba: A Hybrid Transformer-Mamba Language Model28 Mar
  5. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
  6. 2022Simplified State Space Layers for Sequence Modeling9 Aug
  7. 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct

Other references · 35

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  6. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
  7. 2016Pointer Sentinel Mixture Modelscited by 2,992
  8. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  9. 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
  10. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
  11. 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
  12. 2016Pruning Filters for Efficient ConvNetscited by 1,129
23 more
  1. 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
  2. 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 802
  3. 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
  4. 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 697
  5. 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 643
  6. 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
  7. 2020Pruning neural networks without any data by iteratively conserving synaptic flowcited by 475
  8. 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
  9. 2024Hymba: A Hybrid-head Architecture for Small Language Modelscited by 95
  10. 2024SiMBA: Simplified Mamba-Based Architecture for Vision and Multivariate Time seriescited by 81
  11. 2024SlimGPT: Layer-wise Structured Pruning for Large Language Modelscited by 43
  12. 2022The Combinatorial Brain Surgeon: Pruning Weights That Cancel One Another in Neural Networkscited by 41
  13. 2023Fast as CHITA: Neural Network Pruning with Combinatorial Optimizationcited by 40
  14. 2023The LLM Surgeoncited by 36
  15. 2020WoodFisher: Efficient Second-Order Approximation for Neural Network Compressioncited by 21
  16. 2024The Zamba2 Suite: Technical Reportcited by 20
  17. 2024ALPS: Improved Optimization for Highly Sparse One-Shot Pruning for Large Language Modelscited by 18
  18. 2023One-Shot Sensitivity-Aware Mixed Sparsity Pruning for Large Language Modelscited by 16
  19. 2025DarwinLM: Evolutionary Structured Pruning of Large Language Modelscited by 16
  20. 2024Layer-Adaptive State Pruning for Deep State Space Modelscited by 10
  21. 2024The Iterative Optimal Brain Surgeon: Faster Sparse Recovery by Leveraging Second-Order Informationcited by 8
  22. 2025Minitron-SSM: Efficient Hybrid Language Model Compression through Group-Aware SSM Pruningcited by 7
  23. 2025On Pruning State-Space LLMscited by 6

Built on by

Cited by year (2026 to 7 Oct)

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

5 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026AIRE-Prune: Asymptotic Impulse-Response Energy for State Pruning in State Space Models
  2. 2024Beyond Linear Approximations: A Novel Pruning Approach for Attention Matrix
  3. 2026S4oP: Operator-level Pruning of Structured State Space Models for Resource-Constrained Devices
  4. 2026PrunePath: Towards Highly Structured Sparse Language Models
  5. 2023Accurate Neural Network Pruning Requires Rethinking Sparse Optimization
  6. 2026SparseDecoding: Decoding-Aware Pruning for Accurate and Efficient LLM Inference
  7. 2026Flash PD-SSM: Memory-Optimized Structured Sparse State-Space Models
  8. 2025Scaling Linear Attention with Sparse State Expansion