广西师范大学学报（自然科学版）2026，Vol.44Issue(1)：10-22,13.DOI:10.16088/j.issn.1001-6600.2025022702
基于D2STGNN的双向高效多尺度交通流预测
Bidirectional Efficient Multi-scale Traffic Flow Prediction Based on D2STGNN
摘要
Abstract
Due to the complexity of traffic flow and the insufficient extraction of spatio-temporal features,it is difficult for D2STGNN to capture the dynamic changes of traffic networks,which limits the improvement of prediction accuracy.In this paper,a Bi-EMHGRU model combining an efficient multi-head self-attention mechanism(EMHSA)and a bidirectional gated recurrent unit(BiGRU)is proposed.This model captures the sequential dependencies of both forward and backward timings through BiGRU and dynamically allocates weights to each time step by using the multi-head self-attention mechanism to focus on key sequential features.Meanwhile,a multi-scale time feature extraction module is introduced,which enhances the modeling ability for short-term fluctuations and long-term trends and improves the modeling effect of complex spatio-temporal dynamics.The experimental results show that Bi-EMHGRU performs excellently on the PEMS04 and PEMS08 datasets.The root mean square error value has decreased by approximately 0.55～1.55,the mean absolute error has decreased by approximately 0.89～1.40,and the mean absolute percentage error has decreased by approximately 0.86～1.77 percentage points.It can still maintain stable prediction performance when the prediction step length increases and has strong generalization ability.Compared with the existing benchmark models,Bi-EMHGRU can capture the dynamic spatio-temporal features of traffic flow more effectively and significantly improves the prediction accuracy and robustness.关键词
交通流预测/多头自注意力机制/Bi-EMHGRU/动态时空特征/多尺度时间特征Key words
traffic flow prediction/multi-head self-attention/Bi-EMHGRU/dynamic spatio-temporal features/multi-scale temporal features分类
交通工程引用本文复制引用
黄艳国,肖洁,吴水清..基于D2STGNN的双向高效多尺度交通流预测[J].广西师范大学学报（自然科学版）,2026,44(1):10-22,13.基金项目
国家自然科学基金(72061016) （72061016）
