Multi-view Bayesian spatio-temporal graph neural networks for reliable traffic flow prediction

Xia, J. and Wang, S. and Wang, X. and Xia, M. and Xie, K. and Cao, J. (2024) Multi-view Bayesian spatio-temporal graph neural networks for reliable traffic flow prediction. International Journal of Machine Learning and Cybernetics, 15 (1). pp. 65-78.

Text (accepted version)
s13042_022_01689_2.pdf - Accepted Version
Available under License Other.
Download (1MB)

Abstract

Accurate traffic flow prediction is critically essential to transportation safety and Intelligent Transportation Systems (ITS). Existing approaches generally assume the traffic data are complete and reliable. However, in real scenarios, the traffic data are usually sparse and noisy due to the unreliability of the road sensors. Meanwhile, the global semantic traffic correlations among the road links over the road network are largely ignored by existing works. To address these issues, in this paper we study the novel problem of reliable traffic prediction with noisy and sparse traffic data and propose a Multi-View Bayesian Spatio-Temporal Graph Neural Network (MVB-STNet for short) to effectively address it. Specifically, we first construct the traffic flow graphs from two views, the structural traffic graph based on the topological closeness of the road sensors, and the semantic traffic graph which is constructed based on the traffic flow correlations among all the road sensors. Then the features of the two views are learned simultaneously to more broadly capture the spatial correlations. Inspired by the effectiveness of Bayesian neural networks in handling data uncertainty, we design the Bayesian Spatio-Temporal Long Short-Term Memory Net layer to more effectively learn the spatio-temporal features from the sparse and noisy traffic data. Extensive evaluations are conducted over two real traffic datasets. The results show that our proposal significantly improves current state-of-the-arts in terms of traffic flow prediction with sparse and noisy data.