基于多尺度时空融合注意力网络的交通流预测

    Traffic Flow Prediction Based on Multi-scale Spatial-temporal Fusion Attention Network

    • 摘要: 交通流预测作为智能交通系统的关键组成部分,在城市交通治理、交通规划、缓解交通拥堵等方面发挥着重要作用。然而,由于交通流数据固有的高度非线性特征和复杂的时空依赖性,现有的方法难以实现高精度预测。为了进行更加准确的交通流预测,本文提出多尺度时空融合注意力网络(Multi-Scale Spatial-Temporal Fusion Attention Network, MSSTFAN) 来解决交通流预测问题。在时间维度上,本文设计了多尺度时间特征提取模块,通过傅里叶变换对多尺度特征进行分解,基于低频和高频通道注意力建模交通流中的长期趋势和短期波动。在空间维度上,通过多层次空间特征提取模块充分挖掘复杂的空间依赖,采用地理空间特征提取模块捕获交通路网中稳定及时变的空间关系,并基于语义空间特征提取模块建模不同节点之间相似的交通流状态。实验结果表明,本文模型在3个真实的交通数据集上均优于基准模型,相比于次优结果,MAE指标分别降低了2.24%、2.67%和4.09%。

       

      Abstract: As a key component of intelligent transportation systems, traffic flow prediction plays an important role in urban traffic management, transportation planning and alleviating traffic congestion. However, due to the inherent highly nonlinear characteristics and complex spatial-temporal dependencies of traffic flow data, existing methods struggle to achieve high-precision predictions. To make more accurate traffic flow prediction, a Multi-Scale Spatial-Temporal Fusion Attention Network (MSSTFAN) is proposed to solve the issue of traffic flow prediction. In the temporal dimension, a multi-scale temporal feature extraction module is designed. It decomposes multi-scale features through Fourier transform and models long-term trends and short-term fluctuations in traffic flow based on attention mechanisms in low-frequency and high-frequency channels. In the spatial dimension, complicated spatial dependencies are fully explored through a multi-level spatial feature extraction module. A geographic spatial feature extraction module is employed to capture stable and time-varying spatial relationships within the traffic road network. Additionally, a semantic spatial feature extraction module is utilized to model similar traffic flow states among different nodes. The experimental results demonstrate that the proposed model outperforms the baselines on all three real-world traffic datasets, with the MAE metric decreasing by 2.24%, 2.67% and 4.09% respectively, compared with the suboptimal results.

       

    /

    返回文章
    返回