用概率图提升视觉编程的推理能力,实现端到端优化
Enhancing Visual Programming for Visual Reasoning via Probabilistic Graphs
- 构建变量依赖的概率图,将不可导的视觉编程转为可微推断
- 在GQA、NLVRv2、Open Images上显著提升视觉编程性能
- 适合研究视觉推理与大模型程序生成的学者参考
基于大语言模型的视觉编程(VP)在复杂视觉推理(VR)任务中展现出巨大潜力。现有方法主要关注提升LLM生成的视觉程序质量,却忽视了对VP调用的预训练视觉模块的优化。由于仅能获得目标任务的最终标签而非子任务标签,且VP本身不可导,难以使用梯度方法进行端到端优化。为此,我们提出EVPG:通过构建执行过程中变量依赖关系的有向概率图,将不可导的VP执行过程重构为可微的确切概率推断过程,从而利用最终标签实现端到端的梯度优化。大量实验表明,EVPG在GQA、NLVRv2、Open Images三个经典复杂视觉推理任务上均取得显著性能提升。
原文摘要 · Abstract (English)
Recently, Visual Programming (VP) based on large language models (LLMs) has rapidly developed and demonstrated significant potential in complex Visual Reasoning (VR) tasks. Previous works to enhance VP have primarily focused on improving the quality of LLM-generated visual programs. However, they have neglected to optimize the VP-invoked pre-trained models, which serve as modules for the visual sub-tasks decomposed from the targeted tasks by VP. The difficulty is that there are only final labels of targeted VR tasks rather than labels of sub-tasks. Besides, the non-differentiable nature of VP impedes the direct use of efficient gradient-based optimization methods to leverage final labels for end-to-end learning of the entire VP framework. To overcome these issues, we propose EVPG, a method to Enhance Visual Programming for visual reasoning via Probabilistic Graphs. Specifically, we creatively build a directed probabilistic graph according to the variable dependency relationships during the VP executing process, which reconstructs the non-differentiable VP executing process into a differentiable exact probability inference process on this directed probabilistic graph. As a result, this enables the VP framework to utilize the final labels for efficient, gradient-based optimization in end-to-end supervised learning on targeted VR tasks. Extensive and comprehensive experiments demonstrate the effectiveness and advantages of our EVPG, showing significant performance improvements for VP on three classical complex VR tasks: GQA, NLVRv2, and Open Images.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。