道路エージェントの将来の軌道を予測することは、自動運転にとって重要なタスクです。 DenseTNTやPECNetなどの最近の目標ベースの軌道予測方法は、公開データセットの予測タスクで優れたパフォーマンスを示しています。ただし、通常、複雑な目標選択アルゴリズムと最適化が必要です。この作業では、軌道予測のための階層的なエンドツーエンドの深層学習フレームワークであるKEMPを提案します。フレームワークの中核となるのは、キーフレームベースの軌道予測です。キーフレームは、軌道の一般的な方向を追跡する代表的な状態です。 KEMPは、最初に道路コンテキストを条件とするキーフレームを予測し、次にキーフレームと道路コンテキストを条件とする中間状態を入力します。一般的なフレームワークでは、目標条件付きメソッドは、キーフレームの数が1に等しい特殊なケースです。目標条件付きの方法とは異なり、キーフレーム予測子は自動的に学習され、手作りの目標選択アルゴリズムを必要としません。モデルを公開ベンチマークで評価し、Waymo Open Motionデータセットリーダーボードで1位にランク付けされました(2021年9月1日現在)。
Predicting future trajectories of road agents is a critical task for autonomous driving. Recent goal-based trajectory prediction methods, such as DenseTNT and PECNet, have shown good performance on prediction tasks on public datasets. However, they usually require complicated goal-selection algorithms and optimization. In this work, we propose KEMP, a hierarchical end-to-end deep learning framework for trajectory prediction. At the core of our framework is keyframe-based trajectory prediction, where keyframes are representative states that trace out the general direction of the trajectory. KEMP first predicts keyframes conditioned on the road context, and then fills in intermediate states conditioned on the keyframes and the road context. Under our general framework, goal-conditioned methods are special cases in which the number of keyframes equal to one. Unlike goal-conditioned methods, our keyframe predictor is learned automatically and does not require hand-crafted goal-selection algorithms. We evaluate our model on public benchmarks and our model ranked 1st on Waymo Open Motion Dataset Leaderboard (as of September 1, 2021).