Spatial-temporal Dynamic Semantic Graph Neural Network

Zhang Rui,Xie Fei,Sun Rui,Huang Lei,Liu Xixiang,Shi Jianjun
DOI: https://doi.org/10.1007/s00521-022-07285-3
2022-01-01
Abstract:Most existing methods based on graph neural network for traffic flow forecasting cannot effectively exploit potential semantic features, multiple features are aggregated without refining the proportion of their respective weights, and the over-smoothing problem limits the stacked depth of the model. To solve these problems, spatial-temporal dynamic semantic graph neural network is proposed in this paper. Firstly, two different semantic features have been captured by dynamic time warping algorithm and Pearson correlation coefficient for constructing two semantic adjacency matrices. Secondly, a dynamic aggregation method is proposed that learns the weighting ratio corresponding to each feature through training. Thirdly, the injection-stacked structure is designed to solve the over-smoothing problem and allow the network to be stacked with more layer and improve the forecasting accuracy. Finally, the experiments on four PEMS datasets with various methods such as spatio-temporal graph convolutional networks, attention-based spatial-temporal graph convolutional networks, etc. verify that spatial-temporal dynamic semantic graph neural network obtains minimal forecasting errors by capturing the potential semantic features, dynamically aggregating multiple features, and deepening the network layers by injecting-stacked structure. It achieves that root mean square error is 25.59, mean absolute error is 16.12 and mean absolute percentage error is 16.15 on the PEMS03 dataset.
What problem does this paper attempt to address?