让视觉语言模型学会基于图像纠错,提升复杂场景下的推理能力。
Visually Grounded Self-Reflection for Vision-Language Models via Reinforcement Learning

- 用强化学习训练模型在错误后回看图像,实现有依据的自我修正。
- 在分布外图像上准确率比基线提升显著,尤其在图表和导航任务中。
- 适合需要高鲁棒性视觉推理的应用,如智能助手、自动驾驶。
大型视觉语言模型可通过生成文本思维链(CoT)对多模态输入进行推理。其中自省能力——回顾早期决策并纠正错误——尤为关键。然而,现有模型在反思阶段常忽视视觉信息,导致难以将反馈转化为基于图像的修正,尤其在分布外图像上表现差。为此,我们提出一种新的强化学习框架VRRL,包含两个设计:一是随机遮蔽推理路径前缀,强调从错误中间预测中恢复而非避免出错;二是引入经验回放缓冲区中的多样化失败状态,使模型学习纠正各种异常情况。我们在涉及表格与图表的视觉定位任务及空间导航基准上评估该方法。相比现成模型和传统微调模型,在分布偏移下性能大幅下降,而本方法通过有效利用自省显著提升了平均分布外准确率,优于标准强化学习与聚焦自省的微调基线。
原文摘要 · Abstract (English)
Large vision-language models can reason over multimodal inputs by generating textual chains of thought (CoT). A key capability exhibited in CoT reasoning is self-reflection: revisiting earlier decisions and correcting previous errors. However, existing LVLMs often fail to properly attend to visual inputs during reflection, limiting their ability to translate feedback into grounded corrections, especially for out-of-distribution images. To address this issue, we propose a novel reinforcement learning training framework VRRL, with two components explicitly designed to elicit visually grounded self-reflection. First, we randomly mask trajectory prefixes during training to emphasize recovery from incorrect intermediate predictions rather than making early mistakes. Second, we introduce buffered roll-ins from an experience replay buffer to expose the model to diverse failure states that it must learn to correct. We evaluate our approach on visual grounding tasks involving tables and charts, as well as spatial navigation benchmarks. While off-the-shelf and conventionally fine-tuned models degrade substantially under distribution shift, our method substantially improves average out-of-distribution accuracy over standard RL and reflection-oriented fine-tuning baselines by using self-reflection effectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。