DSTGA-Mamba: a disentangled spatio-temporal graph attention Mamba model for traffic flow prediction

DSTGA-Mamba:一种用于交通流量预测的解耦时空图注意力Mamba模型

阅读:1

Abstract

Accurate and robust traffic flow prediction remains a fundamental yet unresolved challenge in intelligent transportation systems, primarily due to the intertwined nature of long-term trends and short-term disruptions, alongside complex, dynamic spatio-temporal dependencies. This study proposes the Disentangled Spatio-Temporal Graph Attention Mamba (DSTGA-Mamba) model, a unified framework that comprises the following key components. First, we introduce a multi-level wavelet-based disentanglement module to explicitly decouple temporal dynamics into trend and event components, reducing distribution shifts and long-horizon error accumulation. Second, we design a dual-channel spatio-temporal encoder combining global temporal attention and causal convolution, integrated with a novel efficient spectral graph attention mechanism to capture scalable and expressive cross-node interactions. Third, we embed a selective state-space model (ST-Mamba) to capture ultra-long temporal dependencies with linear complexity, enhancing both scalability and forecasting fidelity. Complemented by a multi-supervision decoder that fuses trend-guided supervision with adaptive event modeling, DSTGA-Mamba demonstrates superior performance across four benchmark datasets, achieving consistent gains in accuracy, robustness, and computational efficiency over strong baselines.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。