用像素掩码做视觉实体链接,提升细粒度图像理解
Reverse Region-to-Entity Annotation for Pixel-Level Visual Entity Linking
- 通过反向区域-实体标注构建像素级标注数据集
- 在新数据集上模型准确率比零样本提升18个百分点
- 引入语义分块注意力机制,比基线提升5%准确率
视觉实体链接(VEL)是实现细粒度视觉理解的关键任务,旨在将图像中的物体(视觉提及)与知识库中的实体匹配。以往的VEL依赖文本输入,但在复杂场景中编写查询困难。本文提出新的像素级视觉实体链接(PL-VEL)任务,使用视觉输入(如点击或边界框)生成的像素掩码来指代对象,为VEL提供更便捷的参考方式。为此,我们通过完全自动的反向区域-实体标注框架构建了MaskOVEN-Wiki数据集,包含超过500万条像素级区域与实体标签的对齐标注,推动细粒度视觉理解发展。此外,由于像素掩码对应图像中的语义区域,我们采用视觉语义标记化方法,将先前的块交互注意力改进为区域交互注意力。人工评估显示,反向标注框架的成功率达94.8%。实验表明,在该数据集上训练的模型相比零样本模型准确率提升18个百分点;语义标记化方法比基线模型提升5个百分点。
原文摘要 · Abstract (English)
Visual Entity Linking (VEL) is a crucial task for achieving fine-grained visual understanding, matching objects within images (visual mentions) to entities in a knowledge base. Previous VEL tasks rely on textual inputs, but writing queries for complex scenes can be challenging. Visual inputs like clicks or bounding boxes offer a more convenient alternative. Therefore, we propose a new task, Pixel-Level Visual Entity Linking (PL-VEL), which uses pixel masks from visual inputs to refer to objects, supplementing reference methods for VEL. To facilitate research on this task, we have constructed the MaskOVEN-Wiki dataset through an entirely automatic reverse region-entity annotation framework. This dataset contains over 5 million annotations aligning pixel-level regions with entity-level labels, which will advance visual understanding towards fine-grained. Moreover, as pixel masks correspond to semantic regions in an image, we enhance previous patch-interacted attention with region-interacted attention by a visual semantic tokenization approach. Manual evaluation results indicate that the reverse annotation framework achieved a 94.8% annotation success rate. Experimental results show that models trained on this dataset improved accuracy by 18 points compared to zero-shot models. Additionally, the semantic tokenization method achieved a 5-point accuracy improvement over the trained baseline.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。