从自然语言监督中学习可迁移的视觉模型

Learning Transferable Visual Models From Natural Language Supervision

亚历克·拉德福德 Alec Radford · OpenAI · 2021-02-26 · arXiv:2103.00020 ↗ · 被引 51496

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

最先进的计算机视觉系统通常被训练来预测一组固定的预定对象类别。这种受限的监督形式限制了它们的通用性和可用性,因为需要额外的标注数据来指定任何其他视觉概念。直接从关于图像的原始文本中学习是一种有前景的替代方案,它利用了更广泛的监督来源。我们证明,简单的预训练任务——预测哪段文字描述对应哪张图像——是一种高效且可扩展的方法,可以在从互联网收集的 4 亿(图像,文本)对数据集上从零开始学习最先进的图像表示。预训练后,使用自然语言来引用学到的视觉概念(或描述新的概念),使得模型能够零样本迁移到下游任务。我们通过在 30 多个不同的现有计算机视觉数据集上进行基准测试来研究这种方法的性能,这些数据集涵盖 OCR、视频中的动作识别、地理定位以及多种细粒度对象分类等任务。该模型在大多数任务上都能非平凡地迁移,并且通常与完全监督的基线模型竞争,而无需任何特定数据集的训练。例如,我们在 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.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 19)

阅读逐段中英对照全文 →