Injecting Knowledge in Data-driven

Vehicle Trajectory Predictors

Transportation Research Part C: Emerging Technologies, 2021,
Ecole Polytechnique Fédérale de Lausanne (EPFL),
Illustration of our Realistic Residual Block (RRB) model. The knowledge-based model generates a scene-compliant trajectory (yellow dots) which could not effectively account for other agents, hence, is too conservative. Our data-driven RRB improves the prediction (purple dots) by adding confined residuals conditioned on other agents in the scene (illustrated by the arrows). The blue region shows the physically-constrained output space for our RRB predictions.

Abstract

Vehicle trajectory prediction tasks have been commonly tackled from two distinct perspectives: either with knowledge-driven methods or more recently with data-driven ones. On the one hand, we can explicitly implement domain-knowledge or physical priors such as anticipating that vehicles will follow the middle of the roads. While this perspective leads to feasible outputs, it has limited performance due to the difficulty to hand-craft complex interactions in urban environments. On the other hand, recent works use data-driven approaches which can learn complex interactions from the data leading to superior performance. However, generalization, i.e., having accurate predictions on unseen data, is an issue leading to unrealistic outputs. In this paper, we propose to learn a “Realistic Residual Block” (RRB), which effectively connects these two perspectives. Our RRB takes any off-the-shelf knowledge-driven model and finds the required residuals to add to the knowledge-aware trajectory. Our proposed method outputs realistic predictions by confining the residual range and taking into account its uncertainty. We also constrain our output with Model Predictive Control (MPC) to satisfy kinematic constraints. Using a publicly available dataset, we show that our method outperforms previous works in terms of accuracy and generalization to new scenes.

Model Structure

Our proposed Realistic Residual Block (RRB) takes as input (i) the state of the scene, St, and (ii) the knowledge-driven prediction, y^kd. The Residual estimator block builds physically-constrained residuals y^res. Then, the IVW-addition block merges y^kd and y^res by Inverse-Variance Weighted sum and forms y^ref according to equation 3. Finally, Model Predictive Controller (MPC) satisfies the kinematic constraints.
The ’Residual estimator’ block inside our RRB. Three encoders are employed to embed the history, the interactions, and the KD trajectory. The residual decoder integrates the features and estimates the residual distribution. The mean of the distribution is confined to C, a real-word extracted parameter.

Qualitative results

Qualitative results of different baselines. Given the scene and history (Hist), the models predict the future positions. The ground truth (GT) is shown in black. The encoder-decoder neural network (EDN) captures the interactions while is prone to create unrealistic outputs shown in first row. Knowledge-driven (KD) model has realistic predictions as it uses the scene knowledge but cannot reason about the interactions among vehicles as shown in the second row. Our proposed RRB predicts realistic and interaction-aware outputs.

*For the qualitative results, please refer to the paper.

BibTeX

@article{bahari2021injecting,
  title={Injecting Knowledge in Data-driven Vehicle Trajectory Predictors},
  author={Bahari, Mohammadhossein and Nejjar, Ismail and Alahi, Alexandre},
  journal={arXiv preprint arXiv:2103.04854},
  year={2021}
}