用时空图嵌入预测车辆轨迹,提升自动驾驶长时预测能力
Trajectory Prediction for Autonomous Driving using Agent-Interaction Graph Embedding
- 构建车辆间时空交互图,捕捉动态行驶关系
- 在10秒预测上优于现有方法,尤其长时预测表现更优
- 适合自动驾驶轨迹预测与多智能体系统研究者
自动驾驶系统中的轨迹预测模块对决策与安全至关重要。本文提出一种名为AiGem(Agent-Interaction Graph Embedding)的新方法,用于预测自动驾驶车辆周围交通车辆的轨迹。AiGem通过四个步骤实现:首先,在历史帧中每一步通过空间边连接车辆(图节点),形成空间图;随后,利用时间边将各时间步的空间图串联成时空图;接着,采用深度可分离图编码器网络对时空图生成节点嵌入;然后,使用门控循环单元解码器基于当前时刻嵌入输出状态;最后,通过多层感知机输出网络,以解码状态为输入预测轨迹。实验表明,AiGem在较长预测时间(如10秒)上显著优于当前最先进深度学习算法。
原文摘要 · Abstract (English)
Trajectory prediction module in an autonomous driving system is crucial for the decision-making and safety of the autonomous agent car and its surroundings. This work presents a novel scheme called AiGem (Agent-Interaction Graph Embedding) to predict traffic vehicle trajectories around the autonomous car. AiGem tackles this problem in four steps. First, AiGem formulates the historical traffic interaction with the autonomous agent as a graph in two steps: (1) at each time step of the history frames, agent-interactions are captured using spatial edges between the agents (nodes of the graph), and then, (2) connects the spatial graphs in chronological order using temporal edges. Then, AiGem applies a depthwise graph encoder network on the spatial-temporal graph to generate graph embedding, i.e., embedding of all the nodes in the graph. Next, a sequential Gated Recurrent Unit decoder network uses the embedding of the current timestamp to get the decoded states. Finally, an output network comprising a Multilayer Perceptron is used to predict the trajectories utilizing the decoded states as its inputs. Results show that AiGem outperforms the state-of-the-art deep learning algorithms for longer prediction horizons.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。