arXiv:2608.29658physics.soc-phcs.LG2026-08

用巴特沃斯滤波降噪+状态空间模型,高效精准预测交通流量。

ButterMamba: Butterworth-Enhanced Spatial-Temporal Mamba for Efficient Traffic Flow Prediction

  • 先用巴特沃斯滤波器去噪,再用并行Mamba捕捉时空依赖。
  • 在三个数据集上精度超越现有方法,训练耗时和内存显著降低。
  • 适合需要实时高精度交通预测的智慧城市场景。

准确的交通流量预测是智能交通系统的基础,对城市交通优化和智慧城市建设至关重要。尽管融合图神经网络(GNNs)与时间序列预测的方法已展现出潜力,但仍存在两大关键局限:(1) 基于注意力的架构具有二次复杂度,难以在大规模路网中实现实时部署;(2) 传感器数据中的高频噪声严重降低预测可靠性。这些挑战在大都市场景中尤为突出,对计算效率和抗噪能力要求极高。为此,我们提出基于状态空间模型(SSMs)的新型高效框架ButterMamba。该框架包含两个核心组件:(1) 巴特沃斯频谱滤波模块,通过去除高频噪声使模型聚焦于关键趋势;(2) 空间-时序状态混合器,采用并行Mamba架构高效建模长程时序依赖与复杂空间关联。通过将降噪与时空建模解耦,ButterMamba实现线性计算复杂度下的优越预测精度。在三个公开数据集上的大量实验表明,ButterMamba不仅在预测精度上优于现有最先进模型,且显著减少训练时间与内存消耗。

原文摘要 · Abstract (English)

Accurate traffic flow prediction is fundamental to intelligent transportation systems, playing a pivotal role in urban mobility optimization and smart city development. While Graph Neural Networks (GNNs) integrated with time series forecasting have emerged as promising solutions, two critical limitations persist: (1) the quadratic complexity of attention-based architectures hinders real-time deployment in large-scale networks, and (2) high-frequency noise in sensor data significantly degrades prediction reliability. These challenges are particularly acute in metropolitan scenarios where both computational efficiency and noise robustness are paramount. To address these limitations, we introduce \textbf{ButterMamba}, a novel and efficient framework based on State Space Models (SSMs). ButterMamba consists of two key components: (1) a Butterworth Spectral Filtering module that preprocesses the data by removing high-frequency noise, allowing the model to focus on significant underlying trends, and (2) a Spatial-Temporal State Mixer that uses a parallel Mamba architecture to efficiently capture both long-range temporal dependencies and complex spatial correlations across the road network. By decoupling noise filtering from spatial-temporal modeling, ButterMamba achieves superior predictive accuracy with linear computational complexity. Extensive experiments on three public datasets demonstrate that ButterMamba not only outperforms existing state-of-the-art models in terms of prediction accuracy but also considerably reduces training time and memory usage.

交通预测状态空间模型降噪Mamba

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。