重庆理工大学学报(自然科学) ›› 2024, Vol. 38 ›› Issue (8): 191-201.

• 信息·计算机 • 上一篇    下一篇

采用 VAE-CatBoost的高速公路交通事件检测框架

张 兵,邹少权,陆春霖,陈渤文,薛运强   

  1. 华东交通大学 交通运输工程学院,南昌 33000
  • 出版日期:2024-09-09 发布日期:2024-09-09
  • 作者简介:张兵,男,博士,副教授,主要从事交通安全及交通事件智能检测研究,Email:zhangbing@ecjut.edu.cn

  • Online:2024-09-09 Published:2024-09-09

摘要: 为解决交通事件检测中特征稀少与样本失衡的问题,提出一种基于变分自编码器(VAE)随机森林(RF)分类梯度提升树(CatBoost)的交通事件检测框架。依据 4条规则构建较为全面的初始特征集。使用 VAE进行数据平衡。之后,采用 RF算法筛选出最佳输入特征集。最后,采用 CatBoost算法作为分类器检测交通事件。使用真实世界的交通数据集设计实验,并选择 6个有效评价指标对实验结果进行评价。结果表明:所提出的交通事件检测框架除误报率之外,各项评价指标均取得最优结果,表明在交通事件检测方面具备优异性能。

关键词: 交通事件检测, 特征扩展, 数据平衡, 特征选择, 梯度提升树

中图分类号: 

  • U491