arXiv reaDer
自己蒸留によるデータ効率の高い言語教師ありゼロショット学習
Data-Efficient Language-Supervised Zero-Shot Learning with Self-Distillation
従来のコンピュータビジョンモデルは、事前定義されたカテゴリの固定セットを予測するようにトレーニングされています。最近、自然言語は、監視対象の「ゴールド」ラベルよりも視覚的な概念をより詳細に説明する、より広く豊富な監視元であることが示されています。 CLIPなどの以前の作品では、画像とテキストキャプションの組み合わせを予測する簡単な事前トレーニングタスクを使用しています。ただし、CLIPはデータを大量に消費するため、トレーニングには4億を超える画像テキストペアが必要です。ソフトラベルを使用してノイズの多い画像とテキストのペアから学習する、データ効率の高い対照蒸留法を提案します。私たちのモデルは、事前にトレーニングされた画像および文のエンコーダーから知識を転送し、CLIPの133分の1の3Mの画像テキストペアのみで強力なパフォーマンスを実現します。私たちの方法は、ResNet50画像エンコーダーとDeCLUTRテキストエンコーダーを使用した場合、ImageNet 21k + 1kでの一般的なゼロショット学習の以前のSoTAを73%上回っています。また、Google Open Images(19,958クラス)でのゼロショット評価では、CLIPを10.5%上回っています。
Traditional computer vision models are trained to predict a fixed set of predefined categories. Recently, natural language has been shown to be a broader and richer source of supervision that provides finer descriptions to visual concepts than supervised "gold" labels. Previous works, such as CLIP, use a simple pretraining task of predicting the pairings between images and text captions. CLIP, however, is data hungry and requires more than 400M image text pairs for training. We propose a data-efficient contrastive distillation method that uses soft labels to learn from noisy image-text pairs. Our model transfers knowledge from pretrained image and sentence encoders and achieves strong performance with only 3M image text pairs, 133x smaller than CLIP. Our method exceeds the previous SoTA of general zero-shot learning on ImageNet 21k+1k by 73% relatively with a ResNet50 image encoder and DeCLUTR text encoder. We also beat CLIP by 10.5% relatively on zero-shot evaluation on Google Open Images (19,958 classes).
updated: Sun Apr 18 2021 19:55:31 GMT+0000 (UTC)
published: Sun Apr 18 2021 19:55:31 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アソシエイト