DAGCAN: Decoupled Adaptive Graph Convolution Attention Network for Traffic Forecasting

It is necessary to establish a spatio-temporal correlation model in the traffic data to predict the state of the transportation system. Existing research has focused on traditional graph neural networks, which use predefined graphs and have shared parameters. But intuitive predefined graphs introduce biases into prediction tasks and the fine-grained spatio-temporal information can not be obtained by the parameter sharing model. In this paper, we consider it is crucial to learn node-specific parameters and adaptive graphs with complete edge information. To show this, we design a model based on graph structure that decouples nodes and edges into two modules. Each module extracts temporal and spatial features simultaneously. The adaptive node optimization module is used to learn the specific parameter patterns of all nodes, and the adaptive edge optimization module aims to mine the interdependencies among different nodes. Then we propose a Decoupled Adaptive Graph Convolution Attention Network for Traffic Forecasting (DAGCAN), which relies on the above two modules to dynamically capture the fine-grained spatio-temporal relationships in traffic data. Experimental results on four public transportation datasets, demonstrate that our model can further improve the accuracy of traffic prediction.