Apr, 2024
预训练图神经网络基于自动驾驶的智能交通场景理解模型
PreGSU-A Generalized Traffic Scene Understanding Model for Autonomous Driving based on Pre-trained Graph Attention Network
Yuning Wang, Zhiyuan Liu, Haotian Lin, Junkai Jiang, Shaobing Xu...
TL;DR基于图注意力网络的通用预训练场景理解模型能够学习交通场景的通用交互和推理,支持各种下游任务,通过在城市和高速公路场景中进行验证实验,证明了该模型具有广泛应用的潜力,并通过消融实验展示了预训练任务设计的有效性。