通过加权图像相关词元,减少视觉语言模型幻觉。
Prioritizing Image-Related Tokens Enhances Vision-Language Pre-Training
- 用纯文本模型评估词元重要性,仅对图像相关词元加大损失权重。
- 在多个基准上相对标准方法提升19%和8%的平均性能。
- 适合追求高精度、低幻觉的视觉语言模型训练者使用。
在标准大型视觉语言模型(LVLM)预训练中,模型通常通过下一个词预测(NTP)最大化图像条件下的标题联合概率;然而,仅有少数标题词元直接关联视觉内容,这种简单NTP会无意中拟合噪声,增加幻觉风险。本文提出PRIOR,一种通过在NTP损失中引入差异性权重来优先处理图像相关词元的简单预训练方法,基于重要性采样框架。PRIOR利用一个仅用标题训练的纯文本大语言模型(LLM)作为参考模型,根据每个词元在无图像条件下的生成概率为其赋予权重:与视觉内容直接相关的词元在无图像时更难预测,因此在参考模型中概率更低,被赋予更高权重。训练时,基于重要性得分实施词元级重加权,调整各词元的损失。我们在两类设置下实现PRIOR:带视觉编码器的LVLM与无视觉编码器的LVLM。相比NTP,在多个视觉语言基准上分别取得19%和8%的平均相对提升。此外,PRIOR展现出更优的扩展性,其显著更高的扩展系数表明,在计算量与数据量增加时具有更大性能提升潜力。
原文摘要 · Abstract (English)
In standard large vision-language models (LVLMs) pre-training, the model typically maximizes the joint probability of the caption conditioned on the image via next-token prediction (NTP); however, since only a small subset of caption tokens directly relates to the visual content, this naive NTP unintentionally fits the model to noise and increases the risk of hallucination. We present PRIOR, a simple vision-language pre-training approach that addresses this issue by prioritizing image-related tokens through differential weighting in the NTP loss, drawing from the importance sampling framework. PRIOR introduces a reference model-a text-only large language model (LLM) trained on the captions without image inputs, to weight each token based on its probability for LVLMs training. Intuitively, tokens that are directly related to the visual inputs are harder to predict without the image and thus receive lower probabilities from the text-only reference LLM. During training, we implement a token-specific re-weighting term based on the importance scores to adjust each token's loss. We implement PRIOR in two distinct settings: LVLMs with visual encoders and LVLMs without visual encoders. We observe 19% and 8% average relative improvement, respectively, on several vision-language benchmarks compared to NTP. In addition, PRIOR exhibits superior scaling properties, as demonstrated by significantly higher scaling coefficients, indicating greater potential for performance gains compared to NTP given increasing compute and data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。