Traffic prediction based on GCN-LSTM model
Zhizhu Wu, Mingxia Huang, Aiping Zhao, Zhixun Lan
Abstract
Abstract Traffic flow prediction is an important part of intelligent traffic management system. Because there are many irregular data structures in road traffic, in order to improve the accuracy of traffic flow prediction, this paper proposes a combined traffic flow prediction model based on deep learning graph convolution neural network (GCN), long-term memory network (LSTM) and residual network (RESNET). GCN is used to extract the features of topology structure in traffic data, LSTM is used to extract the features of time structure, combined with ResNet to optimize the overall model, reduce the occurrence of gradient disappearance or explosion in network degradation, and finally achieve the prediction of traffic flow. According g to the experimental results, the combined traffic flow prediction model used in this paper is closer to the actual traffic flow occurrence than the traditional convolutional neural network model (CNN), and the accuracy is improved.