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.

Mask- and Contrast-Enhanced Spatio-Temporal Learning for Urban Flow Prediction

Spatio-temporal fusion and contrastive learning for urban flow prediction

Forecasting Fine-Grained Urban Flows Via Spatio-Temporal Contrastive Self-Supervision

Dual-track Spatio-temporal Learning for Urban Flow Prediction with Adaptive Normalization

High-Resolution Urban Flows Forecasting with Coarse-Grained Spatiotemporal Data

A Foundation Model for Unified Urban Spatio-Temporal Flow Prediction

Fuzzy Hybrid Framework with Dynamic Weights for Short‐term Traffic Flow Prediction by Mining Spatio‐temporal Correlations

Fine-Grained Urban Flow Prediction Via a Spatio-Temporal Super-Resolution Scheme

Urban flows prediction from spatial-temporal data using machine learning: A survey

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

CSP-AIT-Net: A contrastive learning-enhanced spatiotemporal graph attention framework for short-term metro OD flow prediction with asynchronous inflow tracking

Spatial–temporal combination and multi-head flow-attention network for traffic flow prediction

Improving Intra-Urban Prediction of Regional Pedestrian Flow Using a Hybrid Deep Learning Approach

Dynamic Spatial-Temporal Representation Learning for Traffic Flow Prediction

Short-term traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework

Online Spatio-Temporal Crowd Flow Distribution Prediction for Complex Metro System

A multi-task spatio-temporal fully convolutional model incorporating interaction patterns for traffic flow prediction

Multi-scale feature enhanced spatio-temporal learning for traffic flow forecasting

Score-based Graph Learning for Urban Flow Prediction

Transport-Hub-Aware Spatial-Temporal Adaptive Graph Transformer for Traffic Flow Prediction

Improving Urban Crowd Flow Prediction on Flexible Region Partition