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

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.

A Novel Spatial–Temporal Deep Learning Method for Metro Flow Prediction Considering External Factors and Periodicity

“…Hongwei Jia et al proposed a network which uses three independent channels with the same structure to model recent, daily periodic, and weekly periodic complicated spatiotemporal correlations, respectively. This model captured not only the steady trend, but also the sudden changes in passenger flow [28]. Peikun Li et al proposed a framework of short-term passenger flow to explore the factors that influence prediction accuracy based on time granularity and station class [29].…”

Graph Based Learning for Building Prediction in Smart Cities

“…In [14], the proposed multi-view graph convolutional network (MVGCN) is used to predict the inflow and outflow in each and every irregular region of a city to integrate the geospatial position via spatial graph convolutions. In [15] the proposed attention-based deep spatio-temporal network, with multi-task learning (ADST-Net) at a citywide level, creates a goal to predict urban traffic flow. ADST-Net furthermore introduces an outside embedding mechanism to extricate the impact of external factors on flow prediction, such as weather conditions.…”

Identifying Subway Passenger Flow under Large-Scale Events Using Symbolic Aggregate Approximation Algorithm

“…Recently, the increasing use of temporal data, in particular time series data, has initiated various research and development attempts in the field of data mining (25)(26)(27)(28). A time series is a collection of observations made chronologically, and is characterized by its numerical and continuous nature (7).…”