A Traffic Congestion Prediction Model Based on Dilated-Dense Network

Abstract

When using the convolutional neural network (CNN) model to predict short-term traffic congestion, due to the convolution pooling operation of the model, part of the data for the information of the target position will be lost, resulting in the decline of the resolution of the output features and the decrease in the predictive ability of the model. To solve this problem, this paper proposes a dilated-dense neural network model. First, it uses dilated convolution to obtain the characteristics of a larger receptive field with fewer network parameters, and fully extracts complex and variable data spatio-temporal characteristics. Then, through down-sampling and equivalent mapping of dense network, it solves the problem of parameter degradation in the process of increasing layers of neural network. Finally, it uses the actual urban road average speed data blocks to verify the validity of the model. The results show that compared with the convolutional neural network model, the average absolute error of the network structure prediction is reduced by 3% to 23%.

Keywords

• dilated-dense network
• spatio-temporal characteristics
• convolutional neural networks
• short-term traffic congestion prediction