让智能体主动探索环境,自监督生成更连贯的图像描述。
Embodied Image Captioning: Self-supervised Learning Agents for Spatially Coherent Image Descriptions
- 通过三阶段流程,用大模型共识生成伪描述标签。
- 相比基线方法,语义相似度提升显著,描述一致性大幅改善。
- 适合研究视觉语言对齐与自主智能体的学者使用。
我们提出一种自监督方法,提升智能体在通用环境中主动探索时对任意物体进行图像描述的能力。当前模型因视角差异和场景杂乱难以生成连贯描述。本文设计三阶段框架:首先智能体探索环境并收集含噪声的图像-描述对;其次利用大语言模型通过共识机制为每个物体实例提炼一致的伪描述;最后以这些伪描述微调现成的描述模型,并引入对比学习。我们在人工标注的测试集上评估了不同描述模型、探索策略、伪标签方法和微调策略的组合表现。结果表明,所提策略能有效挖掘高分歧样本,相比基线更具优势;结合所有策略的伪描述方法在语义相似度上优于现有方法;微调显著提升了描述准确率与跨视角一致性。代码与测试集标注见https://hsp-iit.github.io/embodied-captioning/
原文摘要 · Abstract (English)
We present a self-supervised method to improve an agent's abilities in describing arbitrary objects while actively exploring a generic environment. This is a challenging problem, as current models struggle to obtain coherent image captions due to different camera viewpoints and clutter. We propose a three-phase framework to fine-tune existing captioning models that enhances caption accuracy and consistency across views via a consensus mechanism. First, an agent explores the environment, collecting noisy image-caption pairs. Then, a consistent pseudo-caption for each object instance is distilled via consensus using a large language model. Finally, these pseudo-captions are used to fine-tune an off-the-shelf captioning model, with the addition of contrastive learning. We analyse the performance of the combination of captioning models, exploration policies, pseudo-labeling methods, and fine-tuning strategies, on our manually labeled test set. Results show that a policy can be trained to mine samples with higher disagreement compared to classical baselines. Our pseudo-captioning method, in combination with all policies, has a higher semantic similarity compared to other existing methods, and fine-tuning improves caption accuracy and consistency by a significant margin. Code and test set annotations available at https://hsp-iit.github.io/embodied-captioning/
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。