ビジョン トランスフォーマー (ViTs) は、イメージ パッチを「ビジュアル トークン」として扱い、パッチ間の注意を学習するという前提に基づいて構築されています。パッチ埋め込みベースのトークナイザーには、対応するテキスト トークナイザーとのセマンティック ギャップがあります。パッチパッチへの注意は二次的な複雑さの問題に悩まされており、学習した ViT を説明することも自明ではありません. ViT でこれらの問題に対処するために、この論文では、ViT でパッチからクラスターへの注意 (PaCa) を学習することを提案します.私たちの PaCa-ViT はパッチから始まりますが、キーと値はクラスタリングに直接基づいています (事前に定義された少数のクラスターを使用)。より優れた解釈可能なモデルのためのクラスタリングの注意. 二次複雑さは線形複雑さに緩和されます. 提案されたPaCaモジュールは、効率的で解釈可能なViTバックボーンとセマンティックセグメンテーションヘッドネットワークの設計に使用されます. 実験では, 提案された方法はImageNetでテストされています. 1k 画像分類、MS-COCO オブジェクト検出とインスタンス セグメンテーション、および MIT-ADE20k セマンティック セグメンテーション。従来技術と比較すると、3 つのベンチマークすべてで、ImageNet-1k と MIT-ADE20k で SWin と PVT よりも大幅に優れたパフォーマンスが得られます。また、線形の複雑さにより、MS-COCO および MIT-ADE20k の PVT モデルよりも大幅に効率的です。学習したクラスタは意味的に意味があります。コードとモデルのチェックポイントは、https://github.com/iVMCL/PaCaViT で入手できます。
Vision Transformers (ViTs) are built on the assumption of treating image patches as ``visual tokens" and learn patch-to-patch attention. The patch embedding based tokenizer has a semantic gap with respect to its counterpart, the textual tokenizer. The patch-to-patch attention suffers from the quadratic complexity issue, and also makes it non-trivial to explain learned ViTs. To address these issues in ViT, this paper proposes to learn Patch-to-Cluster attention (PaCa) in ViT. Queries in our PaCa-ViT starts with patches, while keys and values are directly based on clustering (with a predefined small number of clusters). The clusters are learned end-to-end, leading to better tokenizers and inducing joint clustering-for-attention and attention-for-clustering for better and interpretable models. The quadratic complexity is relaxed to linear complexity. The proposed PaCa module is used in designing efficient and interpretable ViT backbones and semantic segmentation head networks. In experiments, the proposed methods are tested on ImageNet-1k image classification, MS-COCO object detection and instance segmentation and MIT-ADE20k semantic segmentation. Compared with the prior art, it obtains better performance in all the three benchmarks than the SWin and the PVTs by significant margins in ImageNet-1k and MIT-ADE20k. It is also significantly more efficient than PVT models in MS-COCO and MIT-ADE20k due to the linear complexity. The learned clusters are semantically meaningful. Code and model checkpoints are available at https://github.com/iVMCL/PaCaViT.