画像分類におけるビジョン トランスフォーマー (ViT) の出現により、視覚表現学習の方法論が変化しました。特に、ViT は、すべての画像パッチにわたってレイヤーごとに完全な受容野で視覚表現を学習します。これは、レイヤーやその他の代替手段 (たとえば、大きなカーネルやアトラス畳み込み) にまたがる CNN の受容野の増加と比較してです。この作業では、初めて、密な視覚的予測 (セマンティック セグメンテーションなど) のための ViT のグローバル コンテキスト学習の可能性を探ります。私たちの動機は、レイヤーごとに完全な受容野でグローバル コンテキストを学習することで、ViT がより強力な長期依存情報を取得できる可能性があることです。これは、高密度の予測タスクに重要です。最初に、画像を一連のパッチとしてエンコードし、ローカル畳み込みと解像度の削減を行わないバニラ ViT を使用して、セマンティック セグメンテーションのより強力な視覚的表現を生成できることを示します。たとえば、SEgmentation TRansformer (SETR) と呼ばれるモデルは、ADE20K (50.28% mIoU、提出日のテスト リーダーボードの最初の位置) と Pascal Context (55.83% mIoU) で優れており、Cityscapes で競争力のあるパフォーマンスを発揮します。費用対効果の高い方法で一般的な高密度の視覚的予測タスクに取り組むために、階層型ローカル グローバル (HLG) トランスフォーマーのファミリーをさらに定式化します。これは、ウィンドウ内のローカル アテンションとピラミッド型アーキテクチャのウィンドウ全体のグローバル アテンションを特徴とします。広範な実験により、私たちの方法がさまざまな高密度予測タスク (オブジェクト検出、インスタンス セグメンテーション、セマンティック セグメンテーションなど) および画像分類で魅力的なパフォーマンスを達成することが示されています。私たちのコードとモデルは、https://github.com/fudan-zvg/SETR で入手できます。
The emergence of vision transformers (ViTs) in image classification has shifted the methodologies for visual representation learning. In particular, ViTs learn visual representation at full receptive field per layer across all the image patches, in comparison to the increasing receptive fields of CNNs across layers and other alternatives (e.g., large kernels and atrous convolution). In this work, for the first time we explore the global context learning potentials of ViTs for dense visual prediction (e.g., semantic segmentation). Our motivation is that through learning global context at full receptive field layer by layer, ViTs may capture stronger long-range dependency information, critical for dense prediction tasks. We first demonstrate that encoding an image as a sequence of patches, a vanilla ViT without local convolution and resolution reduction can yield stronger visual representation for semantic segmentation. For example, our model, termed as SEgmentation TRansformer (SETR), excels on ADE20K (50.28% mIoU, the first position in the test leaderboard on the day of submission) and Pascal Context (55.83% mIoU), and performs competitively on Cityscapes. For tackling general dense visual prediction tasks in a cost-effective manner, we further formulate a family of Hierarchical Local-Global (HLG) Transformers, characterized by local attention within windows and global-attention across windows in a pyramidal architecture. Extensive experiments show that our methods achieve appealing performance on a variety of dense prediction tasks (e.g., object detection and instance segmentation and semantic segmentation) as well as image classification. Our code and models are available at https://github.com/fudan-zvg/SETR.