arXiv:2504.10275cs.CVcs.LG2025-04CVPR被引 5

基于车道结构的Transformer模型,提升自动驾驶轨迹预测精度与可解释性。

LMFormer: Lane based Motion Prediction Transformer

  • 通过动态优先级机制关注关键车道,增强模型可解释性
  • 在nuScenes上达到当前最优性能,多指标领先
  • 支持跨数据集训练,适用于复杂交通场景

运动预测在自动驾驶中至关重要。本文提出LMFormer,一种基于车道感知的Transformer网络,用于轨迹预测。与以往方法不同,该模型引入动态车道优先机制,使网络学习行为更具可解释性。同时,利用交叉口、合流与分叉处的车道连接信息,建模车道结构中的长程依赖关系。此外,提出基于堆叠Transformer层的迭代优化方法,有效提升预测轨迹质量。在nuScenes数据集上的评估表明,LMFormer在多个指标上达到当前最优(SOTA)表现。进一步在Deep Scenario数据集上验证了其跨数据集泛化能力,并展示了联合训练多数据源后性能提升的潜力。

原文摘要 · Abstract (English)

Motion prediction plays an important role in autonomous driving. This study presents LMFormer, a lane-aware transformer network for trajectory prediction tasks. In contrast to previous studies, our work provides a simple mechanism to dynamically prioritize the lanes and shows that such a mechanism introduces explainability into the learning behavior of the network. Additionally, LMFormer uses the lane connection information at intersections, lane merges, and lane splits, in order to learn long-range dependency in lane structure. Moreover, we also address the issue of refining the predicted trajectories and propose an efficient method for iterative refinement through stacked transformer layers. For benchmarking, we evaluate LMFormer on the nuScenes dataset and demonstrate that it achieves SOTA performance across multiple metrics. Furthermore, the Deep Scenario dataset is used to not only illustrate cross-dataset network performance but also the unification capabilities of LMFormer to train on multiple datasets and achieve better performance.

轨迹预测Transformer自动驾驶车道感知

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。