用扩散模型生成未知关系,让视觉关系检测突破预设类别限制。
Generalized Visual Relation Detection with Diffusion Models
- 将视觉关系建模为连续嵌入,通过扩散模型生成新关系。
- 在HOI和SGG任务上超越传统方法,支持未见关系检测。
- 适合研究开放世界关系理解与生成式视觉推理的读者。
视觉关系检测(VRD)旨在识别图像中物体对之间的关系。尽管现有模型性能优异,但均局限于预定义的关系类别,难以应对视觉关系的语义模糊性。与物体不同,视觉关系外观微妙,可从多个视角描述,如“骑”可表述为“赛跑”或“坐在”,分别对应运动和空间位置视角。为此,我们提出将视觉关系建模为连续嵌入,并设计扩散模型以条件生成方式实现广义VRD,称为Diff-VRD。在潜在空间中建模扩散过程,生成图像中所有可能的关系嵌入序列。生成过程中,主体-客体的视觉与文本嵌入作为条件信号,通过交叉注意力注入。生成后,设计匹配阶段基于语义相似性为物体对分配关系词。得益于扩散生成机制,Diff-VRD能生成数据集预定义类别之外的关系。为评估此广义任务,引入两项指标:受图像字幕启发的文本到图像检索与SPICE PR曲线。在人类-物体交互(HOI)与场景图生成(SGG)基准上的大量实验验证了Diff-VRD的优越性与有效性。
原文摘要 · Abstract (English)
Visual relation detection (VRD) aims to identify relationships (or interactions) between object pairs in an image. Although recent VRD models have achieved impressive performance, they are all restricted to pre-defined relation categories, while failing to consider the semantic ambiguity characteristic of visual relations. Unlike objects, the appearance of visual relations is always subtle and can be described by multiple predicate words from different perspectives, e.g., ``ride'' can be depicted as ``race'' and ``sit on'', from the sports and spatial position views, respectively. To this end, we propose to model visual relations as continuous embeddings, and design diffusion models to achieve generalized VRD in a conditional generative manner, termed Diff-VRD. We model the diffusion process in a latent space and generate all possible relations in the image as an embedding sequence. During the generation, the visual and text embeddings of subject-object pairs serve as conditional signals and are injected via cross-attention. After the generation, we design a subsequent matching stage to assign the relation words to subject-object pairs by considering their semantic similarities. Benefiting from the diffusion-based generative process, our Diff-VRD is able to generate visual relations beyond the pre-defined category labels of datasets. To properly evaluate this generalized VRD task, we introduce two evaluation metrics, i.e., text-to-image retrieval and SPICE PR Curve inspired by image captioning. Extensive experiments in both human-object interaction (HOI) detection and scene graph generation (SGG) benchmarks attest to the superiority and effectiveness of Diff-VRD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。