Skip to content
/ MG-TAR Public
forked from kaist-dmlab/MG-TAR

[IEEE T-ITS] MG-TAR: Multi-view Graph Convolutional Networks for Traffic Accident Risk Prediction

License

Notifications You must be signed in to change notification settings

e0jun/MG-TAR

This branch is 402 commits behind kaist-dmlab/MG-TAR:main.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

023b7e9 · Nov 5, 2023

History

29 Commits
Aug 2, 2021
Aug 2, 2021
May 11, 2021
May 11, 2021
Aug 2, 2021
Nov 4, 2023
Jun 6, 2021
Apr 15, 2022
Nov 5, 2023
Aug 2, 2021
Nov 4, 2023
Sep 2, 2021

Repository files navigation

MG-TAR: Multi-view Graph Convolutional Networks for Traffic Accident Risk Prediction

This is the implementation of a paper published in IEEE Transactions on Intelligent Transportation Systems [Paper]

Citation

@article{trirat2023mgtar,
  author={Trirat, Patara and Yoon, Susik and Lee, Jae-Gil},
  journal={IEEE Transactions on Intelligent Transportation Systems}, 
  title={{MG-TAR}: Multi-View Graph Convolutional Networks for Traffic Accident Risk Prediction}, 
  year={2023},
  volume={24},
  number={4},
  pages={3779-3794},
  doi={10.1109/TITS.2023.3237072}}
}

Abstract

Due to the continuing colossal socio-economic losses caused by traffic accidents, it is of prime importance to precisely forecast the traffic accident risk to reduce future accidents. In this paper, we use dangerous driving statistics from driving log data and multi-graph learning to enhance predictive performance. We first conduct geographical and temporal correlation analyses to quantify the relationship between dangerous driving and actual accidents. Then, to learn various dependencies between districts besides the traditional adjacency matrix, we simultaneously model both static and dynamic graphs representing the spatio-temporal contextual relationships with heterogeneous environmental data, including the dangerous driving behavior. A graph is generated for each type of the relationships. Ultimately, we propose an end-to-end framework, called MG-TAR, to effectively learn the association of multiple graphs for accident risk prediction by adopting multi-view graph neural networks with a multi-attention module. Thorough experiments on ten real-world datasets show that, compared with state-of-the-art methods, MG-TAR reduces the error of predicting the accident risk by up to 23% and improves the accuracy of predicting the most dangerous areas by up to 27%.

Note for Driving Record Data

  • Digital Tachograph (Driving Log) Data: cannot be publicly accessible
    • For demonstration purpose, we partially provide the aggregated number of classified dangerous driving cases in the datasets folder.
    • If you are interested in the original data, there is a sample file provided here by Korea Transportation Safety Authority.

Example Run

  • For package installation: pip install -r requirements.txt
  • For graph preprocessing: multi-view_graph_construction.ipynb
  • For model testing: Example Run.ipynb

About

[IEEE T-ITS] MG-TAR: Multi-view Graph Convolutional Networks for Traffic Accident Risk Prediction

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 55.0%
  • Jupyter Notebook 45.0%