arXiv reaDer
SWAT:トークン内およびトークン間の空間構造
SWAT: Spatial Structure Within and Among Tokens
注意メカニズム、フィードフォワードネットワーク、または畳み込みを使用して視覚データをトークン(つまり、画像パッチ)としてモデル化することは、近年非常に効果的です。このようなメソッドには通常、共通のパイプラインがあります。トークン化メソッドの後に、トークン内およびトークン間で情報を混合するための一連のレイヤー/ブロックが続きます。画像パッチがトークンに変換されるとき、それらはしばしば平坦化され、各パッチ内の空間構造を破棄します。その結果、その後の処理(例:マルチヘッドの自己注意)は、そのような情報の回復および/または利益に失敗する可能性があります。このホワイトペーパーでは、トークン化中に空間構造が保持され、混合段階で明示的に使用されると、モデルが大幅に向上する可能性があると主張します。 2つの重要な貢献を提案します:(1)構造を意識したトークン化と(2)構造を意識したミキシング。どちらも最小限の労力で既存のモデルと組み合わせることができます。 ImageNet分類やADE20Kセグメンテーションを含む複数のベンチマークにわたって、DeiT、MLP-Mixer、Swin Transformerなどのモデルよりも優れたモデルファミリー(SWAT)を紹介します。私たちのコードとモデルはオンラインでリリースされます。
Modeling visual data as tokens (i.e., image patches) using attention mechanisms, feed-forward networks or convolutions has been highly effective in recent years. Such methods usually have a common pipeline: a tokenization method, followed by a set of layers/blocks for information mixing, both within and among tokens. When image patches are converted into tokens, they are often flattened, discarding the spatial structure within each patch. As a result, any processing that follows (eg: multi-head self-attention) may fail to recover and/or benefit from such information. In this paper, we argue that models can have significant gains when spatial structure is preserved during tokenization, and is explicitly used during the mixing stage. We propose two key contributions: (1) Structure-aware Tokenization and, (2) Structure-aware Mixing, both of which can be combined with existing models with minimal effort. We introduce a family of models (SWAT), showing improvements over the likes of DeiT, MLP-Mixer and Swin Transformer, across multiple benchmarks including ImageNet classification and ADE20K segmentation. Our code and models will be released online.
updated: Sat Jul 23 2022 14:47:24 GMT+0000 (UTC)
published: Fri Nov 26 2021 18:59:38 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アソシエイト