Spatio-Temporal Deep Learning Framework for Traffic Speed Forecasting in IoT

Abstract

The Internet of Things (IoT) enables us to collect traffic data from multiple sources, which makes it possible to predict traffic speed in time with big traffic data. This article presents a novel spatio-temporal deep learning framework that aims to provide accurate and timely traffic speed forecasting. To extract the spatial and temporal features of traffic data simultaneously, this framework combines two different deep learning models, namely convolutional long short-term memory (ConvLSTM) and the graph convolutional network (GCN). Specifically, the ConvLSTM model is used to learn the temporal dynamics of traffic data to extract temporal features. On the other hand, the GCN model is used to learn the spatial complexity of traffic data to extract spatial features. Experiments show that the prediction performance of our framework at different time horizons outperforms three baseline approaches.