Deep traffic congestion prediction model based on road segment grouping

Yue Tu,Shukuan Lin,Jianzhong Qiao,Bin Liu
DOI: https://doi.org/10.1007/s10489-020-02152-x
IF: 5.3
2021-04-07
Applied Intelligence
Abstract:Intelligent Transportation System (ITS) is now being widely built all over the world. Traffic congestion prediction, as a major part of ITS, not only provides reliable traffic information for travelers to save their time, but also assists traffic management agencies to manage the traffic system. We find that the existing papers, including neural networks, do not perform well due to the complex association features between the road sections. In addition, the lack of a better evaluation standard for traffic congestion also makes the effect of traffic congestion prediction worse. In this paper, we propose a method to generate traffic congestion index by mining free-stream speed and free-stream flow. Considering the association features of road segments in the road network, we propose a road segment grouping method based on association subgraph to pre-train deep learning model and realize information sharing between road segments. Combining the features of traffic data and CNN model, we propose a traffic congestion prediction model named SG-CNN which training process is optimized by road segment grouping algorithm. The experiments demonstrate that our proposed model has higher accuracy rate than other models.
computer science, artificial intelligence
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve The paper primarily aims to address several key issues in traffic congestion prediction: 1. **Comprehensive Data Utilization Issue**: - Most existing methods use only a single type of data (such as traffic flow or vehicle speed) to predict traffic congestion, which can lead to inaccurate judgments. Therefore, how to integrate traffic flow data and vehicle speed data to measure traffic congestion comprehensively becomes an urgent problem to solve. - To this end, the authors propose a new metric called the "Traffic Congestion Index" (TCI), which comprehensively describes traffic congestion by combining flow data and speed data. 2. **Spatiotemporal Correlation Issue**: - Traffic data has spatiotemporal correlations, and data from different road segments are also interrelated. However, many existing models ignore this aspect. - To solve this problem, the authors propose a segment grouping method to mine the spatiotemporal correlations between different road segments and use it for the pre-training of deep learning models. 3. **Deep Neural Network Model Construction Issue**: - How to construct a deep neural network model that can fully utilize the correlations between road segments to improve the model's robustness is also a problem that needs to be addressed. - The authors propose a new model—SG-CNN, which uses lattice theory to generate TCI matrices of related road segments as input, thereby capturing complex interactions. Through the above methods, the authors aim to improve the accuracy of traffic congestion prediction and demonstrate the superior performance of their proposed model on multiple datasets.