摘要
时空网络数据预测在交通管理与城市规划等众多应用中具有重要意义。然而,其内在复杂的时空相关性以及异质性使得该问题极具挑战性。现有方法通常采用独立组件分别捕捉空间与时间相关性,却忽略了时空数据中的异质特性。本文提出一种新型模型——时空同步图卷积网络(Spatial-Temporal Synchronous Graph Convolutional Networks, STSGCN),用于时空网络数据预测。该模型通过精心设计的时空同步建模机制,能够有效捕捉复杂的局部时空相关性;同时,在模型中引入针对不同时间周期的多模块结构,以有效建模局部时空图中的异质性。在四个真实世界数据集上开展的大量实验表明,所提方法达到了当前最优的性能,并持续优于各类基线方法。
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| traffic-prediction-on-bjtaxi | STSGCN | MAE @ in: 12.72 MAE @ out: 12.79 MAPE (%) @ in: 17.22 MAPE (%) @ out: 17.35 |
| traffic-prediction-on-nycbike1 | STSGCN | MAE @ in: 5.81 MAE @ out: 6.10 MAPE (%) @ in: 26.51 MAPE (%) @ out: 27.56 |
| traffic-prediction-on-nycbike2 | STSGCN | MAE @ in: 5.25 MAE @ out: 4.94 MAPE (%) @ in: 29.26 MAPE (%) @ out: 28.02 |
| traffic-prediction-on-nyctaxi | STSGCN | MAE @ in: 13.69 MAE @ out: 10.75 MAPE (%) @ in: 22.91 MAPE (%) @ out: 22.37 |
| traffic-prediction-on-pems07 | STSGCN | MAE@1h: 24.26 |