arXiv:2507.04634cs.CVcs.AI2025-07被引 1

提出LTMSformer模型,提升多智能体轨迹预测的时空依赖建模能力。

LTMSformer: A Local Trend-Aware Attention and Motion State Encoding Transformer for Multi-Agent Trajectory Prediction

  • 设计局部趋势感知注意力机制捕捉时间邻近状态依赖
  • 引入高阶运动状态编码,提升空间交互建模精度
  • 轻量化设计在保持性能同时减少68%模型参数

多智能体轨迹预测面临复杂时空依赖建模难题。由于每个时刻的状态与相邻时间步紧密相关,捕捉局部时序依赖有益于预测,但现有方法常忽略此点。同时,学习高阶运动状态属性(如加速度、急动度、航向等)可增强空间交互建模,但此前研究极少涉及。为此,本文提出轻量级框架LTMSformer,用于提取时空交互特征以实现多模态轨迹预测。具体地,设计局部趋势感知注意力机制,通过分层局部时间窗的卷积注意力捕捉局部时序依赖;构建运动状态编码器,融合加速度、急动度、航向等高阶运动属性以建模空间交互;进一步提出轻量级提议精炼模块,利用多层感知机对轨迹嵌入并生成优化轨迹,显著降低参数量。在Argoverse 1数据集上的实验表明,本方法优于基线HiVT-64,minADE降低约4.35%,minFDE降低8.74%,误率(MR)下降20%;且相较HiVT-128在更高精度下实现68%的模型尺寸缩减。

原文摘要 · Abstract (English)

It has been challenging to model the complex temporal-spatial dependencies between agents for trajectory prediction. As each state of an agent is closely related to the states of adjacent time steps, capturing the local temporal dependency is beneficial for prediction, while most studies often overlook it. Besides, learning the high-order motion state attributes is expected to enhance spatial interaction modeling, but it is rarely seen in previous works. To address this, we propose a lightweight framework, LTMSformer, to extract temporal-spatial interaction features for multi-modal trajectory prediction. Specifically, we introduce a Local Trend-Aware Attention mechanism to capture the local temporal dependency by leveraging a convolutional attention mechanism with hierarchical local time boxes. Next, to model the spatial interaction dependency, we build a Motion State Encoder to incorporate high-order motion state attributes, such as acceleration, jerk, heading, etc. To further refine the trajectory prediction, we propose a Lightweight Proposal Refinement Module that leverages Multi-Layer Perceptrons for trajectory embedding and generates the refined trajectories with fewer model parameters. Experiment results on the Argoverse 1 dataset demonstrate that our method outperforms the baseline HiVT-64, reducing the minADE by approximately 4.35%, the minFDE by 8.74%, and the MR by 20%. We also achieve higher accuracy than HiVT-128 with a 68% reduction in model size.

轨迹预测Transformer多智能体轻量化

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