arXiv reaDer
BoxeR:ボックス-2Dおよび3Dトランスフォーマーへの注意
BoxeR: Box-Attention for 2D and 3D Transformers
この論文では、ボックスアテンションと呼ばれる単純なアテンションメカニズムを提案します。関心のあるボックスからサンプリングされたグリッド機能間の空間的相互作用を可能にし、いくつかのビジョンタスクのトランスフォーマーの学習能力を向上させます。具体的には、Box Transformerの略であるBoxeRを紹介します。これは、入力特徴マップ上の参照ウィンドウからボックスの変換を予測することにより、一連のボックスに対応します。 BoxeRは、グリッド構造を考慮して、これらのボックスの注意の重みを計算します。特に、BoxeR-2Dは、アテンションモジュール内のボックス情報について自然に推論するため、エンドツーエンドのインスタンス検出およびセグメンテーションタスクに適しています。 BoxeR-3Dは、ボックスアテンションモジュールで回転に対する不変性を学習することにより、3Dエンドツーエンドオブジェクト検出のために鳥瞰図平面から識別情報を生成することができます。私たちの実験は、提案されたBoxeR-2DがCOCO検出とインスタンスセグメンテーションに関して最先端の結果を達成することを示しています。さらに、BoxeR-3Dは、エンドツーエンドの3Dオブジェクト検出ベースラインを改善し、クラス固有の最適化を行わなくても、WaymoOpenの車両カテゴリで魅力的なパフォーマンスをすでに実現しています。コードはhttps://github.com/kienduynguyen/BoxeRで入手できます。
In this paper, we propose a simple attention mechanism, we call box-attention. It enables spatial interaction between grid features, as sampled from boxes of interest, and improves the learning capability of transformers for several vision tasks. Specifically, we present BoxeR, short for Box Transformer, which attends to a set of boxes by predicting their transformation from a reference window on an input feature map. The BoxeR computes attention weights on these boxes by considering its grid structure. Notably, BoxeR-2D naturally reasons about box information within its attention module, making it suitable for end-to-end instance detection and segmentation tasks. By learning invariance to rotation in the box-attention module, BoxeR-3D is capable of generating discriminative information from a bird's-eye view plane for 3D end-to-end object detection. Our experiments demonstrate that the proposed BoxeR-2D achieves state-of-the-art results on COCO detection and instance segmentation. Besides, BoxeR-3D improves over the end-to-end 3D object detection baseline and already obtains a compelling performance for the vehicle category of Waymo Open, without any class-specific optimization. Code is available at https://github.com/kienduynguyen/BoxeR.
updated: Fri Mar 25 2022 09:42:32 GMT+0000 (UTC)
published: Thu Nov 25 2021 13:54:25 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アソシエイト