arXiv reaDer
PatchFormer:パッチに注意を払った効率的なポイントトランスフォーマー
PatchFormer: An Efficient Point Transformer with Patch Attention
ポイントクラウド学習コミュニティは、CNNからTransformersへのモデリングの移行を目の当たりにしており、純粋なTransformerアーキテクチャが主要な学習ベンチマークで最高の精度を達成しています。ただし、既存のポイントトランスフォーマーは、入力サイズに関して2次の複雑さ(空間と時間の両方)を持つ大きなアテンションマップを生成する必要があるため、計算コストが高くなります。この欠点を解決するために、パッチアテンション(PAT)を導入して、アテンションマップが計算されるはるかに小さなベースのセットを適応的に学習します。これらのベースに重み付けされた合計により、PATはグローバルな形状コンテキストをキャプチャするだけでなく、入力サイズに対して線形の複雑さを実現します。さらに、軽量のマルチスケールアテンション(MST)ブロックを提案して、さまざまなスケールの機能間でアテンションを構築し、モデルにマルチスケール機能を提供します。 PATとMSTを搭載し、両方のモジュールを点群学習用の共同フレームワークに統合するPatchFormerと呼ばれるニューラルアーキテクチャを構築します。広範な実験により、当社のネットワークは、以前のポイントトランスフォーマーよりも9.2倍高速化された、一般的なポイントクラウド学習タスクで同等の精度を達成することが実証されています。
The point cloud learning community is witnesses a modeling shift from CNNs to Transformers, where pure Transformer architectures have achieved top accuracy on the major learning benchmarks. However, existing point Transformers are computationally expensive since they need to generate a large attention map, which has quadratic complexity (both in space and time) with respect to input size. To solve this shortcoming, we introduce Patch attention (PAT) to adaptively learn a much smaller set of bases upon which the attention maps are computed. By a weighted summation upon these bases, PAT not only captures the global shape context but also achieves linear complexity to input size. In addition, we propose a lightweight Multi-scale attention (MST) block to build attentions among features of different scales, providing the model with multi-scale features. Equipped with the PAT and MST, we construct our neural architecture called PatchFormer that integrates both modules into a joint framework for point cloud learning. Extensive experiments demonstrate that our network achieves comparable accuracy on general point cloud learning tasks with 9.2x speed-up than previous point Transformers.
updated: Thu Dec 02 2021 06:54:02 GMT+0000 (UTC)
published: Sat Oct 30 2021 08:39:55 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アソシエイト