arXiv:2411.12313cs.LGcs.CV2024-11

通过因果干预提升多智能体轨迹预测的泛化能力

C$^{2}$INet: Realizing Incremental Trajectory Prediction with Prior-Aware Continual Causal Intervention

  • 利用变分推断对齐环境先验与潜在空间中的混淆因子后验
  • 在三个真实与合成数据集上表现优于现有方法,稳定抑制场景偏差
  • 适合自动驾驶等需持续学习的复杂场景应用

复杂场景下的多智能体轨迹预测对自动驾驶等应用至关重要。现有方法常忽视环境偏差,导致泛化性能差;硬件限制使大规模跨场景数据难以使用,持续学习更加剧了灾难性遗忘问题。为此,我们提出持续因果干预(C²INet)方法,在持续学习框架下实现可泛化的多智能体轨迹预测。通过变分推断,将环境相关先验与潜在空间中混淆因子的后验估计对齐,从而干预影响轨迹表征的因果关联。此外,采用记忆队列存储不同场景下的最优变分先验,确保增量任务训练中持续去偏。C²INet 提升了对多样化任务的适应性,同时保留旧任务信息以防止遗忘,并引入剪枝策略缓解过拟合。在三个真实与合成复杂数据集上的对比实验表明,该方法始终优于先进方法,有效缓解不同场景特有的混淆因素,凸显其在实际应用中的价值。

原文摘要 · Abstract (English)

Trajectory prediction for multi-agents in complex scenarios is crucial for applications like autonomous driving. However, existing methods often overlook environmental biases, which leads to poor generalization. Additionally, hardware constraints limit the use of large-scale data across environments, and continual learning settings exacerbate the challenge of catastrophic forgetting. To address these issues, we propose the Continual Causal Intervention (C$^{2}$INet) method for generalizable multi-agent trajectory prediction within a continual learning framework. Using variational inference, we align environment-related prior with posterior estimator of confounding factors in the latent space, thereby intervening in causal correlations that affect trajectory representation. Furthermore, we store optimal variational priors across various scenarios using a memory queue, ensuring continuous debiasing during incremental task training. The proposed C$^{2}$INet enhances adaptability to diverse tasks while preserving previous task information to prevent catastrophic forgetting. It also incorporates pruning strategies to mitigate overfitting. Comparative evaluations on three real and synthetic complex datasets against state-of-the-art methods demonstrate that our proposed method consistently achieves reliable prediction performance, effectively mitigating confounding factors unique to different scenarios. This highlights the practical value of our method for real-world applications.

轨迹预测因果干预持续学习多智能体

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