使用 BERT 预测红灯等待时间
机器学习
2021-02-26 v1
摘要
我们提出一种基于 BERT 的模型来近似道路交通仿真结果的方法,其可应用于例如优化交通信号配时,尤其在存在自动驾驶与网联车辆的情况下。实验在采用 Traffic Simulation Framework 软件于真实道路网络上生成的数据集上进行。将基于 BERT 的模型与另外 4 种机器学习模型(LightGBM、全连接神经网络和 2 种图神经网络)进行比较,在所考虑的各项指标上均取得了最佳结果。
引用
@article{arxiv.2102.12896,
title = {Predicting times of waiting on red signals using BERT},
author = {Witold Szejgis and Anna Warno and Paweł Gora},
journal= {arXiv preprint arXiv:2102.12896},
year = {2021}
}
备注
5 pages