arXiv:2411.10440cs.CV2024-11ICCV被引 559

让视觉语言模型像人一样分步推理,显著提升复杂问答能力。

LLaVA-CoT: Let Vision Language Models Reason Step-by-Step

  • 设计分阶段推理框架,自动完成总结、看图、逻辑推导和结论生成。
  • 仅用10万样本训练,在多个基准上比基线高9.4%,超越部分闭源大模型。
  • 提出测试时分阶段回溯搜索,无需额外训练即可增强推理效果。

大型语言模型在推理能力上已取得显著进展,但当前视觉语言模型(VLMs)在处理复杂视觉问答任务时仍难以进行系统性、结构化推理。本文提出LLaVA-CoT,一种支持自主多阶段推理的大规模视觉语言模型。不同于链式思维提示,该模型独立执行摘要、视觉解析、逻辑推理和结论生成等阶段。为实现此目标,我们构建了包含10万条样本的LLaVA-CoT-100k数据集,并提供结构化推理标注。此外,提出测试时分阶段回溯搜索方法(SWIRES),实现高效测试时扩展。结果显示,仅用10万训练样本并结合测试时扩展,LLaVA-CoT在多项多模态推理基准上比基线模型提升9.4%,性能超越Gemini-1.5-pro、GPT-4o-mini和Llama-3.2-90B-Vision-Instruct等更大或闭源模型。代码、数据集及预训练权重已开源。

原文摘要 · Abstract (English)

Large language models have demonstrated substantial advancements in reasoning capabilities. However, current Vision-Language Models (VLMs) often struggle to perform systematic and structured reasoning, especially when handling complex visual question-answering tasks. In this work, we introduce LLaVA-CoT, a large VLM designed to conduct autonomous multistage reasoning. Unlike chain-of-thought prompting, LLaVA-CoT independently engages in sequential stages of summarization, visual interpretation, logical reasoning, and conclusion generation. This structured approach enables LLaVA-CoT to achieve marked improvements on reasoning-intensive tasks. To accomplish this, we construct the LLaVA-CoT-100k dataset, integrating samples from various visual question answering sources and providing structured reasoning annotations. Besides, we propose a test-time stage-wise retracing search method (SWIRES), which enables effective and efficient test-time scaling. Remarkably, with only 100k training samples and test-time scaling, LLaVA-CoT not only outperforms its base model by 9.4% on a wide range of multimodal reasoning benchmarks, but also surpasses the performance of larger and even closed-source models, such as Gemini-1.5-pro, GPT-4o-mini, and Llama-3.2-90B-Vision-Instruct. The code, dataset, and pre-trained weights are publicly available at https://github.com/PKU-YuanGroup/LLaVA-CoT.

视觉推理多模态链式思维模型优化

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