Abstract Accurate prediction of traffic status in real time is critical for advanced traffic management and travel navigation guidance. There are many attempts to predict short-term traffic flows using various deep learning algorithms. Most existing prediction models are only tested on spatiotemporal data assuming no missing data entries. However, this ideal situation rarely exists in real world due to sensor or network transmission failure. Missing data is a nonnegligible problem. Previous studies either remove time series with missing entries or impute missing data before building prediction models. The former may cause insufficient data for model training, while the latter adds extra computational burden and the imputation accuracy has direct impacts on the prediction performance. In this study, we propose an online framework that can make spatiotemporal predictions based on raw incomplete data and impute possible missing values at the same time. We design a novel spatial and temporal regularized matrix factorization model, namely LSTM-GL-ReMF, as the key component of the framework. The Long Short-term Memory (LSTM) model is chosen as the temporal regularizer to capture temporal dependency in time series data and the Graph Laplacian (GL) serves as the spatial regularizer to utilize spatial correlations among network sensors to enhance prediction and imputation performance. The proposed framework integrating with the LSTM-GL-ReMF model are tested and compared with other state-of-the-art matrix factorization models and deep learning models on three uni-variate and multi-variate spatiotemporal traffic datasets. The experimental results show our approach has a robust and accurate performance in terms of prediction and imputation accuracy under various data missing scenarios.

    Highlights This paper addresses the missing data online prediction and imputation problems. Develop an LSTM-GL-ReMF model that incorporates spatiotemporal regularizations. Extend the model to a tensor version LSTM-GL-ReTF following CP decomposition method.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Real-time spatiotemporal prediction and imputation of traffic status based on LSTM and Graph Laplacian regularized matrix factorization


    Contributors:
    Yang, Jin-Ming (author) / Peng, Zhong-Ren (author) / Lin, Lei (author)


    Publication date :

    2021-05-15




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English




    Bayesian Kernelized Matrix Factorization for Spatiotemporal Traffic Data Imputation and Kriging

    Lei, Mengying / Labbe, Aurelie / Wu, Yuankai et al. | IEEE | 2022


    Graph Spectral Regularized Tensor Completion for Traffic Data Imputation

    Deng, Lei / Liu, Xiao-Yang / Zheng, Haifeng et al. | IEEE | 2022


    Spatiotemporal Traffic Flow Prediction with KNN and LSTM

    Xianglong Luo / Danyang Li / Yu Yang et al. | DOAJ | 2019

    Free access

    Spatiotemporal Tensor Completion for Improved Urban Traffic Imputation

    Ben Said, Ahmed / Erradi, Abdelkarim | IEEE | 2022


    Tucker factorization-based tensor completion for robust traffic data imputation

    Lyu, Cheng / Lu, Qing-Long / Wu, Xinhua et al. | Elsevier | 2024