Abstract:Spatio-temporal prediction has drawn much attention given its wide application, of which traffic flow prediction is a typical task. Within the vision of smart cities, traffic flow prediction plays a vital role in traffic control and optimization. The current approaches commonly use a graph convolutional network (GCN) to capture any spatial correlations and a recurrent neural network (RNN) to mine any temporal correlations. However, GCNs cannot detect spatial heterogeneity and time-varying spatial correlations, and RNNs cannot model the periodicity of traffic series data. Further, iterative training of RNNs may come at a high computational cost and result in problems with error propagation. To this end, we propose STSSN, a spatio-temporal sequence-to-sequence network, that not only explores heterogeneous and time-varying spatial correlations, but also efficiently exploits sequential and periodic temporal correlations. STSSN is based on an encoder-decoder framework. In the network, the model’s input is processed to extract the periodic daily and weekly patterns in traffic flows. Both the encoder and decoder mainly consist of an enhanced diffusion convolutional network (EDCN) and a temporal convolutional network (TCN). In the EDCN module, the diffusion convolution incorporates time-varying node representations so as to capture both node-specific patterns and time-varying spatial correlations. In the TCN module, we take full advantage of the parallel computing in the dilated causal convolution to mine local (short-term) temporal correlations. More importantly, global (long-term) temporal correlations are discovered through an encoder-decoder attention (EDA) module. This EDA mechanism directly models the relationship between the encoder and decoder to mitigate problems with error propagation. Experiments on two real-world datasets verify the superiority of STSSN, with STSSN’s MAE at between 3.85%-6.17% lower than the state-of-the-art baselines on the PEMS-BAY dataset.

PSRUNet: a recurrent neural network for spatiotemporal sequence forecasting based on parallel simple recurrent unit

PredRNN: Recurrent Neural Networks for Predictive Learning using Spatiotemporal LSTMs

PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learning

RPP-Net: Rigid Constrained Point Cloud Prediction Network

Temporal Attention Unit: Towards Efficient Spatiotemporal Predictive Learning

SARNN: A Spatiotemporal Prediction Model for Reducing Error Transmissions

RSNN: Recurrent Spiking Neural Networks for Dynamic Spatial-Temporal Information Processing

TSARNet :Integrated Spatio-Temporal Attention Mechanism for Multivariate Time Series Prediction Deep Learning Framework

Enhanced Spatiotemporal Prediction Using Physical-guided And Frequency-enhanced Recurrent Neural Networks

PredRNN++: Towards A Resolution of the Deep-in-Time Dilemma in Spatiotemporal Predictive Learning

STIP: A SpatioTemporal Information-Preserving and Perception-Augmented Model for High-Resolution Video Prediction

A Lightweight Hardware‐efficient Recurrent Network for Video Super‐resolution

Parallel spatio-temporal attention-based TCN for multivariate time series prediction

SPRNN: A spatial–temporal recurrent neural network for crowd flow prediction

SegRNN: Segment Recurrent Neural Network for Long-Term Time Series Forecasting

PGN: The RNN's New Successor is Effective for Long-Range Time Series Forecasting

Costnet: A Concise Overpass Spatiotemporal Network For Predictive Learning

Z-Order Recurrent Neural Networks For Video Prediction

Spatiotemporal Self-Attention-Based LSTNet for Multivariate Time Series Prediction

GSSTU: Generative Spatial Self-Attention Transformer Unit for Enhanced Video Prediction

A Spatio-Temporal Sequence-to-sequence Network for Traffic Flow Prediction