BLIP-2:利用冻结的图像编码器和大型语言模型引导语言-图像预训练

BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Models

李俊男 Junnan Li · Salesforce · 2023-01-30 · arXiv:2301.12597 ↗ · 被引 8480

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

摘要 · Abstract

由于大规模模型的端到端训练,视觉与语言预训练的成本变得越来越高昂。本文提出 BLIP-2,一种通用且高效的预训练策略,它从现成的冻结预训练图像编码器和冻结大型语言模型中引导视觉语言预训练。BLIP-2 通过轻量级的查询 Transformer 弥合模态差距,该 Transformer 分两个阶段进行预训练。第一阶段从冻结的图像编码器引导视觉语言表示学习。第二阶段从冻结的语言模型引导视觉到语言的生成学习。尽管 BLIP-2 的可训练参数远少于现有方法,但在各种视觉语言任务上仍取得了最先进的性能。例如,我们的模型在零样本 VQAv2 上以比 Flamingo80B 少 54 倍的可训练参数,性能提升了 8.7%。我们还展示了模型在零样本图像到文本生成方面的新兴能力,能够遵循自然语言指令。

The cost of vision-and-language pre-training has become increasingly prohibitive due to end-to-end training of large-scale models. This paper proposes BLIP-2, a generic and efficient pre-training strategy that bootstraps vision-language pre-training from off-the-shelf frozen pre-trained image encoders and frozen large language models. BLIP-2 bridges the modality gap with a lightweight Querying Transformer, which is pre-trained in two stages. The first stage bootstraps vision-language representation learning from a frozen image encoder. The second stage bootstraps vision-to-language generative learning from a frozen language model. BLIP-2 achieves state-of-the-art performance on various vision-language tasks, despite having significantly fewer trainable parameters than existing methods. For example, our model outperforms Flamingo80B by 8.7% on zero-shot VQAv2 with 54x fewer trainable parameters. We also demonstrate the model's emerging capabilities of zero-shot image-to-text generation that can follow natural language instructions.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →