Despite advances in trajectory prediction, critical challenges persist in complex scenarios. On the one hand, existing methods struggle to capture intricate multi-agent interactions and long-term temporal dependencies; on the other hand, the high redundancy of traditional generative models in the computational process of gradually denoising makes it challenging to meet real-time requirements. To address these issues, we proposed TrafusionNet, an efficient trajectory prediction framework consisting of a perceptual modeling stage and an accelerated diffusion stage. In the perceptual modeling stage, a temporal-interaction perception module is designed to capture the multi-agent dynamic relationships through interactive cognition. In the accelerated diffusion stage, TrafusionNet samples directly from the latent trajectories generated by the perceptual modeling, skipping redundant denoising steps to optimize sampling efficiency and accuracy. In addition, by generating multiple future latent latent trajectories to address uncertainty, a balance between diversity and accuracy can be achieved. The experimental results on the NBA and ETH- UCY datasets showed that TrafusionNet outperformed existing methods in terms of error, model efficiency, and sampling time, achieving the most advanced predictive performance currently available.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

TrafusionNet: Efficient Multi-agent Trajectory Prediction with Temporal-Social Perception Mamba and Diffusion Model

  • Yiyang Li,
  • Zhenghan Zhang,
  • Hongwei Ye,
  • Qijing Huang,
  • Xinghua Li,
  • Yuntao Guo

摘要

Despite advances in trajectory prediction, critical challenges persist in complex scenarios. On the one hand, existing methods struggle to capture intricate multi-agent interactions and long-term temporal dependencies; on the other hand, the high redundancy of traditional generative models in the computational process of gradually denoising makes it challenging to meet real-time requirements. To address these issues, we proposed TrafusionNet, an efficient trajectory prediction framework consisting of a perceptual modeling stage and an accelerated diffusion stage. In the perceptual modeling stage, a temporal-interaction perception module is designed to capture the multi-agent dynamic relationships through interactive cognition. In the accelerated diffusion stage, TrafusionNet samples directly from the latent trajectories generated by the perceptual modeling, skipping redundant denoising steps to optimize sampling efficiency and accuracy. In addition, by generating multiple future latent latent trajectories to address uncertainty, a balance between diversity and accuracy can be achieved. The experimental results on the NBA and ETH- UCY datasets showed that TrafusionNet outperformed existing methods in terms of error, model efficiency, and sampling time, achieving the most advanced predictive performance currently available.