Hybrid Ensemble Learning Model Combining BERT and CNN for Predicting Urban Rail Transit Accident Consequences
File version
Author(s)
Liu, Weiyi
Li, Xinyao
Chen, Anthony
Schonfeld, Paul M
Du, Bo
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
Size
File type(s)
Location
License
Abstract
Urban Rail Transit (URT) accidents not only seriously affect the safety and reliability of its operations, but also reduce service level to passengers. Based on historical URT accident data, this study develops a hybrid ensemble learning model based on a Convolutional Neural Network (CNN) and Bidirectional Encoder Representations from Transformers (BERT) for predicting accident consequences in URT. The CNN is employed to capture spatial patterns from the diverse accident data, while the BERT is applied to learn complex relations in accident text descriptions. The results of the two models are combined for classifying accident consequences. The proposed hybrid ensemble learning model was applied to predict accident consequences in Chongqing’s URT using historical accident records. It achieved a prediction accuracy of 0.805 on testing data set, which is at least 20% higher than that of commonly used machine learning models, including multilayer perceptrons, support vector machines, and Bayesian networks. Furthermore, the reapplication of the proposed model to historical accident records of the URT in Chengdu demonstrates the generalizability and reusability of the model. This study forecasts the consequences of URT accidents with high accuracy using limited historical data, which supports operators in identifying high-frequency and high-impact accidents. Consequently, targeted maintenance and timely emergency response strategies can be developed to decrease accident rates and mitigate the impacts.
Journal Title
IEEE Transactions on Intelligent Transportation Systems
Conference Title
Book Title
Edition
Volume
26
Issue
8
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject
Transportation, logistics and supply chains
Artificial intelligence
Computer vision and multimedia computation
Persistent link to this record
Citation
Liu, J; Liu, W; Li, X; Chen, A; Schonfeld, PM; Du, B, Hybrid Ensemble Learning Model Combining BERT and CNN for Predicting Urban Rail Transit Accident Consequences, IEEE Transactions on Intelligent Transportation Systems, 2025, 26 (8), pp. 12727-12739