用多模态模型提升自动驾驶决策精度,获PRCV2024冠军
Precise Drive with VLM: First Prize Solution for PRCV 2024 Drive LM challenge
- 用InternVL-2.0模型融合多视角图像,保留原始坐标不变换
- 先预训练再微调,改进损失函数提升坐标预测准确率
- 在DriveLM-nuscenes数据集上实现0.6064高分,适合自动驾驶研究
本文介绍我们在PRCV挑战赛中用于驾驶场景认知与决策的方法。采用开创性的开源多模态模型InternVL-2.0,通过优化输入数据格式与训练策略进行增强。输入方面,将多视角图像按策略拼接并保留原始坐标未做变换;训练阶段,先在公开自动驾驶数据集上预训练以增强任务对齐能力,再在DriveLM-nuscenes数据集上微调,并创新性地修改损失函数以提升坐标预测精度。上述方法使模型具备更强的驾驶场景认知与决策能力,最终在竞赛中取得0.6064的得分,位列第一。
原文摘要 · Abstract (English)
This technical report outlines the methodologies we applied for the PRCV Challenge, focusing on cognition and decision-making in driving scenarios. We employed InternVL-2.0, a pioneering open-source multi-modal model, and enhanced it by refining both the model input and training methodologies. For the input data, we strategically concatenated and formatted the multi-view images. It is worth mentioning that we utilized the coordinates of the original images without transformation. In terms of model training, we initially pre-trained the model on publicly available autonomous driving scenario datasets to bolster its alignment capabilities of the challenge tasks, followed by fine-tuning on the DriveLM-nuscenes Dataset. During the fine-tuning phase, we innovatively modified the loss function to enhance the model's precision in predicting coordinate values. These approaches ensure that our model possesses advanced cognitive and decision-making capabilities in driving scenarios. Consequently, our model achieved a score of 0.6064, securing the first prize on the competition's final results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。