S4oP: Operator-level Pruning of Structured State Space Models for Resource-Constrained Devices
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
Abstract (arXiv)
Structured State Space Models (SSMs), including the S4 and S4D architectures, have recently emerged as powerful alternatives to attention-based models for capturing long-range dependencies in sequential data. Despite their strong empirical performance, deploying these models in time- and resource-constrained settings remains challenging due to their computational and memory demands. In this paper, we propose a novel incremental, operator-level pruning approach for S4- and S4D-based models that significantly reduces inference cost while preserving predictive performance. To the best of our knowledge, this is the first work to systematically investigate structured operator pruning for SSMs. Our method progressively prunes model operators by interleaving structured masking with fine-tuning, while jointly monitoring accuracy and inference latency. We implement this approach within a unified training and evaluation framework that enables systematic exploration of efficiency-accuracy trade-offs. Experiments across multiple benchmark datasets show that pruning up to 70% of the model operators preserves the performance of the original models in most cases, while substantially reducing inference latency. These results demonstrate that structured operator pruning is an effective and previously unexplored strategy for improving the efficiency of SSMs and facilitate their deployment in practical, resource-constrained scenarios.
Key facts. No citing papers in the map yet (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 15 of its 17 references that are in the map. First posted on arXiv 16 Jun 2026.
Builds on
15 of 17 references.
Closest ancestors · 1
Same direction, newest first
- 2025Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models28 Jan
Other references · 14
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2016Pruning Filters for Efficient ConvNetscited by 1,143
- 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 702
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 700
- 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 608
- 2018Rethinking the Value of Network Pruningcited by 599
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
- 2020Comparing Rewinding and Fine-tuning in Neural Network Pruningcited by 240
- 2023A Survey on Deep Neural Network Pruning-Taxonomy, Comparison, Analysis, and Recommendationscited by 202
- 2019Automatic diagnosis of the 12-lead ECG using a deep neural networkcited by 164
- 2023Simple Hardware-Efficient Long Convolutions for Sequence Modelingcited by 62
- 2022Revisiting Random Channel Pruning for Neural Network Compressioncited by 50
2 more
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026AIRE-Prune: Asymptotic Impulse-Response Energy for State Pruning in State Space Models
- 2025SparseSSM: Efficient Selective Structured State Space Models Can Be Pruned in One-Shot
- 2025Efficient Unstructured Pruning of Mamba State-Space Models for Resource-Constrained Environments
- 2024Layer-Adaptive State Pruning for Deep State Space Models
- 2026Beyond FLOPs: Benchmarking Real Inference Acceleration of LLM Pruning under a GEMM-Centric Taxonomy
- 2023Never Train from Scratch: Fair Comparison of Long-Sequence Models Requires Data-Driven Priors
- 2022Simplified State Space Layers for Sequence Modeling
- 2024Enhanced Structured State Space Models via Grouped FIR Filtering and Attention Sink Mechanisms