Latent Gaussian Processes based Graph Learning for Urban Traffic Prediction

Xu Wang,Pengkun Wang,Binwu Wang,Yudong Zhang,Zhengyang Zhou,Lei Bai,Yang Wang
DOI: https://doi.org/10.1109/tvt.2023.3307755
IF: 6.8
2023-01-01
IEEE Transactions on Vehicular Technology
Abstract:Traffic prediction facilitates various applications in the fields of smart vehicles and vehicular communications, and the key of successfully and accurately forecasting urban traffic state is to model the complex spatiotemporal correlations within urban traffic networks. However, even though great efforts have been devoted to modeling the spatiotemporal correlations, such issue remains challenging in the following two loci, i) capturing dynamic spatial correlations and ii) efficiently and holistically modeling temporal trends. To tackle these challenges, in this paper, we propose a novel Adaptive Graph convolution based Autoencoder with Latent Gaussian processes (AGALG). Specifically, a graph fusion module is proposed to fuse learnable graphs for modeling static spatial correlations and data-driven graphs for dynamic spatial correlations. The fused graphs are fed into an adaptive graph convolutional autoencoder whose encoder represents both historical and future data in a latent space where they are holistically modeled by Gaussian processes generated by a latent Gaussian generator. We apply a Gaussian processes regressor to make estimations of latent representations of future data, and the decoder of autoencoder finally generates predictions. Experimental results on several widely used benchmark datasets validate that the proposed model achieves the state-of-the-art performance.
telecommunications,engineering, electrical & electronic,transportation science & technology
What problem does this paper attempt to address?