SparseSSM: Efficient Selective Structured State Space Models Can Be Pruned in One-Shot
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
- 2025Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models28 Jan
- 2024Falcon Mamba: The First Competitive Attention-free 7B Language Model7 Oct
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
- 2024Jamba: A Hybrid Transformer-Mamba Language Model28 Mar
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
- 2022Simplified State Space Layers for Sequence Modeling9 Aug
- 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct
Other references · 35
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
- 2016Pruning Filters for Efficient ConvNetscited by 1,129
23 more
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
- 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 802
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
- 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 697
- 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 643
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
- 2020Pruning neural networks without any data by iteratively conserving synaptic flowcited by 475
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
- 2024Hymba: A Hybrid-head Architecture for Small Language Modelscited by 95
- 2024SiMBA: Simplified Mamba-Based Architecture for Vision and Multivariate Time seriescited by 81
- 2024SlimGPT: Layer-wise Structured Pruning for Large Language Modelscited by 43
- 2022The Combinatorial Brain Surgeon: Pruning Weights That Cancel One Another in Neural Networkscited by 41
- 2023Fast as CHITA: Neural Network Pruning with Combinatorial Optimizationcited by 40
- 2023The LLM Surgeoncited by 36
- 2020WoodFisher: Efficient Second-Order Approximation for Neural Network Compressioncited by 21
- 2024The Zamba2 Suite: Technical Reportcited by 20
- 2024ALPS: Improved Optimization for Highly Sparse One-Shot Pruning for Large Language Modelscited by 18
- 2023One-Shot Sensitivity-Aware Mixed Sparsity Pruning for Large Language Modelscited by 16
- 2025DarwinLM: Evolutionary Structured Pruning of Large Language Modelscited by 16
- 2024Layer-Adaptive State Pruning for Deep State Space Modelscited by 10
- 2024The Iterative Optimal Brain Surgeon: Faster Sparse Recovery by Leveraging Second-Order Informationcited by 8
- 2025Minitron-SSM: Efficient Hybrid Language Model Compression through Group-Aware SSM Pruningcited by 7
- 2025On Pruning State-Space LLMscited by 6
Built on by
Citing papers per year: 2025: 2; 2026: 3.
5 papers.
- 2025OBS-Diff: Accurate Pruning For Diffusion Models in One-Shotcited by 19
- 2025UniQL: Unified Quantization and Low-rank Compression for Adaptive Edge LLMscited by 6
- 2026GHOST: Unmasking Phantom States in Mamba2 via Grouped Hidden-state Output-aware Selection & Truncationcited by 1
- 2026Cross-Resolution Diffusion Models via Network Pruningcited by 0
- 2026Auditing and Mitigating Privacy Leakage in Cloud-Edge Collaborative Decodingcited by 0
- 2026Auditing and Mitigating Privacy Leakage in Cloud-Edge Collaborative Decoding29 Aug 2026
- 2026Cross-Resolution Diffusion Models via Network Pruning7 Apr 2026
- 2026GHOST: Unmasking Phantom States in Mamba2 via Grouped Hidden-state Output-aware Selection & Truncation11 Feb 2026
- 2025UniQL: Unified Quantization and Low-rank Compression for Adaptive Edge LLMs3 Dec 2025
- 2025OBS-Diff: Accurate Pruning For Diffusion Models in One-Shot8 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2026AIRE-Prune: Asymptotic Impulse-Response Energy for State Pruning in State Space Models
- 2024Beyond Linear Approximations: A Novel Pruning Approach for Attention Matrix
- 2026S4oP: Operator-level Pruning of Structured State Space Models for Resource-Constrained Devices
- 2026PrunePath: Towards Highly Structured Sparse Language Models
- 2023Accurate Neural Network Pruning Requires Rethinking Sparse Optimization
- 2026SparseDecoding: Decoding-Aware Pruning for Accurate and Efficient LLM Inference
- 2026Flash PD-SSM: Memory-Optimized Structured Sparse State-Space Models
- 2025Scaling Linear Attention with Sparse State Expansion