Abstract:Traffic forecasting, a core technology within Intelligent Transportation Systems, holds broad application prospects due to its ability to accurately predict future traffic states through the modeling and analysis of complex spatio-temporal traffic data. Nevertheless, due to the complex temporal and spatial heterogeneity of traffic sequences, existing models are difficult to effectively solve the non-stationary problems caused by emergencies. To this end, this paper proposes a Quadratic Memory-Augmented Spatio-Temporal Transformer Graph Recurrent Network (QMAGRN) model based on an encoder-decoder framework. The model consists of three parts: a spatio-temporal Transformer encoder, a quadratic memory-augmented (QMA) module, and a graph convolutional recurrent neural network (GCRU) decoder. Specifically, the spatio-temporal transformer encoder captures the complex spatio-temporal dependencies in traffic data. We designed the QMA module to dynamically update its memory based on incoming data, enabling it to adapt to changing patterns and trends. The QMA module queries the feature information of the memory module on the time and space axis and uses the attention weighting method to perform feature fusion, thereby enhancing the encoder's ability to capture complex spatio-temporal information. This allows the model to maintain information from earlier periods and provide context that helps understand long-term trends and changes, thereby addressing the non-stationarity of traffic data. The GCRU decoder utilizes the features generated by the QMA module as input for its recurrent units. The graph convolutional layers amalgamate historical information from neighboring nodes, thereby enhancing the spatial consistency of predictions. We conducted extensive experiments on five real datasets, and the results demonstrate that our model has achieved state-of-the-art performance. Furthermore, visualizing the learned QMA module enhances the interpretability of the model. Our code and data are accessible via this link: https://anonymous.4open.science/r/QMAGRN-08CD

Quadratic memory-augmented spatio-temporal transformer graph convolutional recurrent network for traffic forecasting

Memory attention enhanced graph convolution long short‐term memory network for traffic forecasting

Augmented Multi-Component Recurrent Graph Convolutional Network for Traffic Flow Forecasting

Meta Attentive Graph Convolutional Recurrent Network for Traffic Forecasting

TSSRGCN: Temporal Spectral Spatial Retrieval Graph Convolutional Network for Traffic Flow Forecasting.

Attention-based Spatial-Temporal Graph Convolutional Recurrent Networks for Traffic Forecasting

Attention-Based Multiple Graph Convolutional Recurrent Network for Traffic Forecasting

Multi-scale attention graph convolutional recurrent network for traffic forecasting

DAGCRN: Graph Convolutional Recurrent Network for Traffic Forecasting with Dynamic Adjacency Matrix

Temporal Metrics Based Aggregated Graph Convolution Network for traffic forecasting

Network Traffic Prediction via Deep Graph-Sequence Spatiotemporal Modeling Based on Mobile Virtual Reality Technology

Spatial-temporal traffic modeling with a fusion graph reconstructed by tensor decomposition

Adaptive Multi-receptive Field Spatial-Temporal Graph Convolutional Network for Traffic Forecasting

Temporal Multi-Graph Convolutional Network for Traffic Flow Prediction

Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting

Spatial dynamic graph convolutional network for traffic flow forecasting

SGRU: A High-Performance Structured Gated Recurrent Unit for Traffic Flow Prediction

Dynamic Spatial-Temporal Memory Augmentation Network for Traffic Prediction

Hierarchical Spatio–Temporal Graph Convolutional Networks and Transformer Network for Traffic Flow Forecasting

MSA-GCN: Multistage Spatio-Temporal Aggregation Graph Convolutional Networks for Traffic Flow Prediction