Fine-grained Graph Convolutional Network with Learning-Based Bi-Relational Graph for Spatiotemporal Forecasting
Hexiang Liu,Qilong Han,Dan Lu,Jingyu Sheng,Shanshan Sui,Hui Sun
DOI: https://doi.org/10.1016/j.eswa.2024.125959
IF: 8.5
2024-01-01
Expert Systems with Applications
Abstract:Accurate spatiotemporal forecasting is crucial for optimizing traffic resource allocation and air pollution warnings. Recent studies describe the spatial correlations between stations with the help of pre-defined graphs and then use graph convolutional network (GCN) to learn node representations by aggregating neighborhood information of all nodes uniformly to capture spatial dependencies. However, pre-defined graphs can not accurately describe the complex spatial correlations between stations. In addition, these GCN-based models not only suffer from over-smoothing issues when stacking multiple GCN layers but also exhibit limited structure-aware (i.e., fewer links) on low-degree nodes, leading to nodes capturing contextual information on local spatial proximity. To this end, in this paper, we propose a novel learning-based spatial graph, which can effectively learn the spatial collaborative correlations between stations, including geographic proximity and functional similarity (e.g., two stations both located in residential areas, thus sharing strong regional dependencies), and represent them as an adjacency matrix. Moreover, we propose a fine-grained graph convolutional network, named FGGCN, aimed to enhance the representations of low-degree nodes in the spatial graph. The core advantage is that FGGCN aggregates neighborhood information for the differences in structure-aware of high- and low-degree nodes, respectively, and adaptively learns the importance weights of different layer representations, allowing the nodes to aggregate the abundance global contextual information. We further incorporate the spatial representations captured by FFGCN into an Encoder-Decoder architecture, recursively capturing spatiotemporal autocorrelation and generating multi-step predictions. We extensive experiments on five real-world spatiotemporal datasets, covering air quality and bike flow. The experimental results demonstrate the state-of-the-art of our propose method.