Clustering Shift Graph Convolutional Network for Taxi Origin-Destination Demand Prediction

Zhilei Peng,Guixing Wu,Fengliang Xia
DOI: https://doi.org/10.1109/ICTAI52525.2021.00044
2021-01-01
Abstract:Taxi Origin-Destination (OD) demand prediction plays an important role in realizing intelligent cities. An accurate OD forecasting for taxi orders helps achieve efficient taxi dispatch, which reduces commuting costs for both drivers and passengers. Recently, graph convolutional networks (GCNs) have been widely used to model an OD Matrix as spatial-temporal graphs and reach remarkable performance. However, the conventional CNNs and GCNs always have fixed receptive fields, as the networks only aggregate nodes' information among the k-th order neighborhood. These methods bring about tremendous computational complexity and parameters by introducing incremental modules and a multi-stream fusion strategy. In the case of taxi OD demand, the origin and destination are often far apart. To address these issues, we propose a novel clustering shift graph convolutional network (CSGCN). Specifically, we employ lightweight point-wise convolutions to extract the features, and a Shift operation is proposed to make the features circulate among every node which expands the receptive fields to the whole graph. Extensive experiments on two datasets demonstrate the superiority of our CSGCN over other GCN-based methods.
What problem does this paper attempt to address?