SSL-Lanes: Self-Supervised Learning for Motion Forecasting in Autonomous Driving

28 Jun 2022  ·  Prarthana Bhattacharyya, Chengjie Huang, Krzysztof Czarnecki ·

Self-supervised learning (SSL) is an emerging technique that has been successfully employed to train convolutional neural networks (CNNs) and graph neural networks (GNNs) for more transferable, generalizable, and robust representation learning. However its potential in motion forecasting for autonomous driving has rarely been explored. In this study, we report the first systematic exploration and assessment of incorporating self-supervision into motion forecasting. We first propose to investigate four novel self-supervised learning tasks for motion forecasting with theoretical rationale and quantitative and qualitative comparisons on the challenging large-scale Argoverse dataset. Secondly, we point out that our auxiliary SSL-based learning setup not only outperforms forecasting methods which use transformers, complicated fusion mechanisms and sophisticated online dense goal candidate optimization algorithms in terms of performance accuracy, but also has low inference time and architectural complexity. Lastly, we conduct several experiments to understand why SSL improves motion forecasting. Code is open-sourced at \url{https://github.com/AutoVision-cloud/SSL-Lanes}.

PDF Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Motion Forecasting Argoverse CVPR 2020 SSL-Lanes MR (K=6) 0.1326 # 248
minADE (K=1) 1.6342 # 258
minFDE (K=1) 3.5643 # 263
MR (K=1) 0.5671 # 254
minADE (K=6) 0.8401 # 237
minFDE (K=6) 1.2493 # 253
DAC (K=6) 0.9844 # 97
brier-minFDE (K=6) 1.9433 # 59

Methods


No methods listed for this paper. Add relevant methods here