Robust Traffic Prediction From Spatial-Temporal Data Based on Conditional Distribution Learning

Page view(s)
29
Checked on Jun 08, 2024
Robust Traffic Prediction From Spatial-Temporal Data Based on Conditional Distribution Learning
Title:
Robust Traffic Prediction From Spatial-Temporal Data Based on Conditional Distribution Learning
Journal Title:
IEEE Transactions on Cybernetics
Publication Date:
17 December 2021
Citation:
Zeng, Z., Zhao, W., Qian, P., Zhou, Y., Zhao, Z., Chen, C., & Guan, C. (2021). Robust Traffic Prediction From Spatial-Temporal Data Based on Conditional Distribution Learning. IEEE Transactions on Cybernetics, 1–14. https://doi.org/10.1109/tcyb.2021.3131285
Abstract:
Traffic prediction based on massive speed data collected from traffic sensors plays an important role in traffic management. However, it is still challenging to obtain satisfactory performance due to the complex and dynamic spatial-temporal correlations among the data. Recently, many research works have demonstrated the effectiveness of graph neural networks (GNNs) for spatial-temporal modeling. However, such models are restricted by conditional distribution during training, and may not perform well when the target is outside the primary region of interest in the distribution. In this article, we address this problem with a stagewise learning mechanism, in which we redefine speed prediction as a conditional distribution learning followed by speed regression. We first perform a conditional distribution learning for each observed speed class, and then obtain speed prediction by optimizing regression learning, based on the learned conditional distribution. To effectively learn the conditional distribution, we introduce a mean-residue loss, consisting of two parts: 1) a mean loss, which penalizes the differences between the mean of the estimated conditional distribution and the ground truth and 2) a residue loss, which penalizes residue errors of the long tails in the distribution. To optimize the subsequent regression based on distribution information, we combine the mean absolute error (MAE) as another part of the loss function. We also incorporate a GNN-based architecture with our proposed learning mechanism. Mean-residue loss is employed to supervise the hidden speed representation in the network at each time interval, followed by a shared layer to recalibrate the hidden temporal dependencies in the conditional distribution. The experimental results based on three public traffic datasets have demonstrated that the effectiveness of the proposed method outperforms state-of-the-art methods.
License type:
Publisher Copyright
Funding Info:
This research / project is supported by the National Research Foundation Singapore and the National Natural Science Foundation of China (NSFC) - Singapore-China NRF-NSFC Grant
Grant Reference no. : NRF2016NRF-NSFC001-111

Overseas Funding: National Natural Science Foundation of China (NSFC) with No. 61801315
Description:
© 2021 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works
ISSN:
2168-2267
2168-2275
Files uploaded:

File Size Format Action
weusepublicdataset.zip 178 bytes ZIP Request a copy