GTR框架解决视觉语言模型推理崩溃问题,提升任务成功率。
GTR: Guided Thought Reinforcement Prevents Thought Collapse in RL-based VLM Agent Training
- 引入过程引导机制,自动修正每步推理
- 在24点等任务中实现3-5倍成功率提升
- 无需人工标注,适合轻量级模型部署
基于可验证结果奖励的强化学习(RLVR)已有效扩展大语言模型的思维链(CoT)推理。然而,其在训练视觉语言模型(VLM)代理进行视觉环境下的目标导向动作推理方面的效果尚不明确。本研究通过在复杂纸牌游戏(如24点)和ALFWorld中的沉浸式任务上进行大量实验发现,仅依赖动作结果奖励时,RL会导致VLM产生“推理崩溃”现象:思维多样性迅速丧失、推理与状态无关且不完整,进而导致无效动作和负奖励。为应对这一问题,我们强调过程引导的重要性,并提出一种自动化校正器,在每个RL步骤中评估并优化代理推理。该简单且可扩展的GTR(Guided Thought Reinforcement)框架可同时训练推理与动作,无需密集的人工逐步标注。实验表明,GTR显著提升了LLaVA-7b模型在多种视觉环境中的性能与泛化能力,任务成功率比当前最优模型高出3-5倍,且模型规模显著更小。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable outcome rewards (RLVR) has effectively scaled up chain-of-thought (CoT) reasoning in large language models (LLMs). Yet, its efficacy in training vision-language model (VLM) agents for goal-directed action reasoning in visual environments is less established. This work investigates this problem through extensive experiments on complex card games, such as 24 points, and embodied tasks from ALFWorld. We find that when rewards are based solely on action outcomes, RL fails to incentivize CoT reasoning in VLMs, instead leading to a phenomenon we termed thought collapse, characterized by a rapid loss of diversity in the agent's thoughts, state-irrelevant and incomplete reasoning, and subsequent invalid actions, resulting in negative rewards. To counteract thought collapse, we highlight the necessity of process guidance and propose an automated corrector that evaluates and refines the agent's reasoning at each RL step. This simple and scalable GTR (Guided Thought Reinforcement) framework trains reasoning and action simultaneously without the need for dense, per-step human labeling. Our experiments demonstrate that GTR significantly enhances the performance and generalization of the LLaVA-7b model across various visual environments, achieving 3-5 times higher task success rates compared to SoTA models with notably smaller model sizes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。