关键词: 交通拥堵预测, 图神经网络, 动态自适应门控, 聚类算法, 门控时间卷积网络

Abstract:

With the continual rise in the number of motor vehicles in urban areas, traffic congestion has become increasingly severe, adversely affecting environmental protection and urban operational efficiency. Consequently, it is of critical importance to accurately predict traffic congestion for traffic management and optimization. However, existing research still faces limitations in modeling the dynamic, time-varying characteristics of traffic flow and the complex interactions among road segments. To address these challenges, a gated spatiotemporal convolutional network model based on graph neural networks was proposed to more effectively capture and predict traffic congestion. Firstly, an improved K-means clustering algorithm was employed to divide the raw data into multiple congestion-state categories, which are then incorporated as auxiliary features to enhance feature representation. Next, a gated temporal convolutional network was introduced to capture the temporal properties and dynamic dependencies in traffic data, and a dynamic adaptive gated graph convolutional network was constructed to achieve feature fusion and dynamic weight allocation through a signal generation module and a dual-modulation mechanism, thereby facilitating effective extraction of spatiotemporal features. Finally, residual connections were incorporated to improve training stability, and skip connections were utilized to integrate multi-level and multi-scale features. Experimental results on real-world PeMS08 and PeMS04 datasets demonstrate that the proposed model achieves superior prediction accuracy compared with other baseline methods.

Key words: traffic congestion prediction, graph neural network, dynamic adaptive gating, clustering algorithm, gated temporal convolutional network