用更少数据实现噪声鲁棒的视觉语言预训练,性能领先
NEVLP: Noise-Robust Framework for Efficient Vision-Language Pre-training
- 通过可学习变换器桥接冻结图像编码器与大语言模型
- 在多个任务上以更少数据达到顶尖效果,如图文检索、图像描述
- 适合资源有限但需高效利用噪声数据的研究者
视觉语言模型(VLMs)在各类视觉语言任务中的成功高度依赖大规模网络爬取数据的预训练。然而,网络数据的噪声和不完整性使得数据规模成为性能关键,导致端到端训练越来越不可行。本文提出NEVLP,一种噪声鲁棒的高效视觉语言预训练框架,所需预训练数据更少。具体地,我们通过一个Transformer桥接冻结的图像编码器与大语言模型,并引入两种创新学习策略:噪声自适应学习与概念增强学习以缓解噪声影响。在噪声自适应学习中,基于Transformer的记忆效应估计每对图文的噪声概率,并对图像-文本对比学习施加噪声自适应正则化,以调节跨模态对齐。在概念增强学习中,通过引入图像中的视觉概念(如物体)来丰富不完整的文本,为图文匹配和图像引导的文本生成提供先验信息,从而缓解文本不完整问题。该框架有效利用噪声网络数据,在包括图文检索、图像描述和视觉问答在内的多种视觉语言任务中,以更少预训练数据达到当前最优性能。
原文摘要 · Abstract (English)
The success of Vision Language Models (VLMs) on various vision-language tasks heavily relies on pre-training with large scale web-crawled datasets. However, the noisy and incomplete nature of web data makes dataset scale crucial for performance, rendering end-to-end training increasingly prohibitive. In this paper, we propose NEVLP, a noise-robust framework for efficient vision-language pre-training that requires less pre-training data. Specifically, we bridge the modality gap between a frozen image encoder and a large language model with a transformer and introduce two innovative learning strategies: noise-adaptive learning and concept-enhanced learning to mitigate the impact of noise. In noise-adaptive learning, we estimate the noise probability of each image-text pair based on the transformer's memorization effect and employ noise-adaptive regularization on image-text contrastive learning to condition cross-modal alignment. In concept-enhanced learning, we enrich incomplete text by incorporating visual concepts (objects in the image) to provide prior information about existing objects for image-text matching and image-grounded text generation, thereby mitigating text incompletion. Our framework effectively utilizes noisy web data and achieves state-of-the-art performance with less pre-training data across a wide range of vision-language tasks, including image-text retrieval, image captioning, and visual question answering.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。