STLGRU: Spatio-Temporal Lightweight Graph GRU for Traffic Flow Prediction

Bhaumik, Kishor Kumar; Niloy, Fahim Faisal; Mahmud, Sakil; Woo, Simon S.

doi:10.48550/arxiv.2212.04548

Cited by 1 publication

(1 citation statement)

References 0 publications

Supporting

Mentioning

Contrasting

Order By: Relevance

“…STLGRU [36]: A spatio-temporal graph gated recurrent unit, which uses memory-enhanced attention and gating mechanisms to capture local and global spatio-temporal relationships of traffic data.…”

Section: Compare the Proposed Model With The Following Representative...mentioning

confidence: 99%

Spatial–Temporal Traffic Flow Prediction With Fusion Graph Convolution Network and Enhanced Gated Recurrent Units

Cai,

Qu,

et al. 2024

IEEE Access

View full text Add to dashboard Cite

Accurately predicting traffic flow is paramount for the efficient operation of transportation systems. The key to enhancing prediction accuracy lies in effectively mining the intricate spatio-temporal correlations within traffic flow data. However, traditional traffic flow prediction methods that combine Graph Convolutional Network and Recurrent Neural Network have limitations in capturing comprehensive spatial correlation information and face challenges in modeling long-term temporal dependencies, consequently leading to suboptimal prediction performance. This study proposes a hybrid traffic flow prediction model based on fusion graph convolutional network and enhanced gate recurrent unit. Initially, a fusion graph structure is constructed based on adjacency graph and adaptive graph to better represent the correlations between nodes in the road network. Subsequently, the stacked fusion graph convolution module is utilized to capture multi-level spatial correlations and the enhanced gated recurrent unit is applied to extract multi-scale temporal correlations. In addition, the model integrates the extracted spatio-temporal features with the direct features through residual connection units, and utilizes the fused features for prediction, achieving superior predictive performance. The experimental results from four authentic datasets demonstrate that our proposed model outperforms state-of-the-art baseline models, showcasing an average enhancement of 3% in Mean Absolute Error(MAE), 3.3% in Root Mean Square Error(RMSE), and 2.7% in Mean Absolute Percentage Error(MAPE) across the four datasets.

show abstract

Section: Compare the Proposed Model With The Following Representative...mentioning

confidence: 99%