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.

Spatio-Temporal Attention Hybrid Network for Short-Term Origin-Destination Flow Prediction in Urban Subways

Space-Time Adaptive Network for Origin-Destination Passenger Demand Prediction

Hybrid Approach Combining Modified Gravity Model and Deep Learning for Short-Term Forecasting of Metro Transit Passenger Flows

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

Spatio-Temporal Self-Attention Network for Origin–Destination Matrix Prediction in Urban Rail Transit

Short-term origin-destination demand prediction in urban rail transit systems: A channel-wise attentive split-convolutional neural network method

Deep Learning for Metro Short-Term Origin-Destination Passenger Flow Forecasting Considering Section Capacity Utilization Ratio

Completion and Augmentation based Spatiotemporal Deep Learning Approach for Short-Term Metro Origin-Destination Matrix Prediction under Limited Observable Data

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

Spatial-Temporal Attention Fusion Network for short-term passenger flow prediction on holidays in urban rail transit systems

Attention Mechanism with Spatial-Temporal Joint Deep Learning Model for the Forecasting of Short-Term Passenger Flow Distribution at the Railway Station

Learning spatial–temporal pairwise and high-order relationships for short-term passenger flow prediction in urban rail transit

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

Short-Term Prediction of Origin–Destination Passenger Flow in Urban Rail Transit Systems with Multi-Source Data: A Deep Learning Method Fusing High-Dimensional Features

Short-Term Passenger Flow Prediction in Urban Rail Transit Based on Points of Interest

Deep spatio-temporal neural network based on interactive attention for traffic flow prediction

A spatiotemporal hierarchical attention mechanism-based model for multi-step station-level crowd flow prediction

Short-Term Online Forecasting for Passenger Origin–Destination (OD) Flows of Urban Rail Transit: A Graph–Temporal Fused Deep Learning Method

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

Multi-sequence spatio-temporal feature fusion network for peak-hour passenger flow prediction in urban rail transit

Multi-level spatial-temporal fusion neural network for traffic flow prediction