GLSNN Network: A Multi-Scale Spatiotemporal Prediction Model for Urban Traffic Flow.

Journal: Sensors (Basel, Switzerland)
Published Date:

Abstract

Traffic flow prediction is a key issue in intelligent transportation systems. The growing trend in data disclosure has created more potential sources for the input for predictive models, posing new challenges to the prediction of traffic flow in the era of big data. In this study, the prediction of urban traffic flow was regarded as a spatiotemporal prediction problem, focusing on the traffic speed. A Graph LSTM (Long Short-Term Memory) Spatiotemporal Neural Network (GLSNN) model was constructed to perform a multi-scale spatiotemporal fusion prediction based on the multi-source input data. The GLSNN model consists of three parts: MS-LSTM, LZ-GCN, and LSTM-GRU. We used the MS-LSTM module to scale the traffic timing data, and then used the LZ-GCN network and the LSTM-GRU network to capture both the time and space dependencies. The model was tested on a real traffic dataset, and the experiment results verified the superior performance of the GLSNN model on both a high-precision and multi-scale prediction of urban traffic flow.

Authors

  • Benhe Cai
    Key Laboratory of 3-Dimensional Information Acquisition and Application, Ministry of Education, Capital Normal University, Beijing 100048, China.
  • Yanhui Wang
    Department of Traditional Chinese Medicine, Medical College, Xiamen University, Xiamen, China.
  • Chong Huang
    State Key Laboratory of Resources and Environmental Information System, Institute of Geographic Science and Natural Resources Research, Chinese Academy of Sciences, Beijing 100101, China. huangch@lreis.ac.cn.
  • Jiahao Liu
    School of Artificial Intelligence, Hebei University of Technology, Tianjin, 300130, China.
  • Wenxin Teng
    Key Laboratory of 3-Dimensional Information Acquisition and Application, Ministry of Education, Capital Normal University, Beijing 100048, China.