基于卷积神经网络和注意力机制的交通事故智能预测方法OA
Intelligent Prediction Method for Traffic Accidents Based on Convolutional Neural Network and Attention Mechanism
针对复杂环境下传统研究方法在处理高维复杂数据特征时存在局限性,难以实现高精度鲁棒性预测的问题,提出一种基于卷积神经网络和注意力机制的交通事故严重程度智能预测方法.通过融合卷积神经网络和注意力机制的优势,构建了通道与多头注意力网络(channel and multi-head attention network,CMHANet)模型,其中卷积层用于有效提取数据中的空间特征,通道注意力机制能够对重要特征进行加权增强,抑制不重要特征,强化模型对关键数据点的关注.此外,为了捕捉不同特征之间的复杂依赖关系,还引入了多头注意力机制.最后,在US-Accidents数据集上开展实验.实验结果表明,以该模型为骨干的预测框架在F1分数、精确率、召回率和准确率上均有所提升.该模型在提高高维复杂数据的特征提取与关联建模效果的同时,也为交通事故的智能预测提供了一种新的思路.
In order to solve the problems that traditional research methods often have limitations in dealing with high-dimensional and complex data features in complex environments,and it is difficult to achieve high-preci-sion and robust prediction,a traffic accidents severity intelligent prediction method based on convolutional neu-ral network and attention mechanism is proposed.A multi-scale feature extraction model with attention fusion,which is called channel and multi-head attention network(CMHANet),is constructed to make full use of the ad-vantages of convolution and attention mechanism.The convolution layer is used to effectively extract spatial fea-tures in the data,while the channel attention mechanism can weight and enhance important features,suppress un-important features,and emphasize the focus on key data points.In addition,in order to capture the complex de-pendencies between different features,a multi-head attention mechanism is also introduced.Finally,experiments are conducted on the US-Accidents dataset.The experimental results show that the prediction framework with this model as the backbone achieves improvements in F1-score,precision,recall and accuracy.While improving the effect of feature extraction and association modeling for high-dimensional and complex data,the proposed model also provides a new idea for intelligent prediction of traffic accidents.
严丽平;徐嘉悦;吴康来;唐仁杰;宋凯
华东交通大学信息与软件工程学院,江西 南昌 330013华东交通大学信息与软件工程学院,江西 南昌 330013华东交通大学信息与软件工程学院,江西 南昌 330013华东交通大学信息与软件工程学院,江西 南昌 330013华东交通大学信息与软件工程学院,江西 南昌 330013
信息技术与安全科学
卷积神经网络注意力机制交通事故严重程度预测
convolutional neural networkattention mechanismtraffic accidentsseverity prediction
《华东交通大学学报》 2026 (2)
38-44,7
国家自然科学基金项目(62362031,62262022)江西省自然科学基金项目(20224BAB202021)
评论