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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Traffic Forecasting | PeMSD7 (test) | MAE32.77 | 83 | |
| Traffic Flow Forecasting | PeMSD7 M | RMSE6.51 | 60 | |
| Traffic Flow Forecasting | PeMSD7 (L) | RMSE7.12 | 60 | |
| Traffic Forecasting | PeMSD3 (test) | MAE19.03 | 53 | |
| Traffic Forecasting | PeMSD8 (test) | MAE20.17 | 53 | |
| Traffic Forecasting | PeMSD4 (test) | MAE25.2 | 53 | |
| Traffic Demand Prediction | NYC-Bike 16 | RMSE3.7843 | 13 | |
| Traffic Demand Prediction | NYC-Taxi 16 | RMSE19.2077 | 13 | |
| Traffic Demand Prediction | NYC-Taxi 15 | RMSE39.4318 | 13 | |
| Traffic Demand Prediction | NYC-Bike 14 | RMSE10.8561 | 13 |