arXiv:2410.04734cs.LGcs.CL2024-10ICLR被引 16

为视觉语言模型设计细粒度奖励模型,精准识别文本生成中的幻觉

TLDR: Token-Level Detective Reward Model for Large Vision Language Models

  • 基于扰动生成合成难例,实现逐词级标注的奖励机制
  • 使模型自纠正生成错误,显著提升基础模型性能
  • 可将人工标注效率提升3倍,适用于高质量数据构建

尽管奖励模型在提升多模态大语言模型方面取得成功,但现有奖励模型仍过于粗粒度,仅提供整体二元反馈,缺乏细节信息。在处理图像与文本的多模态任务中,这种粗粒度反馈可能导致模型偏向文本而忽视图像,产生幻觉。为此,本文提出一种逐词级侦探奖励模型(TLDR),对每个文本标记进行细粒度标注。首先,采用基于扰动的方法生成合成难例及其逐词标签以训练TLDR模型;其次,证明其在辅助现成模型自我修正生成结果、作为幻觉评估工具方面的有效性。结果显示,TLDR能自动实现逐词概率优化,显著提升基线模型性能;最后,该模型可将人工标注效率提高3倍,有效拓展高质量视觉语言数据的获取范围。

原文摘要 · Abstract (English)

Although reward models have been successful in improving multimodal large language models, the reward models themselves remain brutal and contain minimal information. Notably, existing reward models only mimic human annotations by assigning only one binary feedback to any text, no matter how long the text is. In the realm of multimodal language models, where models are required to process both images and texts, a naive reward model may learn implicit biases toward texts and become less grounded in images. In this paper, we propose a $\textbf{T}$oken-$\textbf{L}$evel $\textbf{D}$etective $\textbf{R}$eward Model ($\textbf{TLDR}$) to provide fine-grained annotations to each text token. We first introduce a perturbation-based method to generate synthetic hard negatives and their token-level labels to train TLDR models. Then we show the rich usefulness of TLDR models both in assisting off-the-shelf models to self-correct their generations, and in serving as a hallucination evaluation tool. We show that TLDR automatically trains a token-level likelihood optimization, and can improve the base model's performance significantly. Finally, we show that TLDR models can significantly speed up human annotation by 3 times to acquire a broader range of high-quality vision language data.

视觉语言模型奖励模型幻觉检测自纠正

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。