arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1910.08233cs.CVcs.LGcs.ROeess.SP

Spatially-Aware Graph Neural Networks for Relational Behavior Forecasting from Sensor Data

Uber Advanced Technologies Group(优步先进技术集团);University of Toronto(多伦多大学)

Sergio Casas, Cole Gulino, Renjie Liao, Raquel Urtasun

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑