A hybrid machine learning approach for congestion prediction and warning

Dongxue Li Yao Hu Chuliang Wu Wangyong Chen Feiyun Wang School of Mathematics and Statistics,Guizhou University,Guiyang,People's Republic of China
DOI: https://doi.org/10.1080/03081060.2024.2367751
2024-06-27
Transportation Planning and Technology
Abstract:Global traffic management encounters a significant challenge in traffic congestion. This paper presents a hybrid machine learning method for predicting traffic congestion. It leverages Convolutional Neural Network-Gated Recurrent Unit (CNN-GRU) network for parameter prediction and combines clustering and classification models, using the K-means algorithm to categorize traffic states into four levels and constructing a KNN classification model based on this segmentation. This results in the K-means-KNN model. Predicted parameters are inputted into the K-means-KNN model for congestion level prediction. Validation with real traffic flow data shows that the CNN-GRU network can capture spatiotemporal features more effectively. For instance, in traffic flow prediction, it reduces the MAPE by 7.39% and 51.14% compared to CNN and GRU, respectively. K-means-KNN excels in traffic state discrimination, achieving a congestion prediction accuracy of 91.8%. These results underscore the efficacy of the hybrid machine learning method in assessing and predicting urban traffic congestion.
transportation science & technology
What problem does this paper attempt to address?