arXiv reaDer
3Dオブジェクト検出のためのモデルにとらわれない階層的注意
Model-Agnostic Hierarchical Attention for 3D Object Detection
汎用ネットワーク アーキテクチャとしてのトランスフォーマーは、最近、3D ポイント クラウド オブジェクト検出で大きな成功を収めています。ただし、単純なトランスフォーマーに階層がないため、さまざまなスケールで特徴を学習することが難しくなり、ローカライズされた特徴を抽出する能力が制限されます。このような制限により、サイズの異なるオブジェクトではパフォーマンスが不均衡になり、小さいオブジェクトではパフォーマンスが低下します。この作業では、変圧器ベースの 3D 検出器のモジュール化された階層設計として、2 つの新しい注意メカニズムを提案します。さまざまなスケールでの機能学習を有効にするために、単一スケールの入力機能からマルチスケール トークンを構築する Simple Multi-Scale Attention を提案します。ローカライズされた機能集約のために、すべてのバウンディング ボックスの提案に対して適応的な注意範囲を持つサイズ適応型のローカル 注意を提案します。 Attention モジュールはどちらもモデルに依存しないネットワーク レイヤーであり、既存のポイント クラウド トランスフォーマーにプラグインして、エンドツーエンドのトレーニングを行うことができます。広く使用されている 2 つの屋内 3D ポイント クラウド オブジェクト検出ベンチマークで、この方法を評価します。提案されたモジュールを最先端のトランスベースの 3D 検出器に接続することで、両方のベンチマークで以前の最良の結果を改善し、小さなオブジェクトで最大の改善マージンを実現しました。
Transformers as versatile network architectures have recently seen great success in 3D point cloud object detection. However, the lack of hierarchy in a plain transformer makes it difficult to learn features at different scales and restrains its ability to extract localized features. Such limitation makes them have imbalanced performance on objects of different sizes, with inferior performance on smaller ones. In this work, we propose two novel attention mechanisms as modularized hierarchical designs for transformer-based 3D detectors. To enable feature learning at different scales, we propose Simple Multi-Scale Attention that builds multi-scale tokens from a single-scale input feature. For localized feature aggregation, we propose Size-Adaptive Local Attention with adaptive attention ranges for every bounding box proposal. Both of our attention modules are model-agnostic network layers that can be plugged into existing point cloud transformers for end-to-end training. We evaluate our method on two widely used indoor 3D point cloud object detection benchmarks. By plugging our proposed modules into the state-of-the-art transformer-based 3D detector, we improve the previous best results on both benchmarks, with the largest improvement margin on small objects.
updated: Fri Jan 06 2023 18:52:12 GMT+0000 (UTC)
published: Fri Jan 06 2023 18:52:12 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アソシエイト