Vision Transformersは、多くのコンピュータービジョンタスクで卓越したパフォーマンスを実現しました。 ViTやDeiTなどの初期のVisionTransformerは、グローバルな自己注意を採用しています。これは、パッチの数が多い場合に計算コストが高くなります。効率を向上させるために、最近のVision Transformerは、ローカルウィンドウ内で自己注意が計算されるローカル自己注意メカニズムを採用しています。ウィンドウベースのローカル自己注意が効率を大幅に向上させるという事実にもかかわらず、画像平面内の離れているが類似したパッチ間の関係をキャプチャすることはできません。画像空間の局所的注意のこの制限を克服するために、この論文では、特徴空間におけるパッチの局所性をさらに活用します。パッチをその機能を使用して複数のクラスターにグループ化し、すべてのクラスター内で自己注意を計算します。このような機能空間のローカルアテンションは、異なるローカルウィンドウ間のパッチ間の接続を効果的にキャプチャしますが、それでも関連性があります。特徴空間の局所的注意と画像空間の局所的注意を統合するバイラテラルローカルアテンションビジョントランスフォーマー(BOAT)を提案します。さらに、BOATをSwinモデルとCSWinモデルの両方と統合し、いくつかのベンチマークデータセットでの広範な実験により、BOAT-CSWinモデルが既存の最先端のCNNモデルとビジョントランスフォーマーを明確かつ一貫して上回っていることを示しています。
Vision Transformers achieved outstanding performance in many computer vision tasks. Early Vision Transformers such as ViT and DeiT adopt global self-attention, which is computationally expensive when the number of patches is large. To improve efficiency, recent Vision Transformers adopt local self-attention mechanisms, where self-attention is computed within local windows. Despite the fact that window-based local self-attention significantly boosts efficiency, it fails to capture the relationships between distant but similar patches in the image plane. To overcome this limitation of image-space local attention, in this paper, we further exploit the locality of patches in the feature space. We group the patches into multiple clusters using their features, and self-attention is computed within every cluster. Such feature-space local attention effectively captures the connections between patches across different local windows but still relevant. We propose a Bilateral lOcal Attention vision Transformer (BOAT), which integrates feature-space local attention with image-space local attention. We further integrate BOAT with both Swin and CSWin models, and extensive experiments on several benchmark datasets demonstrate that our BOAT-CSWin model clearly and consistently outperforms existing state-of-the-art CNN models and vision Transformers.