Abstract:Passenger flow prediction has drawn increasing attention in the deep learning research field due to its great importance in traffic management and public safety. The major challenge of this essential task lies in multiple spatiotemporal correlations that exhibit complex non-linear correlations. Although both the spatial and temporal perspectives have been considered in modeling, most existing works have ignored complex temporal correlations or underlying spatial similarity. In this paper, we identify the unique spatiotemporal correlation of urban metro flow, and propose an attention-based deep spatiotemporal network with multi-task learning (ADST-Net) at a citywide level to predict the future flow from historical observations. ADST-Net uses three independent channels with the same structure to model the recent, daily-periodic and weekly-periodic complicated spatiotemporal correlations, respectively. Specifically, each channel uses the framework of residual networks, the rectified block and the multi-scale convolutions to mine spatiotemporal correlations. The residual networks can effectively overcome the gradient vanishing problem. The rectified block adopts an attentional mechanism to automatically reweigh measurements at different time intervals, and the multi-scale convolutions are used to extract explicit spatial relationships. ADST-Net also introduces an external embedding mechanism to extract the influence of external factors on flow prediction, such as weather conditions. Furthermore, we enforce multi-task learning to utilize transition passenger flow volume prediction as an auxiliary task during the training process for generalization. Through this model, we can not only capture the steady trend, but also the sudden changes of passenger flow. Extensive experimental results on two real-world traffic flow datasets demonstrate the obvious improvement and superior performance of our proposed algorithm compared with state-of-the-art baselines.

Disentanglement-Guided Spatial-Temporal Graph Neural Network for Metro Flow Forecasting (Student Abstract)

Deep Graph Gaussian Processes for Short-Term Traffic Flow Forecasting from Spatiotemporal Data

SPATIOTEMPORAL GRAPH CONVOLUTIONAL NEURAL NETWORKS FOR METRO FLOW PREDICTION

Spatiotemporal Virtual Graph Convolution Network for Key Origin-Destination Flow Prediction in Metro System

Transformer Based Spatial-Temporal Fusion Network for Metro Passenger Flow Forecasting.

Multi-STGCnet: A Graph Convolution Based Spatial-Temporal Framework for Subway Passenger Flow Forecasting

Metro Traffic Flow Prediction via Knowledge Graph and Spatiotemporal Graph Neural Network

MGSTCN: A Multi-Graph Spatio-Temporal Convolutional Network for Metro Passenger Flow Prediction

Spatial-Temporal Fusion Graph Neural Networks for Traffic Flow Forecasting

Traffic Flow Forecasting with Spatial-Temporal Graph Diffusion Network

ADST: Forecasting Metro Flow Using Attention-Based Deep Spatial-Temporal Networks with Multi-Task Learning

Spatio-Temporal Dynamic Graph Relation Learning for Urban Metro Flow Prediction

Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecasting

Dynamic Relational Graph Convolutional Network for Metro Passenger Flow Forecasting

Multi-Spatio-Temporal Convolutional Neural Network for Short-Term Metro Passenger Flow Prediction

Adaptive spatial-temporal graph attention networks for traffic flow forecasting

QuoGNN: Quotient Graph Neural Network for Urban Flow Forecasting.

Topology augmented dynamic spatial-temporal network for passenger flow forecasting in urban rail transit

Metro Passenger Flow Prediction Via Dynamic Hypergraph Convolution Networks

Dynamic spatial aware graph transformer for spatiotemporal traffic flow forecasting

Short-Term Metro Origin-Destination Passenger Flow Prediction via Spatio-Temporal Dynamic Attentive Multi-Hypergraph Network