arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.06186cs.CV
收藏

LlamaV-o1:重新思考大语言模型中的逐步视觉推理

LlamaV-o1: Rethinking Step-by-step Visual Reasoning in LLMs

Mohamed bin Zayed University of AI(穆罕默德·本·扎耶德人工智能大学);University of Central Florida(中佛罗里达大学);Linköping University(林雪平大学);Australian National University(澳大利亚国立大学)
Omkar Thawakar, Dinura Dissanayake, Ketan More, Ritesh Thawkar, Ahmed Heakl, Noor Ahsan, Yuhao Li, Mohammed Zumri, Jean Lahoud, Rao Muhammad Anwer, Hisham Chola… 展开作者

Omkar Thawakar, Dinura Dissanayake, Ketan More, Ritesh Thawkar, Ahmed Heakl, Noor Ahsan, Yuhao Li, Mohammed Zumri, Jean Lahoud, Rao Muhammad Anwer, Hisham Cholakkal, Ivan Laptev, Mubarak Shah, Fahad Shahbaz Khan, Salman Khan

AI总结:

针对多步视觉推理缺乏系统评估与训练框架的问题,本文提出含八类任务的基准、逐步推理质量指标及课程学习模型LlamaV-o1,实验显示其性能和推理效率显著提升。

摘要:

推理是解决复杂多步问题的一项基本能力,尤其是在视觉场景中,顺序化、逐步理解至关重要。现有方法缺乏用于评估视觉推理的综合框架,也不重视分步解决问题。为此,我们提出了一个综合框架,通过三项关键贡献推进大语言模型(LMM)中的逐步视觉推理。首先,我们引入一个专门用于评估多步推理任务的视觉推理基准。该基准呈现了涵盖八个不同类别的多样化挑战,范围从复杂视觉感知到科学推理,总计包含超过4k个推理步骤,从而能够稳健评估LLM在多个步骤中执行准确且可解释的视觉推理的能力。其次,我们提出一种新颖指标,以单个步骤为粒度评估视觉推理质量,同时强调正确性和逻辑连贯性。与传统的终端任务准确率指标相比,该指标能够更深入地洞察推理表现。第三,我们提出一个名为LlamaV-o1的新型多模态视觉推理模型,采用多步课程学习方法进行训练,其中任务被渐进组织,以促进增量式技能获取和问题解决。所提出的LlamaV-o1面向多步推理设计,并通过结构化训练范式逐步学习。大量实验表明,我们的LlamaV-o1优于现有开源模型,并且与闭源专有模型相比表现具有竞争力。与近期的Llava-CoT相比,我们的LlamaV-o1在六个基准上取得67.3的平均得分,绝对提升3.8%,同时在推理扩展时速度快5倍。我们的基准、模型和代码均公开发布。

英文摘要:

Reasoning is a fundamental capability for solving complex multi-step problems, particularly in visual contexts where sequential step-wise understanding is essential. Existing approaches lack a comprehensive framework for evaluating visual reasoning and do not emphasize step-wise problem-solving. To this end, we propose a comprehensive framework for advancing step-by-step visual reasoning in large language models (LMMs) through three key contributions. First, we introduce a visual reasoning benchmark specifically designed to evaluate multi-step reasoning tasks. The benchmark presents a diverse set of challenges with eight different categories ranging from complex visual perception to scientific reasoning with over 4k reasoning steps in total, enabling robust evaluation of LLMs' abilities to perform accurate and interpretable visual reasoning across multiple steps. Second, we propose a novel metric that assesses visual reasoning quality at the granularity of individual steps, emphasizing both correctness and logical coherence. The proposed metric offers deeper insights into reasoning performance compared to traditional end-task accuracy metrics. Third, we present a new multimodal visual reasoning model, named LlamaV-o1, trained using a multi-step curriculum learning approach, where tasks are progressively organized to facilitate incremental skill acquisition and problem-solving. The proposed LlamaV-o1 is designed for multi-step reasoning and learns step-by-step through a structured training paradigm. Extensive experiments show that our LlamaV-o1 outperforms existing open-source models and performs favorably against close-source proprietary models. Compared to the recent Llava-CoT, our LlamaV-o1 achieves an average score of 67.3 with an absolute gain of 3.8\% across six benchmarks while being 5 times faster during inference scaling. Our benchmark, model, and code are publicly available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑