SpAGNN - Spatially-Aware Graph Neural Networks for Relational Behavior Forecasting from Sensor Data.
arxiv(2020)
摘要
In this paper, we tackle the problem of relational behavior forecasting from sensor data. Towards this goal, we propose a novel spatially-aware graph neural network (SpAGNN) that models the interactions between agents in the scene. Specifically, we exploit a convolutional neural network to detect the actors and compute their initial states. A graph neural network then iteratively updates the actor states via a message passing process. Inspired by Gaussian belief propagation, we design the messages to be spatially-transformed parameters of the output distributions from neighboring agents. Our model is fully differentiable, thus enabling end-to-end training. Importantly, our probabilistic predictions can model uncertainty at the trajectory level. We demonstrate the effectiveness of our approach by achieving significant improvements over the state-of-the-art on two real-world self-driving datasets: ATG4D and nuScenes.
更多查看译文
关键词
self-driving,probabilistic predictions,iterative actor state update,spatially-aware graph neural network,convolutional neural network,sensor data,relational behavior forecasting,SpAGNN,model uncertainty,Gaussian belief propagation,message passing
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要