关键词: 交通流预测, 图自编码器, 门控循环单元, 分层, 时空依赖

Abstract: Accurate traffic flow prediction information not only provides traffic administrator with a strong foundation for traffic decisions,but also eases congestion.In traffic flow forecasting tasks,obtaining valid spatiotemporal characteristics of the traffic flow is a prerequisite to ensure the effectiveness of the forecast.Most of the existing methods use data from future moments for supervised learning,and the extracted features have limitations.To address the problem that existing prediction models cannot fully exploit the spatiotemporal characteristics of traffic flows,this paper proposes a hierarchical traffic prediction model based on an improved graph autoencoder and gated recurrent unit.The graph attention autoencoder is first used to deeply explore the spatial characteristics of the traffic flow in an unsupervised manner,and then the gated recurrent unit is used to extract temporal features.The hierarchical structure uses separate training for learning spatio-temporal dependencies,aiming to capture the naturally existing spatial topological features of the road network and make it compatible with traffic flow prediction tasks at different time steps.Extensive experiments demonstrate that the proposed GAE-GRU model achieves excellent performance in traffic prediction tasks on different datasets,with MAE,RMSE and MAPE outperforming the baseline model.

Key words: Traffic flow forecasting, Graph autoencoder, Gated recurrent neural unit, Hierarchical, Spatiotemporal dependencies