Interpretable Traffic Accident Prediction: Attention Spatial–Temporal Multi-Graph Traffic Stream Learning Approach

Abstract

Traffic accident prediction plays a vital role in Intelligent Transportation Systems (ITS), where a large number of traffic streaming data are generated on a daily basis for spatiotemporal big data analysis. The rarity of accidents and the absent interconnection information make it hard for spatiotemporal modeling. Moreover, the inherent characteristic of the black box predictive model makes it difficult to interpret the reliability and effectiveness of the deep learning model. To address these issues, a novel self-explanatory spatial-temporal deep learning model–Attention Spatial-Temporal Multi-Graph Convolutional Network (ASTMGCN) is proposed for traffic accident prediction. The original recorded rare accident data is formulated as a multivariate irregularly interval-aligned dataset, and the temporal discretization method is used to transfer into regularly sampled time series. Multiple graphs are defined to construct edge features and represent spatial relationships when node-related information is missing. Multi-graph convolutional operators and attention mechanisms are integrated into a Sequence-to-Sequence (Seq2Seq) framework to effectively capture dynamic spatial and temporal features and correlations in multi-step prediction. Comparative experiments and interpretability analysis are conducted on a real-world data set, and results indicate that our model can not only yield superior prediction performance but also has the advantage of interpretability.