ASGMamba: Adaptive Spectral Gating Mamba for Multivariate Time Series Forecasting
This provides a scalable solution for high-throughput forecasting in resource-constrained supercomputing environments, such as energy grid management and traffic flow simulation, though it appears incremental as it builds on existing Mamba and spectral methods.
The paper tackles the challenge of long-term multivariate time series forecasting by addressing the trade-offs between Transformer-based models' quadratic complexity and linear State Space Models' noise sensitivity, proposing ASGMamba which achieves state-of-the-art accuracy with O(L) complexity and reduced memory usage.
Long-term multivariate time series forecasting (LTSF) plays a crucial role in various high-performance computing applications, including real-time energy grid management and large-scale traffic flow simulation. However, existing solutions face a dilemma: Transformer-based models suffer from quadratic complexity, limiting their scalability on long sequences, while linear State Space Models (SSMs) often struggle to distinguish valuable signals from high-frequency noise, leading to wasted state capacity. To bridge this gap, we propose ASGMamba, an efficient forecasting framework designed for resource-constrained supercomputing environments. ASGMamba integrates a lightweight Adaptive Spectral Gating (ASG) mechanism that dynamically filters noise based on local spectral energy, enabling the Mamba backbone to focus its state evolution on robust temporal dynamics. Furthermore, we introduce a hierarchical multi-scale architecture with variable-specific Node Embeddings to capture diverse physical characteristics. Extensive experiments on nine benchmarks demonstrate that ASGMamba achieves state-of-the-art accuracy. While keeping strictly $$\mathcal{O}(L)$$ complexity we significantly reduce the memory usage on long-horizon tasks, thus establishing ASGMamba as a scalable solution for high-throughput forecasting in resource limited environments.The code is available at https://github.com/hit636/ASGMamba