Short-Term Traffic Flow Prediction Based on Wavelet Analysis and XGBoost

Author:

Wang Xin1ORCID,Fang Fang1

Affiliation:

1. School of Applied Science, Beijing Information Science and Technology University, Beijing, China

Abstract

Traffic flow prediction is of great significance for urban planning and alleviating traffic congestion. Due to the randomness and high volatility of urban road network short-term traffic flow, it is difficult for a single model to accurately estimate traffic flow and travel time. In order to obtain more ideal prediction accuracy, a combined prediction model based on wavelet decomposition and reconstruction (WDR) and the extreme gradient boosting (XGBoost) model is developed in this paper. Firstly, the Mallat algorithm is applied to perform multi-scale wavelet decomposition on the average travel time series of the original traffic data, and single branch reconstruction is performed on the components at each scale. Secondly, XGBoost is used to predict each reconstructed single-branch sequence, so as to obtain multiple sub-models, and the Bayesian algorithm is used to optimize the hyperparameters of the sub-models. Finally, the algebraic sum of the predicted values of all sub-models is used to obtain the overall traffic prediction result. To test the performance of the proposed model, actual traffic flow data has been collected from a certain link of the Brooklyn area in New York, USA. The performance of proposed WDR-XGBoost model has been compared with other existing machine learning models, e.g., support vector regression model (SVR) and single XGBoost model. Experimental findings demonstrated that the proposed WDR-XGBoost model performs better on multiple evaluation indicators and has significantly outperformed the other models in terms of accuracy and stability.

Publisher

Science Publishing Group

Reference30 articles.

1. Zhao, H., Zhai, D. M., Shi, Z. H. Review of short-term traffic flow forecasting models. Urban Rapid Rail Transit. 2019, 32(4), 50-54. https://doi.org/10.3969/j.issn.1672-6073.2019.04.009

2. Li, W., L, J. Z., Wang, T. Improved ARIMA model traffic flow prediction method based on box-cox exponential transformation. Journal of Wuhan University of Technology (Transportation Science & Engineering). 2020, 44(6), 974-977. https://doi.org/10.3963/j.issn.2095-3844.2020.06.006

3. Zhou, T. Jiang, D. Lin, Z. et al. Hybrid dual Kalman filtering model for short-term traffic flow forecasting. IET Intelligent Transport Systems. 2019, 13(6), 1023-1032. https://doi.org/10.1049/iet-its.2018.5385

4. Cai, L. Zhang, Z. Yang, J. et al. A noise-immune Kalman filter for short-term traffic flow forecasting. Physica A: Statistical Mechanics and its Applications. 2019, 536, 122601. https://doi.org/10.1016/j.physa.2019.122601

5. Liao, R. H., Lan, S., Liu, Z. X. Short-term traffic flow forecasting based on local prediction method in chaotic time series. Computer Technology and Development. 2015, 25 (1), 1-5. https://doi.org/10.3969/j.issn.1673-629X.2015.01.001

同舟云学术

1.学者识别学者识别

2.学术分析学术分析

3.人才评估人才评估

"同舟云学术"是以全球学者为主线,采集、加工和组织学术论文而形成的新型学术文献查询和分析系统,可以对全球学者进行文献检索和人才价值评估。用户可以通过关注某些学科领域的顶尖人物而持续追踪该领域的学科进展和研究前沿。经过近期的数据扩容,当前同舟云学术共收录了国内外主流学术期刊6万余种,收集的期刊论文及会议论文总量共计约1.5亿篇,并以每天添加12000余篇中外论文的速度递增。我们也可以为用户提供个性化、定制化的学者数据。欢迎来电咨询!咨询电话:010-8811{复制后删除}0370

www.globalauthorid.com

TOP

Copyright © 2019-2024 北京同舟云网络信息技术有限公司
京公网安备11010802033243号  京ICP备18003416号-3