Traffic flow prediction by an ensemble framework with data denoising and deep learning model

• Published in: Physica A 2021-03, Vol.565, p.125574, Article 125574
• Main Authors: Chen, Xinqiang, Chen, Huixing, Yang, Yongsheng, Wu, Huafeng, Zhang, Wenhui, Zhao, Jiansen, Xiong, Yong
• Format: Article
• Language: English
• Subjects: Data quality control; Ensemble framework; LSTM network; Traffic flow prediction
• ISSN: 0378-4371; 1873-2119
• DOI: 10.1016/j.physa.2020.125574

Accurate traffic flow data is important for traffic flow state estimation, real-time traffic management and control, etc. Raw traffic flow data collected from inductive detectors may be contaminated by different noises (e.g., sharp data increase/decrease, trivial anomaly oscillations) under various unexpected interference (caused by roadway maintenance, loop detector damage, etc.). To address the issue, we introduced data denoising schemes (i.e., Empirical Mode Decomposition (EMD), Ensemble Empirical Mode Decomposition (EEMD) and Wavelet (WL)) to suppress the potential data outliers. After that, the Long Short-Term Memory (LSTM) neural network was introduced to fulfill the traffic flow prediction task. We have tested the proposed framework performance on three traffic flow datasets, which were downloaded from Caltrans Performance Measurement System (PeMS). The experimental results showed that the LSTM+EEMD scheme obtained higher accuracy considering that the average Root Mean Square Error (RMSE), Mean Absolute Error (MAE) and Mean Absolute Percentage Error (MAPE) are 0.79, 0.60 and 2.14.