用动词嵌入提升图像关系推理,让模型更懂物体间的互动
Dynamic Relation Inference via Verb Embeddings
- 基于动词嵌入构建动态关系推理框架,利用语言监督学习关系表征
- 在多个数据集上实现零样本关系推理准确率显著提升,超越CLIP和现有模型
- 适用于需要理解图像中物体间复杂关系的场景,如视觉问答与内容生成
CLIP凭借对比学习训练,在图像-文本匹配任务中表现优异。然而当匹配依赖于图像中物体间的关系推理时,其性能下降。以往仅通过语言监督在关系检测数据集上微调CLIP的方法效果有限。本文提出动态关系推理方法DRIVE:扩充COCO数据集,使用硬负样本(主语-关系-宾语)三元组及其对应图像微调CLIP,并引入新损失函数以增强关系检测能力。在多个基于CLIP的模型上评估,该方法在冻结和微调设置下均显著提升零样本关系推理准确率,优于CLIP及当前最优模型,且在未见数据上具有良好泛化能力。
原文摘要 · Abstract (English)
CLIP has demonstrated exceptional image-text matching capabilities due to its training on contrastive learning tasks. Past research has suggested that whereas CLIP effectively matches text to images when the matching can be achieved just by matching the text with the objects in the image, CLIP struggles when the matching depends on representing the relationship among the objects in the images (i.e., inferring relations). Previous attempts to address this limitation by training CLIP on relation detection datasets with only linguistic supervision have met with limited success. In this paper, we offer insights and practical methods to advance the field of relation inference from images. This paper approaches the task of creating a model that effectively detects relations among the objects in images by producing text and image embeddings that capture relationships through linguistic supervision. To this end, we propose Dynamic Relation Inference via Verb Embeddings (DRIVE), which augments the COCO dataset, fine-tunes CLIP with hard negatives subject-relation-object triples and corresponding images, and introduces a novel loss function to improve relation detection. Evaluated on multiple CLIP-based models, our method significantly improves zero-shot relation inference accuracy in both frozen and fine-tuned settings, significantly outperforming CLIP and state-of-the-art models while generalizing well on unseen data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。