Accurate traffic flow forecasting is pivotal for transportation systems. However, existing deep learning methods, including Graph Neural Networks (GNNs) and Transformers, often struggle with several challenges: 1) Efficiently modeling long-range spatio-temporal dependencies, hindered by inherent GNN limitations or the quadratic computational complexity (O(N 2)) of Transformers; 2) Adaptively capturing dynamic spatio-temporal heterogeneity inherent in real-world traffic; 3) Effectively disentangling the complex superposition of mixed traffic patterns and multi-source influences. To address these limitations, we propose the Spatio-Temporal Selective State Expert Network (S3E-Net). This novel framework uniquely integrates a Mamba-inspired backbone, enabling efficient processing of long sequences with linear complexity, and a dynamic multi-expert mechanism, termed Dynamic Spatio-Temporal Expert Routing (DynSTER). DynSTER employs specialized experts to adaptively model diverse traffic phenomena. Extensive experiments on benchmark datasets demonstrate that S3E-Net achieves state-of-the-art performance, significantly outperforming existing methods in prediction accuracy and robustness across various horizons.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

S3E-Net: Spatio-Temporal Selective State Expert Network for Traffic Flow Prediction

  • Ning Xu,
  • Jing Yu

摘要

Accurate traffic flow forecasting is pivotal for transportation systems. However, existing deep learning methods, including Graph Neural Networks (GNNs) and Transformers, often struggle with several challenges: 1) Efficiently modeling long-range spatio-temporal dependencies, hindered by inherent GNN limitations or the quadratic computational complexity (O(N 2)) of Transformers; 2) Adaptively capturing dynamic spatio-temporal heterogeneity inherent in real-world traffic; 3) Effectively disentangling the complex superposition of mixed traffic patterns and multi-source influences. To address these limitations, we propose the Spatio-Temporal Selective State Expert Network (S3E-Net). This novel framework uniquely integrates a Mamba-inspired backbone, enabling efficient processing of long sequences with linear complexity, and a dynamic multi-expert mechanism, termed Dynamic Spatio-Temporal Expert Routing (DynSTER). DynSTER employs specialized experts to adaptively model diverse traffic phenomena. Extensive experiments on benchmark datasets demonstrate that S3E-Net achieves state-of-the-art performance, significantly outperforming existing methods in prediction accuracy and robustness across various horizons.