Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Coupled Layer-wise Graph Convolution for Transportation Demand Prediction

About

Graph Convolutional Network (GCN) has been widely applied in transportation demand prediction due to its excellent ability to capture non-Euclidean spatial dependence among station-level or regional transportation demands. However, in most of the existing research, the graph convolution was implemented on a heuristically generated adjacency matrix, which could neither reflect the real spatial relationships of stations accurately, nor capture the multi-level spatial dependence of demands adaptively. To cope with the above problems, this paper provides a novel graph convolutional network for transportation demand prediction. Firstly, a novel graph convolution architecture is proposed, which has different adjacency matrices in different layers and all the adjacency matrices are self-learned during the training process. Secondly, a layer-wise coupling mechanism is provided, which associates the upper-level adjacency matrix with the lower-level one. It also reduces the scale of parameters in our model. Lastly, a unitary network is constructed to give the final prediction result by integrating the hidden spatial states with gated recurrent unit, which could capture the multi-level spatial dependence and temporal dynamics simultaneously. Experiments have been conducted on two real-world datasets, NYC Citi Bike and NYC Taxi, and the results demonstrate the superiority of our model over the state-of-the-art ones.

Junchen Ye, Leilei Sun, Bowen Du, Yanjie Fu, Hui Xiong• 2020

Related benchmarks

TaskDatasetResultRank
Traffic speed forecastingMETR-LA (test)--
195
Traffic ForecastingMETR-LA
MAE2.85
127
Traffic ForecastingMETR-LA 30min horizon 6
MAE3.24
44
Traffic ForecastingPEMS-BAY 15min horizon 3
MAE1.38
25
Traffic ForecastingPEMS-BAY 30min horizon 6
MAE1.74
24
Traffic ForecastingPEMS-BAY 60min horizon 12
MAE2.07
24
Traffic Speed PredictionPEMS-BAY
MAE (15 min)1.38
15
Traffic ForecastingEXPY-TKY 10min horizon 1
MAE5.9
13
Traffic Demand PredictionNYC-Bike 16
RMSE2.7674
13
Traffic Demand PredictionNYC-Taxi 16
RMSE9.8744
13
Showing 10 of 15 rows

Other info

Follow for update