arXiv:2503.13360cs.CVcs.AI2025-03ACL被引 34

让AI在多模态推理中不丢掉视觉信息,提升数学题解题准确率。

Mitigating Visual Forgetting via Take-along Visual Conditioning for Multi-modal Long CoT Reasoning

  • 将视觉信息动态植入关键推理阶段,减少冗余图像令牌。
  • 在五个数学基准上平均提升3.4分,达到当前最优性能。
  • 适合需要长期视觉推理的复杂任务,如几何问题求解。

近年来,大语言模型在推理能力上取得进展,从链式思维(CoT)提示演进为如OpenAI o1等面向产品的解决方案。在复现该模型时,我们发现,在需视觉输入的多模态任务(如几何问题)中,多模态大模型(MLLMs)会逐渐忽略视觉信息,即随着推理过程推进,对视觉内容的关注度持续下降,导致输出过度依赖文本。为验证此现象,我们在长链推理中截断并移除图像输入后重新完成推理,发现在MathVista测试-难子集上仅出现约2%的准确率下降,表明后续推理主要由文本输出驱动。为此,我们提出随身视觉条件化(Take-along Visual Conditioning, TVC),将图像输入引导至关键推理阶段,并通过动态剪枝压缩冗余视觉令牌。该方法有效帮助模型在推理全程保持对视觉信息的关注。实验显示,TVC在五个数学推理基准上平均表现优于先前最优方案3.4分,证明其在增强多模态推理系统中的有效性。

原文摘要 · Abstract (English)

Recent advancements in Large Language Models (LLMs) have demonstrated enhanced reasoning capabilities, evolving from Chain-of-Thought (CoT) prompting to advanced, product-oriented solutions like OpenAI o1. During our re-implementation of this model, we noticed that in multimodal tasks requiring visual input (e.g., geometry problems), Multimodal LLMs (MLLMs) struggle to maintain focus on the visual information, in other words, MLLMs suffer from a gradual decline in attention to visual information as reasoning progresses, causing text-over-relied outputs. To investigate this, we ablate image inputs during long-chain reasoning. Concretely, we truncate the reasoning process midway, then re-complete the reasoning process with the input image removed. We observe only a ~2% accuracy drop on MathVista's test-hard subset, revealing the model's textual outputs dominate the following reasoning process. Motivated by this, we propose Take-along Visual Conditioning (TVC), a strategy that shifts image input to critical reasoning stages and compresses redundant visual tokens via dynamic pruning. This methodology helps the model retain attention to the visual components throughout the reasoning. Our approach achieves state-of-the-art performance on average across five mathematical reasoning benchmarks (+3.4 points vs previous sota), demonstrating the effectiveness of TVC in enhancing multimodal reasoning systems.

多模态推理视觉遗忘链式思维数学推理

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