用图模型提升复杂视觉查询的定位准确率
SceneProp: Combining Neural Network and Markov Random Field for Scene-Graph Grounding
- 将场景图接地转化为马尔可夫随机场的全局推断问题
- 在四个数据集上优于现有方法,且查询越复杂效果越好
- 适合需要精准理解物体关系的视觉语言任务
将复杂、组合型视觉查询(含多个物体及关系)的定位作为视觉-语言模型的核心挑战。标准短语定位方法虽能精确定位单个物体,却缺乏解析复杂关系描述的结构归纳偏置,导致查询越详细,性能反而下降。为此,本文聚焦于场景图接地这一强大但研究较少的范式——将查询显式表示为物体及其关系的图结构。然而,现有方法同样面临困境:随着查询图规模增大,性能不升反降,未能利用本应使定位更简单的结构信息。本文提出SceneProp,通过将场景图接地重新建模为马尔可夫随机场(MRF)中的最大后验(MAP)推断问题,实现对整个查询图的全局推理,寻找最优的图像区域与节点匹配方案。该方法在端到端框架内通过可微分信念传播算法实现。在四个基准测试上的实验表明,专用于场景图接地的设定使SceneProp显著超越先前工作。关键突破在于:其准确率随查询图规模与复杂度增加而持续提升,首次证明更多关系上下文不仅能带来更好定位,也应被充分利用。代码已开源。
原文摘要 · Abstract (English)
Grounding complex, compositional visual queries with multiple objects and relationships is a fundamental challenge for vision-language models. While standard phrase grounding methods excel at localizing single objects, they lack the structural inductive bias to parse intricate relational descriptions, often failing as queries become more descriptive. To address this structural deficit, we focus on scene-graph grounding, a powerful but less-explored formulation where the query is an explicit graph of objects and their relationships. However, existing methods for this task also struggle, paradoxically showing decreased performance as the query graph grows -- failing to leverage the very information that should make grounding easier. We introduce SceneProp, a novel method that resolves this issue by reformulating scene-graph grounding as a Maximum a Posteriori (MAP) inference problem in a Markov Random Field (MRF). By performing global inference over the entire query graph, SceneProp finds the optimal assignment of image regions to nodes that jointly satisfies all constraints. This is achieved within an end-to-end framework via a differentiable implementation of the Belief Propagation algorithm. Experiments on four benchmarks show that our dedicated focus on the scene-graph grounding formulation allows SceneProp to significantly outperform prior work. Critically, its accuracy consistently improves with the size and complexity of the query graph, demonstrating for the first time that more relational context can, and should, lead to better grounding. Codes are available at https://github.com/keitaotani/SceneProp.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。