arXiv reaDer
ビジョン トランスフォーマーのためのウェーブレット ニューラル オペレーターによるマルチスケール アテンション
Multiscale Attention via Wavelet Neural Operators for Vision Transformers
トランスフォーマーは、コンピューター ビジョンで広く成功を収めています。その中心にあるのは、自己注意 (SA) メカニズムです。これは、加重基準によって入力内の各トークンを他のすべてのトークンに関連付ける誘導バイアスです。標準的な SA メカニズムは、シーケンスの長さに関して 2 次の複雑さを持っているため、高解像度の視覚に現れる長いシーケンスへの有用性が妨げられています。最近、PDE のオペレーター学習に触発されて、FFT を介して効率的に実装されるグローバル畳み込みに基づく高解像度アテンションのために、適応フーリエ ニューラル オペレーター (AFNO) が導入されました。ただし、AFNO グローバル フィルタリングは、自然画像によく見られる小規模および中規模の構造をうまく表現できません。粗いスケールから細かいスケールの構造を活用するために、ウェーブレット ニューラル オペレーターを活用してマルチスケール ウェーブレット アテンション (MWA) を導入します。 ViT のアテンションを MWA に置き換え、CIFAR と ImageNet 分類を使用した実験では、AFNO やグローバル フィルター ネットワーク (GFN) などのフーリエ ベースのアテンションに比べて大幅な改善が見られます。
Transformers have achieved widespread success in computer vision. At their heart, there is a Self-Attention (SA) mechanism, an inductive bias that associates each token in the input with every other token through a weighted basis. The standard SA mechanism has quadratic complexity with the sequence length, which impedes its utility to long sequences appearing in high resolution vision. Recently, inspired by operator learning for PDEs, Adaptive Fourier Neural Operators (AFNO) were introduced for high resolution attention based on global convolution that is efficiently implemented via FFT. However, the AFNO global filtering cannot well represent small and moderate scale structures that commonly appear in natural images. To leverage the coarse-to-fine scale structures we introduce a Multiscale Wavelet Attention (MWA) by leveraging wavelet neural operators which incurs linear complexity in the sequence size. We replace the attention in ViT with MWA and our experiments with CIFAR and ImageNet classification demonstrate significant improvement over alternative Fourier-based attentions such as AFNO and Global Filter Network (GFN).
updated: Sun Mar 26 2023 10:20:16 GMT+0000 (UTC)
published: Wed Mar 22 2023 09:06:07 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アソシエイト