arXiv reaDer
MLPに注目する
Pay Attention to MLPs
トランスフォーマーは、深層学習における最も重要なアーキテクチャのイノベーションの一つとなっており、過去数年間で多くのブレイクスルーを可能にしてきた。ここでは、ゲーティングを用いたMLPのみをベースにしたシンプルなアテンション非依存のネットワークアーキテクチャgMLPを提案し、主要な言語および視覚アプリケーションにおいてトランスフォーマーと同等の性能を発揮できることを示す。この比較から、gMLPはビジョントランスフォーマーと同等の精度を達成できるため、自己注意は重要ではないことがわかった。BERTでは、我々のモデルは、事前学習パープレキシティでトランスフォーマーと同等の性能を達成し、いくつかの下流のタスクではより優れている。gMLPの性能が低いファインチューニングタスクでは、gMLPモデルを大幅に大きくすることで、トランスフォーマーとの差を縮めることができる。一般的に、今回の実験では、gMLPはデータや計算量が増えてもトランスフォーマーと同等のスケーリングが可能であることを示している。
Transformers have become one of the most important architectural innovations in deep learning and have enabled many breakthroughs over the past few years. Here we propose a simple attention-free network architecture, gMLP, based solely on MLPs with gating, and show that it can perform as well as Transformers in key language and vision applications. Our comparisons show that self-attention is not critical for Vision Transformers, as gMLP can achieve the same accuracy. For BERT, our model achieves parity with Transformers on pretraining perplexity and is better on some downstream tasks. On finetuning tasks where gMLP performs worse, making the gMLP model substantially larger can close the gap with Transformers. In general, our experiments show that gMLP can scale as well as Transformers over increased data and compute.
updated: Mon May 17 2021 17:55:04 GMT+0000 (UTC)
published: Mon May 17 2021 17:55:04 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アソシエイト