基于卷积神经网络和注意力机制的交通事故智能预测方法
DOI:
作者:
作者单位:

华东交通大学信息与软件工程学院,江西 南昌 330013

作者简介:

通讯作者:

严丽平(1980—),女,教授,博士,硕士生导师,研究方向为智能交通。E-mail:csyanliping@163.com。

中图分类号:

TP391.4;U491.3

基金项目:

国家自然科学基金项目(62362031,62262022);江西省自然科学基金项目(20224BAB202021)


Intelligent Prediction Method for Traffic Accidents Based on Convolutional Neural Network and Attention Mechanism
Author:
Affiliation:

School of Information and Software Engineering, East China Jiaotong University, Nanchang 330013 , China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对复杂环境下传统研究方法在处理高维复杂数据特征时存在局限性,难以实现高精度鲁棒性预测的问题,提出一种基于卷积神经网络和注意力机制的交通事故严重程度智能预测方法。通过融合卷积神经网络和注意力机制的优势,构建了通道与多头注意力网络(channel and multi-head attention network, CMHANet)模型,其中卷积层用于有效提取数据中的空间特征,通道注意力机制能够对重要特征进行加权增强,抑制不重要特征,强化模型对关键数据点的关注。此外,为了捕捉不同特征之间的复杂依赖关系,还引入了多头注意力机制。最后,在US-Accidents数据集上开展实验。实验结果表明,以该模型为骨干的预测框架在F1分数、精确率、召回率和准确率上均有所提升。该模型在提高高维复杂数据的特征提取与关联建模效果的同时,也为交通事故的智能预测提供了一种新的思路。

    Abstract:

    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-precision and robust prediction, a traffic accidents severity intelligent prediction method based on convolutional neural 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 advantages of convolution and attention mechanism. The convolution layer is used to effectively extract spatial features in the data, while the channel attention mechanism can weight and enhance important features, suppress unimportant features, and emphasize the focus on key data points. In addition, in order to capture the complex dependencies 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.

    参考文献
    相似文献
    引证文献
引用本文

严丽平,徐嘉悦,吴康来,等. 基于卷积神经网络和注意力机制的交通事故智能预测方法[J]. 华东交通大学学报,2026,43(2):38-44.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-02-28
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-05-20
  • 出版日期:
关闭