軌道予測は、起こりうる危険を予測して信頼性の高い自動運転システムを構築するための重要なタスクです。重要な問題の 1 つは、衝突することなく一貫した軌道予測を生成することです。この課題を克服するために、運転環境におけるエージェントの複雑な行動をより適切に表現する、軌道予測用の効率的なマスク付きオートエンコーダー (Traj-MAE) を提案します。具体的には、Traj-MAE は、さまざまなマスキング戦略を採用して、軌跡エンコーダーとマップ エンコーダーを事前トレーニングし、複数の粒度から環境の影響を活用しながら、エージェント間の社会的および時間的情報をキャプチャできるようにします。複数のマスキング戦略を使用してネットワークを事前トレーニングするときに発生する壊滅的な忘却の問題に対処するために、Traj-MAE がさまざまな戦略から貴重で多様な情報を効率的に学習するのに役立つ、継続的な事前トレーニング フレームワークを導入します。マルチエージェント設定とシングルエージェント設定の両方での実験結果は、Traj-MAE が最先端の方法で競争力のある結果を達成し、ベースライン モデルを大幅に上回ることを示しています。
Trajectory prediction has been a crucial task in building a reliable autonomous driving system by anticipating possible dangers. One key issue is to generate consistent trajectory predictions without colliding. To overcome the challenge, we propose an efficient masked autoencoder for trajectory prediction (Traj-MAE) that better represents the complicated behaviors of agents in the driving environment. Specifically, our Traj-MAE employs diverse masking strategies to pre-train the trajectory encoder and map encoder, allowing for the capture of social and temporal information among agents while leveraging the effect of environment from multiple granularities. To address the catastrophic forgetting problem that arises when pre-training the network with multiple masking strategies, we introduce a continual pre-training framework, which can help Traj-MAE learn valuable and diverse information from various strategies efficiently. Our experimental results in both multi-agent and single-agent settings demonstrate that Traj-MAE achieves competitive results with state-of-the-art methods and significantly outperforms our baseline model.