Conference proceeding
Hetero-ConvLSTM: A Deep Learning Approach to Traffic Accident Prediction on Heterogeneous Spatio-Temporal Data
Proceedings of the 24th ACM SIGKDD International Conference on knowledge discovery & data mining, pp.984-992
KDD '18
07/19/2018
DOI: 10.1145/3219819.3219922
Abstract
Predicting traffic accidents is a crucial problem to improving transportation and public safety as well as safe routing. The problem is also challenging due to the rareness of accidents in space and time and spatial heterogeneity of the environment (e.g., urban vs. rural). Most previous research on traffic accident prediction conducted by domain researchers simply applied classical prediction models on limited data without addressing the above challenges properly, thus leading to unsatisfactory performance. A small number of recent works have attempted to use deep learning for traffic accident prediction. However, they either ignore time information or use only data from a small and homogeneous study area (a city), without handling spatial heterogeneity and temporal auto-correlation properly at the same time. In this paper we perform a comprehensive study on the traffic accident prediction problem using the Convolutional Long Short-Term Memory (ConvLSTM) neural network model. A number of detailed features such as weather, environment, road condition, and traffic volume are extracted from big datasets over the state of Iowa across 8 years. To address the spatial heterogeneity challenge in the data, we propose a Hetero-ConvLSTM framework, where a few novel ideas are implemented on top of the basic ConvLSTM model, such as incorporating spatial graph features and spatial model ensemble. Extensive experiments on the 8-year data over the entire state of Iowa show that the proposed framework makes reasonably accurate predictions and significantly improves the prediction accuracy over baseline approaches.
Details
- Title: Subtitle
- Hetero-ConvLSTM: A Deep Learning Approach to Traffic Accident Prediction on Heterogeneous Spatio-Temporal Data
- Creators
- Zhuoning YuanXun ZhouTianbao Yang
- Resource Type
- Conference proceeding
- Publication Details
- Proceedings of the 24th ACM SIGKDD International Conference on knowledge discovery & data mining, pp.984-992
- Series
- KDD '18
- DOI
- 10.1145/3219819.3219922
- Publisher
- ACM
- Grant note
- DOI: 10.13039/100000001, name: National Science Foundation, award: 1566386
- Language
- English
- Date published
- 07/19/2018
- Academic Unit
- Computer Science; Business Analytics
- Record Identifier
- 9984083844302771
Metrics
61 Record Views