arXiv reaDer
BEiT:イメージトランスフォーマーのBERT事前トレーニング
BEiT: BERT Pre-Training of Image Transformers
ImageTransformersの双方向エンコーダー表現を表す自己監視ビジョン表現モデルBEiTを紹介します。自然言語処理領域で開発されたBERTに続いて、ビジョントランスフォーマーを事前トレーニングするためのマスクされた画像モデリングタスクを提案します。具体的には、各画像には、事前トレーニングで2つのビュー、つまり画像パッチ(16x16ピクセルなど)とビジュアルトークン(つまり、個別のトークン)があります。まず、元の画像を視覚的なトークンに「トークン化」します。次に、いくつかの画像パッチをランダムにマスクし、それらをバックボーンTransformerにフィードします。事前トレーニングの目的は、破損した画像パッチに基づいて元のビジュアルトークンを復元することです。 BEiTを事前トレーニングした後、事前トレーニングされたエンコーダーにタスクレイヤーを追加することにより、ダウンストリームタスクのモデルパラメーターを直接微調整します。画像分類とセマンティックセグメンテーションに関する実験結果は、私たちのモデルが以前の事前トレーニング方法と競争力のある結果を達成することを示しています。たとえば、ベースサイズのBEiTは、ImageNet-1Kで83.2%のトップ1精度を達成し、同じセットアップでゼロからのDeiTトレーニング(81.8%)を大幅に上回っています。さらに、大型のBEiTは、ImageNet-1Kのみを使用して86.3%を取得し、ImageNet-22Kでの監視付き事前トレーニングを使用したViT-Lを上回っています(85.2%)。コードと事前トレーニング済みモデルは、https://aka.ms/beitで入手できます。
We introduce a self-supervised vision representation model BEiT, which stands for Bidirectional Encoder representation from Image Transformers. Following BERT developed in the natural language processing area, we propose a masked image modeling task to pretrain vision Transformers. Specifically, each image has two views in our pre-training, i.e, image patches (such as 16x16 pixels), and visual tokens (i.e., discrete tokens). We first "tokenize" the original image into visual tokens. Then we randomly mask some image patches and fed them into the backbone Transformer. The pre-training objective is to recover the original visual tokens based on the corrupted image patches. After pre-training BEiT, we directly fine-tune the model parameters on downstream tasks by appending task layers upon the pretrained encoder. Experimental results on image classification and semantic segmentation show that our model achieves competitive results with previous pre-training methods. For example, base-size BEiT achieves 83.2% top-1 accuracy on ImageNet-1K, significantly outperforming from-scratch DeiT training (81.8%) with the same setup. Moreover, large-size BEiT obtains 86.3% only using ImageNet-1K, even outperforming ViT-L with supervised pre-training on ImageNet-22K (85.2%). The code and pretrained models are available at https://aka.ms/beit.
updated: Sat Sep 03 2022 14:11:33 GMT+0000 (UTC)
published: Tue Jun 15 2021 16:02:37 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アソシエイト