トランスフォーマーは、長距離の依存関係をキャプチャする機能のおかげで、多くのビジョンタスクで成功しています。ただし、それらの2次計算の複雑さは、オブジェクト検出、特徴マッチング、ステレオなどの高密度予測を必要とするビジョンタスクにそれらを適用する上で大きな障害となります。QuadTreeAttentionを導入し、計算の複雑さを2次から線形に減らします。私たちの四分木トランスフォーマーはトークンピラミッドを構築し、粗い方法から細かい方法で注意を計算します。各レベルで、注意スコアが最も高い上位Kパッチが選択され、次のレベルでは、これらの上位Kパッチに対応する関連領域内でのみ注意が評価されます。クアッドツリーアテンションがさまざまなビジョンタスクで最先端のパフォーマンスを実現することを示します。たとえば、ScanNetでの機能マッチングが4.0%向上し、ステレオマッチングが約50%フロップス削減され、トップ1の精度が0.4〜1.5%向上します。 ImageNet分類、COCOオブジェクト検出の1.2〜1.8%の改善、および以前の最先端のトランスフォーマーに対するセマンティックセグメンテーションの0.7〜2.4%の改善。コードはhttps://github.com/Tangshitao/QuadtreeAttentionで入手できます。
Transformers have been successful in many vision tasks, thanks to their capability of capturing long-range dependency. However, their quadratic computational complexity poses a major obstacle for applying them to vision tasks requiring dense predictions, such as object detection, feature matching, stereo, etc. We introduce QuadTree Attention, which reduces the computational complexity from quadratic to linear. Our quadtree transformer builds token pyramids and computes attention in a coarse-to-fine manner. At each level, the top K patches with the highest attention scores are selected, such that at the next level, attention is only evaluated within the relevant regions corresponding to these top K patches. We demonstrate that quadtree attention achieves state-of-the-art performance in various vision tasks, e.g. with 4.0% improvement in feature matching on ScanNet, about 50% flops reduction in stereo matching, 0.4-1.5% improvement in top-1 accuracy on ImageNet classification, 1.2-1.8% improvement on COCO object detection, and 0.7-2.4% improvement on semantic segmentation over previous state-of-the-art transformers. The codes are available at https://github.com/Tangshitao/QuadtreeAttention.