Traffic Flow Prediction Based on Multimodal Spatio-Temporal Bayesian Neural Network
摘要
This paper constructs a prediction model based on multimodal spatio-temporal Bayesian neural network. The model integrates traffic flow, image and meteorological data, uses graph convolution to model spatial relationships, combines variational inference to achieve uncertainty quantification, and is built on the basis of GCN and LSTM for structural optimization. The experimental results show that the proposed model outperforms the traditional LSTM, Transformer and non-Bayesian multimodal models in terms of RMSE, MAE and R2, with the lowest RMSE of 12.03, the highest R2 of 0.941, and the coverage of 0.98 in the confidence interval prediction, which demonstrates good generalization and credibility.