用排列增强强化学习,让模型更好理解多图间位置关系。
PeRL: Permutation-Enhanced Reinforcement Learning for Interleaved Vision-Language Reasoning
- 通过打乱图像顺序模拟不同位置关系,扩大空间探索范围。
- 在5个多图基准上显著超越现有模型,单图任务性能相当。
- 适合需要跨图像推理的复杂视觉语言任务研究者。
受DeepSeek-R1等强化学习方法在推理能力上的启发,近期研究开始探索用强化学习(RL)提升视觉语言模型(VLMs)在多模态推理任务中的表现。然而,现有方法大多局限于单图内的空间推理,难以泛化到涉及多图位置关系的复杂真实场景,而这类场景中理解图像间的关联至关重要。为此,我们提出一种专为交错式多模态任务设计的通用强化学习框架PeRL,以及一种多阶段策略以优化探索与利用的平衡,从而提升学习效率和任务性能。具体而言,我们引入图像序列的排列机制,模拟多样化的相对位置关系,以增强空间与位置多样性。同时,设计了一种回放过滤机制进行重采样,聚焦于对学习最优行为贡献最大的轨迹,有效利用已学策略。我们在5个广泛使用的多图基准和3个单图基准上评估了该模型。实验结果表明,使用PeRL训练的模型在多图基准上持续大幅超越R1相关及交错式VLM基线,在多个任务上达到最新水平,同时在单图任务上保持相当的性能。
原文摘要 · Abstract (English)
Inspired by the impressive reasoning capabilities demonstrated by reinforcement learning approaches like DeepSeek-R1, recent emerging research has begun exploring the use of reinforcement learning (RL) to enhance vision-language models (VLMs) for multimodal reasoning tasks. However, most existing multimodal reinforcement learning approaches remain limited to spatial reasoning within single-image contexts, yet still struggle to generalize to more complex and real-world scenarios involving multi-image positional reasoning, where understanding the relationships across images is crucial. To address this challenge, we propose a general reinforcement learning approach PeRL tailored for interleaved multimodal tasks, and a multi-stage strategy designed to enhance the exploration-exploitation trade-off, thereby improving learning efficiency and task performance. Specifically, we introduce permutation of image sequences to simulate varied positional relationships to explore more spatial and positional diversity. Furthermore, we design a rollout filtering mechanism for resampling to focus on trajectories that contribute most to learning optimal behaviors to exploit learned policies effectively. We evaluate our model on 5 widely-used multi-image benchmarks and 3 single-image benchmarks. Our experiments confirm that PeRL trained model consistently surpasses R1-related and interleaved VLM baselines by a large margin, achieving state-of-the-art performance on multi-image benchmarks, while preserving comparable performance on single-image tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。