IEDSFAN: information enhancement and dynamic-static fusion attention network for traffic flow forecasting

Abstract

Abstract Accurate forecasting of traffic flow in the future period is very important for planning traffic routes and alleviating traffic congestion. However, traffic flow forecasting still faces serious challenges. Most of the existing traffic flow forecasting methods are static graph convolutional networks based on prior knowledge, ignoring the special spatial–temporal dynamics of spatial–temporal data. Using only adaptive dynamic graphs completely discards the objective and real spatial connectivity information in static graphs. To this end, we propose a novel information enhancement and dynamic-static fusion attention network (IEDSFAN). Firstly, the Multi-Graph Fusion Gating mechanism (MGFG) designed in IEDSFAN effectively fuses dynamic and static graphs to dynamically capture the hidden spatial–temporal correlation. Secondly, we construct a novel Gated Multi-head Self-Attention (GMHSA), which maps the input through the MGFG module to capture the complex spatial–temporal interactions in the features. Finally, we generate adaptive parameters to solve the problem that shared parameters cannot learn multiple traffic patterns, and enhance the expression of sequence information through the peak flag module. We conducted extensive experiments on five real-world traffic datasets, and the experimental results show that the performance of IEDSFAN is significantly better than all baselines.

Keywords

• Traffic flow forecasting
• Spatial–temporal correlation
• Graph convolutional networks
• Information enhancement
• Self-attention