Self-attention Based Multimodule Fusion Graph Convolution Network for Traffic Flow Prediction

Lijie Li,Hongyang Shao,Junhao Chen,Ye Wang
DOI: https://doi.org/10.1007/978-981-19-5194-7_1
2022-01-01
Abstract:With rapid economic development, the per capita ownership of automobiles in our country has begun to rise year by year. More researchers have paid attention to using scientific methods to solve traffic flow problems. Traffic flow prediction is not simply affected by the number of vehicles, but also contains various complex factors, such as time, road conditions, and people flow. However, the existing methods ignore the complexity of road conditions and the correlation between individual nodes, which leads to the poor performance. In this study, a deep learning model SAMGCN is proposed to effectively capture the correlation between individual nodes to improve the performance of traffic flow prediction. First, the theory of spatiotemporal decoupling is used to divide each time of each node into finer particles. Second, multimodule fusion is used to mine the potential periodic relationships in the data. Finally, GRU is used to obtain the potential time relationship of the three modules. Extensive experiments were conducted on two traffic flow datasets, PeMSO4 and PeMS08 in the Caltrans Performance Measurement System to prove the validity of the proposed model.
What problem does this paper attempt to address?