arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1811.08075cs.CV

基于条件随机场的场景图生成

Scene Graph Generation via Conditional Random Fields

Weilin Cong, William Wang, Wang-Chien Lee

AI总结:

本文提出 SG-CRF 场景图生成模型,通过建模关系三元组主宾语顺序及节点语义兼容性,在 CLEVR、VRD 和 Visual Genome 上提升 Recall@100。

摘要:

尽管目标检测和分割模型在识别图像中的单个目标方面取得了巨大成功,但它们在图像描述、语义图像检索和视觉问答等认知任务上的表现仍远不能令人满意。为了在这些认知任务上取得更好的性能,仅仅识别单个目标实例是不够的。相反,需要捕捉目标实例之间的交互,以促进对图像中视觉场景的推理和理解。场景图是一种图像的图表示,能够捕捉目标实例及其关系,从而提供对图像的全面理解。然而,现有的场景图生成技术无法区分图像视觉场景中的主语和宾语,因此在存在模糊目标实例的真实世界数据集上表现不佳。在本工作中,我们提出了一种新颖的场景图生成模型,用于预测图像中的目标实例及其对应关系。我们的模型 SG-CRF 能够学习关系三元组中主语和宾语的顺序,并高效学习场景图中目标实例节点与关系节点之间的语义兼容性。实验从经验上表明,SG-CRF 在三个不同的数据集(即 CLEVR、VRD 和 Visual Genome)上优于最先进的方法,将 Recall@100 分别从 24.99% 提升到 49.95%、从 41.92% 提升到 50.47%,以及从 54.69% 提升到 54.77%。

英文摘要:

Despite the great success object detection and segmentation models have achieved in recognizing individual objects in images, performance on cognitive tasks such as image caption, semantic image retrieval, and visual QA is far from satisfactory. To achieve better performance on these cognitive tasks, merely recognizing individual object instances is insufficient. Instead, the interactions between object instances need to be captured in order to facilitate reasoning and understanding of the visual scenes in an image. Scene graph, a graph representation of images that captures object instances and their relationships, offers a comprehensive understanding of an image. However, existing techniques on scene graph generation fail to distinguish subjects and objects in the visual scenes of images and thus do not perform well with real-world datasets where exist ambiguous object instances. In this work, we propose a novel scene graph generation model for predicting object instances and its corresponding relationships in an image. Our model, SG-CRF, learns the sequential order of subject and object in a relationship triplet, and the semantic compatibility of object instance nodes and relationship nodes in a scene graph efficiently. Experiments empirically show that SG-CRF outperforms the state-of-the-art methods, on three different datasets, i.e., CLEVR, VRD, and Visual Genome, raising the Recall@100 from 24.99% to 49.95%, from 41.92% to 50.47%, and from 54.69% to 54.77%, respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑