Mixed-Order Relation-Aware Recurrent Neural Networks for Spatio-Temporal Forecasting

Abstract

Spatio-temporal forecasting has a wide range of applications in smart city efforts, such as traffic forecasting and air quality prediction. Graph Convolutional Recurrent Neural Networks (GCRNN) are the state-of-the-art methods for this problem, which learn temporal dependencies by RNNs and exploit pairwise node proximity to model spatial dependencies. However, the spatial relations in real data are not simply pairwise but sometimes in a higher order among multiple nodes. Moreover, spatio-temporal sequences deriving from nature are often regulated by known or unknown physical laws. GCRNNs rarely take into account the underlying physics in real-world systems, which may result in degenerated performance. To address these issues, we devise a general model called Mixed-Order Relation-Aware RNN (MixRNN+) for spatio-temporal forecasting. Specifically, our MixRNN+ captures the complex mixed-order spatial relations of nodes through a newly proposed building block called Mixer, and simultaneously addressing underlying physics by the integration of a new residual update strategy. Experimental results on three forecasting tasks in smart city applications (i.e., traffic speed, taxi flow, and air quality prediction) demonstrate the superiority of our model against the state-of-the-art methods.

Publication
IEEE Transactions on Knowledge and Data Engineering, 2022

Related