摘要In order to improve the accuracy of urban road network traffic flow prediction with missing values, the generator and discriminator of the generative adversarial network are reconstructed, the loss function is improved, and the traffic generative adversarial imputation network (TGAIN) is proposed for the completion of the missing data of traffic flow. Based on empirical mode decomposition (EMD), graph convolutional networks (GCN) and gated recurrent unit (GRU), EMD-GCN-GRU model is designed for urban road network traffic flow prediction. First, the traffic flow data is processed by empirical mode decomposition and each component of the same level is reconstructed as the input of the subsequent prediction model. Then, the graph convolutional networks are used to learn the road network topology to capture the spatial characteristic of traffic flow, and the gated recurrent unit is employed to capture the temporal characteristic of traffic flow. For the road network traffic flow data with missing values, TGAIN is used to complete the data, and then EMD-GCN-GRU is used to predict the traffic flow. The Shenzhen average vehicle speed data set is used to construct a variety of typical traffic flow data with different missing patterns and different missing rates to simulate the actual missing situation. The effectiveness of the method is verified on the ModelArts development platform. The results show that compared with the commonly used matrix factorization imputation method, the TGAIN model has higher completion accuracy in the random missing mode of the dataset and has better completion performance when the non-random missing rate is lower than 50%. Compared with other seven prediction algorithms, the proposed prediction method has higher prediction accuracy. Combining the data imputation method TGAIN with the traffic flow prediction method EMD-GCN-GRU for urban road network traffic flow prediction with missing values can significantly reduce the negative impact of missing data and data noise on traffic flow prediction and capture the spatial and temporal correlation of network traffic flow, which improves the accuracy of urban road network traffic flow prediction.
Abstract：In order to improve the accuracy of urban road network traffic flow prediction with missing values, the generator and discriminator of the generative adversarial network are reconstructed, the loss function is improved, and the traffic generative adversarial imputation network (TGAIN) is proposed for the completion of the missing data of traffic flow. Based on empirical mode decomposition (EMD), graph convolutional networks (GCN) and gated recurrent unit (GRU), EMD-GCN-GRU model is designed for urban road network traffic flow prediction. First, the traffic flow data is processed by empirical mode decomposition and each component of the same level is reconstructed as the input of the subsequent prediction model. Then, the graph convolutional networks are used to learn the road network topology to capture the spatial characteristic of traffic flow, and the gated recurrent unit is employed to capture the temporal characteristic of traffic flow. For the road network traffic flow data with missing values, TGAIN is used to complete the data, and then EMD-GCN-GRU is used to predict the traffic flow. The Shenzhen average vehicle speed data set is used to construct a variety of typical traffic flow data with different missing patterns and different missing rates to simulate the actual missing situation. The effectiveness of the method is verified on the ModelArts development platform. The results show that compared with the commonly used matrix factorization imputation method, the TGAIN model has higher completion accuracy in the random missing mode of the dataset and has better completion performance when the non-random missing rate is lower than 50%. Compared with other seven prediction algorithms, the proposed prediction method has higher prediction accuracy. Combining the data imputation method TGAIN with the traffic flow prediction method EMD-GCN-GRU for urban road network traffic flow prediction with missing values can significantly reduce the negative impact of missing data and data noise on traffic flow prediction and capture the spatial and temporal correlation of network traffic flow, which improves the accuracy of urban road network traffic flow prediction.
基金资助:Supported by the National Natural Science Foundation of China ( No. 11772264); the Natural Science Basis Research Plan in Shaanxi Province of China (No. 2020JM-119)
通讯作者:
CHEN Jian-zhong
引用本文:
陈建忠, 吕泽凯, 蔺皓萌. Prediction Model for Traffic Flow with Missing Values Based on Generative Adversarial and Graph Convolutional Networks[J]. Journal of Highway and Transportation Research and Development, 2023, 17(3): 62-74.
CHEN Jian-zhong, LV Ze-kai, LIN Hao-meng. Prediction Model for Traffic Flow with Missing Values Based on Generative Adversarial and Graph Convolutional Networks. Journal of Highway and Transportation Research and Development, 2023, 17(3): 62-74.
[1] ZHU L, YU F R, WANG Y G, et al. Big Data Analytics in Intelligent Transportation Systems:A Survey[J]. IEEE Transactions on Intelligent Transportation Systems, 2019, 20(1):383-398. [2] VLAHOGIANNI E I, KARLAFTIS M G, GOLIAS J C. Short-term Traffic Forecasting:Where We Are and Where We're Going[J]. Transportation Research Part C:Emerging Technologies, 2014, 43:3-19. [3] TIAN Y, ZHANG K L, LI J Y, et al. LSTM-based Traffic Flow Prediction with Missing Data[J]. Neurocomputing, 2018, 318:297-305. [4] DUAN Y J, LV Y S, LIU Y L, et al. An Efficient Realization of Deep Learning for Traffic Data Imputation[J]. Transportation Research Part C:Emerging Technologies, 2016, 72:168-181. [5] TAN H C, WU Y K, FENG J S, et al. Traffic Missing Data Completion with Spatial-temporal Correlations[C]//93rd Annual Meeting of the Transportation Research Board. Washington, D.C.:TRB, 2014. [6] DELL'ACQUA P, BELLOTTI F, BERTA R, et al. Time-aware Multivariate Nearest Neighbor Regression Methods for Traffic Flow Prediction[J]. IEEE Transactions on Intelligent Transportation Systems, 2015, 16(6):3393-3402. [7] QI Wei, LI Ye, WANG Zuo-xin. Utilizing Seasonal ARiMA Model for Traffic Estimation in Sparseness[J]. Journal of Highway and Transportation Research and Development, 2014, 31(4):130-135.(in Chinese) [8] WU Jin-wu, ZHANG Hai-feng, RAN Xu-dong. Nonparametric Regressive Short-term Traffic Flow Forecast Algorithm Based on Data Reduction and SVM[J]. Journal of Highway and Transportation Research and Development, 2020, 37(7):129-134.(in Chinese) [9] LV Y S, DUAN Y J, KANG W W, et al. Traffic Flow Prediction with Big Data:A Deep Learning Approach[J]. IEEE Transactions on Intelligent Transportation Systems, 2015, 16(2):865-873. [10] HUANG W H, SONG G J, HONG H K, et al. Deep Architecture for Traffic Flow Prediction:Deep Belief Networks with Multitask Learning[J]. IEEE Transactions on Intelligent Transportation Systems, 2014, 15(5):2191-2201. [11] MA X L, TAO Z M, WANG Y H, et al. Long Short-term Memory Neural Network for Traffic Speed Prediction Using Remote Microwave Sensor Data[J]. Transportation Research Part C:Emerging Technologies, 2015, 54:187-197. [12] ZHAO L, SONG Y J, ZHANG C, et al. T-GCN:A Temporal Graph Convolutional Network for Traffic Prediction[J]. IEEE Transactions on Intelligent Transportation Systems, 2020, 21(9):3848-3858. [13] YU B, YIN H T, ZHU Z X. Spatio-temporal Graph Convolutional Networks:A Deep Learning Framework for Traffic Forecasting[C]//Proceedings of the 27th International Joint Conference on Artificial IntelligenceJuly. Stockholm:eprint arXiv, 2018:3634-3640. [14] TANG Zhi-hui, ZHENG Wei-hao, DONG Wei, et al. A Method for Predicting Short-term Traffic Flow Based on Interactive IMM-BP-UKF Model[J]. Journal of Highway and Transportation Research and Development, 2019, 36(4):117-124,134.(in Chinese) [15] YANG Chun-xia, QIN Jia-peng, WANG Qing, et al. Study on Short-term Traffic Flow Prediction Based on Multi-lane Weighted Fusion[J]. Journal of Highway and Transportation Research and Development, 2021, 38(1):121-127.(in Chinese) [16] CHEN Meng, GAN Ke, LI Kai, et al. A Spatial-temporal Graph Convolutional Network Model for Accurate and Robust Traffic Flow Prediction Based on Real-time Multimodal Spatial-temporal Data[J]. Journal of Highway and Transportation Research and Development, 2021, 38(8):134-139,158.(in Chinese) [17] GOODFELLOW I J, POUGET-ABADIE J, MIRZA M, et al. Generative Adversarial Networks[J]. Advances in Neural Information Processing Systems, 2014, 3:2672-2680. [18] LÜ Y S, CHEN Y Y, LI L, et al. Generative Adversarial Networks for Parallel Transportation Systems[J]. IEEE Intelligent Transportation Systems Magazine, 2018, 10(3):4-10. [19] LIU S Z, TRENKLER G. Hadamard, Khatri-Rao, Kronecker and Other Matrix Products[J]. International Journal of Information and Systems Sciences, 2008, 4(1):160-177. [20] HUANG N E, SHEN Z, LONG S R, et al. The Empirical Mode Decomposition and the Hilbert Spectrum for Nonlinear and Non-stationary Time Series Analysis[J]. Proceedings of the Royal Society of London A, 1998, 454(1971):903-995. [21] KIPF T N, WELLING M. Semi-supervised Classification with Graph Convolutional Networks[C]//5th International Conference on Leaning Representations. Toulon:arXiv, 2017. [22] FU R, ZHANG Z, LI L. Using LSTM and GRU Neural Network Methods for Traffic Flow Prediction[C]//31st Youth Academic Annual Conference of Chinese Association of Automation (YAC). Wuhan:IEEE, 2016. [23] CHEN X Y, HE Z C, WANG J W. Spatial-temporal Traffic Speed Patterns Discovery and Incomplete Data Recovery via SVD-combined Tensor Decomposition[J]. Transportation Research Part C:Emerging Technologies, 2018, 86:59-77. [24] CHEN X Y, HE Z C, CHEN Y X, et al. Missing Traffic Data Imputation and Pattern Discovery with a Bayesian Augmented Tensor Factorization Model[J]. Transportation Research Part C:Emerging Technologies, 2019, 104:66-77. [25] GULTEKIN S, PAISLEY J. Online Forecasting Matrix Factorization[J]. IEEE Transactions on Signal Processing, 2018, 67(5):1223-1236.
2023, Vol. 17 
