<p>Accurate traffic forecasting is pivotal in the domain of intelligent transportation systems and urban planning. Although deep learning-based methods for traffic forecasting have shown significant advancements, it remains challenging to simultaneously capture both temporal and spatial dependencies in traffic flow problems. Moreover, existing research often overlooks the complex spatio-temporal characteristics of intersections within discrete vehicle trajectories in urban traffic scenarios. To address these challenges, this paper proposes a novel spatio-temporal graph transformer (ST-Graphormer) for end-to-end traffic forecasting, which effectively extracts complex spatio-temporal features from urban traffic flow data. First, to tackle the complexities of urban traffic scenarios, we designed a bidirectional long short-term memory (BiLSTM)-based temporal trajectory encoder and a graph attention network (GAT)-based spatial topology encoder. Specifically, the temporal trajectory encoder is utilized to extract and fuse long-term and short-term temporal dependencies across various intersections from continuous vehicle trajectory information. The spatial topology encoder captures the spatial dependencies between intersections based on the intersection network topology graph. Subsequently, temporal trajectory encoding and spatial topology encoding are obtained, allowing for the effective modeling of spatio-temporal representations within complex traffic intersection networks. Additionally, a multi-head attention fusion mechanism is developed to integrate prior knowledge derived from temporal trajectory and spatial topology encodings, thereby enhancing traffic forecasting accuracy. Extensive experiments on real-world traffic datasets demonstrate that the proposed method outperforms existing state-of-the-art baseline methods.</p>

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

St-Graphormer: spatio-temporal graph transformer for end-to-end traffic forecasting

  • Zhanchi Wang,
  • Yi Hu,
  • Ruirui Zhong,
  • Bingtao Hu,
  • Dinghao Cheng,
  • Yixiong Feng,
  • Jianrong Tan

摘要

Accurate traffic forecasting is pivotal in the domain of intelligent transportation systems and urban planning. Although deep learning-based methods for traffic forecasting have shown significant advancements, it remains challenging to simultaneously capture both temporal and spatial dependencies in traffic flow problems. Moreover, existing research often overlooks the complex spatio-temporal characteristics of intersections within discrete vehicle trajectories in urban traffic scenarios. To address these challenges, this paper proposes a novel spatio-temporal graph transformer (ST-Graphormer) for end-to-end traffic forecasting, which effectively extracts complex spatio-temporal features from urban traffic flow data. First, to tackle the complexities of urban traffic scenarios, we designed a bidirectional long short-term memory (BiLSTM)-based temporal trajectory encoder and a graph attention network (GAT)-based spatial topology encoder. Specifically, the temporal trajectory encoder is utilized to extract and fuse long-term and short-term temporal dependencies across various intersections from continuous vehicle trajectory information. The spatial topology encoder captures the spatial dependencies between intersections based on the intersection network topology graph. Subsequently, temporal trajectory encoding and spatial topology encoding are obtained, allowing for the effective modeling of spatio-temporal representations within complex traffic intersection networks. Additionally, a multi-head attention fusion mechanism is developed to integrate prior knowledge derived from temporal trajectory and spatial topology encodings, thereby enhancing traffic forecasting accuracy. Extensive experiments on real-world traffic datasets demonstrate that the proposed method outperforms existing state-of-the-art baseline methods.