Dual-Transformer: A General Model for Traffic Accident Prediction

Dongkun Wang,Jieyang Peng,Junkai Zhao,Yunfei Teng,Wenjing Xue,Xiaoming Tao
DOI: https://doi.org/10.1109/vtc2023-fall60731.2023.10333613
2023-01-01
Abstract:Traffic accident prediction aims to estimate the future traffic accident risk for improving urban public safety. Extensive efforts have been dedicated to leveraging spatial correlations and temporal dependencies in the evolving patterns of traffic accidents which are achieved by utilizing explicit spatial relations, such as geographical proximity, through predefined geographical structures. However, current methods for predicting traffic accident still neglect the implicit spatial correlations influenced by an area’s inherent geographic characteristics. To tackle this issue, we propose a Dual-Transformer model that automatically explores implicit spatial correlations, including a Spatial Transformer and a Temporal Transformer. The Spatial Transformer autonomously learns the implicit spatial correlations among road segments, extending beyond the boundaries of geographical structures. Meanwhile, the Temporal Transformer aims to capture the dynamic patterns of these implicit spatial correlations. Through extensive experiments conducted on two real-world traffic accident datasets, our proposed model demonstrates significant improvements compared to existing methods.
What problem does this paper attempt to address?