Traffic accident severity prediction and analysis via spatio-temporal deep learning: A ConvLSTM-transformer approach

Kun Li, Kaiyang Guo, Zhantu Liang, Haocheng Xu and Xinlong Duan

Chaos, Solitons & Fractals, 2025, vol. 200, issue P1

Abstract: Traffic accidents pose a significant challenge to public safety, and predicting their causation and severity is crucial to traffic safety research. Traditional machine learning methods, however, mainly depend on direct records from traffic datasets, therefore ignoring the spatial and temporal heterogeneity of traffic accidents. Unfortunately, most existing deep learning algorithms, which excel at extracting spatiotemporal features of accidents, often overlook semantic features like driver and vehicle characteristics and fail to differentiate accident severity levels. To address the limitations of existing models, we propose Spatio-Temporal RiskFormer (ST-RiskFormer), a deep learning model designed to accurately predict traffic accident severity (slight, serious, fatal) by integrating ConvLSTM for capturing spatio-temporal correlations with a Transformer-based module for processing semantic features, including driver demographics and vehicle attributes. Moreover, Gradient-weighted Class Activation Mapping (Grad-CAM) is applied to data visualization for ST-RiskFormer. Simulation results show that medium-term traffic data can uncover the most significant spatial-temporal information of traffic accidents. More importantly, as compared to other typical machine learning algorithms, ST-RiskFormer, which highlights the spatio-temporal distribution of traffic accidents, performs better in predicting traffic accident severity. We hope to provide some insights into deeper-level traffic data mining through this work.

Keywords: Data visualization; Grad-CAM; ST-RiskFormer; Spatio-temporal correlations; Severity of traffic accidents (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0960077925010604
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:chsofr:v:200:y:2025:i:p1:s0960077925010604

DOI: 10.1016/j.chaos.2025.117047

Access Statistics for this article

Chaos, Solitons & Fractals is currently edited by Stefano Boccaletti and Stelios Bekiros

More articles in Chaos, Solitons & Fractals from Elsevier
Bibliographic data for series maintained by Thayer, Thomas R. ().