突破封闭词汇限制,实现任意物体关系的场景图生成
From Data to Modeling: Fully Open-vocabulary Scene Graph Generation
- 基于Transformer架构,联合预测未知物体与关系
- 在VG150上全面超越现有方法,全开集性能最优
- 利用大模型自动生成标注,减少人工依赖
我们提出OvSGTR,一种基于Transformer的全开集场景图生成框架,克服传统闭集模型的局限。传统方法将物体和关系识别限制在固定词汇表中,难以应对现实场景中不断出现的新概念。本文方法可同时预测超出预定义类别的物体(节点)及其相互关系(边)。OvSGTR采用类似DETR的架构,使用冻结的图像主干和文本编码器提取高质量视觉与语义特征,并通过Transformer解码器实现端到端场景图预测。为增强对复杂视觉关系的理解,我们设计了一种关系感知的弱监督预训练策略,通过三种路径——基于场景解析器、大语言模型(LLM)及多模态大语言模型——合成可迁移的监督信号,几乎无需人工标注。此外,针对开集设置下的灾难性遗忘问题,引入视觉-概念保留机制与知识蒸馏策略,确保微调过程中保留丰富的语义信息。在VG150基准上的大量实验表明,OvSGTR在闭集、基于物体检测的开集、基于关系的开集以及全开集等多种设置下均达到当前最佳性能。结果证明大规模关系感知预训练与Transformer架构在推动场景图生成向更泛化、更可靠的视觉理解迈进方面具有巨大潜力。
原文摘要 · Abstract (English)
We present OvSGTR, a novel transformer-based framework for fully open-vocabulary scene graph generation that overcomes the limitations of traditional closed-set models. Conventional methods restrict both object and relationship recognition to a fixed vocabulary, hindering their applicability to real-world scenarios where novel concepts frequently emerge. In contrast, our approach jointly predicts objects (nodes) and their inter-relationships (edges) beyond predefined categories. OvSGTR leverages a DETR-like architecture featuring a frozen image backbone and text encoder to extract high-quality visual and semantic features, which are then fused via a transformer decoder for end-to-end scene graph prediction. To enrich the model's understanding of complex visual relations, we propose a relation-aware pre-training strategy that synthesizes scene graph annotations in a weakly supervised manner. Specifically, we investigate three pipelines--scene parser-based, LLM-based, and multimodal LLM-based--to generate transferable supervision signals with minimal manual annotation. Furthermore, we address the common issue of catastrophic forgetting in open-vocabulary settings by incorporating a visual-concept retention mechanism coupled with a knowledge distillation strategy, ensuring that the model retains rich semantic cues during fine-tuning. Extensive experiments on the VG150 benchmark demonstrate that OvSGTR achieves state-of-the-art performance across multiple settings, including closed-set, open-vocabulary object detection-based, relation-based, and fully open-vocabulary scenarios. Our results highlight the promise of large-scale relation-aware pre-training and transformer architectures for advancing scene graph generation towards more generalized and reliable visual understanding.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。