arXiv reaDer
画像分類のためのグローバルフィルターネットワーク
Global Filter Networks for Image Classification
視覚のための自己注意および純粋な多層パーセプトロン(MLP)モデルの最近の進歩は、より少ない誘導バイアスで有望なパフォーマンスを達成する上で大きな可能性を示しています。これらのモデルは通常、生データから空間位置間の相互作用を学習することに基づいています。自己注意とMLPの複雑さは、画像サイズが大きくなるにつれて二次関数的に増大します。そのため、高解像度の機能が必要な場合、これらのモデルをスケールアップするのは困難です。この論文では、対数線形の複雑さで周波数領域の長期的な空間依存性を学習する、概念的に単純でありながら計算効率の高いアーキテクチャであるグローバルフィルターネットワーク(GFNet)を紹介します。私たちのアーキテクチャは、ビジョントランスフォーマーの自己注意層を3つの主要な操作に置き換えます。2D離散フーリエ変換、周波数領域の特徴と学習可能なグローバルフィルター間の要素ごとの乗算、および2D逆フーリエ変換です。 ImageNetとダウンストリームタスクの両方で、モデルの精度と複雑さのトレードオフが良好です。私たちの結果は、GFNetが、効率、一般化能力、および堅牢性において、トランススタイルモデルおよびCNNの非常に競争力のある代替手段になり得ることを示しています。コードはhttps://github.com/raoyongming/GFNetで入手できます。
Recent advances in self-attention and pure multi-layer perceptrons (MLP) models for vision have shown great potential in achieving promising performance with fewer inductive biases. These models are generally based on learning interaction among spatial locations from raw data. The complexity of self-attention and MLP grows quadratically as the image size increases, which makes these models hard to scale up when high-resolution features are required. In this paper, we present the Global Filter Network (GFNet), a conceptually simple yet computationally efficient architecture, that learns long-term spatial dependencies in the frequency domain with log-linear complexity. Our architecture replaces the self-attention layer in vision transformers with three key operations: a 2D discrete Fourier transform, an element-wise multiplication between frequency-domain features and learnable global filters, and a 2D inverse Fourier transform. We exhibit favorable accuracy/complexity trade-offs of our models on both ImageNet and downstream tasks. Our results demonstrate that GFNet can be a very competitive alternative to transformer-style models and CNNs in efficiency, generalization ability and robustness. Code is available at https://github.com/raoyongming/GFNet
updated: Tue Oct 26 2021 13:21:45 GMT+0000 (UTC)
published: Thu Jul 01 2021 17:58:16 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アソシエイト