arXiv:2410.21411cs.CV2024-10NeurIPS被引 20

用大模型生成社交关系推理的可解释答案,无需训练。

SocialGPT: Prompting LLMs for Social Relation Reasoning via Greedy Segment Optimization

  • 结合视觉模型与大模型,将图像转为文本故事后推理关系
  • 零样本下在两个数据集上表现媲美训练模型,且输出可解释
  • 提出贪心段优化法自动调优提示词,提升效果并适配不同风格

社交关系推理旨在从图像中识别朋友、配偶、同事等关系类别。现有方法多采用端到端训练专用网络,存在泛化性差、可解释性不足的问题。为此,本文提出名为SocialGPT的模块化框架,融合视觉基础模型(VFMs)的感知能力与大语言模型(LLMs)的推理能力:先由VFMs将图像内容转化为文本形式的社交故事,再由LLMs进行基于文本的推理。该框架遵循系统性设计原则,分别适配两类模型并弥合其差异。无需额外训练,即可在两个数据库上实现具有竞争力的零样本性能,并生成语言解释以增强可解释性。然而,人工设计推理阶段的提示词过程繁琐,亟需自动化优化。由于将视觉分类任务转化为大模型生成任务,自动提示优化面临长提示词优化难题。为此,我们进一步提出贪心段提示优化(GSPO),通过段级梯度信息执行贪心搜索。实验表明,GSPO显著提升性能,且方法对不同图像风格具备良好泛化性。代码已开源:https://github.com/Mengzibin/SocialGPT。

原文摘要 · Abstract (English)

Social relation reasoning aims to identify relation categories such as friends, spouses, and colleagues from images. While current methods adopt the paradigm of training a dedicated network end-to-end using labeled image data, they are limited in terms of generalizability and interpretability. To address these issues, we first present a simple yet well-crafted framework named {\name}, which combines the perception capability of Vision Foundation Models (VFMs) and the reasoning capability of Large Language Models (LLMs) within a modular framework, providing a strong baseline for social relation recognition. Specifically, we instruct VFMs to translate image content into a textual social story, and then utilize LLMs for text-based reasoning. {\name} introduces systematic design principles to adapt VFMs and LLMs separately and bridge their gaps. Without additional model training, it achieves competitive zero-shot results on two databases while offering interpretable answers, as LLMs can generate language-based explanations for the decisions. The manual prompt design process for LLMs at the reasoning phase is tedious and an automated prompt optimization method is desired. As we essentially convert a visual classification task into a generative task of LLMs, automatic prompt optimization encounters a unique long prompt optimization issue. To address this issue, we further propose the Greedy Segment Prompt Optimization (GSPO), which performs a greedy search by utilizing gradient information at the segment level. Experimental results show that GSPO significantly improves performance, and our method also generalizes to different image styles. The code is available at https://github.com/Mengzibin/SocialGPT.

社交推理大模型零样本提示优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。