通过空间感知去噪网络提升开放词汇场景图中关系预测的准确性。
SPADE: Spatial-Aware Denoising Network for Open-vocabulary Panoptic Scene Graph Generation with Long- and Local-range Context Reasoning

- 利用扩散模型反演过程校准预训练网络,保留图像空间结构。
- 引入长程与局部上下文推理机制,显著提升空间关系识别效果。
- 适合关注复杂场景理解与视觉语言模型改进的研究者。
全景场景图生成(PSG)结合实例分割与关系理解,以捕捉复杂场景中的像素级结构关系。尽管近期基于预训练视觉-语言模型(VLMs)的方法在开放词汇设置下性能显著提升,但它们普遍忽视了VLMs在空间关系推理上的固有局限,例如难以区分物体相对位置,导致关系预测不佳。受去噪扩散模型反演过程在保持图像空间结构方面的启发,本文提出SPADE(空间感知去噪网络)框架——一种新颖的开放词汇PSG方法。SPADE包含两个关键步骤:(1) 基于反演过程的交叉注意力图,通过轻量级LoRA微调策略将通用预训练教师扩散模型校准为特定于PSG的去噪网络;(2) 设计空间感知关系图变换器,捕获局部与长距离上下文信息,促进高质量关系查询生成。在基准PSG和Visual Genome数据集上的大量实验表明,SPADE在封闭与开放设置下均优于现有最优方法,尤其在空间关系预测方面表现突出。
原文摘要 · Abstract (English)
Panoptic Scene Graph Generation (PSG) integrates instance segmentation with relation understanding to capture pixel-level structural relationships in complex scenes. Although recent approaches leveraging pre-trained vision-language models (VLMs) have significantly improved performance in the open-vocabulary setting, they commonly ignore the inherent limitations of VLMs in spatial relation reasoning, such as difficulty in distinguishing object relative positions, which results in suboptimal relation prediction. Motivated by the denoising diffusion model's inversion process in preserving the spatial structure of input images, we propose SPADE (SPatial-Aware Denoising-nEtwork) framework -- a novel approach for open-vocabulary PSG. SPADE consists of two key steps: (1) inversion-guided calibration for the UNet adaptation, and (2) spatial-aware context reasoning. In the first step, we calibrate a general pre-trained teacher diffusion model into a PSG-specific denoising network with cross-attention maps derived during inversion through a lightweight LoRA-based fine-tuning strategy. In the second step, we develop a spatial-aware relation graph transformer that captures both local and long-range contextual information, facilitating the generation of high-quality relation queries. Extensive experiments on benchmark PSG and Visual Genome datasets demonstrate that SPADE outperforms state-of-the-art methods in both closed- and open-set scenarios, particularly for spatial relationship prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。