School of Civil Engineering and Transportation, South China University of Technology, Guangzhou 510641, China.
College of Computer Science and Technology, Harbin Institute of Technology, Shenzhen 518055, China.
Sensors (Basel). 2022 Aug 5;22(15):5877. doi: 10.3390/s22155877.
Despite the extensive efforts, accurate traffic time series forecasting remains challenging. By taking into account the non-linear nature of traffic in-depth, we propose a novel ST-CRMF model consisting of the Compensated Residual Matrix Factorization with Spatial-Temporal regularization for graph-based traffic time series forecasting. Our model inherits the benefits of MF and regularizer optimization and further carries out the compensatory modeling of the spatial-temporal correlations through a well-designed bi-directional residual structure. Of particular concern is that MF modeling and later residual learning share and synchronize iterative updates as equal training parameters, which considerably alleviates the error propagation problem that associates with rolling forecasting. Besides, most of the existing prediction models have neglected the difficult-to-avoid issue of missing traffic data; the ST-CRMF model can repair the possible missing value while fulfilling the forecasting tasks. After testing the effects of key parameters on model performance, the numerous experimental results confirm that our ST-CRMF model can efficiently capture the comprehensive spatial-temporal dependencies and significantly outperform those state-of-the-art models in the short-to-long terms (5-/15-/30-/60-min) traffic forecasting tasks on the open Seattle-Loop and METR-LA traffic datasets.
尽管已经付出了广泛的努力,但准确的交通时间序列预测仍然具有挑战性。通过深入考虑交通的非线性性质,我们提出了一种新颖的 ST-CRMF 模型,该模型由具有空间-时间正则化的补偿残差矩阵分解和基于图的交通时间序列预测组成。我们的模型继承了 MF 和正则化优化的优势,并通过精心设计的双向残差结构进一步对时空相关性进行补偿建模。值得特别关注的是,MF 建模和后来的残差学习共享和同步迭代更新作为相等的训练参数,这极大地减轻了与滚动预测相关的误差传播问题。此外,大多数现有的预测模型都忽略了难以避免的交通数据缺失问题;ST-CRMF 模型可以在完成预测任务的同时修复可能的缺失值。在测试模型性能的关键参数的效果后,大量实验结果证实,我们的 ST-CRMF 模型可以有效地捕捉全面的时空依赖性,并在短至长(5-/15-/30-/60-分钟)的开放西雅图环路和 METR-LA 交通数据集的交通预测任务中显著优于那些最先进的模型。