STGAT: A Spatio-Temporal Graph Attention Network for Travel Demand Prediction

Abstract

Forecasting travel demand is a challenging task due to the complex spatial dependencies and dynamic temporal correlation of the traffic data. Furthermore, a limited representation of the given spatial graph structure may restrict the model's ability to effectively learn spatial-temporal dependencies. To extract latent semantic features from traffic data, this paper proposes a method to construct a traffic graph using the Markov cluster algorithm. The traffic semantic correlation matrix is constructed based on a traffic graph to obtain the deep semantic information. Specifically, this study proposes a Spatio-Temporal Graph Attention Network(STGAT) for travel demand prediction. STGAT uses a graph attention layer based on the Node2Vec graph embedding algorithm, two convolutional layers based on the Markov cluster algorithm, and a long short-term memory network to capture spatial-temporal dependencies. Experimental results on the NYC Taxi dataset and the Chengdu online car dataset demonstrate that STGAT achieved state-of-the-art performance compared to other baseline models.