Spatiotemporal Multi-Graph Convolution Network for Ride-Hailing Demand Forecasting

  • Xu Geng The Hong Kong University of Science and Technology
  • Yaguang Li University of Southern California
  • Leye Wang The Hong Kong University of Science and Technology
  • Lingyu Zhang Didichuxing Inc.
  • Qiang Yang The Hong Kong University of Science and Technology
  • Jieping Ye Didichuxing Inc.
  • Yan Liu University of Southern California


Region-level demand forecasting is an essential task in ridehailing services. Accurate ride-hailing demand forecasting can guide vehicle dispatching, improve vehicle utilization, reduce the wait-time, and mitigate traffic congestion. This task is challenging due to the complicated spatiotemporal dependencies among regions. Existing approaches mainly focus on modeling the Euclidean correlations among spatially adjacent regions while we observe that non-Euclidean pair-wise correlations among possibly distant regions are also critical for accurate forecasting. In this paper, we propose the spatiotemporal multi-graph convolution network (ST-MGCN), a novel deep learning model for ride-hailing demand forecasting. We first encode the non-Euclidean pair-wise correlations among regions into multiple graphs and then explicitly model these correlations using multi-graph convolution. To utilize the global contextual information in modeling the temporal correlation, we further propose contextual gated recurrent neural network which augments recurrent neural network with a contextual-aware gating mechanism to re-weights different historical observations. We evaluate the proposed model on two real-world large scale ride-hailing demand datasets and observe consistent improvement of more than 10% over stateof-the-art baselines.

AAAI Technical Track: Machine Learning