基于GCN-AGRU的短时交通流预测OA
Short-Term Traffic Flow Preiction Based on GCN-AGRU
针对现有短时交通流预测方法未考虑交通流频域特征的问题,文中构建了一种新型GCN-AGRU(Graph Convolutional Neural Network-Attention Gated Recurrent Unit)短时交通流预测模型.将基于自适应图结构的图卷积神经网络(Graph Convolutional Neural Network,GCN)与门控循环单元(Gated Recurrent Unit,GRU)相结合,通过使用频域通道注意力(Frequency Channel Attention,FCA)对GRU的隐藏状态进行频域分析,从而能够更准确地捕捉交通流中的周期性特征.基于武汉市路网数据集和PeMS04数据集进行实验,并对比分析了其他模型的预测结果.相较于最优基线模型,所提模型的MAE(Mean Absolute Error)、RMSE(Root Mean Square Error)和MAPE(Mean Absolute Percentage Error)分别降低了9.84百分点、1.87百分点和4.88百分点,显著提升了短时交通流预测的准确性和稳定性.
In view of the problem that the existing short-term traffic flow prediction methods do not consider the frequency-domain characteristics of traffic flow,a new GCN-AGRU(Graph Convolutional Neural Network-At-tention Gated Recurrent Unit)short-term traffic flow prediction model is constructedr.Combining the GCN(Graph Convolutional Neural Network)network based on adaptive graph structure with the GRU(Gated Recurrent Unit),and conducting frequency-domain analysis of the hidden state of the GRU by using FCA(Frequency Channel Atten-tion)attention,the periodic characteristics in traffic flow can be captured more accurately.Experiments were con-ducted based on the Wuhan Road network dataset and the PeMS04 dataset,and the prediction results of other mod-els were compared and analyzed.Compared with the optimal baseline model,the MAE(Mean Absolute Error)、RMSE(Root Mean Square Error)and MAPE(Mean Absolute Percentage Error)of the proposed model decreased by 9.84 percentage points,1.87 percentage points and 4.88 percentage points respectively,significantly improving the accuracy and stability of short-term traffic flow prediction.
王庆国;赵磊
武汉科技大学 汽车与交通工程学院,湖北 武汉 430081武汉科技大学 汽车与交通工程学院,湖北 武汉 430081
信息技术与安全科学
交通流预测GCNGRU频域分析周期性自适应图空间信息注意力机制
traffic flow predictionGCNGRUfrequency domain analysisperiodicityadaptive graphspatial informationattention mechanism
《电子科技》 2026 (8)
62-68,7
国家自然科学基金(41571396)武汉市重点研发计划(2024050702030122)National Natural Science Foundation of China(41571396)Wuhan Key Research and Development Program(2024050702030122)
评论