提升视觉语言模型知识推理能力,抗干扰更强。
RoRA-VLM: Robust Retrieval-Augmented Vision Language Models
- 两阶段检索+图像锚定文本扩展,融合多模态查询信息
- 注入对抗噪声并过滤无关视觉信息,显著提升鲁棒性
- 小样本训练即有效,支持零样本跨域迁移
当前视觉语言模型在知识密集型任务上表现仍不佳,主要因难以准确编码视觉对象与场景与其对应实体及背景知识的关联。尽管检索增强方法能高效引入外部知识,但在多模态查询中存在检索不精准、易受冗余噪声影响等挑战。本文提出RORA-VLM,一种专为视觉语言模型设计的鲁棒检索增强框架,包含两项创新:(1) 两阶段检索流程结合图像锚定的文本查询扩展,协同利用查询中的视觉与文本信息,精准检索最相关的多模态知识片段;(2) 通过在检索增强训练中注入对抗噪声,并采用查询导向的视觉标记精炼策略,过滤无关视觉内容(如图像中无关实体),增强模型对冗余信息的抵抗力。我们在三个主流基准数据集上进行了广泛实验,结果表明,仅需少量训练样本,RORA-VLM即可使基础模型实现显著性能提升,在所有基准上持续优于现有最优检索增强视觉语言模型,并展现出新颖的零样本领域迁移能力。
原文摘要 · Abstract (English)
Current vision-language models (VLMs) still exhibit inferior performance on knowledge-intensive tasks, primarily due to the challenge of accurately encoding all the associations between visual objects and scenes to their corresponding entities and background knowledge. While retrieval augmentation methods offer an efficient way to integrate external knowledge, extending them to vision-language domain presents unique challenges in (1) precisely retrieving relevant information from external sources due to the inherent discrepancy within the multimodal queries, and (2) being resilient to the irrelevant, extraneous and noisy information contained in the retrieved multimodal knowledge snippets. In this work, we introduce RORA-VLM, a novel and robust retrieval augmentation framework specifically tailored for VLMs, with two key innovations: (1) a 2-stage retrieval process with image-anchored textual-query expansion to synergistically combine the visual and textual information in the query and retrieve the most relevant multimodal knowledge snippets; and (2) a robust retrieval augmentation method that strengthens the resilience of VLMs against irrelevant information in the retrieved multimodal knowledge by injecting adversarial noises into the retrieval-augmented training process, and filters out extraneous visual information, such as unrelated entities presented in images, via a query-oriented visual token refinement strategy. We conduct extensive experiments to validate the effectiveness and robustness of our proposed methods on three widely adopted benchmark datasets. Our results demonstrate that with a minimal amount of training instance, RORA-VLM enables the base model to achieve significant performance improvement and constantly outperform state-of-the-art retrieval-augmented VLMs on all benchmarks while also exhibiting a novel zero-shot domain transfer capability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。