Prediction of High-Speed Traffic Flow around City Based on BO-XGBoost Model

• Published in: Symmetry (Basel) Vol. 15; no. 7; p. 1453
• Main Authors: Lu, Xin, Chen, Cai, Gao, RuiDan, Xing, ZhenZhen
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.07.2023
• Subjects: Accuracy; Algorithms; Bayesian analysis; bayesian optimization; Bi-LSTM; China; Cities; Datasets; Distance learning; Economic development; Economic planning; Forecasts and trends; Freeways; High speed; Literature reviews; Machine learning; Neural networks; Optimization; Roads & highways; spatiotemporal characteristics; Time series; Traffic congestion; Traffic flow; traffic flow prediction; Traffic models; Traffic speed; Urban planning; XGBoost algorithm; China
• ISSN: 2073-8994; 2073-8994
• DOI: 10.3390/sym15071453

The prediction of high-speed traffic flow around the city is affected by multiple factors, which have certain particularity and difficulty. This study devised an asymmetric Bayesian optimization extreme gradient boosting (BO-XGBoost) model based on Bayesian optimization for the spatiotemporal and multigranularity prediction of high-speed traffic flow around a city. First, a traffic flow dataset for a ring expressway was constructed, and the data features were processed based on the original data. The data were then visualized, and their spatiotemporal distribution exhibited characteristics such as randomness, continuity, periodicity, and rising fluctuations. Secondly, a feature matrix was constructed monthly for the dataset, and the BO-XGBoost model was used for traffic flow prediction. The proposed model BO-XGBoost was compared with the symmetric model bidirectional long short-term memory and integrated models (random forest, extreme gradient boosting, and categorical boosting) that directly input temporal data. The R-squared (R2) of the BO XGBoost model for predicting TF and PCU reached 0.90 and 0.87, respectively, with an average absolute percentage error of 2.88% and 3.12%, respectively. Thus, the proposed model achieved an accurate prediction of high-speed traffic flow around the province, providing a theoretical basis and data support for the development of central-city planning.