arXiv reaDer
CRAT-Pred:クリスタルグラフ畳み込みニューラルネットワークとマルチヘッド自己注意による車両軌道予測
CRAT-Pred: Vehicle Trajectory Prediction with Crystal Graph Convolutional Neural Networks and Multi-Head Self-Attention
自動運転車は、周囲の車両の動作計画を管理するため、周囲の車両の動作を予測することが不可欠です。現在の最先端の車両予測モデルは、地図情報に大きく依存しています。ただし、実際には、この情報が常に利用できるとは限りません。したがって、マップ情報に依存することなく、車両間の社会的相互作用を効果的にモデル化するように特別に設計された、マルチモーダルで非ラスタライズベースの軌道予測モデルであるCRAT-Predを提案します。 CRAT-Predは、材料科学の分野に由来するグラフ畳み込み法を車両予測に適用し、エッジ機能を効率的に活用できるようにし、マルチヘッドの自己注意と組み合わせます。他のマップフリーアプローチと比較して、このモデルは、モデルパラメータの数を大幅に減らして最先端のパフォーマンスを実現します。それに加えて、自己注意メカニズムが車両間の社会的相互作用を学習できることを定量的に示します。重みは測定可能な相互作用スコアを表します。ソースコードは公開されています。
Predicting the motion of surrounding vehicles is essential for autonomous vehicles, as it governs their own motion plan. Current state-of-the-art vehicle prediction models heavily rely on map information. In reality, however, this information is not always available. We therefore propose CRAT-Pred, a multi-modal and non-rasterization-based trajectory prediction model, specifically designed to effectively model social interactions between vehicles, without relying on map information. CRAT-Pred applies a graph convolution method originating from the field of material science to vehicle prediction, allowing to efficiently leverage edge features, and combines it with multi-head self-attention. Compared to other map-free approaches, the model achieves state-of-the-art performance with a significantly lower number of model parameters. In addition to that, we quantitatively show that the self-attention mechanism is able to learn social interactions between vehicles, with the weights representing a measurable interaction score. The source code is publicly available.
updated: Thu Feb 10 2022 06:57:54 GMT+0000 (UTC)
published: Wed Feb 09 2022 14:36:36 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アソシエイト