Convolutional Social Pooling for Vehicle Trajectory Prediction
About
Forecasting the motion of surrounding vehicles is a critical ability for an autonomous vehicle deployed in complex traffic. Motion of all vehicles in a scene is governed by the traffic context, i.e., the motion and relative spatial configuration of neighboring vehicles. In this paper we propose an LSTM encoder-decoder model that uses convolutional social pooling as an improvement to social pooling layers for robustly learning interdependencies in vehicle motion. Additionally, our model outputs a multi-modal predictive distribution over future trajectories based on maneuver classes. We evaluate our model using the publicly available NGSIM US-101 and I-80 datasets. Our results show improvement over the state of the art in terms of RMS values of prediction error and negative log-likelihoods of true future trajectories under the model's predictive distribution. We also present a qualitative analysis of the model's predicted distributions for various traffic scenarios.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Trajectory Prediction | HighD | RMSE0.24 | 64 | |
| Trajectory Prediction | SHRP2 | minADE0.4 | 40 | |
| Trajectory Prediction | HighD | minADE0.14 | 40 | |
| Trajectory Prediction | NGSIM | ADE (Avg)2.29 | 26 | |
| Vehicle Trajectory Prediction | HighD (test) | RMSE0.22 | 25 | |
| Vehicle Trajectory Prediction | NGSIM (test) | RMSE0.61 | 25 | |
| Trajectory Prediction | NGSIM | RMSE (1s)1.17 | 18 | |
| Trajectory Prediction | HighD | RMSE (1s)0.76 | 18 | |
| Trajectory Prediction | HighD | ADE (Avg)2.14 | 16 | |
| Trajectory Prediction | NGSIM | RMSE (1s)0.58 | 11 |