arXiv reaDer
kMaX-DeepLab: k-means マスク変換器
kMaX-DeepLab: k-means Mask Transformer
ビジョンタスクにおけるトランスフォーマーの台頭は、ネットワークバックボーン設計を進歩させるだけでなく、エンドツーエンドの画像認識(物体検出やパノプティックセグメンテーションなど)を実現するためのまったく新しいページを開始します。自然言語処理 (NLP) に由来するトランスフォーマー アーキテクチャは、セルフ アテンションとクロス アテンションで構成され、シーケンス内の要素間の長距離の相互作用を効果的に学習します。しかし、既存のトランスフォーマーベースの視覚モデルのほとんどは、単に NLP からアイデアを借用しており、言語と画像の間の決定的な違い、特に空間的に平坦化されたピクセル特徴の非常に長いシーケンス長を無視していることがわかります。これにより、ピクセルの特徴とオブジェクトのクエリの間のクロスアテンションの学習が妨げられます。この論文では、ピクセルとオブジェクト クエリの関係を再考し、クロスアテンション学習をクラスタリング プロセスとして再定式化することを提案します。従来の K 平均法クラスタリング アルゴリズムからインスピレーションを得て、セグメンテーション タスク用の K 平均法マスク Xformer (kMaX-DeepLab) を開発しました。これは最先端技術を向上させるだけでなく、シンプルでエレガントなデザインも備えています。その結果、当社の kMaX-DeepLab は、58.0% PQ の COCO val セット、68.4% PQ、44.0% AP、および 83.5% mIoU の Cityscapes val セット、および ADE20K val セットで新しい最先端のパフォーマンスを達成しました。テスト時の拡張または外部データセットなしの場合、50.9% の PQ および 55.2% の mIoU。私たちの研究が、ビジョンタスクに合わせた変圧器の設計に光を当てることができれば幸いです。 TensorFlow のコードとモデルは https://github.com/google-research/deeplab2 で入手できます。PyTorch の再実装も https://github.com/bytedance/kmax-deeplab で入手できます。
The rise of transformers in vision tasks not only advances network backbone designs, but also starts a brand-new page to achieve end-to-end image recognition (e.g., object detection and panoptic segmentation). Originated from Natural Language Processing (NLP), transformer architectures, consisting of self-attention and cross-attention, effectively learn long-range interactions between elements in a sequence. However, we observe that most existing transformer-based vision models simply borrow the idea from NLP, neglecting the crucial difference between languages and images, particularly the extremely large sequence length of spatially flattened pixel features. This subsequently impedes the learning in cross-attention between pixel features and object queries. In this paper, we rethink the relationship between pixels and object queries and propose to reformulate the cross-attention learning as a clustering process. Inspired by the traditional k-means clustering algorithm, we develop a k-means Mask Xformer (kMaX-DeepLab) for segmentation tasks, which not only improves the state-of-the-art, but also enjoys a simple and elegant design. As a result, our kMaX-DeepLab achieves a new state-of-the-art performance on COCO val set with 58.0% PQ, Cityscapes val set with 68.4% PQ, 44.0% AP, and 83.5% mIoU, and ADE20K val set with 50.9% PQ and 55.2% mIoU without test-time augmentation or external dataset. We hope our work can shed some light on designing transformers tailored for vision tasks. TensorFlow code and models are available at https://github.com/google-research/deeplab2 A PyTorch re-implementation is also available at https://github.com/bytedance/kmax-deeplab
updated: Mon Jul 10 2023 20:59:46 GMT+0000 (UTC)
published: Fri Jul 08 2022 17:59:01 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アソシエイト