References & Citations
Computer Science > Machine Learning
Title: Spatial-Temporal Dynamic Graph Attention Networks for Ride-hailing Demand Prediction
(Submitted on 7 Jun 2020 (v1), last revised 17 Apr 2022 (this version, v4))
Abstract: Ride-hailing demand prediction is an essential task in spatial-temporal data mining. Accurate Ride-hailing demand prediction can help to pre-allocate resources, improve vehicle utilization and user experiences. Graph Convolutional Networks (GCN) is commonly used to model the complicated irregular non-Euclidean spatial correlations. However, existing GCN-based ride-hailing demand prediction methods only assign the same importance to different neighbor regions, and maintain a fixed graph structure with static spatial relationships throughout the timeline when extracting the irregular non-Euclidean spatial correlations. In this paper, we propose the Spatial-Temporal Dynamic Graph Attention Network (STDGAT), a novel ride-hailing demand prediction method. Based on the attention mechanism of GAT, STDGAT extracts different pair-wise correlations to achieve the adaptive importance allocation for different neighbor regions. Moreover, in STDGAT, we design a novel time-specific commuting-based graph attention mode to construct a dynamic graph structure for capturing the dynamic time-specific spatial relationships throughout the timeline. Extensive experiments are conducted on a real-world ride-hailing demand dataset, and the experimental results demonstrate the significant improvement of our method on three evaluation metrics RMSE, MAPE and MAE over state-of-the-art baselines.
Submission history
From: Weiguo Pian [view email][v1] Sun, 7 Jun 2020 13:00:19 GMT (1958kb,D)
[v2] Wed, 9 Dec 2020 08:30:26 GMT (1958kb,D)
[v3] Fri, 28 May 2021 19:53:56 GMT (1960kb,D)
[v4] Sun, 17 Apr 2022 01:41:27 GMT (3922kb,D)
Link back to: arXiv, form interface, contact.