arXiv:2409.03025cs.CV2024-09被引 6

提升图像描述的细粒度,让模型更准确捕捉细节。

No Detail Left Behind: Revisiting Self-Retrieval for Fine-Grained Image Captioning

  • 用视觉描述增强框架注入细粒度信息,保持与人工标注一致
  • 设计课程训练策略,提升自检索奖励的利用效率
  • 提出新评测基准TrueMatch,检验模型对细微差异的感知能力

图像描述系统难以生成细粒度描述,因训练数据或嘈杂(替代文本)或通用(人工标注),且最大似然训练倾向生成高频短语。先前工作通过自检索(SR)奖励微调解决此问题,但发现其易降低描述忠实度甚至导致幻觉。本文通过改进MLE初始化并设计课程训练流程,提出两个关键方法:(1) 视觉描述增强(Visual Caption Boosting),在保持与人类标注一致的前提下,向通用数据集注入细粒度信息;(2) BagCurri,一种精心设计的训练课程,更有效地利用自检索的对比特性。联合使用使模型既能描述图像细粒度特征,又保持对真实描述的忠实性。在对抗99个随机干扰项(RD100)的SR任务上,性能提升+8.9%;在ImageCoDe上提升+7.6%。此外,现有评估指标无法衡量多样性或细粒度理解能力。为此,本文引入TrueMatch基准,包含高度相似图像组,以自检索方式评估模型捕捉细微视觉差异的能力。我们在多个主流开源多模态大模型上评估,结果表明本方法显著优于现有模型(如比Cambrian高4.8%–7.1%),且参数量仅为其1-2个数量级。

原文摘要 · Abstract (English)

Image captioning systems are unable to generate fine-grained captions as they are trained on data that is either noisy (alt-text) or generic (human annotations). This is further exacerbated by maximum likelihood training that encourages generation of frequently occurring phrases. Previous works have tried to address this limitation by fine-tuning captioners with a self-retrieval (SR) reward. However, we find that SR fine-tuning has a tendency to reduce caption faithfulness and even hallucinate. In this work, we circumvent this bottleneck by improving the MLE initialization of the captioning system and designing a curriculum for the SR fine-tuning process. To this extent, we present (1) Visual Caption Boosting, a novel framework to instill fine-grainedness in generic image captioning datasets while remaining anchored in human annotations; and (2) BagCurri, a carefully designed training curriculum that more optimally leverages the contrastive nature of the self-retrieval reward. Jointly, they enable the captioner to describe fine-grained aspects in the image while preserving faithfulness to ground-truth captions. Our approach outperforms previous work by +8.9% on SR against 99 random distractors (RD100) (Dessi et al., 2023); and +7.6% on ImageCoDe. Additionally, existing metrics to evaluate captioning systems fail to reward diversity or evaluate a model's fine-grained understanding ability. Our third contribution addresses this by proposing self-retrieval from the lens of evaluation. We introduce TrueMatch, a benchmark comprising bags of highly similar images that uses SR to assess the captioner's ability to capture subtle visual distinctions. We evaluate and compare several state-of-the-art open-source MLLMs on TrueMatch, and find that our SR approach outperforms them all by a significant margin (e.g. +4.8% - 7.1% over Cambrian) while having 1-2 orders of magnitude fewer parameters.

图像描述细粒度自检索评测基准

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