Bayesian graph convolutional network for traffic prediction

Fu, Jun, Zhou, Wei and Chen, Zhibo 2024. Bayesian graph convolutional network for traffic prediction. Neurocomputing 582 , 127507. 10.1016/j.neucom.2024.127507 Item availability restricted.

PDF - Accepted Post-Print Version Restricted to Repository staff only until 11 March 2025 due to copyright restrictions. Download (2MB)

Abstract

Recently, adaptive graph convolutional network based traffic prediction methods, learning a latent graph structure from traffic data via various attention-based mechanisms, have achieved impressive performance. However, they are still limited to finding a better description of spatial relationships between traffic conditions due to: (1) ignoring the prior of the observed road network topology; (2) neglecting the presence of negative spatial relationships; and (3) lacking investigation on the uncertainty of the graph structure. In this paper, we propose a Bayesian Graph Convolutional Network (BGCN) framework to alleviate these issues. Under this framework, the graph structure is viewed as a random realization from a parametric generative model, and its posterior is inferred using the observed topology of the road network and traffic data. Specifically, the parametric generative model is comprised of two parts: (1) a constant adjacency matrix that discovers potential spatial relationships from the observed physical connections between roads using a Bayesian approach; (2) a learnable adjacency matrix that learns globally shared spatial correlations from traffic data in an end-to-end fashion and can model negative spatial correlations. The posterior of the graph structure is then approximated by performing Monte Carlo dropout on the parametric graph structure. We verify the effectiveness of our method on five real-world datasets, and the experimental results demonstrate that BGCN attains superior performance compared with state-of-the-art methods. The source code is available at https://github.com/JunFu1995/BGCN.git

Item Type:	Article
Date Type:	Publication
Status:	Published
Schools:	Computer Science & Informatics
Publisher:	Elsevier
ISSN:	0925-2312
Date of First Compliant Deposit:	29 April 2024
Date of Acceptance:	4 March 2024
Last Modified:	29 Apr 2024 21:57
URI:	https://orca.cardiff.ac.uk/id/eprint/167241

Actions (repository staff only)

Edit Item