⬅ Home / Publications

Towards Large-Scale Graph-Based Traffic Forecasting: A Data-Driven Network Partitioning Approach

Chenhan Zhang, Shuyu Zhang, Xiexin Zou, Shui Yu, and James J.Q. Yu*

IEEE Internet of Things Journal, Volume 10, Issue 5, Mar. 2023, Pages 4506-4519

Network partitioning is recognized as an effective auxiliary approach for solving transportation tasks on largescale traffic networks in a domain-decomposition manner. Most of the existing related partitioning algorithms are explicitly designed to traffic management problems and merely focus on the implied topology of the networks. In this paper, towards the practical problems that happened to traffic forecasting tasks, we propose a network-partitioning-based domain-decomposition framework to improve GCN-based predictors' performance on large-scale transportation networks. Particularly, we devise a data-driven network-partitioning approach, namely, SpeedMatching-Partitioning, which employs not only the topological features but also the traffic speed observations of traffic networks for partitioning. Additionally, we propose a data-parallel training strategy that feeds partitioned sub-networks into independent predictors for parallel training. The proposed approach is tested by comprehensive case studies on three real-world datasets to evaluate its effectiveness. The results indicate that the proposed approach can help improve GCN-based predictors' accuracy and training efficiency on both small and relatively large traffic datasets. Furthermore, we investigate the model sensitivity to the selection of graph representations and framework parameters, and the learning efficiency of the data-parallel training strategy.