TAE: A Semi-supervised Controllable Behavior-aware Trajectory Generator and Predictor
TAE: A Semi-supervised Controllable Behavior-aware Trajectory Generator and Predictor
复制标题
DOI:
10.1109/iros47612.2022.9981029
复制
发表时间:
2022-03
期刊:
影响因子:
--
通讯作者:
Ruochen Jiao;Xiangguo Liu;Bowen Zheng;Davis Liang;Qi Zhu
中科院分区:
文献类型:
--
作者:
Ruochen Jiao;Xiangguo Liu;Bowen Zheng;Davis Liang;Qi Zhu
Trajectory generation and prediction are two in-terwoven tasks that play important roles in planner evaluation and decision making for intelligent vehicles. Most existing methods focus on one of the two and are optimized to directly output the final generated/predicted trajectories, which only contain limited information for critical scenario augmentation and safe planning. In this work, we propose a novel behavior-aware Trajectory Autoencoder (TAE) that explicitly models drivers' behavior such as aggressiveness and intention in the latent space, using semi-supervised adversarial autoencoder and domain knowledge in transportation. Our model addresses trajectory generation and prediction in a unified architecture and benefits both tasks: the model can generate diverse, controllable and realistic trajectories to enhance planner op-timization in safety-critical and long-tailed scenarios, and it can provide prediction of critical behavior in addition to the final trajectories for decision making. Experimental results demonstrate that our method achieves promising performance on both trajectory generation and prediction.