arXiv reaDer
自然言語の監督から転送可能な視覚モデルを学ぶ
Learning Transferable Visual Models From Natural Language Supervision
最先端のコンピュータビジョンシステムは、事前に決定されたオブジェクトカテゴリの固定セットを予測するようにトレーニングされています。この制限された形式の監視は、他の視覚的概念を指定するために追加のラベル付きデータが必要になるため、それらの一般性と使いやすさを制限します。画像に関する生のテキストから直接学習することは、はるかに幅広い監視ソースを活用する有望な代替手段です。どのキャプションがどの画像に対応するかを予測する単純な事前トレーニングタスクが、インターネットから収集された4億(画像、テキスト)ペアのデータセットでSOTA画像表現を最初から学習する効率的でスケーラブルな方法であることを示します。事前トレーニング後、自然言語を使用して、学習した視覚的概念を参照(または新しい概念を説明)し、モデルをダウンストリームタスクにゼロショットで転送できるようにします。 OCR、ビデオでの行動認識、ジオローカリゼーション、さまざまな種類のきめ細かいオブジェクト分類などのタスクにまたがる、30を超える既存のコンピュータービジョンデータセットのベンチマークによって、このアプローチのパフォーマンスを調査します。モデルはほとんどのタスクに自明ではなく転送され、データセット固有のトレーニングを必要とせずに、完全に監視されたベースラインと競合することがよくあります。たとえば、ImageNetゼロショットでの元のResNet-50の精度と一致しますが、トレーニングされた128万のトレーニング例を使用する必要はありません。コードと事前トレーニング済みのモデルの重みをhttps://github.com/OpenAI/CLIPでリリースします。
State-of-the-art computer vision systems are trained to predict a fixed set of predetermined object categories. This restricted form of supervision limits their generality and usability since additional labeled data is needed to specify any other visual concept. Learning directly from raw text about images is a promising alternative which leverages a much broader source of supervision. We demonstrate that the simple pre-training task of predicting which caption goes with which image is an efficient and scalable way to learn SOTA image representations from scratch on a dataset of 400 million (image, text) pairs collected from the internet. After pre-training, natural language is used to reference learned visual concepts (or describe new ones) enabling zero-shot transfer of the model to downstream tasks. We study the performance of this approach by benchmarking on over 30 different existing computer vision datasets, spanning tasks such as OCR, action recognition in videos, geo-localization, and many types of fine-grained object classification. The model transfers non-trivially to most tasks and is often competitive with a fully supervised baseline without the need for any dataset specific training. For instance, we match the accuracy of the original ResNet-50 on ImageNet zero-shot without needing to use any of the 1.28 million training examples it was trained on. We release our code and pre-trained model weights at https://github.com/OpenAI/CLIP.
updated: Fri Feb 26 2021 19:04:58 GMT+0000 (UTC)
published: Fri Feb 26 2021 19:04:58 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アソシエイト