Graph Neural Network-Driven Traffic Forecasting for the Connected Internet of Vehicles

Qin Zhang,Keping Yu,Zhiwei Guo,Sahil Garg,Joel J. P. C. Rodrigues,Mohammad Mehedi Hassan,Mohsen Guizani
DOI: https://doi.org/10.1109/tnse.2021.3126830
IF: 6.6
2021-01-01
IEEE Transactions on Network Science and Engineering
Abstract:Due to great advances in wireless communication, the connected Internet of vehicles (CIoVs) has become prevalent. Naturally, internal connections among active vehicles are an indispensable factor in traffic forecasting. Although many related research studies have been conducted in the past few years, they mainly designed and/or developed single forecasting models for traffic forecasting. Such models may show ideal performance in some scenarios but lack satisfactory robustness to dynamic scenario changes. To address this challenge, a graph neural network-driven traffic forecasting model for CIoVs is proposed in this work, which is denoted as Gra-TF. In this paper, we regard the dynamics of traffic data as a temporal evolution scenario. With the assistance of ensemble learning, three typical graph-level prediction methods are employed to construct an integrated and enhanced forecasting model. This design utilizes several methods to minimize uncertainty in CIoVs. Finally, we use a real-world dataset to build an experimental scenario for further assessment. Numerical results indicate that the proposed Gra-TF improves the prediction accuracy by 30% to 40% compared with several baseline methods.
What problem does this paper attempt to address?