Abstract:Short-term passenger flow prediction is an important but challenging task for better managing urban rail transit (URT) systems. Some emerging deep learning models provide good insights to improve short-term prediction accuracy. However, there exist many complex spatiotemporal dependencies in URT systems. Most previous methods only consider the absolute error between ground truth and predictions as the optimization objective, which fails to account for spatial and temporal constraints on the predictions. Furthermore, a large number of existing prediction models introduce complex neural network layers to improve accuracy while ignoring their training efficiency and memory occupancy, decreasing the chances to be applied to the real world. To overcome these limitations, we propose a novel deep learning-based spatiotemporal graph generative adversarial network (STG-GAN) model with higher prediction accuracy, higher efficiency, and lower memory occupancy to predict short-term passenger flows of the URT network. Our model consists of two major parts, which are optimized in an adversarial learning manner: (1) a generator network including gated temporal conventional networks (TCN) and weight sharing graph convolution networks (GCN) to capture structural spatiotemporal dependencies and generate predictions with a relatively small computational burden; (2) a discriminator network including a spatial discriminator and a temporal discriminator to enhance the spatial and temporal constraints of the predictions. The STG-GAN is evaluated on two large-scale real-world datasets from Beijing Subway. A comparison with those of several state-of-the-art models illustrates its superiority and robustness. This study can provide critical experience in conducting short-term passenger flow predictions, especially from the perspective of real-world applications.

Traffic Flow Online Prediction Based on a Generative Adversarial Network with Multi-Source Data

DyGGAN: Traffic Flow Prediction Based on Generative Adversarial Network and Dynamic Graph Convolution

GCN-GAN: Integrating Graph Convolutional Network and Generative Adversarial Network for Traffic Flow Prediction

Generative Adversarial Network-Based Regional Epitaxial Traffic Flow Prediction

Road Traffic Network State Prediction Based on a Generative Adversarial Network

SATP-GAN: self-attention based generative adversarial network for traffic flow prediction

A Traffic-Weather Generative Adversarial Network for Traffic Flow Prediction for Road Networks under Bad Weather

TrafficPro: A Road Speed Prediction Framework for Urban Signal-Controlled Road Networks

Network Traffic Prediction for Intelligent Transportation Systems: A Reinforcement Learning Approach

Traffic Prediction Based on Auto Spatiotemporal Multi-graph Adversarial Neural Network

TFGAN: Traffic Forecasting Using Generative Adversarial Network with Multi-Graph Convolutional Network

Daily Long-Term Traffic Flow Forecasting Based on a Deep Neural Network

An Intelligent Deep Learning Framework for Traffic Flow Imputation and Short-term Prediction Based on Dynamic Features

Traffic Flow Prediction Based on Deep Neural Networks

STG-GAN: A spatiotemporal graph generative adversarial networks for short-term passenger flow prediction in urban rail transit systems

A General Traffic Flow Prediction Approach Based on Spatial-Temporal Graph Attention

A Short-Term Traffic Flow Prediction Model Based on an Improved Gate Recurrent Unit Neural Network

Spatial-Temporal Similarity Fusion Graph Adversarial Convolutional Networks for traffic flow forecasting

Traffic flow prediction based on graph convolutional networks with a parallel attention network and stacked gate recurrent units

Convolutional Neural Network-Based Bidirectional Gated Recurrent Unit–Additive Attention Mechanism Hybrid Deep Neural Networks for Short-Term Traffic Flow Prediction

Traffic Flow Data Prediction Using Residual Deconvolution Based Deep Generative Network