arXiv reaDer
EqMotion: 不変相互作用推論による同変マルチエージェント動作予測
EqMotion: Equivariant Multi-agent Motion Prediction with Invariant Interaction Reasoning
関係推論を使用してエージェントの動きを予測することを学習することは、多くのアプリケーションにとって重要です。モーション予測タスクでは、ユークリッドの幾何学的変換とエージェントの相互作用の不変性の下でモーションの同等性を維持することが重要かつ基本的な原則です。ただし、そのような等分散性と不変性のプロパティは、ほとんどの既存の方法で見落とされています。このギャップを埋めるために、不変の相互作用推論を備えた効率的な等変モーション予測モデルである EqMotion を提案します。運動の等価性を達成するために、等価演算の専用設計を通じてユークリッド変換可能な特徴を学習するための等価幾何学的特徴学習モジュールを提案します。エージェントの相互作用を推論するために、より安定した相互作用モデリングを実現する不変相互作用推論モジュールを提案します。より包括的なモーション機能をさらに促進するために、不変パターン機能学習モジュールを提案し、不変パターン機能を学習します。これは、等変幾何学的機能と連携してネットワークの表現力を高めます。粒子動力学、分子動力学、人間の骨格の動き予測、歩行者の軌道予測の 4 つの異なるシナリオで、提案されたモデルの実験を行います。実験結果は、私たちの方法が一般的に適用可能であるだけでなく、4 つのタスクすべてで最先端の予測パフォーマンスを達成し、24.0/30.1/8.6/9.2% 向上することを示しています。コードは https://github.com/MediaBrain-SJTU/EqMotion で入手できます。
Learning to predict agent motions with relationship reasoning is important for many applications. In motion prediction tasks, maintaining motion equivariance under Euclidean geometric transformations and invariance of agent interaction is a critical and fundamental principle. However, such equivariance and invariance properties are overlooked by most existing methods. To fill this gap, we propose EqMotion, an efficient equivariant motion prediction model with invariant interaction reasoning. To achieve motion equivariance, we propose an equivariant geometric feature learning module to learn a Euclidean transformable feature through dedicated designs of equivariant operations. To reason agent's interactions, we propose an invariant interaction reasoning module to achieve a more stable interaction modeling. To further promote more comprehensive motion features, we propose an invariant pattern feature learning module to learn an invariant pattern feature, which cooperates with the equivariant geometric feature to enhance network expressiveness. We conduct experiments for the proposed model on four distinct scenarios: particle dynamics, molecule dynamics, human skeleton motion prediction and pedestrian trajectory prediction. Experimental results show that our method is not only generally applicable, but also achieves state-of-the-art prediction performances on all the four tasks, improving by 24.0/30.1/8.6/9.2%. Code is available at https://github.com/MediaBrain-SJTU/EqMotion.
updated: Mon Mar 27 2023 04:51:17 GMT+0000 (UTC)
published: Mon Mar 20 2023 05:23:46 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アソシエイト