JS']JS-STDGN: A Spatial-Temporal Dynamic Graph Network Using JS']JS-Graph for Traffic Prediction

Traffic prediction is a fundamental operation in real-time traffic analysis. A precise prediction of traffic condition can benefit both road users and traffic management agencies. However, since road traffic is decided by multiple static and dynamic factors, traffic prediction is still a challenging task. As the core indicator of traffic condition, many works focus on traffic speed prediction using time-series forecasting approaches. Although current methods take into account the static road topology while modelling, they fail to consider (1) the semantic closeness between road components and (2) congestion caused by upstream/downstream traffic propagation. In this paper, we introduce a Spatial-Temporal Dynamic Graph Network using JS-Graph, which considers both static road features and dynamic traffic flows when forecasting. Specifically, we first propose a data-driven 'JS-Graph' method that describes the semantic similarity between road nodes. It models the complex spatial correlations that cannot be captured by the traditional spatial adjacency graph. Secondly, we design a dynamic graph attention network that considers the traffic dynamics that happened in previous time slices when predicting the current one to capture the congestion propagation phenomena. Extensive experiments conducted on real-world datasets show that our proposed method is significantly better than baselines.