Traj-LLM: A New Exploration for Empowering Trajectory Prediction with Pre-trained Large Language Models

LLM PTSAD Embodied AI and Robotics Trajectory Prediction ML Mamba/SSM
2024年05月08日
自动驾驶中一个基本的任务是预测动态交通参与者的未来轨迹。尽管现有的研究已经取得了显著的表现提升,但在场景认知和复杂交通语义理解方面仍存在差距。本文提出了Traj-LLM,首次探讨使用大型语言模型(LLMs)生成未来运动的潜力,而无需显式提示工程来分析代理的过去/观察轨迹和场景语义。Traj-LLM从稀疏上下文联合编码开始,将代理和场景特征解析为LLMs可以理解的形式。在此基础上,我们创新地探索了LLMs的强大理解能力,以捕捉高级场景知识和交互信息的各种光谱。为了模拟类人的车道焦点认知功能并增强Traj-LLM的场景理解能力,我们引入了由开创性的Mamba模块驱动的车道感知概率学习。最后,设计了多模式拉普拉斯解码器,以实现符合场景的多模式预测。广泛的实验表明,Traj-LLM,借助LLMs的强大先验知识和理解能力,以及车道感知概率学习,超过了评估指标的最新方法。此外,少样本分析进一步证实了Traj-LLM的性能,在仅使用数据集的50%的情况下,它的表现优于依赖完整数据利用的大多数基准。本研究探索了为轨迹预测任务提供LLMs内在先进能力的方法,为预测代理运动提供了一种更通用和适应性更强的解决方案。
Predicting the future trajectories of dynamic traffic actors is a cornerstone task in autonomous driving. Though existing notable efforts have resulted in impressive performance improvements, a gap persists in scene cognitive and understanding of the complex traffic semantics. This paper proposes Traj-LLM, the first to investigate the potential of using Large Language Models (LLMs) without explicit prompt engineering to generate future motion from agents' past/observed trajectories and scene semantics. Traj-LLM starts with sparse context joint coding to dissect the agent and scene features into a form that LLMs understand. On this basis, we innovatively explore LLMs' powerful comprehension abilities to capture a spectrum of high-level scene knowledge and interactive information. Emulating the human-like lane focus cognitive function and enhancing Traj-LLM's scene comprehension, we introduce lane-aware probabilistic learning powered by the pioneering Mamba module. Finally, a multi-modal Laplace decoder is designed to achieve scene-compliant multi-modal predictions. Extensive experiments manifest that Traj-LLM, fortified by LLMs' strong prior knowledge and understanding prowess, together with lane-aware probability learning, outstrips state-of-the-art methods across evaluation metrics. Moreover, the few-shot analysis further substantiates Traj-LLM's performance, wherein with just 50% of the dataset, it outperforms the majority of benchmarks relying on complete data utilization. This study explores equipping the trajectory prediction task with advanced capabilities inherent in LLMs, furnishing a more universal and adaptable solution for forecasting agent motion in a new way.
许愿