余剑峤
James Jianqiao Yu
首页 论文 服务 ENG

教授、博士生导师

计算机科学与技术学院

哈尔滨工业大学(深圳)

广东省深圳市南山区深圳大学城

jqyu(at)hit.edu.cn jqyu(at)ieee.org Google Scholar
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, March 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.