Tigc-Net: Transformer-Improved Graph Convolution Network for Spatio-Temporal Prediction

Kai Chen,Zhengyuan Zhou,Yao Liu,Tianjiao Ji,Weiya Sun,Chunfeng Yang,Yang Chen,Xiao Lu
DOI: https://doi.org/10.2139/ssrn.4290302
IF: 5.1
2025-01-01
Biomedical Signal Processing and Control
Abstract:Modeling spatio-temporal sequences is an important topic yet challenging for existing neural networks. Most of the current spatio-temporal sequence prediction methods usually capture features separately in temporal and spatial dimensions or employ multiple mutually independent local spatio-temporal graphs to represent a spatio-temporal sequence. The first kind of method mentioned above is difficult to mine the complex spatio-temporal correlations, while the other is limited for the accuracy of long-term predictions. To handle these issues, this paper proposes a Transformer-Improved Graph Convolution Network for spatio-temporal prediction. Specifically, the temporal location encoding method is exploited to derive the spatio-temporal characteristics of the sequence utilizing a spatio-temporal feature fusion network. In addition, a spatio-temporal attention network is developed to enhance the spatio-temporal correlation of the sequence, and the dynamic spatial features of sequence are further extracted through the adaptive graph convolution network. A private dataset and a public dataset are employed to demonstrate the performance of the proposed TIGC-Net. The qualitative and quantitative results show that the proposed TIGC-Net can extract dynamic spatiotemporal properties more effectively, enhance the spatio-temporal correlation of sequences and improve the prediction accuracy compared with four stateof-the-art.
What problem does this paper attempt to address?