arXiv reaDer
コンパクトトランスフォーマーでビッグデータパラダイムを回避
Escaping the Big Data Paradigm with Compact Transformers
言語処理の標準としてのトランスフォーマーの台頭、コンピュータービジョンの進歩、および前例のないサイズと量のトレーニングデータにより、多くの人がトランスフォーマーは小さなデータセットには適さないと信じるようになりました。この傾向は、特定の科学分野でのデータの利用可能性の制限、およびこの分野の研究からのリソースが限られているデータの除外を含むがこれらに限定されない大きな懸念につながります。このホワイトペーパーでは、トランスフォーマーは「データを大量に消費する」ため、大量のデータセットにのみ適用できるという神話を払拭します。適切なサイズとトークン化により、トランスフォーマーが小さなデータセットで最先端のCNNと直接実行できることを初めて示しました。多くの場合、精度が高く、パラメーターが少なくなっています。私たちのモデルは、新しいシーケンスプーリング戦略と畳み込みの使用により、クラストークンと位置埋め込みの要件を排除します。モデルサイズの点で柔軟性があり、良好な結果を達成しながら、わずか0.28Mのパラメーターを持つことができます。私たちのモデルは、CIFAR-10で最初からトレーニングすると、98.00%の精度に達する可能性があります。これは、以前のTransformerベースのモデルよりも大幅に改善されています。また、ResNetなどの多くの最新のCNNベースのアプローチや、Proxyless-NASなどの最近のNASベースのアプローチよりも優れています。私たちのシンプルでコンパクトな設計は、限られたコンピューティングリソースを持つ人々や小さなデータセットを扱う人々がトランスフォーマーにアクセスできるようにすることで、トランスフォーマーを民主化します。私たちの方法は、ImageNet(ViTの29%のパラメーターで82.71%の精度)やNLPタスクなどのより大きなデータセットでも機能します。コードと事前トレーニング済みモデルは、https://github.com/SHI-Labs/Compact-Transformersで公開されています。
With the rise of Transformers as the standard for language processing, and their advancements in computer vision, along with their unprecedented size and amounts of training data, many have come to believe that they are not suitable for small sets of data. This trend leads to great concerns, including but not limited to: limited availability of data in certain scientific domains and the exclusion of those with limited resource from research in the field. In this paper, we dispel the myth that transformers are "data hungry" and therefore can only be applied to large sets of data. We show for the first time that with the right size and tokenization, transformers can perform head-to-head with state-of-the-art CNNs on small datasets, often with better accuracy and fewer parameters. Our model eliminates the requirement for class token and positional embeddings through a novel sequence pooling strategy and the use of convolution/s. It is flexible in terms of model size, and can have as little as 0.28M parameters while achieving good results. Our model can reach 98.00% accuracy when training from scratch on CIFAR-10, which is a significant improvement over previous Transformer based models. It also outperforms many modern CNN based approaches, such as ResNet, and even some recent NAS-based approaches, such as Proxyless-NAS. Our simple and compact design democratizes transformers by making them accessible to those with limited computing resources and/or dealing with small datasets. Our method also works on larger datasets, such as ImageNet (82.71% accuracy with 29% parameters of ViT), and NLP tasks as well. Our code and pre-trained models are publicly available at https://github.com/SHI-Labs/Compact-Transformers.
updated: Fri Aug 13 2021 17:02:45 GMT+0000 (UTC)
published: Mon Apr 12 2021 17:58:56 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アソシエイト