RT-GCN: Gaussian-based spatiotemporal graph convolutional network for robust traffic prediction

Abstract

Traffic forecasting plays a critical role in intelligent transportation systems (ITS) in smart cities. Travelers as well as urban managers rely on reliable traffic information to make their decisions for route choice and traffic management. However, noisy or missing traffic data poses a problem for accurate and robust traffic forecasting. While data-driven models such as deep neural networks can achieve high prediction accuracy with complete datasets, sensor malfunctions, and environmental effects degrade the performance of such models, as these models rely heavily on precise traffic measurements for model training and estimation. Consequently, incomplete traffic data poses a challenge for robust model design that can make accurate traffic forecasts with noisy/missing data. This research proposes the Robust Spatiotemporal Graph Convolutional Network (RT-GCN), a traffic prediction model that handles noise perturbations and missing data using a Gaussian distributed node representation and a variance based attention mechanism. Through experiments conducted on four real-world traffic datasets using diverse noisy and missing scenarios, the proposed RT-GCN model has demonstrated its ability to handle noise perturbations and missing values and provide high accuracy prediction.