中国物理B ›› 2022, Vol. 31 ›› Issue (4): 40502-040502.doi: 10.1088/1674-1056/ac3647

• • 上一篇    下一篇

Traffic flow prediction based on BILSTM model and data denoising scheme

Zhong-Yu Li(李中昱)1,2,3, Hong-Xia Ge(葛红霞)1,2,3,†, and Rong-Jun Cheng(程荣军)1,2,3   

  1. 1 Faculty of Maritime and Transportation, Ningbo University, Ningbo 315211, China;
    2 Jiangsu Provincial Collaborative Innovation Center for Modern Urban Traffic Technologies, Nanjing 210096, China;
    3 National Traffic Management Engineering and Technology Research Center Ningbo University Subcenter, Ningbo 315211, China
  • 收稿日期:2021-07-13 修回日期:2021-10-12 接受日期:2021-11-04 出版日期:2022-03-16 发布日期:2022-03-21
  • 通讯作者: Hong-Xia Ge E-mail:gehongxia@nbu.edu.cn
  • 基金资助:
    Project supported by the Program of Humanities and Social Science of the Education Ministry of China (Grant No. 20YJA630008), the Natural Science Foundation of Zhejiang Province, China (Grant No. LY20G010004), and the K C Wong Magna Fund in Ningbo University, China.

Traffic flow prediction based on BILSTM model and data denoising scheme

Zhong-Yu Li(李中昱)1,2,3, Hong-Xia Ge(葛红霞)1,2,3,†, and Rong-Jun Cheng(程荣军)1,2,3   

  1. 1 Faculty of Maritime and Transportation, Ningbo University, Ningbo 315211, China;
    2 Jiangsu Provincial Collaborative Innovation Center for Modern Urban Traffic Technologies, Nanjing 210096, China;
    3 National Traffic Management Engineering and Technology Research Center Ningbo University Subcenter, Ningbo 315211, China
  • Received:2021-07-13 Revised:2021-10-12 Accepted:2021-11-04 Online:2022-03-16 Published:2022-03-21
  • Contact: Hong-Xia Ge E-mail:gehongxia@nbu.edu.cn
  • Supported by:
    Project supported by the Program of Humanities and Social Science of the Education Ministry of China (Grant No. 20YJA630008), the Natural Science Foundation of Zhejiang Province, China (Grant No. LY20G010004), and the K C Wong Magna Fund in Ningbo University, China.

摘要: Accurate prediction of road traffic flow is a significant part in the intelligent transportation systems. Accurate prediction can alleviate traffic congestion, and reduce environmental pollution. For the management department, it can make effective use of road resources. For individuals, it can help people plan their own travel paths, avoid congestion, and save time. Owing to complex factors on the road, such as damage to the detector and disturbances from environment, the measured traffic volume can contain noise. Reducing the influence of noise on traffic flow prediction is a piece of very important work. Therefore, in this paper we propose a combination algorithm of denoising and BILSTM to effectively improve the performance of traffic flow prediction. At the same time, three denoising algorithms are compared to find the best combination mode. In this paper, the wavelet (WL) denoising scheme, the empirical mode decomposition (EMD) denoising scheme, and the ensemble empirical mode decomposition (EEMD) denoising scheme are all introduced to suppress outliers in traffic flow data. In addition, we combine the denoising schemes with bidirectional long short-term memory (BILSTM) network to predict the traffic flow. The data in this paper are cited from performance measurement system (PeMS). We choose three kinds of road data (mainline, off ramp, on ramp) to predict traffic flow. The results for mainline show that data denoising can improve prediction accuracy. Moreover, prediction accuracy of BILSTM+EEMD scheme is the highest in the three methods (BILSTM+WL, BILSTM+EMD, BILSTM+EEMD). The results for off ramp and on ramp show the same performance as the results for mainline. It is indicated that this model is suitable for different road sections and long-term prediction.

关键词: traffic flow prediction, bidirectional long short-term memory network, data denoising

Abstract: Accurate prediction of road traffic flow is a significant part in the intelligent transportation systems. Accurate prediction can alleviate traffic congestion, and reduce environmental pollution. For the management department, it can make effective use of road resources. For individuals, it can help people plan their own travel paths, avoid congestion, and save time. Owing to complex factors on the road, such as damage to the detector and disturbances from environment, the measured traffic volume can contain noise. Reducing the influence of noise on traffic flow prediction is a piece of very important work. Therefore, in this paper we propose a combination algorithm of denoising and BILSTM to effectively improve the performance of traffic flow prediction. At the same time, three denoising algorithms are compared to find the best combination mode. In this paper, the wavelet (WL) denoising scheme, the empirical mode decomposition (EMD) denoising scheme, and the ensemble empirical mode decomposition (EEMD) denoising scheme are all introduced to suppress outliers in traffic flow data. In addition, we combine the denoising schemes with bidirectional long short-term memory (BILSTM) network to predict the traffic flow. The data in this paper are cited from performance measurement system (PeMS). We choose three kinds of road data (mainline, off ramp, on ramp) to predict traffic flow. The results for mainline show that data denoising can improve prediction accuracy. Moreover, prediction accuracy of BILSTM+EEMD scheme is the highest in the three methods (BILSTM+WL, BILSTM+EMD, BILSTM+EEMD). The results for off ramp and on ramp show the same performance as the results for mainline. It is indicated that this model is suitable for different road sections and long-term prediction.

Key words: traffic flow prediction, bidirectional long short-term memory network, data denoising

中图分类号:  (Transport processes)

  • 05.60.-k
45.70.Vn (Granular models of complex systems; traffic flow) 02.30.Oz (Bifurcation theory)