Modeling spatial dependency is crucial to solving traffic prediction tasks; thus, spatial-temporal graph-based models have been widely used in this area in recent years. Existing approaches either rely on a fixed pre-defined graph (e.g., a road network) or learn the correlations between locations. However, most methods suffer from spurious correlation and do not sufficiently consider the traffic's causal relationships. This study proposes a Spatiotemporal Causal Graph Inference (ST-CGI) framework for traffic prediction tasks that learn both the causal graph and autoregressive processes. We decouple the spatiotemporal traffic prediction process into two steps; the causal graph inference step and the autoregressive step, where the latter relies on the former. Optimizing the entire framework on the autoregressive task approximates the Granger causality test and thus enables excellent interpretability of the prediction. Extensive experimentation using two real-world datasets demonstrates the outstanding performance of the proposed models.
Granger Causal Inference for Interpretable Traffic Prediction
2022-10-08
2418234 byte
Conference paper
Electronic Resource
English
Traffic flow prediction method based on space-time Granger causality
European Patent Office | 2023
|Interpretable space-time analysis method for traffic jam prediction
European Patent Office | 2022
|