arXiv reaDer
PTP:グラフニューラルネットワークとダイバーシティサンプリングによる並列化された追跡と予測
PTP: Parallelized Tracking and Prediction with Graph Neural Networks and Diversity Sampling
マルチオブジェクトトラッキング(MOT)と軌道予測は、マルチエージェントの相互作用の正確なモデリングを必要とする最新の3D知覚システムの2つの重要なコンポーネントです。エージェントの相互作用の共有機能表現を学習するために、1つのフレームワークの下で両方のタスクを統合することが有益であると仮定します。さらに、追跡から予測にエラーを伝播する可能性のある追跡と予測を順番に実行する代わりに、問題を軽減するための並列化されたフレームワークを提案します。また、並列トラック予測フレームワークには、2つの新しい計算ユニットが組み込まれています。まず、グラフニューラルネットワーク(GNN)を導入することにより、機能の相互作用手法を使用して、エージェントが相互作用する方法をキャプチャします。 GNNは、MOTアソシエーションの識別的特徴学習を改善し、軌道予測のための社会的認識コンテキストを提供することができます。次に、ダイバーシティサンプリング機能を使用して、予測される軌道の品質とダイバーシティを向上させます。学習したサンプリング関数は、生成された軌道分布からさまざまな結果を効率的に抽出するようにトレーニングされており、重複した軌道サンプルを生成する問題を回避するのに役立ちます。 KITTIとnuScenesのデータセットで評価し、社会を意識した特徴学習と多様性サンプリングを使用した方法が、3DMOTと軌道予測で新しい最先端のパフォーマンスを実現することを示しています。プロジェクトのウェブサイトは次のとおりです:https://www.xinshuoweng.com/projects/PTP
Multi-object tracking (MOT) and trajectory prediction are two critical components in modern 3D perception systems that require accurate modeling of multi-agent interaction. We hypothesize that it is beneficial to unify both tasks under one framework in order to learn a shared feature representation of agent interaction. Furthermore, instead of performing tracking and prediction sequentially which can propagate errors from tracking to prediction, we propose a parallelized framework to mitigate the issue. Also, our parallel track-forecast framework incorporates two additional novel computational units. First, we use a feature interaction technique by introducing Graph Neural Networks (GNNs) to capture the way in which agents interact with one another. The GNN is able to improve discriminative feature learning for MOT association and provide socially-aware contexts for trajectory prediction. Second, we use a diversity sampling function to improve the quality and diversity of our forecasted trajectories. The learned sampling function is trained to efficiently extract a variety of outcomes from a generative trajectory distribution and helps avoid the problem of generating duplicate trajectory samples. We evaluate on KITTI and nuScenes datasets showing that our method with socially-aware feature learning and diversity sampling achieves new state-of-the-art performance on 3D MOT and trajectory prediction. Project website is: https://www.xinshuoweng.com/projects/PTP
updated: Sat Apr 03 2021 13:56:15 GMT+0000 (UTC)
published: Tue Mar 17 2020 17:53:33 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト