用图像检索增强偏好优化,减少视觉语言模型幻觉。
Re-Align: Aligning Vision Language Models via Retrieval-Augmented Direct Preference Optimization
- 通过图像检索构建图文双重偏好数据集
- 在VQA任务上比之前方法降低30%以上幻觉率
- 适配多种模型规模,适合跨模态应用开发者
大型视觉语言模型(VLM)通过融合视觉模态扩展了单模态大语言模型的能力,推动了多种现实场景中的跨模态应用。尽管性能优异,但VLM仍易产生显著幻觉,尤其表现为跨模态不一致。受强化学习人类反馈(RLHF)成功启发,近期研究聚焦于在精心构建的数据集上应用直接偏好优化(DPO)以缓解该问题。然而,现有方法多采用粗暴方式引入偏好信号,忽视了视觉信息在对齐过程中的关键作用。本文提出Re-Align框架,利用图像检索构建双重视觉与文本偏好数据集,并引入rDPO,在微调中加入额外的视觉偏好目标。实验表明,Re-Align不仅能更有效地抑制幻觉,还在通用视觉问答(VQA)任务上取得显著提升。此外,该方法在不同规模与架构的VLM上均表现出鲁棒性与可扩展性。本工作为多模态大模型对齐提供了新路径,助力更可靠、高效的跨模态应用。代码已开源:https://github.com/taco-group/Re-Align。
原文摘要 · Abstract (English)
The emergence of large Vision Language Models (VLMs) has broadened the scope and capabilities of single-modal Large Language Models (LLMs) by integrating visual modalities, thereby unlocking transformative cross-modal applications in a variety of real-world scenarios. Despite their impressive performance, VLMs are prone to significant hallucinations, particularly in the form of cross-modal inconsistencies. Building on the success of Reinforcement Learning from Human Feedback (RLHF) in aligning LLMs, recent advancements have focused on applying direct preference optimization (DPO) on carefully curated datasets to mitigate these issues. Yet, such approaches typically introduce preference signals in a brute-force manner, neglecting the crucial role of visual information in the alignment process. In this paper, we introduce Re-Align, a novel alignment framework that leverages image retrieval to construct a dual-preference dataset, effectively incorporating both textual and visual preference signals. We further introduce rDPO, an extension of the standard direct preference optimization that incorporates an additional visual preference objective during fine-tuning. Our experimental results demonstrate that Re-Align not only mitigates hallucinations more effectively than previous methods but also yields significant performance gains in general visual question-answering (VQA) tasks. Moreover, we show that Re-Align maintains robustness and scalability across a wide range of VLM sizes and architectures. This work represents a significant step forward in aligning multimodal LLMs, paving the way for more reliable and effective cross-modal applications. We release all the code in https://github.com/taco-group/Re-Align.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。