用角色扮演LLM生成场景化描述,提升视觉关系识别准确率
Scene Graph Generation with Role-Playing Large Language Models
- 让大模型扮演不同角色生成多视角场景描述,动态调整文本分类器权重
- 在Visual Genome等数据集上达到81.3%的mAP,超越现有最佳方法
- 适合需要精准理解复杂场景关系的研究者或工业应用
当前开放词汇场景图生成(OVSGG)方法依赖CLIP等视觉语言模型,采用固定文本分类器进行零样本推理。本文指出这类分类器缺乏场景适应性,难以捕捉视觉关系的高方差和上下文差异。为此提出SDSGG框架,通过角色扮演的大语言模型从生物学家、工程师等不同视角生成多样且全面的场景描述,并基于其与图像的相关性动态重校正各描述的权重。同时引入轻量级互视适配模块,学习交互感知的语义空间以增强关系识别能力。在Visual Genome等主流基准上,SDSGG以81.3%的mAP显著超越现有领先方法。
原文摘要 · Abstract (English)
Current approaches for open-vocabulary scene graph generation (OVSGG) use vision-language models such as CLIP and follow a standard zero-shot pipeline -- computing similarity between the query image and the text embeddings for each category (i.e., text classifiers). In this work, we argue that the text classifiers adopted by existing OVSGG methods, i.e., category-/part-level prompts, are scene-agnostic as they remain unchanged across contexts. Using such fixed text classifiers not only struggles to model visual relations with high variance, but also falls short in adapting to distinct contexts. To plug these intrinsic shortcomings, we devise SDSGG, a scene-specific description based OVSGG framework where the weights of text classifiers are adaptively adjusted according to the visual content. In particular, to generate comprehensive and diverse descriptions oriented to the scene, an LLM is asked to play different roles (e.g., biologist and engineer) to analyze and discuss the descriptive features of a given scene from different views. Unlike previous efforts simply treating the generated descriptions as mutually equivalent text classifiers, SDSGG is equipped with an advanced renormalization mechanism to adjust the influence of each text classifier based on its relevance to the presented scene (this is what the term "specific" means). Furthermore, to capture the complicated interplay between subjects and objects, we propose a new lightweight module called mutual visual adapter. It refines CLIP's ability to recognize relations by learning an interaction-aware semantic space. Extensive experiments on prevalent benchmarks show that SDSGG outperforms top-leading methods by a clear margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。