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

Chaojie Li, Borui Zhang, Zeyu Wang, Yin Yang, Xiaojun Zhou, Shirui Pan, Xinghuo Yu

Research output: Contribution to journalArticlepeer-review

7 Citations (Scopus)

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 cannot only yield superior prediction performance but also has the advantage of interpretability.
Original languageEnglish
Pages (from-to)15574-15586
Number of pages13
JournalIEEE Transactions on Intelligent Transportation Systems
Volume25
Issue number11
Early online dateSept 2024
DOIs
Publication statusPublished - 25 Sept 2024

Keywords

  • Accidents
  • Correlation
  • Deep learning
  • Feature extraction
  • Interpretability
  • Irregular interval
  • Multiple graphs
  • Predictive models
  • Spatiotemporal phenomena
  • Time series analysis
  • Traffic accident prediction
  • Traffic stream learning

Fingerprint

Dive into the research topics of 'Interpretable Traffic Accident Prediction: Attention Spatial–Temporal Multi-Graph Traffic Stream Learning Approach'. Together they form a unique fingerprint.

Cite this