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

STG2Seq: Spatial-temporal Graph to Sequence Model for Multi-step Passenger Demand Forecasting

About

Multi-step passenger demand forecasting is a crucial task in on-demand vehicle sharing services. However, predicting passenger demand over multiple time horizons is generally challenging due to the nonlinear and dynamic spatial-temporal dependencies. In this work, we propose to model multi-step citywide passenger demand prediction based on a graph and use a hierarchical graph convolutional structure to capture both spatial and temporal correlations simultaneously. Our model consists of three parts: 1) a long-term encoder to encode historical passenger demands; 2) a short-term encoder to derive the next-step prediction for generating multi-step prediction; 3) an attention-based output module to model the dynamic temporal and channel-wise information. Experiments on three real-world datasets show that our model consistently outperforms many baseline methods and state-of-the-art models.

Lei Bai, Lina Yao, Salil.S Kanhere, Xianzhi Wang, Quan.Z Sheng• 2019

Related benchmarks

TaskDatasetResultRank
Traffic ForecastingPeMSD7 (test)
MAE32.77
83
Traffic Flow ForecastingPeMSD7 M
RMSE6.51
60
Traffic Flow ForecastingPeMSD7 (L)
RMSE7.12
60
Traffic ForecastingPeMSD3 (test)
MAE19.03
53
Traffic ForecastingPeMSD8 (test)
MAE20.17
53
Traffic ForecastingPeMSD4 (test)
MAE25.2
53
Traffic Demand PredictionNYC-Bike 16
RMSE3.7843
13
Traffic Demand PredictionNYC-Taxi 16
RMSE19.2077
13
Traffic Demand PredictionNYC-Taxi 15
RMSE39.4318
13
Traffic Demand PredictionNYC-Bike 14
RMSE10.8561
13
Showing 10 of 11 rows

Other info

Follow for update