Short-term traffic flow prediction based on SAE and its parallel training

The alleviation of traffic congestion relies on efficient traffic control and traffic guidance, which are based on real-time short-term traffic flow prediction. In this paper, the stacked autoencoder (SAE) deep learning model with powerful feature learning capability is selected to predict the traffic flow on road sections. The process of training SAE includes the pre-training phase and the fine-tuning phase, which mainly apply the BP algorithm. However, the process of training SAE is time-consuming and cannot meet the real-time performance of modern application systems. This paper proposes a parallel training strategy for the SAE prediction model based on data parallel mode. The gradient solution process in our algorithm satisfies the conditions of parallel computing, so the training process can be designed in a parallel manner. The original dataset is distributed to some computing nodes, which are work nodes. The work node is responsible for gradient calculation using the local data. The task of the sole master node is to synthesize the gradient calculation results and then broadcast the updated gradient to each work node. The simulation results show that the SAE-based prediction model achieves better results than the traditional model, and the parallel algorithm reduces the running time of training processes.