MS-SSM: A Multi-Scale State Space Model for Efficient Sequence Modeling

TL;DR

The paper proposes MS-SSM, a multi-scale state space model that captures multi-resolution dependencies in sequences, improving long-range modeling and efficiency over traditional SSMs.

Contribution

It introduces a multi-scale framework with specialized dynamics and a scale-mixer for dynamic resolution fusion, enhancing sequence modeling capabilities.

Findings

Outperforms prior SSM models on benchmarks

Improves long-range and hierarchical sequence modeling

Maintains computational efficiency

Abstract

State-space models (SSMs) have recently attention as an efficient alternative to computationally expensive attention-based models for sequence modeling. They rely on linear recurrences to integrate information over time, enabling fast inference, parallelizable training, and control over recurrence stability. However, traditional SSMs often suffer from limited effective memory, requiring larger state sizes for improved recall. Moreover, existing SSMs struggle to capture multi-scale dependencies, which are essential for modeling complex structures in time series, images, and natural language. This paper introduces a multi-scale SSM framework that addresses these limitations by representing sequence dynamics across multiple resolution and processing each resolution with specialized state-space dynamics. By capturing both fine-grained, high-frequency patterns and coarse, global trends, MS-SSM enhances memory efficiency and long-range modeling. We further introduce an input-dependent scale-mixer, enabling dynamic information fusion across resolutions. The proposed approach significantly improves sequence modeling, particularly in long-range and hierarchical tasks, while maintaining computational efficiency. Extensive experiments on benchmarks, including Long Range Arena, hierarchical reasoning, time series classification, and image recognition, demonstrate that MS-SSM consistently outperforms prior SSM-based models, highlighting the benefits of multi-resolution processing in state-space architectures.