Mar, 2024

DriveCoT:将思维链推理与端到端驾驶相结合

TL;DR在本文中,我们收集了一个全面的端到端驾驶数据集,名为 DriveCoT,利用 CARLA 模拟器,通过传感器数据、控制决策和思维过程标签来指示推理过程。我们利用 CARLA 2.0 榜单中的具有挑战性的驾驶场景,提出了一种基于规则的专家策略来控制汽车并生成推理过程和最终决策的真实标签。此数据集可作为一个开环端到端驾驶基准,评估各种思维链路的准确性和最终决策。此外,我们提出了一个名为 DriveCoT-Agent 的基线模型,通过我们的数据集进行训练,来生成思维链路预测和最终决策。训练好的模型在开环和闭环评估中表现出很强的性能,证明了我们提出的数据集的有效性。