STGNN-FAM: A Traffic Flow Prediction Model for Spatiotemporal Graph Networks Based on Fusion of Attention Mechanisms

Abstract

Network traffic state prediction has been constantly challenged by complex spatiotemporal features of traffic information as well as imperfection in streaming data. This paper proposes a traffic flow prediction model for spatiotemporal graph networks based on fusion of attention mechanisms (STGNN-FAM) to simultaneously tackle these challenges. This model contains a spatial feature extraction layer, a bidirectional temporal feature extraction layer, and an attention fusion layer, which not only fully considers the temporal and spatial features of the traffic flow problem but also uses the attention mechanism to enhance the critical temporal and spatial features to achieve more accurate and robust predictions. Experimental results on a network traffic speed dataset PeMSD7 show that the proposed STGNN-FAM outperforms several important benchmarks in prediction accuracy and the ability to withstand interference in the data stream, especially for mid- and long-term prediction of 30 minutes and 45 minutes.